2019-02-18 18:44:27 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"gomark/parsing"
|
|
|
|
"os"
|
|
|
|
|
|
|
|
"git.sp4ke.com/sp4ke/gum"
|
2019-02-26 18:41:02 +00:00
|
|
|
|
|
|
|
cli "gopkg.in/urfave/cli.v1"
|
2019-02-18 18:44:27 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var startServerCmd = cli.Command{
|
|
|
|
Name: "server",
|
|
|
|
Aliases: []string{"s"},
|
|
|
|
Usage: "run browser watchers",
|
|
|
|
Action: startServer,
|
|
|
|
}
|
|
|
|
|
|
|
|
func startServer(c *cli.Context) {
|
|
|
|
manager := gum.NewManager()
|
|
|
|
manager.ShutdownOn(os.Interrupt)
|
|
|
|
|
|
|
|
api := NewApi()
|
|
|
|
manager.AddUnit(api)
|
|
|
|
|
|
|
|
go manager.Run()
|
|
|
|
|
|
|
|
// Initialize sqlite database available in global `cacheDB` variable
|
|
|
|
initDB()
|
|
|
|
|
2019-02-22 18:50:26 +00:00
|
|
|
var browsers []IBrowser
|
|
|
|
|
|
|
|
ff := NewFFBrowser()
|
|
|
|
if ff != nil {
|
|
|
|
browsers = append(browsers, ff)
|
|
|
|
}
|
|
|
|
|
|
|
|
cr := NewChromeBrowser()
|
|
|
|
if cr != nil {
|
|
|
|
browsers = append(browsers, cr)
|
2019-02-18 18:44:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for _, b := range browsers {
|
|
|
|
defer b.Shutdown()
|
|
|
|
b.RegisterHooks(parsing.ParseTags)
|
2019-03-01 17:30:50 +00:00
|
|
|
|
|
|
|
err := b.Init()
|
|
|
|
if err != nil {
|
2019-03-01 18:41:44 +00:00
|
|
|
log.Criticalf("<%s> %s", b, err)
|
2019-03-01 17:30:50 +00:00
|
|
|
b.Shutdown()
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
err = b.Load()
|
|
|
|
if err != nil {
|
2019-03-01 18:41:44 +00:00
|
|
|
log.Criticalf("<%s> %s", b, err)
|
2019-03-01 17:30:50 +00:00
|
|
|
b.Shutdown()
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
2019-02-18 18:44:27 +00:00
|
|
|
b.Watch()
|
|
|
|
}
|
|
|
|
|
|
|
|
<-manager.Quit
|
|
|
|
}
|