bbgo_origin/cmd/bbgo-webview/main.go
2023-04-25 18:30:23 +08:00

166 lines
3.6 KiB
Go

package main
import (
"context"
"flag"
"net"
"os"
"os/signal"
"path/filepath"
"strconv"
"time"
"github.com/joho/godotenv"
"github.com/webview/webview"
log "github.com/sirupsen/logrus"
"github.com/c9s/bbgo/pkg/bbgo"
"github.com/c9s/bbgo/pkg/server"
)
func main() {
noChangeDir := false
portNum := 0
flag.BoolVar(&noChangeDir, "no-chdir", false, "do not change directory")
flag.IntVar(&portNum, "port", 0, "server port")
flag.Parse()
if !noChangeDir {
ep, err := os.Executable()
if err != nil {
log.Fatalln("failed to find the current executable:", err)
}
resourceDir := filepath.Join(filepath.Dir(ep), "..", "Resources")
if _, err := os.Stat(resourceDir); err == nil {
err = os.Chdir(resourceDir)
if err != nil {
log.Fatalln("chdir error:", err)
}
}
}
dotenvFile := ".env.local"
if _, err := os.Stat(dotenvFile); err == nil {
if err := godotenv.Load(dotenvFile); err != nil {
log.WithError(err).Error("error loading dotenv file")
return
}
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
debug, _ := strconv.ParseBool(os.Getenv("DEBUG_WEBVIEW"))
view := webview.New(debug)
defer view.Destroy()
view.SetTitle("BBGO")
view.SetSize(1024, 780, webview.HintNone)
configFile := "bbgo.yaml"
var setup *server.Setup
var userConfig *bbgo.Config
_, err := os.Stat(configFile)
if os.IsNotExist(err) {
setup = &server.Setup{
Context: ctx,
Cancel: cancel,
Token: "",
BeforeRestart: func() {
view.Destroy()
},
}
userConfig = &bbgo.Config{
Notifications: nil,
Persistence: nil,
Sessions: nil,
ExchangeStrategies: nil,
}
} else {
userConfig, err = bbgo.Load(configFile, true)
if err != nil {
log.WithError(err).Error("can not load config file")
return
}
}
environ := bbgo.NewEnvironment()
trader := bbgo.NewTrader(environ)
// we could initialize the environment from the settings
if setup == nil {
if err := bbgo.BootstrapEnvironment(ctx, environ, userConfig); err != nil {
log.WithError(err).Error("failed to bootstrap environment")
return
}
// we could initialize the environment from the settings
go func() {
if err := environ.Sync(ctx); err != nil {
log.WithError(err).Error("failed to sync data")
return
}
if err := trader.Configure(userConfig); err != nil {
log.WithError(err).Error("failed to configure trader")
return
}
if err := trader.LoadState(ctx); err != nil {
log.WithError(err).Error("failed to load strategy states")
return
}
// for setup mode, we don't start the trader
if err := trader.Run(ctx); err != nil {
log.WithError(err).Error("failed to start trader")
}
}()
}
// find a free port for binding the server
ln, err := net.Listen("tcp", "127.0.0.1:"+strconv.Itoa(portNum))
if err != nil {
log.WithError(err).Error("can not bind listener")
return
}
defer ln.Close()
baseURL := "http://" + ln.Addr().String()
srv := &server.Server{
Config: userConfig,
Environ: environ,
Trader: trader,
OpenInBrowser: false,
Setup: setup,
}
go func() {
if err := srv.RunWithListener(ctx, ln); err != nil {
log.WithError(err).Errorf("server error")
}
}()
log.Infof("pinging the server at %s", baseURL)
server.PingUntil(ctx, time.Second, baseURL, func() {
log.Infof("got pong, navigate to %s", baseURL)
view.Navigate(baseURL)
view.Run()
})
// Wait until the interrupt signal arrives or browser window is closed
sigc := make(chan os.Signal)
signal.Notify(sigc, os.Interrupt)
select {
case <-sigc:
}
log.Println("exiting...")
}