|
|
@@ -1,87 +1,75 @@
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
- "flag"
|
|
|
+ "os"
|
|
|
|
|
|
- "github.com/ayntgl/astatine"
|
|
|
"github.com/ayntgl/discordo/config"
|
|
|
"github.com/ayntgl/discordo/ui"
|
|
|
"github.com/gdamore/tcell/v2"
|
|
|
"github.com/rivo/tview"
|
|
|
+ "github.com/urfave/cli/v2"
|
|
|
"github.com/zalando/go-keyring"
|
|
|
)
|
|
|
|
|
|
-const name = "discordo"
|
|
|
+const (
|
|
|
+ name = "discordo"
|
|
|
+ usage = "A lightweight, secure, and feature-rich Discord terminal client"
|
|
|
+)
|
|
|
|
|
|
func main() {
|
|
|
- var token string
|
|
|
- var cfg string
|
|
|
- flag.StringVar(&token, "token", "", "The authentication token.")
|
|
|
- flag.StringVar(&cfg, "config", "", "The path of the configuration file.")
|
|
|
- flag.Parse()
|
|
|
-
|
|
|
- if token == "" {
|
|
|
- token, _ = keyring.Get(name, "token")
|
|
|
- }
|
|
|
-
|
|
|
- c := config.New()
|
|
|
- if cfg != "" {
|
|
|
- c.Path = cfg
|
|
|
+ t, _ := keyring.Get(name, "token")
|
|
|
+
|
|
|
+ cliApp := &cli.App{
|
|
|
+ Name: name,
|
|
|
+ Usage: usage,
|
|
|
+ EnableBashCompletion: true,
|
|
|
+ Flags: []cli.Flag{
|
|
|
+ &cli.StringFlag{
|
|
|
+ Name: "token",
|
|
|
+ Usage: "The client authentication token.",
|
|
|
+ Value: t,
|
|
|
+ DefaultText: "From keyring",
|
|
|
+ Aliases: []string{"t"},
|
|
|
+ },
|
|
|
+ &cli.StringFlag{
|
|
|
+ Name: "config",
|
|
|
+ Usage: "The path of the configuration file.",
|
|
|
+ Value: config.DefaultPath(),
|
|
|
+ Aliases: []string{"c"},
|
|
|
+ },
|
|
|
+ },
|
|
|
}
|
|
|
- c.Load()
|
|
|
-
|
|
|
- app := ui.NewApp(c)
|
|
|
- if token != "" {
|
|
|
- app.Session = astatine.New(token)
|
|
|
- err := app.Connect()
|
|
|
- if err != nil {
|
|
|
- panic(err)
|
|
|
- }
|
|
|
|
|
|
- app.DrawMainFlex()
|
|
|
- app.SetFocus(app.GuildsList)
|
|
|
- } else {
|
|
|
- app.Session = astatine.New(token)
|
|
|
-
|
|
|
- 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
|
|
|
- }
|
|
|
+ cliApp.Action = func(ctx *cli.Context) error {
|
|
|
+ c := config.New()
|
|
|
+ c.Load(ctx.String("config"))
|
|
|
|
|
|
- // Login using the email and password
|
|
|
- lr, err := app.Session.Login(email, password)
|
|
|
+ token := ctx.String("token")
|
|
|
+ app := ui.NewApp(token, c)
|
|
|
+ if token != "" {
|
|
|
+ err := app.Connect()
|
|
|
if err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
|
|
|
- if lr.Token != "" && !lr.Mfa {
|
|
|
- app.Session.Identify.Token = lr.Token
|
|
|
- err = app.Connect()
|
|
|
+ app.DrawMainFlex()
|
|
|
+ app.SetFocus(app.GuildsList)
|
|
|
+ } 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
|
|
|
+ lr, err := app.Session.Login(email, password)
|
|
|
if err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
|
|
|
- app.DrawMainFlex()
|
|
|
- app.SetFocus(app.GuildsList)
|
|
|
-
|
|
|
- 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 := loginForm.GetFormItem(0).(*tview.InputField).GetText()
|
|
|
- if code == "" {
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- lr, err = app.Session.Totp(code, lr.Ticket)
|
|
|
- if err != nil {
|
|
|
- panic(err)
|
|
|
- }
|
|
|
-
|
|
|
+ if lr.Token != "" && !lr.Mfa {
|
|
|
app.Session.Identify.Token = lr.Token
|
|
|
err = app.Connect()
|
|
|
if err != nil {
|
|
|
@@ -92,31 +80,63 @@ func main() {
|
|
|
app.SetFocus(app.GuildsList)
|
|
|
|
|
|
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 := loginForm.GetFormItem(0).(*tview.InputField).GetText()
|
|
|
+ if code == "" {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ lr, err = app.Session.Totp(code, lr.Ticket)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ app.Session.Identify.Token = lr.Token
|
|
|
+ err = app.Connect()
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ app.DrawMainFlex()
|
|
|
+ app.SetFocus(app.GuildsList)
|
|
|
+
|
|
|
+ go keyring.Set(name, "token", lr.Token)
|
|
|
+ })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ app.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
|
|
|
+
|
|
|
+ tview.Styles.PrimitiveBackgroundColor = tcell.GetColor(app.Config.Theme.Background)
|
|
|
+ tview.Styles.BorderColor = tcell.GetColor(app.Config.Theme.Border)
|
|
|
+ tview.Styles.TitleColor = tcell.GetColor(app.Config.Theme.Title)
|
|
|
+
|
|
|
+ err := app.Run()
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
|
|
|
- app.SetRoot(loginForm, true)
|
|
|
+ return nil
|
|
|
}
|
|
|
|
|
|
- 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
|
|
|
-
|
|
|
- tview.Styles.PrimitiveBackgroundColor = tcell.GetColor(app.Config.Theme.Background)
|
|
|
- tview.Styles.BorderColor = tcell.GetColor(app.Config.Theme.Border)
|
|
|
- tview.Styles.TitleColor = tcell.GetColor(app.Config.Theme.Title)
|
|
|
-
|
|
|
- err := app.Run()
|
|
|
+ err := cliApp.Run(os.Args)
|
|
|
if err != nil {
|
|
|
panic(err)
|
|
|
}
|
ayntgl