repl: handle buffered string logic on finish

nodejs · Nov 18, 2018 · 2c179f3 · 2c179f3
1 parent d58a0c6
commit 2c179f3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/lib/readline.js b/lib/readline.js
@@ -348,9 +348,9 @@ Interface.prototype.undoHistory = function() {
 
 // If it's a multiline code, then add history
 // accordingly.
-Interface.prototype.multilineHistory = function() {
-  // check if we got a multiline code
-  if (this.multiline !== '' && this.terminal) {
+Interface.prototype.multilineHistory = function(clearBuffer) {
+  // if not clear buffer, add multiline history
+  if (!clearBuffer && this.terminal) {
     const dupIndex = this.history.indexOf(this.multiline);
     if (dupIndex !== -1) this.history.splice(dupIndex, 1);
     // Remove the last entered line as multiline

diff --git a/lib/repl.js b/lib/repl.js
@@ -668,6 +668,13 @@ function REPLServer(prompt,
         }
       }
 
+      // handle multiline history
+      if (self[kBufferedCommandSymbol].length)
+        REPLServer.super_.prototype.multilineHistory.call(self, false);
+      else {
+        REPLServer.super_.prototype.multilineHistory.call(self, true);
+      }
+
       // Clear buffer if no SyntaxErrors
       self.clearBufferedCommand();
       sawCtrlD = false;
@@ -774,7 +781,6 @@ exports.start = function(prompt,
 
 REPLServer.prototype.clearBufferedCommand = function clearBufferedCommand() {
   this[kBufferedCommandSymbol] = '';
-  REPLServer.super_.prototype.multilineHistory.call(this);
 };
 
 REPLServer.prototype.close = function close() {