nashorn/src/jdk.scripting.nashorn.shell/share/classes/jdk/nashorn/tools/jjs/Console.java
changeset 32152 705d4af0d4d7
child 32240 d7c7a5dc92d8
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/nashorn/src/jdk.scripting.nashorn.shell/share/classes/jdk/nashorn/tools/jjs/Console.java	Thu Aug 13 19:07:27 2015 +0530
@@ -0,0 +1,103 @@
+/*
+ * 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.nashorn.tools.jjs;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.PrintStream;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.List;
+import java.util.prefs.BackingStoreException;
+import java.util.prefs.Preferences;
+import jdk.internal.jline.console.ConsoleReader;
+import jdk.internal.jline.console.history.History.Entry;
+import jdk.internal.jline.console.history.MemoryHistory;
+
+class Console implements AutoCloseable {
+    private final ConsoleReader in;
+    private final PersistentHistory history;
+
+    Console(InputStream cmdin, PrintStream cmdout, Preferences prefs) throws IOException {
+        in = new ConsoleReader(cmdin, cmdout);
+        in.setExpandEvents(false);
+        in.setHandleUserInterrupt(true);
+        in.setHistory(history = new PersistentHistory(prefs));
+        Runtime.getRuntime().addShutdownHook(new Thread(()->close()));
+    }
+
+    String readLine(String prompt) throws IOException {
+        return in.readLine(prompt);
+    }
+
+
+    @Override
+    public void close() {
+        history.save();
+    }
+
+    public static class PersistentHistory extends MemoryHistory {
+
+        private final Preferences prefs;
+
+        protected PersistentHistory(Preferences prefs) {
+            this.prefs = prefs;
+            load();
+        }
+
+        private static final String HISTORY_LINE_PREFIX = "HISTORY_LINE_";
+
+        public final void load() {
+            try {
+                List<String> keys = new ArrayList<>(Arrays.asList(prefs.keys()));
+                Collections.sort(keys);
+                for (String key : keys) {
+                    if (!key.startsWith(HISTORY_LINE_PREFIX))
+                        continue;
+                    CharSequence line = prefs.get(key, "");
+                    add(line);
+                }
+            } catch (BackingStoreException ex) {
+                throw new IllegalStateException(ex);
+            }
+        }
+
+        public void save() {
+            Iterator<Entry> entries = iterator();
+            if (entries.hasNext()) {
+                int len = (int) Math.ceil(Math.log10(size()+1));
+                String format = HISTORY_LINE_PREFIX + "%0" + len + "d";
+                while (entries.hasNext()) {
+                    Entry entry = entries.next();
+                    prefs.put(String.format(format, entry.index()), entry.value().toString());
+                }
+            }
+        }
+
+    }
+}