all repos — disgord @ 8cb81ef762818bdc9e468ba1f1e5e18f7565aa95

A simple Discord bot in Go.

initial commit
Marco Andronaco andronacomarco@gmail.com
Mon, 30 Sep 2024 12:04:15 +0200
commit

8cb81ef762818bdc9e468ba1f1e5e18f7565aa95

8 files changed, 360 insertions(+), 0 deletions(-)

jump to
A .env.example 
@@ -0,0 +1,2 @@
+APP_ID=
+AUTH_TOKEN=
A .gitignore 
@@ -0,0 +1,1 @@
+.env
A .vscode/launch.json 
@@ -0,0 +1,15 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Launch Package",
+            "type": "go",
+            "request": "launch",
+            "mode": "auto",
+            "program": "${fileDirname}"
+        },
+    ]
+}
A constants.go 
@@ -0,0 +1,17 @@
+package main
+
+import (
+	"os"
+
+	"github.com/BiRabittoh/disgord/mylog"
+	"github.com/kkdai/youtube/v2"
+)
+
+var (
+	discordToken string
+	appID        string
+	prefix       string = "$"
+
+	logger = mylog.NewLogger(os.Stdout, "main", mylog.DEBUG)
+	yt     = youtube.Client{}
+)
A go.mod 
@@ -0,0 +1,21 @@
+module github.com/BiRabittoh/disgord
+
+go 1.23.1
+
+require (
+	github.com/bwmarrin/discordgo v0.28.1
+	github.com/joho/godotenv v1.5.1
+)
+
+require (
+	github.com/bitly/go-simplejson v0.5.1 // indirect
+	github.com/dlclark/regexp2 v1.11.0 // indirect
+	github.com/dop251/goja v0.0.0-20240220182346-e401ed450204 // indirect
+	github.com/go-sourcemap/sourcemap v2.1.3+incompatible // indirect
+	github.com/google/pprof v0.0.0-20240227163752-401108e1b7e7 // indirect
+	github.com/gorilla/websocket v1.4.2 // indirect
+	github.com/kkdai/youtube/v2 v2.10.1 // indirect
+	golang.org/x/crypto v0.0.0-20210921155107-089bfa567519 // indirect
+	golang.org/x/sys v0.18.0 // indirect
+	golang.org/x/text v0.14.0 // indirect
+)
A go.sum 
@@ -0,0 +1,75 @@
+github.com/bitly/go-simplejson v0.5.1 h1:xgwPbetQScXt1gh9BmoJ6j9JMr3TElvuIyjR8pgdoow=
+github.com/bitly/go-simplejson v0.5.1/go.mod h1:YOPVLzCfwK14b4Sff3oP1AmGhI9T9Vsg84etUnlyp+Q=
+github.com/bwmarrin/discordgo v0.28.1 h1:gXsuo2GBO7NbR6uqmrrBDplPUx2T3nzu775q/Rd1aG4=
+github.com/bwmarrin/discordgo v0.28.1/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
+github.com/chzyer/logex v1.2.0/go.mod h1:9+9sk7u7pGNWYMkh0hdiL++6OeibzJccyQU4p4MedaY=
+github.com/chzyer/readline v1.5.0/go.mod h1:x22KAscuvRqlLoK9CsoYsmxoXZMMFVyOl86cAH8qUic=
+github.com/chzyer/test v0.0.0-20210722231415-061457976a23/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
+github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
+github.com/dlclark/regexp2 v1.4.1-0.20201116162257-a2a8dda75c91/go.mod h1:2pZnwuY/m+8K6iRw6wQdMtk+rH5tNGR1i55kozfMjCc=
+github.com/dlclark/regexp2 v1.7.0/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
+github.com/dlclark/regexp2 v1.11.0 h1:G/nrcoOa7ZXlpoa/91N3X7mM3r8eIlMBBJZvsz/mxKI=
+github.com/dlclark/regexp2 v1.11.0/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
+github.com/dop251/goja v0.0.0-20211022113120-dc8c55024d06/go.mod h1:R9ET47fwRVRPZnOGvHxxhuZcbrMCuiqOz3Rlrh4KSnk=
+github.com/dop251/goja v0.0.0-20240220182346-e401ed450204 h1:O7I1iuzEA7SG+dK8ocOBSlYAA9jBUmCYl/Qa7ey7JAM=
+github.com/dop251/goja v0.0.0-20240220182346-e401ed450204/go.mod h1:QMWlm50DNe14hD7t24KEqZuUdC9sOTy8W6XbCU1mlw4=
+github.com/dop251/goja_nodejs v0.0.0-20210225215109-d91c329300e7/go.mod h1:hn7BA7c8pLvoGndExHudxTDKZ84Pyvv+90pbBjbTz0Y=
+github.com/dop251/goja_nodejs v0.0.0-20211022123610-8dd9abb0616d/go.mod h1:DngW8aVqWbuLRMHItjPUyqdj+HWPvnQe8V8y1nDpIbM=
+github.com/go-sourcemap/sourcemap v2.1.3+incompatible h1:W1iEw64niKVGogNgBN3ePyLFfuisuzeidWPMPWmECqU=
+github.com/go-sourcemap/sourcemap v2.1.3+incompatible/go.mod h1:F8jJfvm2KbVjc5NqelyYJmf/v5J0dwNLS2mL4sNA1Jg=
+github.com/google/pprof v0.0.0-20230207041349-798e818bf904/go.mod h1:uglQLonpP8qtYCYyzA+8c/9qtqgA3qsXGYqCPKARAFg=
+github.com/google/pprof v0.0.0-20240227163752-401108e1b7e7 h1:y3N7Bm7Y9/CtpiVkw/ZWj6lSlDF3F74SfKwfTCer72Q=
+github.com/google/pprof v0.0.0-20240227163752-401108e1b7e7/go.mod h1:czg5+yv1E0ZGTi6S6vVK1mke0fV+FaUhNGcd6VRS9Ik=
+github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
+github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
+github.com/ianlancetaylor/demangle v0.0.0-20220319035150-800ac71e25c2/go.mod h1:aYm2/VgdVmcIU8iMfdMvDMsRAQjcfZSKFby6HOFvi/w=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
+github.com/kkdai/youtube/v2 v2.10.1 h1:jdPho4R7VxWoRi9Wx4ULMq4+hlzSVOXxh4Zh83f2F9M=
+github.com/kkdai/youtube/v2 v2.10.1/go.mod h1:qL8JZv7Q1IoDs4nnaL51o/hmITXEIvyCIXopB0oqgVM=
+github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
+github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
+github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
+github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
+github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=
+github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg=
+golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=
+golang.org/x/crypto v0.0.0-20210921155107-089bfa567519 h1:7I4JAnoQBe7ZtJcBaYHi5UtiO8tQHbUSXxL+pnGRANg=
+golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
+golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
+golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
+golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 h1:nxC68pudNYkKU6jWhgrqdreuFiOQWj1Fs7T3VrH4Pjw=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220310020820-b874c991c1a5/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
+golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
+golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
+golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
+golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
+golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
+golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
+gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
A main.go 
@@ -0,0 +1,89 @@
+package main
+
+import (
+	"os"
+	"os/signal"
+	"strings"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/joho/godotenv"
+)
+
+func parseUserMessage(messageContent string) (command string, args []string, ok bool) {
+	after, found := strings.CutPrefix(messageContent, prefix)
+	if !found {
+		return
+	}
+
+	userInput := strings.Split(after, " ")
+	if len(userInput) == 0 {
+		return
+	}
+
+	return userInput[0], userInput[1:], true
+}
+
+func messageHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
+	logger.Debug("got a message: " + m.Content)
+	command, args, ok := parseUserMessage(m.Content)
+	if !ok {
+		return
+	}
+
+	var response string
+	switch command {
+	case "e", "echo":
+		response = strings.Join(args, " ")
+	default:
+		response = "unknown command: " + command
+	}
+
+	_, err := s.ChannelMessageSend(m.ChannelID, response)
+	if err != nil {
+		logger.Errorf("could not send message: %s", err)
+	}
+}
+
+func readyHandler(s *discordgo.Session, r *discordgo.Ready) {
+	logger.Infof("Logged in as %s", r.User.String())
+}
+
+func main() {
+	err := godotenv.Load()
+	if err != nil {
+		logger.Errorf("Error loading .env file: %s", err)
+	}
+
+	appID = os.Getenv("APP_ID")
+	if appID == "" {
+		logger.Fatal("Could not find Discord App ID.")
+	}
+
+	discordToken = os.Getenv("AUTH_TOKEN")
+	if discordToken == "" {
+		logger.Fatal("Could not find Discord bot token.")
+	}
+
+	session, err := discordgo.New("Bot " + discordToken)
+	if err != nil {
+		logger.Fatalf("could not create bot session: %s", err)
+	}
+
+	session.AddHandler(messageHandler)
+	session.AddHandler(readyHandler)
+
+	err = session.Open()
+	if err != nil {
+		logger.Fatalf("could not open session: %s", err)
+	}
+
+	sigch := make(chan os.Signal, 1)
+	signal.Notify(sigch, os.Interrupt)
+	<-sigch
+
+	logger.Info("Stopping gracefully...")
+	err = session.Close()
+	if err != nil {
+		logger.Errorf("could not close session: %s", err)
+	}
+}
A mylog/log.go 
@@ -0,0 +1,140 @@
+package mylog
+
+import (
+	"io"
+	"log"
+)
+
+type Level uint
+
+const (
+	FATAL Level = iota
+	ERROR
+	WARN
+	INFO
+	DEBUG
+)
+
+type Logger struct {
+	logger      *log.Logger
+	fatalLogger *log.Logger
+	errorLogger *log.Logger
+	warnLogger  *log.Logger
+	infoLogger  *log.Logger
+	debugLogger *log.Logger
+	level       Level
+	name        string
+}
+
+func NewLogger(out io.Writer, name string, level Level) (l *Logger) {
+	l = &Logger{
+		level: level,
+
+		logger:      log.New(out, "", 3),
+		fatalLogger: log.New(out, "", 3),
+		errorLogger: log.New(out, "", 3),
+		warnLogger:  log.New(out, "", 3),
+		infoLogger:  log.New(out, "", 3),
+		debugLogger: log.New(out, "", 3),
+	}
+
+	l.SetName(name)
+	return
+
+}
+
+func parsePrefix(name string, level string) string {
+	return level + " " + name + " "
+}
+
+func (l Logger) GetLevel() Level {
+	return l.level
+}
+
+func (l *Logger) SetLevel(lvl Level) {
+	l.level = lvl
+}
+
+func (l *Logger) SetName(name string) {
+	l.name = name
+	l.logger.SetPrefix(parsePrefix(name, "LOG  "))
+	l.fatalLogger.SetPrefix(parsePrefix(name, "FATAL"))
+	l.errorLogger.SetPrefix(parsePrefix(name, "ERROR"))
+	l.warnLogger.SetPrefix(parsePrefix(name, "WARN "))
+	l.infoLogger.SetPrefix(parsePrefix(name, "INFO "))
+	l.debugLogger.SetPrefix(parsePrefix(name, "DEBUG"))
+}
+
+// log functions
+
+func (l Logger) Printf(format string, v ...any) {
+	l.logger.Printf(format, v...)
+}
+
+func (l Logger) Println(v ...any) {
+	l.logger.Println(v...)
+}
+
+func (l Logger) Fatal(v ...any) {
+	l.fatalLogger.Fatal(v...)
+}
+
+func (l Logger) Fatalf(format string, v ...any) {
+	l.fatalLogger.Fatalf(format, v...)
+}
+
+func (l Logger) Error(v ...any) {
+	if l.level < ERROR {
+		return
+	}
+	l.errorLogger.Println(v...)
+}
+
+func (l Logger) Errorf(format string, v ...any) {
+	if l.level < ERROR {
+		return
+	}
+	l.errorLogger.Printf(format, v...)
+}
+
+func (l Logger) Warn(v ...any) {
+	if l.level < WARN {
+		return
+	}
+	l.warnLogger.Println(v...)
+}
+
+func (l Logger) Warnf(format string, v ...any) {
+	if l.level < WARN {
+		return
+	}
+	l.warnLogger.Printf(format, v...)
+}
+
+func (l Logger) Info(v ...any) {
+	if l.level < INFO {
+		return
+	}
+	l.infoLogger.Println(v...)
+}
+
+func (l Logger) Infof(format string, v ...any) {
+	if l.level < INFO {
+		return
+	}
+	l.infoLogger.Printf(format, v...)
+}
+
+func (l Logger) Debug(v ...any) {
+	if l.level < DEBUG {
+		return
+	}
+	l.debugLogger.Println(v...)
+}
+
+func (l Logger) Debugf(format string, v ...any) {
+	if l.level < DEBUG {
+		return
+	}
+	l.debugLogger.Printf(format, v...)
+}