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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
package main
import (
"github.com/spf13/pflag"
"os"
"pik/cache"
"pik/crawl"
"pik/flags"
"pik/indexers/pikdex"
"pik/menu"
"pik/model"
"pik/run"
"pik/runner/python"
"pik/runner/shell"
"pik/search"
"pik/spool"
)
var initializers = []model.HasInit{
python.Python,
}
var indexers = []model.Indexer{
pikdex.Indexer,
}
var runners = []model.Runner{
shell.Runner,
python.Python,
}
var hydrators = []model.Hydrator{
pikdex.Indexer,
}
var ForceConfirm = false
func main() {
pflag.Parse()
for _, i := range initializers {
err := i.Init()
if err != nil {
_, _ = spool.Warn("%v\n", err)
}
}
here, err := os.Getwd()
if err != nil {
panic(err)
}
locs := crawl.RichLocations(here)
last := locs[len(locs)-1]
root, err := os.OpenRoot(last)
if root == nil {
panic(err)
}
fs := root.FS()
if err != nil {
panic(err)
}
var st *model.State
var stateError error
if !*flags.All {
st, stateError = model.NewState(fs, locs, indexers, runners)
} else {
c, err := cache.Load()
if err != nil {
panic(err)
}
st, stateError = cache.LoadState(fs, c, indexers, runners)
}
if stateError != nil {
panic(stateError)
}
err = cache.Save(st)
if err != nil {
_, _ = spool.Warn("%v", err)
}
args := pflag.Args()
if len(args) == 0 {
source, target, err := menu.Show(st, hydrators)
if err != nil {
panic(err)
}
err = run.Run(source.Source, target, args...)
if err != nil {
panic(err)
}
return
}
target, src, confirm, _, args := search.Search(st, args...)
if !*flags.All && target == nil && len(args) > 0 {
err := pflag.Set("all", "true")
ForceConfirm = true
if err != nil {
panic(err)
}
main()
return
}
if target == nil {
_, _ = spool.Print("no target found.")
os.Exit(1)
return
}
if confirm || ForceConfirm {
if !menu.Confirm(os.Stdin, src, target, args...) {
os.Exit(0)
}
}
err = run.Run(src, target, args...)
if err != nil {
panic(err)
}
}
|
