147 lines
3.5 KiB
Go
147 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/alecthomas/kong"
|
|
"github.com/ayntgl/discordo/ui"
|
|
"github.com/gdamore/tcell/v2"
|
|
"github.com/rivo/tview"
|
|
lua "github.com/yuin/gopher-lua"
|
|
"github.com/zalando/go-keyring"
|
|
)
|
|
|
|
const (
|
|
name = "discordo"
|
|
usage = "A lightweight, secure, and feature-rich Discord terminal client"
|
|
)
|
|
|
|
var cli struct {
|
|
Token string `help:"The authentication token."`
|
|
Config string `help:"The path to the configuration directory." type:"path"`
|
|
}
|
|
|
|
func main() {
|
|
kong.Parse(&cli, kong.Name(name), kong.Description(usage), kong.UsageOnError())
|
|
|
|
// If the authentication token is provided via a flag, store it in the default keyring.
|
|
if cli.Token != "" {
|
|
go keyring.Set(name, "token", cli.Token)
|
|
}
|
|
|
|
// Defaults
|
|
if cli.Config == "" {
|
|
path, err := os.UserConfigDir()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
cli.Config = filepath.Join(path, name)
|
|
}
|
|
|
|
if cli.Token == "" {
|
|
cli.Token, _ = keyring.Get(name, "token")
|
|
}
|
|
|
|
c := ui.NewCore(cli.Config)
|
|
if cli.Token != "" {
|
|
err := c.Run(cli.Token)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
c.DrawMainFlex()
|
|
c.Application.SetRoot(c.MainFlex, true)
|
|
c.Application.SetFocus(c.GuildsTree)
|
|
} else {
|
|
loginForm := ui.NewLoginForm(false)
|
|
loginForm.AddButton("Login", func() {
|
|
email := loginForm.GetFormItem(0).(*tview.InputField).GetText()
|
|
password := loginForm.GetFormItem(1).(*tview.InputField).GetText()
|
|
if email == "" || password == "" {
|
|
return
|
|
}
|
|
|
|
// Login using the email and password only
|
|
lr, err := c.State.Login(email, password)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if lr.Token != "" && !lr.MFA {
|
|
err = c.Run(lr.Token)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
c.DrawMainFlex()
|
|
c.Application.SetRoot(c.MainFlex, true)
|
|
c.Application.SetFocus(c.GuildsTree)
|
|
|
|
go keyring.Set(name, "token", lr.Token)
|
|
} else {
|
|
// The account has MFA enabled, reattempt login with MFA code and ticket.
|
|
mfaLoginForm := ui.NewLoginForm(true)
|
|
mfaLoginForm.AddButton("Login", func() {
|
|
code := mfaLoginForm.GetFormItem(0).(*tview.InputField).GetText()
|
|
if code == "" {
|
|
return
|
|
}
|
|
|
|
lr, err = c.State.TOTP(code, lr.Ticket)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = c.Run(lr.Token)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
c.DrawMainFlex()
|
|
c.Application.SetRoot(c.MainFlex, true)
|
|
c.Application.SetFocus(c.GuildsTree)
|
|
|
|
go keyring.Set(name, "token", lr.Token)
|
|
})
|
|
|
|
c.Application.SetRoot(mfaLoginForm, true)
|
|
}
|
|
})
|
|
|
|
c.Application.SetRoot(loginForm, true)
|
|
}
|
|
|
|
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
|
|
|
|
themeTable, ok := c.Config.State.GetGlobal("theme").(*lua.LTable)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
background := themeTable.RawGetString("background")
|
|
border := themeTable.RawGetString("border")
|
|
title := themeTable.RawGetString("title")
|
|
|
|
tview.Styles.PrimitiveBackgroundColor = tcell.GetColor(lua.LVAsString(background))
|
|
tview.Styles.BorderColor = tcell.GetColor(lua.LVAsString(border))
|
|
tview.Styles.TitleColor = tcell.GetColor(lua.LVAsString(title))
|
|
|
|
err := c.Application.Run()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|