discordo-plus/main.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)
	}
}