forked from u-root/u-root
-
Notifications
You must be signed in to change notification settings - Fork 0
/
fuser.go
78 lines (66 loc) · 1.33 KB
/
fuser.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package history
import (
"sync"
"github.com/u-root/u-root/cmds/elvish/store/storedefs"
)
// Fuser provides a unified view into a shared storage-backed command history
// and per-session history.
type Fuser struct {
store storedefs.Store
storeUpper int
*sync.RWMutex
// Per-session history.
cmds []string
seqs []int
}
func NewFuser(store storedefs.Store) (*Fuser, error) {
upper, err := store.Next()
if err != nil {
return nil, err
}
return &Fuser{
store: store,
storeUpper: upper,
RWMutex: &sync.RWMutex{},
}, nil
}
func (f *Fuser) FastForward() error {
f.Lock()
defer f.Unlock()
upper, err := f.store.Next()
if err != nil {
return err
}
f.storeUpper = upper
f.cmds = nil
f.seqs = nil
return nil
}
func (f *Fuser) AddCmd(cmd string) error {
f.Lock()
defer f.Unlock()
seq, err := f.store.Add(cmd)
if err != nil {
return err
}
f.cmds = append(f.cmds, cmd)
f.seqs = append(f.seqs, seq)
return nil
}
func (f *Fuser) AllCmds() ([]string, error) {
f.RLock()
defer f.RUnlock()
cmds, err := f.store.List(0, f.storeUpper)
if err != nil {
return nil, err
}
return append(cmds, f.cmds...), nil
}
func (f *Fuser) SessionCmds() []string {
return f.cmds
}
func (f *Fuser) Walker(prefix string) *Walker {
f.RLock()
defer f.RUnlock()
return NewWalker(f.store, f.storeUpper, f.cmds, f.seqs, prefix)
}