98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/ayntgl/discordo/config"
|
|
"github.com/ayntgl/discordo/ui"
|
|
"github.com/diamondburned/arikawa/v3/api"
|
|
"github.com/diamondburned/arikawa/v3/gateway"
|
|
"github.com/rivo/tview"
|
|
"github.com/zalando/go-keyring"
|
|
)
|
|
|
|
var (
|
|
flagToken string
|
|
flagConfig string
|
|
flagLog string
|
|
)
|
|
|
|
func init() {
|
|
tview.Borders.TopLeftFocus = tview.Borders.TopLeft
|
|
tview.Borders.TopRightFocus = tview.Borders.TopRight
|
|
tview.Borders.BottomLeftFocus = tview.Borders.BottomLeft
|
|
tview.Borders.BottomRightFocus = tview.Borders.BottomRight
|
|
tview.Borders.HorizontalFocus = tview.Borders.Horizontal
|
|
tview.Borders.VerticalFocus = tview.Borders.Vertical
|
|
tview.Borders.TopLeft = 0
|
|
tview.Borders.TopRight = 0
|
|
tview.Borders.BottomLeft = 0
|
|
tview.Borders.BottomRight = 0
|
|
tview.Borders.Horizontal = 0
|
|
tview.Borders.Vertical = 0
|
|
|
|
api.UserAgent = fmt.Sprintf("%s/%s %s/%s", config.Name, "0.1", "arikawa", "v3")
|
|
gateway.DefaultIdentity = gateway.IdentifyProperties{
|
|
OS: runtime.GOOS,
|
|
Browser: config.Name,
|
|
Device: "",
|
|
}
|
|
|
|
flag.StringVar(&flagToken, "token", "", "The authentication token.")
|
|
flag.StringVar(&flagConfig, "config", config.DefaultConfigPath(), "The path to the configuration file.")
|
|
flag.StringVar(&flagLog, "log", config.DefaultLogPath(), "The path to the log file.")
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if flagLog != "" {
|
|
// Set the standard logger output to the provided log file.
|
|
f, err := os.OpenFile(flagLog, os.O_CREATE|os.O_WRONLY, 0666)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
log.SetOutput(f)
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
}
|
|
|
|
cfg := config.New()
|
|
err := cfg.Load(flagConfig)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var token string
|
|
if flagToken != "" {
|
|
token = flagToken
|
|
go keyring.Set(config.Name, "token", token)
|
|
} else {
|
|
token, err = keyring.Get(config.Name, "token")
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
c := ui.NewCore(cfg)
|
|
if token != "" {
|
|
err = c.Run(token)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
c.Draw()
|
|
} else {
|
|
loginView := ui.NewLoginView(c)
|
|
c.App.SetRoot(loginView, true)
|
|
}
|
|
|
|
err = c.App.Run()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|