F

stateful · Jan 5, 2023 · 405aafa · 405aafa
1 parent 42bdb5d
commit 405aafa
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/internal/kernel/prompt.go b/internal/kernel/prompt.go
@@ -45,5 +45,5 @@ func detectPrompt(cmdName string) ([]byte, error) {
 			break
 		}
 	}
-	return prompt, nil
+	return dropANSIEscape(prompt), nil
 }
diff --git a/internal/kernel/service.go b/internal/kernel/service.go
@@ -30,6 +30,7 @@ func (m *kernelServiceServer) PostSession(ctx context.Context, req *kernelv1.Pos
 	if len(prompt) == 0 {
 		var err error
 		prompt, err = DetectPrompt(req.CommandName)
+		m.logger.Info("detected prompt", zap.Error(err), zap.ByteString("prompt", prompt))
 		if err != nil {
 			return nil, fmt.Errorf("failed to detect prompt: %w", err)
 		}

diff --git a/internal/kernel/session.go b/internal/kernel/session.go
@@ -252,7 +252,7 @@ func scanLinesPrompt(data []byte, atEOF bool, prompt []byte) (advance int, token
 	newLineIdx := bytes.IndexByte(data, '\n')
 	if promptIdx >= 0 && (newLineIdx < 0 || promptIdx < newLineIdx) {
 		m := promptIdx + len(prompt)
-		if len(data) >= m && data[m] == ' ' {
+		if len(data) > m && data[m] == ' ' {
 			m++
 		}
 		return m, data[0 : promptIdx+len(prompt)], nil