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
|
package main
import (
"flag"
"fmt"
"io"
"log"
"os"
"strings"
"github.com/chzyer/readline"
)
var cmdMode = flag.String("c", "", "")
var helpMode = flag.Bool("h", false, "")
func main() {
conf, err := getConfig()
if err != nil {
log.Fatal(err)
}
if err := getTofuStore(); err != nil {
log.Fatal(err)
}
state := NewBrowserState(conf)
marks, err := getMarks()
if err != nil {
log.Fatal(err)
}
state.Marks = marks
tours, err := getTours()
if err != nil {
log.Fatal(err)
}
state.NamedTours = tours
idents, err := getIdentities()
if err != nil {
log.Fatal(err)
}
state.Identities = idents
flag.Parse()
if *helpMode {
if err := Help(state, "cli"); err != nil {
writeError(err.Error())
}
return
}
if *cmdMode != "" {
conf.Quiet = true
state.Quiet = true
if err := handleCmdLine(state, conf, *cmdMode); err != nil {
writeError(err.Error())
}
return
}
rl, err := readline.New(Prompt)
if err != nil {
log.Fatal(err)
}
if conf.VimKeys {
rl.SetVimMode(true)
}
state.Readline = rl
if urls := flag.Args(); len(urls) > 0 {
if err := Go(state, urls[0], conf); err != nil {
writeError(err.Error())
}
}
for {
rl.SetPrompt(Prompt)
line, err := rl.Readline()
if err == io.EOF {
break
}
if err != nil {
log.Fatal(err)
}
if err := handleCmdLine(state, conf, line); err != nil {
writeError(err.Error())
}
}
}
func handleCmdLine(state *BrowserState, conf *Config, line string) error {
for _, cmd := range strings.Split(line, ";") {
if c, err := ParseCommand(strings.TrimSpace(cmd)); err != nil {
return err
} else if err := RunCommand(conf, c, state); err != nil {
return err
}
}
return nil
}
const Prompt = promptStyle + "X-1" + ansiClear + "> "
func writeError(msg string) {
fmt.Fprintf(os.Stdout, "\x1b[31m%s\x1b[0m\n", msg)
}
|
