readline/operation.go

169 lines
3.1 KiB
Go
Raw Normal View History

2015-09-21 08:30:10 +03:00
package readline
import (
"io"
"os"
)
type Operation struct {
2015-09-22 13:16:24 +03:00
cfg *Config
2015-09-21 08:30:10 +03:00
t *Terminal
buf *RuneBuffer
outchan chan []rune
2015-09-22 13:16:24 +03:00
*opHistory
2015-09-22 18:01:15 +03:00
*opSearch
2015-09-21 08:30:10 +03:00
}
type wrapWriter struct {
r *Operation
target io.Writer
}
func (w *wrapWriter) Write(b []byte) (int, error) {
buf := w.r.buf
buf.Clean()
n, err := w.target.Write(b)
2015-09-22 12:41:14 +03:00
w.r.buf.Refresh()
if w.r.IsSearchMode() {
w.r.SearchRefresh(-1)
}
2015-09-21 08:30:10 +03:00
return n, err
}
2015-09-22 13:16:24 +03:00
func NewOperation(t *Terminal, cfg *Config) *Operation {
2015-09-21 08:30:10 +03:00
op := &Operation{
2015-09-22 13:16:24 +03:00
cfg: cfg,
t: t,
buf: NewRuneBuffer(t, cfg.Prompt),
outchan: make(chan []rune),
opHistory: newOpHistory(cfg.HistoryFile),
2015-09-21 08:30:10 +03:00
}
2015-09-22 18:01:15 +03:00
op.opSearch = newOpSearch(op.buf.w, op.buf, op.opHistory)
2015-09-21 08:30:10 +03:00
go op.ioloop()
return op
}
2015-09-23 08:52:26 +03:00
func (o *Operation) ioloop() {
2015-09-21 08:30:10 +03:00
for {
2015-09-22 18:01:15 +03:00
keepInSearchMode := false
2015-09-23 08:52:26 +03:00
r := o.t.ReadRune()
2015-09-21 08:30:10 +03:00
switch r {
2015-09-22 18:01:15 +03:00
case CharCannel:
2015-09-23 08:52:26 +03:00
if o.IsSearchMode() {
o.ExitSearchMode(true)
o.buf.Refresh()
2015-09-22 18:01:15 +03:00
}
case CharBckSearch:
2015-09-23 08:52:26 +03:00
o.SearchMode(S_DIR_BCK)
2015-09-22 18:01:15 +03:00
keepInSearchMode = true
case CharFwdSearch:
2015-09-23 08:52:26 +03:00
o.SearchMode(S_DIR_FWD)
2015-09-22 18:01:15 +03:00
keepInSearchMode = true
2015-09-21 17:51:48 +03:00
case CharKill:
2015-09-23 08:52:26 +03:00
o.buf.Kill()
2015-09-21 08:30:10 +03:00
case MetaNext:
2015-09-23 08:52:26 +03:00
o.buf.MoveToNextWord()
2015-09-23 08:03:13 +03:00
case CharTranspose:
2015-09-23 08:52:26 +03:00
o.buf.Transpose()
2015-09-21 08:30:10 +03:00
case MetaPrev:
2015-09-23 08:52:26 +03:00
o.buf.MoveToPrevWord()
2015-09-21 08:30:10 +03:00
case MetaDelete:
2015-09-23 08:52:26 +03:00
o.buf.DeleteWord()
2015-09-21 08:30:10 +03:00
case CharLineStart:
2015-09-23 08:52:26 +03:00
o.buf.MoveToLineStart()
2015-09-21 08:30:10 +03:00
case CharLineEnd:
2015-09-23 08:52:26 +03:00
o.buf.MoveToLineEnd()
2015-09-23 06:46:56 +03:00
case CharDelete:
2015-09-23 08:52:26 +03:00
o.buf.Delete()
2015-09-23 06:46:56 +03:00
case CharBackspace, CharCtrlH:
2015-09-23 08:52:26 +03:00
if o.IsSearchMode() {
o.SearchBackspace()
2015-09-22 18:01:15 +03:00
keepInSearchMode = true
} else {
2015-09-23 08:52:26 +03:00
o.buf.Backspace()
2015-09-22 18:01:15 +03:00
}
2015-09-23 06:46:56 +03:00
case MetaBackspace, CharCtrlW:
2015-09-23 08:52:26 +03:00
o.buf.BackEscapeWord()
2015-09-23 06:46:56 +03:00
case CharEnter, CharCtrlJ:
2015-09-23 08:52:26 +03:00
if o.IsSearchMode() {
o.ExitSearchMode(false)
2015-09-23 06:10:36 +03:00
}
2015-09-23 08:52:26 +03:00
o.buf.MoveToLineEnd()
o.buf.WriteRune('\n')
data := o.buf.Reset()
2015-09-21 08:30:10 +03:00
data = data[:len(data)-1] // trim \n
2015-09-23 08:52:26 +03:00
o.outchan <- data
o.NewHistory(data)
2015-09-21 08:30:10 +03:00
case CharBackward:
2015-09-23 08:52:26 +03:00
o.buf.MoveBackward()
2015-09-21 08:30:10 +03:00
case CharForward:
2015-09-23 08:52:26 +03:00
o.buf.MoveForward()
2015-09-21 08:30:10 +03:00
case CharPrev:
2015-09-23 08:52:26 +03:00
buf := o.PrevHistory()
2015-09-21 08:30:10 +03:00
if buf != nil {
2015-09-23 08:52:26 +03:00
o.buf.Set(buf)
2015-09-21 08:30:10 +03:00
}
case CharNext:
2015-09-23 08:52:26 +03:00
buf, ok := o.NextHistory()
2015-09-21 16:00:48 +03:00
if ok {
2015-09-23 08:52:26 +03:00
o.buf.Set(buf)
2015-09-21 08:30:10 +03:00
}
2015-09-23 06:46:56 +03:00
case CharInterrupt:
2015-09-23 08:52:26 +03:00
if o.IsSearchMode() {
o.ExitSearchMode(false)
2015-09-22 18:01:15 +03:00
}
2015-09-23 08:52:26 +03:00
o.buf.MoveToLineEnd()
o.buf.Refresh()
o.buf.WriteString("^C\n")
o.outchan <- nil
2015-09-21 08:30:10 +03:00
default:
2015-09-23 08:52:26 +03:00
if o.IsSearchMode() {
o.SearchChar(r)
2015-09-22 18:01:15 +03:00
keepInSearchMode = true
} else {
2015-09-23 08:52:26 +03:00
o.buf.WriteRune(r)
2015-09-22 18:01:15 +03:00
}
}
2015-09-23 08:52:26 +03:00
if !keepInSearchMode && o.IsSearchMode() {
o.ExitSearchMode(false)
o.buf.Refresh()
2015-09-21 08:30:10 +03:00
}
2015-09-23 08:52:26 +03:00
if !o.IsSearchMode() {
o.UpdateHistory(o.buf.Runes(), false)
2015-09-23 06:10:36 +03:00
}
2015-09-21 08:30:10 +03:00
}
}
2015-09-23 08:52:26 +03:00
func (o *Operation) Stderr() io.Writer {
return &wrapWriter{target: os.Stderr, r: o}
2015-09-21 08:30:10 +03:00
}
2015-09-23 08:52:26 +03:00
func (o *Operation) String() (string, error) {
r, err := o.Runes()
2015-09-21 08:30:10 +03:00
if err != nil {
return "", err
}
return string(r), nil
}
2015-09-23 08:52:26 +03:00
func (o *Operation) Runes() ([]rune, error) {
o.buf.Refresh() // print prompt
r := <-o.outchan
2015-09-21 08:30:10 +03:00
if r == nil {
return nil, io.EOF
}
return r, nil
}
2015-09-23 08:52:26 +03:00
func (o *Operation) Slice() ([]byte, error) {
r, err := o.Runes()
2015-09-21 08:30:10 +03:00
if err != nil {
return nil, err
}
return []byte(string(r)), nil
}
2015-09-22 13:16:24 +03:00
2015-09-23 08:52:26 +03:00
func (o *Operation) Close() {
o.opHistory.Close()
2015-09-22 13:16:24 +03:00
}