98 lines
2.0 KiB
Go
98 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"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"
|
|
)
|
|
|
|
func main() {
|
|
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
|
|
}
|
|
}
|
|
|
|
state.Setup()
|
|
commands.Setup()
|
|
|
|
// TODO: user account support
|
|
client, err := discordgo.New("Bot " + token)
|
|
if err != nil {
|
|
fmt.Println("% Failed to create client:", err)
|
|
os.Exit(1)
|
|
return
|
|
}
|
|
|
|
// TODO: dont set for user accounts(? never really tested if it matters)
|
|
client.Identify.Intents = discordgo.IntentsAll
|
|
|
|
client.AddHandlerOnce(events.Ready)
|
|
client.AddHandler(events.MessageCreate)
|
|
|
|
err = client.Open()
|
|
if err != nil {
|
|
fmt.Println("% Failed to connect to Discord:", err)
|
|
os.Exit(1)
|
|
return
|
|
}
|
|
|
|
fmt.Println("COMcord (c)left 2023")
|
|
fmt.Println("Type 'h' for Commands")
|
|
|
|
keyboard.Listen(func(key keys.Key) (stop bool, err error) {
|
|
if !state.IsInPrompt() {
|
|
if key.Code == keys.CtrlC {
|
|
client.Close()
|
|
os.Exit(0)
|
|
return true, nil
|
|
} else {
|
|
command, has := commands.GetCommand(key.String())
|
|
if has {
|
|
command.Run(client)
|
|
} else {
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
return false, nil
|
|
})
|
|
}
|