forked from sstarcher/hound
/
main.go
121 lines (98 loc) · 2.31 KB
/
main.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
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
package main // import "github.com/etsy/hound"
import (
"encoding/json"
"flag"
"log"
"net/http"
"os"
"runtime"
"strings"
"github.com/etsy/hound/api"
"github.com/etsy/hound/config"
"github.com/etsy/hound/searcher"
"github.com/etsy/hound/ui"
)
var (
info_log *log.Logger
error_log *log.Logger
)
func makeSearchers(cfg *config.Config) (map[string]*searcher.Searcher, bool, error) {
// Ensure we have a dbpath
if _, err := os.Stat(cfg.DbPath); err != nil {
if err := os.MkdirAll(cfg.DbPath, os.ModePerm); err != nil {
return nil, false, err
}
}
searchers, errs, err := searcher.MakeAll(cfg)
if err != nil {
return nil, false, err
}
if len(errs) > 0 {
// NOTE: This mutates the original config so the repos
// are not even seen by other code paths.
for name, _ := range errs {
delete(cfg.Repos, name)
}
return searchers, false, nil
}
return searchers, true, nil
}
func makeTemplateData(cfg *config.Config) (interface{}, error) {
var data struct {
ReposAsJson string
}
res := map[string]*config.Repo{}
for name, repo := range cfg.Repos {
res[strings.ToLower(name)] = repo
}
b, err := json.Marshal(res)
if err != nil {
return nil, err
}
data.ReposAsJson = string(b)
return &data, nil
}
func runHttp(
addr string,
dev bool,
cfg *config.Config,
idx map[string]*searcher.Searcher) error {
m := http.DefaultServeMux
h, err := ui.Content(dev, cfg)
if err != nil {
return err
}
m.Handle("/", h)
api.Setup(m, idx)
return http.ListenAndServe(addr, m)
}
func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
info_log = log.New(os.Stdout, "", log.LstdFlags)
error_log = log.New(os.Stderr, "", log.LstdFlags)
flagConf := flag.String("conf", "config.json", "")
flagAddr := flag.String("addr", ":6080", "")
flagDev := flag.Bool("dev", false, "")
flag.Parse()
var cfg config.Config
if err := cfg.LoadFromFile(*flagConf); err != nil {
panic(err)
}
idx, ok, err := makeSearchers(&cfg)
if err != nil {
log.Panic(err)
}
if !ok {
info_log.Println("Some repos failed to index, see output above")
} else {
info_log.Println("All indexes built!")
}
host := *flagAddr
if strings.HasPrefix(host, ":") {
host = "localhost" + host
}
info_log.Printf("running server at http://%s...\n", host)
if err := runHttp(*flagAddr, *flagDev, &cfg, idx); err != nil {
panic(err)
}
}