--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/src/jdk.jshell/share/classes/jdk/jshell/MaskCommentsAndModifiers.java Mon Oct 19 19:15:16 2015 +0200
@@ -0,0 +1,202 @@
+/*
+ * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation. Oracle designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Oracle in the LICENSE file that accompanied this code.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+package jdk.jshell;
+
+import java.util.Set;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+/**
+ * Within a String, mask code comments and ignored modifiers (within context).
+ *
+ * @author Robert Field
+ */
+class MaskCommentsAndModifiers {
+
+ private final static Set<String> IGNORED_MODIFERS =
+ Stream.of( "public", "protected", "private", "static", "final" )
+ .collect( Collectors.toSet() );
+
+ private final StringBuilder sbCleared = new StringBuilder();
+ private final StringBuilder sbMask = new StringBuilder();
+ private final String str;
+ private final int length;
+ private final boolean maskModifiers;
+ private int next = 0;
+ private boolean wasMasked = false;
+ private boolean inside = false;
+
+ @SuppressWarnings("empty-statement")
+ public MaskCommentsAndModifiers(String s, boolean maskModifiers) {
+ this.str = s;
+ this.length = s.length();
+ this.maskModifiers = maskModifiers;
+ do { } while (next());
+ }
+
+ public String cleared() {
+ return sbCleared.toString();
+ }
+
+ public String mask() {
+ return sbMask.toString();
+ }
+
+ public boolean wasMasked() {
+ return wasMasked;
+ }
+
+ /**
+ * Read the next character
+ */
+ private int read() {
+ if (next >= length) {
+ return -1;
+ }
+ return str.charAt(next++);
+ }
+
+ private void write(StringBuilder sb, int ch) {
+ sb.append((char)ch);
+ }
+
+ private void write(int ch) {
+ write(sbCleared, ch);
+ write(sbMask, Character.isWhitespace(ch) ? ch : ' ');
+ }
+
+ private void writeMask(int ch) {
+ wasMasked = true;
+ write(sbMask, ch);
+ write(sbCleared, Character.isWhitespace(ch) ? ch : ' ');
+ }
+
+ private void write(CharSequence s) {
+ for (int cp : s.chars().toArray()) {
+ write(cp);
+ }
+ }
+
+ private void writeMask(CharSequence s) {
+ for (int cp : s.chars().toArray()) {
+ writeMask(cp);
+ }
+ }
+
+ private boolean next() {
+ return next(read());
+ }
+
+ private boolean next(int c) {
+ if (c < 0) {
+ return false;
+ }
+
+ if (c == '\'' || c == '"') {
+ inside = true;
+ write(c);
+ int match = c;
+ c = read();
+ while (c != match) {
+ if (c < 0) {
+ return false;
+ }
+ if (c == '\n' || c == '\r') {
+ write(c);
+ return true;
+ }
+ if (c == '\\') {
+ write(c);
+ c = read();
+ }
+ write(c);
+ c = read();
+ }
+ write(c);
+ return true;
+ }
+
+ if (c == '/') {
+ c = read();
+ if (c == '*') {
+ writeMask('/');
+ writeMask(c);
+ int prevc = 0;
+ while ((c = read()) != '/' || prevc != '*') {
+ if (c < 0) {
+ return false;
+ }
+ writeMask(c);
+ prevc = c;
+ }
+ writeMask(c);
+ return true;
+ } else if (c == '/') {
+ writeMask('/');
+ writeMask(c);
+ while ((c = read()) != '\n' && c != '\r') {
+ if (c < 0) {
+ return false;
+ }
+ writeMask(c);
+ }
+ writeMask(c);
+ return true;
+ } else {
+ inside = true;
+ write('/');
+ // read character falls through
+ }
+ }
+
+ if (Character.isJavaIdentifierStart(c)) {
+ if (maskModifiers && !inside) {
+ StringBuilder sb = new StringBuilder();
+ do {
+ write(sb, c);
+ c = read();
+ } while (Character.isJavaIdentifierPart(c));
+ String id = sb.toString();
+ if (IGNORED_MODIFERS.contains(id)) {
+ writeMask(sb);
+ } else {
+ write(sb);
+ if (id.equals("import")) {
+ inside = true;
+ }
+ }
+ return next(c); // recurse to handle left-over character
+ }
+ } else if (!Character.isWhitespace(c)) {
+ inside = true;
+ }
+
+ if (c < 0) {
+ return false;
+ }
+ write(c);
+ return true;
+ }
+}