forked from mirror/readline
443 lines
8.6 KiB
Go
443 lines
8.6 KiB
Go
package readline
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
|
|
"golang.org/x/crypto/ssh/terminal"
|
|
)
|
|
|
|
var (
|
|
ErrInterrupt = errors.New("Interrupt")
|
|
)
|
|
|
|
type Operation struct {
|
|
cfg *Config
|
|
t *Terminal
|
|
buf *RuneBuffer
|
|
outchan chan []rune
|
|
errchan chan error
|
|
w io.Writer
|
|
|
|
history *opHistory
|
|
*opSearch
|
|
*opCompleter
|
|
*opPassword
|
|
*opVim
|
|
}
|
|
|
|
type wrapWriter struct {
|
|
r *Operation
|
|
t *Terminal
|
|
target io.Writer
|
|
}
|
|
|
|
func (w *wrapWriter) Write(b []byte) (int, error) {
|
|
if !w.t.IsReading() {
|
|
return w.target.Write(b)
|
|
}
|
|
|
|
var (
|
|
n int
|
|
err error
|
|
)
|
|
w.r.buf.Refresh(func() {
|
|
n, err = w.target.Write(b)
|
|
})
|
|
|
|
if w.r.IsSearchMode() {
|
|
w.r.SearchRefresh(-1)
|
|
}
|
|
if w.r.IsInCompleteMode() {
|
|
w.r.CompleteRefresh()
|
|
}
|
|
return n, err
|
|
}
|
|
|
|
func NewOperation(t *Terminal, cfg *Config) *Operation {
|
|
op := &Operation{
|
|
t: t,
|
|
buf: NewRuneBuffer(t, cfg.Prompt, cfg.MaskRune, cfg),
|
|
outchan: make(chan []rune),
|
|
errchan: make(chan error),
|
|
}
|
|
op.w = op.buf.w
|
|
op.SetConfig(cfg)
|
|
op.opVim = newVimMode(op)
|
|
op.opCompleter = newOpCompleter(op.buf.w, op)
|
|
op.opPassword = newOpPassword(op)
|
|
go op.ioloop()
|
|
return op
|
|
}
|
|
|
|
func (o *Operation) SetPrompt(s string) {
|
|
o.buf.SetPrompt(s)
|
|
}
|
|
|
|
func (o *Operation) SetMaskRune(r rune) {
|
|
o.buf.SetMask(r)
|
|
}
|
|
|
|
func (o *Operation) ioloop() {
|
|
for {
|
|
keepInSearchMode := false
|
|
keepInCompleteMode := false
|
|
r := o.t.ReadRune()
|
|
if r == 0 { // io.EOF
|
|
o.buf.Clean()
|
|
select {
|
|
case o.errchan <- io.EOF:
|
|
}
|
|
break
|
|
}
|
|
isUpdateHistory := true
|
|
|
|
if o.IsInCompleteSelectMode() {
|
|
keepInCompleteMode = o.HandleCompleteSelect(r)
|
|
if keepInCompleteMode {
|
|
continue
|
|
}
|
|
|
|
o.buf.Refresh(nil)
|
|
switch r {
|
|
case CharEnter, CharCtrlJ:
|
|
o.history.Update(o.buf.Runes(), false)
|
|
fallthrough
|
|
case CharInterrupt:
|
|
o.t.KickRead()
|
|
fallthrough
|
|
case CharBell:
|
|
continue
|
|
}
|
|
}
|
|
|
|
if o.IsEnableVimMode() {
|
|
r = o.HandleVim(r, o.t.ReadRune)
|
|
if r == 0 {
|
|
continue
|
|
}
|
|
}
|
|
|
|
switch r {
|
|
case CharBell:
|
|
if o.IsSearchMode() {
|
|
o.ExitSearchMode(true)
|
|
o.buf.Refresh(nil)
|
|
}
|
|
if o.IsInCompleteMode() {
|
|
o.ExitCompleteMode(true)
|
|
o.buf.Refresh(nil)
|
|
}
|
|
case CharTab:
|
|
if o.cfg.AutoComplete == nil {
|
|
o.t.Bell()
|
|
break
|
|
}
|
|
o.OnComplete()
|
|
keepInCompleteMode = true
|
|
case CharBckSearch:
|
|
o.SearchMode(S_DIR_BCK)
|
|
keepInSearchMode = true
|
|
case CharCtrlU:
|
|
o.buf.KillFront()
|
|
case CharFwdSearch:
|
|
o.SearchMode(S_DIR_FWD)
|
|
keepInSearchMode = true
|
|
case CharKill:
|
|
o.buf.Kill()
|
|
keepInCompleteMode = true
|
|
case MetaForward:
|
|
o.buf.MoveToNextWord()
|
|
case CharTranspose:
|
|
o.buf.Transpose()
|
|
case MetaBackward:
|
|
o.buf.MoveToPrevWord()
|
|
case MetaDelete:
|
|
o.buf.DeleteWord()
|
|
case CharLineStart:
|
|
o.buf.MoveToLineStart()
|
|
case CharLineEnd:
|
|
o.buf.MoveToLineEnd()
|
|
case CharBackspace, CharCtrlH:
|
|
if o.IsSearchMode() {
|
|
o.SearchBackspace()
|
|
keepInSearchMode = true
|
|
break
|
|
}
|
|
|
|
if o.buf.Len() == 0 {
|
|
o.t.Bell()
|
|
break
|
|
}
|
|
o.buf.Backspace()
|
|
if o.IsInCompleteMode() {
|
|
o.OnComplete()
|
|
}
|
|
case MetaBackspace, CharCtrlW:
|
|
o.buf.BackEscapeWord()
|
|
case CharEnter, CharCtrlJ:
|
|
if o.IsSearchMode() {
|
|
o.ExitSearchMode(false)
|
|
}
|
|
o.buf.MoveToLineEnd()
|
|
var data []rune
|
|
if !o.cfg.UniqueEditLine {
|
|
o.buf.WriteRune('\n')
|
|
data = o.buf.Reset()
|
|
data = data[:len(data)-1] // trim \n
|
|
} else {
|
|
o.buf.Clean()
|
|
data = o.buf.Reset()
|
|
}
|
|
o.outchan <- data
|
|
if !o.cfg.DisableAutoSaveHistory {
|
|
// ignore IO error
|
|
_ = o.history.New(data)
|
|
} else {
|
|
isUpdateHistory = false
|
|
}
|
|
case CharBackward:
|
|
o.buf.MoveBackward()
|
|
case CharForward:
|
|
o.buf.MoveForward()
|
|
case CharPrev:
|
|
buf := o.history.Prev()
|
|
if buf != nil {
|
|
o.buf.Set(buf)
|
|
} else {
|
|
o.t.Bell()
|
|
}
|
|
case CharNext:
|
|
buf, ok := o.history.Next()
|
|
if ok {
|
|
o.buf.Set(buf)
|
|
} else {
|
|
o.t.Bell()
|
|
}
|
|
case CharDelete:
|
|
if o.buf.Len() > 0 || !o.IsNormalMode() {
|
|
o.t.KickRead()
|
|
if !o.buf.Delete() {
|
|
o.t.Bell()
|
|
}
|
|
break
|
|
}
|
|
|
|
// treat as EOF
|
|
o.buf.WriteString(o.cfg.EOFPrompt + "\n")
|
|
o.buf.Reset()
|
|
isUpdateHistory = false
|
|
o.history.Revert()
|
|
o.errchan <- io.EOF
|
|
case CharInterrupt:
|
|
if o.IsSearchMode() {
|
|
o.t.KickRead()
|
|
o.ExitSearchMode(true)
|
|
break
|
|
}
|
|
if o.IsInCompleteMode() {
|
|
o.t.KickRead()
|
|
o.ExitCompleteMode(true)
|
|
o.buf.Refresh(nil)
|
|
break
|
|
}
|
|
o.buf.MoveToLineEnd()
|
|
o.buf.Refresh(nil)
|
|
o.buf.WriteString(o.cfg.InterruptPrompt + "\n")
|
|
o.buf.Reset()
|
|
isUpdateHistory = false
|
|
o.history.Revert()
|
|
o.errchan <- ErrInterrupt
|
|
default:
|
|
if o.IsSearchMode() {
|
|
o.SearchChar(r)
|
|
keepInSearchMode = true
|
|
break
|
|
}
|
|
o.buf.WriteRune(r)
|
|
if o.IsInCompleteMode() {
|
|
o.OnComplete()
|
|
keepInCompleteMode = true
|
|
}
|
|
}
|
|
|
|
if o.cfg.Listener != nil {
|
|
newLine, newPos, ok := o.cfg.Listener.OnChange(o.buf.Runes(), o.buf.Pos(), r)
|
|
if ok {
|
|
o.buf.SetWithIdx(newPos, newLine)
|
|
}
|
|
}
|
|
|
|
if !keepInSearchMode && o.IsSearchMode() {
|
|
o.ExitSearchMode(false)
|
|
o.buf.Refresh(nil)
|
|
} else if o.IsInCompleteMode() {
|
|
if !keepInCompleteMode {
|
|
o.ExitCompleteMode(false)
|
|
o.Refresh()
|
|
} else {
|
|
o.buf.Refresh(nil)
|
|
o.CompleteRefresh()
|
|
}
|
|
}
|
|
if isUpdateHistory && !o.IsSearchMode() {
|
|
// it will cause null history
|
|
o.history.Update(o.buf.Runes(), false)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (o *Operation) Stderr() io.Writer {
|
|
return &wrapWriter{target: o.cfg.Stderr, r: o, t: o.t}
|
|
}
|
|
|
|
func (o *Operation) Stdout() io.Writer {
|
|
return &wrapWriter{target: o.cfg.Stdout, r: o, t: o.t}
|
|
}
|
|
|
|
func (o *Operation) String() (string, error) {
|
|
r, err := o.Runes()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(r), nil
|
|
}
|
|
|
|
func (o *Operation) Runes() ([]rune, error) {
|
|
o.t.EnterRawMode()
|
|
defer o.t.ExitRawMode()
|
|
|
|
if o.cfg.Listener != nil {
|
|
o.cfg.Listener.OnChange(nil, 0, 0)
|
|
}
|
|
o.buf.Refresh(nil) // print prompt
|
|
o.t.KickRead()
|
|
select {
|
|
case r := <-o.outchan:
|
|
return r, nil
|
|
case err := <-o.errchan:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func (o *Operation) PasswordEx(prompt string, l Listener) ([]byte, error) {
|
|
cfg := o.GenPasswordConfig()
|
|
cfg.Prompt = prompt
|
|
cfg.Listener = l
|
|
return o.PasswordWithConfig(cfg)
|
|
}
|
|
|
|
func (o *Operation) GenPasswordConfig() *Config {
|
|
return o.opPassword.PasswordConfig()
|
|
}
|
|
|
|
func (o *Operation) PasswordWithConfig(cfg *Config) ([]byte, error) {
|
|
if err := o.opPassword.EnterPasswordMode(cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
defer o.opPassword.ExitPasswordMode()
|
|
return o.Slice()
|
|
}
|
|
|
|
func (o *Operation) Password(prompt string) ([]byte, error) {
|
|
w := o.Stdout()
|
|
if prompt != "" {
|
|
fmt.Fprintf(w, prompt)
|
|
}
|
|
o.t.EnterRawMode()
|
|
defer o.t.ExitRawMode()
|
|
|
|
b, err := terminal.ReadPassword(int(o.cfg.Stdin.Fd()))
|
|
fmt.Fprint(w, "\r\n")
|
|
return b, err
|
|
}
|
|
|
|
func (o *Operation) SetTitle(t string) {
|
|
o.w.Write([]byte("\033[2;" + t + "\007"))
|
|
}
|
|
|
|
func (o *Operation) Slice() ([]byte, error) {
|
|
r, err := o.Runes()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return []byte(string(r)), nil
|
|
}
|
|
|
|
func (o *Operation) Close() {
|
|
o.history.Close()
|
|
}
|
|
|
|
func (o *Operation) SetHistoryPath(path string) {
|
|
if o.history != nil {
|
|
o.history.Close()
|
|
}
|
|
o.cfg.HistoryFile = path
|
|
o.history = newOpHistory(o.cfg)
|
|
}
|
|
|
|
func (o *Operation) IsNormalMode() bool {
|
|
return !o.IsInCompleteMode() && !o.IsSearchMode()
|
|
}
|
|
|
|
func (op *Operation) SetConfig(cfg *Config) (*Config, error) {
|
|
if op.cfg == cfg {
|
|
return op.cfg, nil
|
|
}
|
|
if err := cfg.Init(); err != nil {
|
|
return op.cfg, err
|
|
}
|
|
old := op.cfg
|
|
op.cfg = cfg
|
|
op.SetPrompt(cfg.Prompt)
|
|
op.SetMaskRune(cfg.MaskRune)
|
|
op.buf.SetConfig(cfg)
|
|
|
|
if cfg.opHistory == nil {
|
|
op.SetHistoryPath(cfg.HistoryFile)
|
|
cfg.opHistory = op.history
|
|
cfg.opSearch = newOpSearch(op.buf.w, op.buf, op.history, cfg)
|
|
}
|
|
op.history = cfg.opHistory
|
|
|
|
// SetHistoryPath will close opHistory which already exists
|
|
// so if we use it next time, we need to reopen it by `InitHistory()`
|
|
op.history.Init()
|
|
|
|
if op.cfg.AutoComplete != nil {
|
|
op.opCompleter = newOpCompleter(op.buf.w, op)
|
|
}
|
|
|
|
op.opSearch = cfg.opSearch
|
|
return old, nil
|
|
}
|
|
|
|
// if err is not nil, it just mean it fail to write to file
|
|
// other things goes fine.
|
|
func (o *Operation) SaveHistory(content string) error {
|
|
return o.history.New([]rune(content))
|
|
}
|
|
|
|
func (o *Operation) Refresh() {
|
|
if o.t.IsReading() {
|
|
o.buf.Refresh(nil)
|
|
}
|
|
}
|
|
|
|
func FuncListener(f func(line []rune, pos int, key rune) (newLine []rune, newPos int, ok bool)) Listener {
|
|
return &DumpListener{f: f}
|
|
}
|
|
|
|
type DumpListener struct {
|
|
f func(line []rune, pos int, key rune) (newLine []rune, newPos int, ok bool)
|
|
}
|
|
|
|
func (d *DumpListener) OnChange(line []rune, pos int, key rune) (newLine []rune, newPos int, ok bool) {
|
|
return d.f(line, pos, key)
|
|
}
|
|
|
|
type Listener interface {
|
|
OnChange(line []rune, pos int, key rune) (newLine []rune, newPos int, ok bool)
|
|
}
|