mirror of https://github.com/curusarn/resh
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.2 KiB
47 lines
1.2 KiB
package strat
|
|
|
|
import "github.com/curusarn/resh/pkg/records"
|
|
|
|
// DirectorySensitive prediction/recommendation strategy
|
|
type DirectorySensitive struct {
|
|
history map[string][]string
|
|
lastPwd string
|
|
}
|
|
|
|
// Init see name
|
|
func (s *DirectorySensitive) Init() {
|
|
s.history = map[string][]string{}
|
|
}
|
|
|
|
// GetTitleAndDescription see name
|
|
func (s *DirectorySensitive) GetTitleAndDescription() (string, string) {
|
|
return "directory sensitive (recent)", "Use recent commands executed is the same directory"
|
|
}
|
|
|
|
// GetCandidates see name
|
|
func (s *DirectorySensitive) GetCandidates() []string {
|
|
return s.history[s.lastPwd]
|
|
}
|
|
|
|
// AddHistoryRecord see name
|
|
func (s *DirectorySensitive) AddHistoryRecord(record *records.EnrichedRecord) error {
|
|
// work on history for PWD
|
|
pwd := record.Pwd
|
|
// remove previous occurance of record
|
|
for i, cmd := range s.history[pwd] {
|
|
if cmd == record.CmdLine {
|
|
s.history[pwd] = append(s.history[pwd][:i], s.history[pwd][i+1:]...)
|
|
}
|
|
}
|
|
// append new record
|
|
s.history[pwd] = append([]string{record.CmdLine}, s.history[pwd]...)
|
|
s.lastPwd = record.PwdAfter
|
|
return nil
|
|
}
|
|
|
|
// ResetHistory see name
|
|
func (s *DirectorySensitive) ResetHistory() error {
|
|
s.Init()
|
|
s.history = map[string][]string{}
|
|
return nil
|
|
}
|
|
|