repl: proper setPrompt() and multiline support

fix #8031
nodejs · Jul 31, 2014 · 1a84ba2 · 1a84ba2
1 parent 1a52d6a
commit 1a84ba2
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/lib/repl.js b/lib/repl.js
@@ -389,16 +389,22 @@ REPLServer.prototype.resetContext = function() {
 };
 
 REPLServer.prototype.displayPrompt = function(preserveCursor) {
-  var initial = this._prompt;
-  var prompt = initial;
+  var prompt = this._initialPrompt;
   if (this.bufferedCommand.length) {
     prompt = '...';
     var levelInd = new Array(this.lines.level.length).join('..');
     prompt += levelInd + ' ';
   }
-  this.setPrompt(prompt);
+
+  // Do not overwrite `_initialPrompt` here
+  REPLServer.super_.prototype.setPrompt.call(this, prompt);
   this.prompt(preserveCursor);
-  this.setPrompt(initial);
+};
+
+// When invoked as an API method, overwrite _initialPrompt
+REPLServer.prototype.setPrompt = function setPrompt(prompt) {
+  this._initialPrompt = prompt;
+  REPLServer.super_.prototype.setPrompt.call(this, prompt);
 };
 
 // A stream to push an array into a REPL