package main import ( "fmt" "os" "runtime" "strings" "atomicgo.dev/keyboard" "atomicgo.dev/keyboard/keys" "github.com/Cynosphere/comcord/commands" "github.com/Cynosphere/comcord/events" "github.com/Cynosphere/comcord/rcfile" "github.com/Cynosphere/comcord/state" "github.com/bwmarrin/discordgo" "golang.org/x/term" ) func main() { oldState, err := term.MakeRaw(int(os.Stdin.Fd())) if err != nil { panic(err) } defer term.Restore(int(os.Stdin.Fd()), oldState) var config map[string]string = make(map[string]string) var token string homeDir, homeErr := os.UserHomeDir() if homeErr != nil { panic(homeErr) } RCPATH := rcfile.GetPath() _, rcErr := os.Stat(RCPATH) if !os.IsNotExist(rcErr) { fmt.Printf("%% Reading %s ...\n", strings.Replace(RCPATH, homeDir, "~", 1)) config = rcfile.Load() } if len(os.Args) > 1 { token = os.Args[1] if os.IsNotExist(rcErr) { fmt.Println("% Writing token to ~/.comcordrc") config["token"] = token rcfile.Save(config) } } else { configToken, tokenInConfig := config["token"] if tokenInConfig { token = configToken } else { fmt.Println("No token provided.") os.Exit(1) return } } fmt.Println("\rCOMcord (c)left 2023") fmt.Println("\rType 'h' for Commands") fmt.Print("\r") state.Setup(config) commands.Setup() // TODO: user account support client, err := discordgo.New("Bot " + token) if err != nil { fmt.Println("% Failed to create client:", err) fmt.Print("\r") os.Exit(1) return } // TODO: dont set for user accounts(? never really tested if it matters) client.Identify.Intents = discordgo.IntentsAll if config["useMobile"] == "true" { client.Identify.Properties = discordgo.IdentifyProperties{ OS: "Android", Browser: "Discord Android", Device: "Pixel, raven", } } else { // TODO: user account support client.Identify.Properties = discordgo.IdentifyProperties{ OS: runtime.GOOS, Browser: "comcord", Device: "comcord", } } events.Setup(client) err = client.Open() if err != nil { fmt.Println("% Failed to connect to Discord:", err) fmt.Print("\r") os.Exit(1) return } keyboard.Listen(func(key keys.Key) (stop bool, err error) { if !state.IsInPrompt() { if key.Code == keys.CtrlC { commands.QuitCommand(client) return true, nil } else { command, has := commands.GetCommand(key.String()) if has { command.Run(client) } else { commands.SendMode(client) } } } return false, nil }) /*sc := make(chan os.Signal, 1) signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt) <-sc client.Close()*/ }