From 8e6ed1e9cecfc4e1d5067e0121707f5ef182f055 Mon Sep 17 00:00:00 2001 From: Jack Merrill Date: Fri, 18 Aug 2023 22:22:20 -0500 Subject: [PATCH] initial commit :tada: --- .gitignore | 3 + .vscode/hampbot-snippets.code-snippets | 62 +++++++ .vscode/launch.json | 20 +++ go.mod | 37 +++++ go.sum | 195 +++++++++++++++++++++++ internal/commands/fun/ai.go | 100 ++++++++++++ internal/commands/studentlife/laundry.go | 122 ++++++++++++++ internal/commands/util/ping.go | 79 +++++++++ internal/commands/util/steal.go | 98 ++++++++++++ internal/utils/config/config.go | 47 ++++++ internal/utils/embed/embed.go | 103 ++++++++++++ internal/utils/embed/prebuild_embeds.go | 24 +++ main.go | 67 ++++++++ 13 files changed, 957 insertions(+) create mode 100644 .gitignore create mode 100644 .vscode/hampbot-snippets.code-snippets create mode 100644 .vscode/launch.json create mode 100644 go.mod create mode 100644 go.sum create mode 100644 internal/commands/fun/ai.go create mode 100644 internal/commands/studentlife/laundry.go create mode 100644 internal/commands/util/ping.go create mode 100644 internal/commands/util/steal.go create mode 100644 internal/utils/config/config.go create mode 100644 internal/utils/embed/embed.go create mode 100644 internal/utils/embed/prebuild_embeds.go create mode 100644 main.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..17f7671 --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +.env +__debug_bin.* +__debug_bin diff --git a/.vscode/hampbot-snippets.code-snippets b/.vscode/hampbot-snippets.code-snippets new file mode 100644 index 0000000..28216bb --- /dev/null +++ b/.vscode/hampbot-snippets.code-snippets @@ -0,0 +1,62 @@ +{ + // Place your hampbot workspace snippets here. Each snippet is defined under a snippet name and has a scope, prefix, body and + // description. Add comma separated ids of the languages where the snippet is applicable in the scope field. If scope + // is left empty or omitted, the snippet gets applied to all languages. The prefix is what is + // used to trigger the snippet and the body will be expanded and inserted. Possible variables are: + // $1, $2 for tab stops, $0 for the final cursor position, and ${1:label}, ${2:another} for placeholders. + // Placeholders with the same ids are connected. + // Example: + // "Print to console": { + // "scope": "javascript,typescript", + // "prefix": "log", + // "body": [ + // "console.log('$1');", + // "$2" + // ], + // "description": "Log output to console" + // } + "Command": { + "prefix": "command", + "body": [ + "import (", + " \"github.com/jackmerrill/hampbot/internal/utils/config\"", + " \"github.com/zekroTJA/shireikan\"", + ")", + "", + "type ${1:CommandName} struct {", + "}", + "", + "func (c *${1:CommandName}) GetInvokes() []string {", + " return []string{\"${1}\"}", + "}", + "", + "func (c *${1:CommandName}) GetDescription() string {", + " return \"${1:CommandName} description\"", + "}", + "", + "func (c *${1:CommandName}) GetHelp() string {", + " return \"`${1}` - `${1}`\"", + "}", + "", + "func (c *${1:CommandName}) GetGroup() string {", + " return config.GroupUtil", + "}", + "", + "func (c *${1:CommandName}) GetDomainName() string {", + " return \"hamp.util.${1}\"", + "}", + "", + "func (c *${1:CommandName}) GetSubPermissionRules() []shireikan.SubPermission {", + " return nil", + "}", + "func (c *${1:CommandName}) IsExecutableInDMChannels() bool {", + " return true", + "}", + "", + "func (c *${1:CommandName}) Exec(ctx shireikan.Context) error {", + " return nil", + "}" + ], + "description": "Create a new command" + } +} diff --git a/.vscode/launch.json b/.vscode/launch.json new file mode 100644 index 0000000..ba3d9aa --- /dev/null +++ b/.vscode/launch.json @@ -0,0 +1,20 @@ +{ + // 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": "main.go", + "envFile": "${workspaceFolder}/.env", + "args": [ + "-ldflags", + "\"-X github.com/jackmerrill/hampbot/internal/utils/config.Version=dev -X github.com/jackmerrill/hampbot/internal/utils/config.Build=$(date -u +.%Y%m%d.%H%M%S)\"" + ] + } + ] +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..0d5d522 --- /dev/null +++ b/go.mod @@ -0,0 +1,37 @@ +module github.com/jackmerrill/hampbot + +go 1.19 + +require ( + github.com/PuerkitoBio/goquery v1.8.1 // indirect + github.com/andybalholm/cascadia v1.3.1 // indirect + github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect + github.com/bwmarrin/discordgo v0.27.1 // indirect + github.com/cespare/xxhash/v2 v2.1.1 // indirect + github.com/charmbracelet/lipgloss v0.7.1 // indirect + github.com/charmbracelet/log v0.2.3 // indirect + github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect + github.com/go-logfmt/logfmt v0.6.0 // indirect + github.com/go-redis/redis/v8 v8.11.0 // indirect + github.com/gorilla/websocket v1.5.0 // indirect + github.com/jackmerrill/hamp-api v0.0.0-20230818235104-8d222c9674c9 // indirect + github.com/labstack/echo/v4 v4.9.1 // indirect + github.com/labstack/gommon v0.4.0 // indirect + github.com/lucasb-eyer/go-colorful v1.2.0 // indirect + github.com/mattn/go-colorable v0.1.11 // indirect + github.com/mattn/go-isatty v0.0.18 // indirect + github.com/mattn/go-runewidth v0.0.14 // indirect + github.com/muesli/reflow v0.3.0 // indirect + github.com/muesli/termenv v0.15.2 // indirect + github.com/rivo/uniseg v0.2.0 // indirect + github.com/sarulabs/di/v2 v2.4.2 // indirect + github.com/sashabaranov/go-openai v1.14.2 // indirect + github.com/valyala/bytebufferpool v1.0.0 // indirect + github.com/valyala/fasttemplate v1.2.1 // indirect + github.com/zekroTJA/shireikan v0.7.0 // indirect + github.com/zekrotja/dgrs v0.3.1 // indirect + golang.org/x/crypto v0.12.0 // indirect + golang.org/x/net v0.10.0 // indirect + golang.org/x/sys v0.11.0 // indirect + golang.org/x/text v0.12.0 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..8d2929e --- /dev/null +++ b/go.sum @@ -0,0 +1,195 @@ +github.com/PuerkitoBio/goquery v1.8.1 h1:uQxhNlArOIdbrH1tr0UXwdVFgDcZDrZVdcpygAcwmWM= +github.com/PuerkitoBio/goquery v1.8.1/go.mod h1:Q8ICL1kNUJ2sXGoAhPGUdYDJvgQgHzJsnnd3H7Ho5jQ= +github.com/andybalholm/cascadia v1.3.1 h1:nhxRkql1kdYCc8Snf7D5/D3spOX+dBgjA6u8x004T2c= +github.com/andybalholm/cascadia v1.3.1/go.mod h1:R4bJ1UQfqADjvDa4P6HZHLh/3OxWWEqc0Sk8XGwHqvA= +github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k= +github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8= +github.com/bwmarrin/discordgo v0.23.2 h1:BzrtTktixGHIu9Tt7dEE6diysEF9HWnXeHuoJEt2fH4= +github.com/bwmarrin/discordgo v0.23.2/go.mod h1:c1WtWUGN6nREDmzIpyTp/iD3VYt4Fpx+bVyfBG7JE+M= +github.com/bwmarrin/discordgo v0.27.1 h1:ib9AIc/dom1E/fSIulrBwnez0CToJE113ZGt4HoliGY= +github.com/bwmarrin/discordgo v0.27.1/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY= +github.com/cespare/xxhash/v2 v2.1.1 h1:6MnRN8NT7+YBpUIWxHtefFZOKTAPgGjpQSxqLNn0+qY= +github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/charmbracelet/lipgloss v0.7.1 h1:17WMwi7N1b1rVWOjMT+rCh7sQkvDU75B2hbZpc5Kc1E= +github.com/charmbracelet/lipgloss v0.7.1/go.mod h1:yG0k3giv8Qj8edTCbbg6AlQ5e8KNWpFujkNawKNhE2c= +github.com/charmbracelet/log v0.2.3 h1:YVmBhJtpGL7nW/nlf5u+SEloU8XYljxozGzZpgwIvhs= +github.com/charmbracelet/log v0.2.3/go.mod h1:ZApwwzDbbETVTIRTk7724yQRJAXIktt98yGVMMaa3y8= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= +github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ= +github.com/go-logfmt/logfmt v0.6.0 h1:wGYYu3uicYdqXVgoYbvnkrPVXkuLM1p1ifugDMEdRi4= +github.com/go-logfmt/logfmt v0.6.0/go.mod h1:WYhtIu8zTZfxdn5+rREduYbwxfcBr/Vr6KEVveWlfTs= +github.com/go-redis/redis/v8 v8.10.0/go.mod h1:vXLTvigok0VtUX0znvbcEW1SOt4OA9CU1ZfnOtKOaiM= +github.com/go-redis/redis/v8 v8.11.0 h1:O1Td0mQ8UFChQ3N9zFQqo6kTU2cJ+/it88gDB+zg0wo= +github.com/go-redis/redis/v8 v8.11.0/go.mod h1:DLomh7y2e3ggQXQLd1YgmvIfecPJoFl7WU5SOQ/r06M= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= +github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= +github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= +github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= +github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= +github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/gorilla/websocket v1.4.0/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ= +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/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc= +github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= +github.com/jackmerrill/hamp-api v0.0.0-20230818235104-8d222c9674c9 h1:ZP4cMycNDHXMk5LQ+VbDq7So0e7gbYPetPpvqbD2XLo= +github.com/jackmerrill/hamp-api v0.0.0-20230818235104-8d222c9674c9/go.mod h1:KS8EBP1su7OHA4Kq5+wZkx162SEVDbIbUV9yo0MMRac= +github.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +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/labstack/echo/v4 v4.9.1 h1:GliPYSpzGKlyOhqIbG8nmHBo3i1saKWFOgh41AN3b+Y= +github.com/labstack/echo/v4 v4.9.1/go.mod h1:Pop5HLc+xoc4qhTZ1ip6C0RtP7Z+4VzRLWZZFKqbbjo= +github.com/labstack/gommon v0.4.0 h1:y7cvthEAEbU0yHOf4axH8ZG2NH8knB9iNSoTO8dyIk8= +github.com/labstack/gommon v0.4.0/go.mod h1:uW6kP17uPlLJsD3ijUYn3/M5bAxtlZhMI6m3MFxTMTM= +github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY= +github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0= +github.com/mattn/go-colorable v0.1.11 h1:nQ+aFkoE2TMGc0b68U2OKSexC+eq46+XwZzWXHRmPYs= +github.com/mattn/go-colorable v0.1.11/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4= +github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94= +github.com/mattn/go-isatty v0.0.18 h1:DOKFKCQ7FNG2L1rbrmstDN4QVRdS89Nkh85u68Uwp98= +github.com/mattn/go-isatty v0.0.18/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/mattn/go-runewidth v0.0.12/go.mod h1:RAqKPSqVFrSLVXbA8x7dzmKdmGzieGRCM46jaSJTDAk= +github.com/mattn/go-runewidth v0.0.14 h1:+xnbZSEeDbOIg5/mE6JF0w6n9duR1l3/WmbinWVwUuU= +github.com/mattn/go-runewidth v0.0.14/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= +github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s= +github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8= +github.com/muesli/termenv v0.15.2 h1:GohcuySI0QmI3wN8Ok9PtKGkgkFIk7y6Vpb5PvrY+Wo= +github.com/muesli/termenv v0.15.2/go.mod h1:Epx+iuz8sNs7mNKhxzH4fWXGNpZwUaJKRS1noLXviQ8= +github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A= +github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/ginkgo v1.12.1/go.mod h1:zj2OWP4+oCPe1qIXoGWkgMRwljMUYCdkwsT2108oapk= +github.com/onsi/ginkgo v1.15.0/go.mod h1:hF8qUzuuC8DJGygJH3726JnCZX4MYbRB8yFfISqnKUg= +github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY= +github.com/onsi/gomega v1.10.1/go.mod h1:iN09h71vgCQne3DLsj+A5owkum+a2tYe+TOCB1ybHNo= +github.com/onsi/gomega v1.10.5/go.mod h1:gza4q3jKQJijlu05nKWRCW/GavJumGt8aNRxWg7mt48= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY= +github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/sarulabs/di/v2 v2.4.2 h1:A/PDVU41gHYeUbZZKco8dOwPAB2rrFfiwWLJrZsi+h8= +github.com/sarulabs/di/v2 v2.4.2/go.mod h1:trZu4KPwNLE623mBIIsljn1LLkNE6ee/Pk24b7yzSf8= +github.com/sashabaranov/go-openai v1.14.2 h1:5DPTtR9JBjKPJS008/A409I5ntFhUPPGCmaAihcPRyo= +github.com/sashabaranov/go-openai v1.14.2/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw= +github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc= +github.com/valyala/fasttemplate v1.2.1 h1:TVEnxayobAdVkhQfrfes2IzOB6o+z4roRkPF52WA1u4= +github.com/valyala/fasttemplate v1.2.1/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ= +github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +github.com/zekroTJA/shireikan v0.7.0 h1:tjsitvaj81Ioeq/EcSIp1aIHkrIH7K5f0Ml4tD8jdIM= +github.com/zekroTJA/shireikan v0.7.0/go.mod h1:W9mFLVTrq+mdfLCRqsg9qg8Dd569FxhAeHva4tEzl5s= +github.com/zekroTJA/timedmap v1.4.0/go.mod h1:Go4uPxMN1Wjl5IgO6HYD1tM9IQhkYEVqcrrdsI4ljXo= +github.com/zekrotja/dgrs v0.3.1 h1:HYKa1wX3WIjUVAj4RhGKhVRvBRhCyofGpHeKyY5W/eg= +github.com/zekrotja/dgrs v0.3.1/go.mod h1:I73ryIMJJXCQxBHyBdRALG+0+mIEXF008EF7A/fYV4I= +go.opentelemetry.io/otel v0.20.0/go.mod h1:Y3ugLH2oa81t5QO+Lty+zXf8zC9L26ax4Nzoxm/dooo= +go.opentelemetry.io/otel v1.0.0-RC1/go.mod h1:x9tRa9HK4hSSq7jf2TKbqFbtt58/TGk0f9XiEYISI1I= +go.opentelemetry.io/otel/internal/metric v0.22.0/go.mod h1:7qVuMihW/ktMonEfOvBXuh6tfMvvEyoIDgeJNRloYbQ= +go.opentelemetry.io/otel/metric v0.20.0/go.mod h1:598I5tYlH1vzBjn+BTuhzTCSb/9debfNp6R3s7Pr1eU= +go.opentelemetry.io/otel/metric v0.22.0/go.mod h1:KcsUkBiYGW003DJ+ugd2aqIRIfjabD9jeOUXqsAtrq0= +go.opentelemetry.io/otel/oteltest v0.20.0/go.mod h1:L7bgKf9ZB7qCwT9Up7i9/pn0PWIa9FqQ2IQ8LoxiGnw= +go.opentelemetry.io/otel/oteltest v1.0.0-RC1/go.mod h1:+eoIG0gdEOaPNftuy1YScLr1Gb4mL/9lpDkZ0JjMRq4= +go.opentelemetry.io/otel/trace v0.20.0/go.mod h1:6GjCW8zgDjwGHGa6GkyeB8+/5vjT16gUEi0Nf1iBdgw= +go.opentelemetry.io/otel/trace v1.0.0-RC1/go.mod h1:86UHmyHWFEtWjfWPSbu0+d0Pf9Q6e1U+3ViBOc+NXAg= +golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4= +golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97 h1:/UOmuWzQfxxo9UtlXMwuQU8CMgg1eZXqTRwkSQJWKOI= +golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= +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/crypto v0.12.0 h1:tFM/ta59kqch6LlvYnPa0yx5a83cL2nHflFhYKvv9Yk= +golang.org/x/crypto v0.12.0/go.mod h1:NF0Gs7EO5K4qLn+Ylc+fih8BSTeIjAP05siRnAh98yw= +golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= +golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200520004742-59133d7f0dd7/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= +golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20201202161906-c7110b5ffcbb/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20210916014120-12bc252f5db8/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g= +golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M= +golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/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-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190904154756-749cb33beabd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191120155948-bd437916bb0e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210112080510-489259a85091/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210423082822-04245dca01da/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-20210630005230-0f9fa26af87c h1:F1jZWGFhYfh0Ci55sIpILtKKK8p3i2/krTr0H1rg74I= +golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20211103235746-7861aae1554b/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.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.7.0 h1:3jlCCIQZPdOYu1h8BkNvLz8Kgwtae2cagcG/VamtZRU= +golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM= +golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +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/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= +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.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc= +golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= +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.0.0-20201224043029-2b0845dc783e/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA= +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= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= +google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= +google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= +google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= +google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= +google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +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/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys= +gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw= +gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/internal/commands/fun/ai.go b/internal/commands/fun/ai.go new file mode 100644 index 0000000..742a980 --- /dev/null +++ b/internal/commands/fun/ai.go @@ -0,0 +1,100 @@ +package fun + +import ( + "context" + "fmt" + "os" + + "github.com/jackmerrill/hampbot/internal/utils/config" + "github.com/jackmerrill/hampbot/internal/utils/embed" + "github.com/sashabaranov/go-openai" + "github.com/zekroTJA/shireikan" +) + +type AI struct { +} + +func (c *AI) GetInvokes() []string { + return []string{"ai", "gpt", "ask", "question"} +} + +func (c *AI) GetDescription() string { + return "Ask AI anything!" +} + +func (c *AI) GetHelp() string { + return "`ai [prompt]` - Ask GPT-3.5 anything." +} + +func (c *AI) GetGroup() string { + return config.GroupFun +} + +func (c *AI) GetDomainName() string { + return "hamp.fun.AI" +} + +func (c *AI) GetSubPermissionRules() []shireikan.SubPermission { + return nil +} +func (c *AI) IsExecutableInDMChannels() bool { + return true +} + +func (c *AI) Exec(ctx shireikan.Context) error { + openaiToken := os.Getenv("OPENAI_TOKEN") + + if openaiToken == "" { + ctx.ReplyEmbed(embed.NewErrorEmbed(ctx).SetDescription("No OpenAI token set.").MessageEmbed) + return fmt.Errorf("no openai token set") + } + + client := openai.NewClient(openaiToken) + + messages := []openai.ChatCompletionMessage{ + { + Role: openai.ChatMessageRoleSystem, + Content: "You are HampBot. You are a bot on Discord that can answer any question. Multiple users may ask you questions at the same time. There will be context given. Don't add any decorations (such as `AI:` or `User:`) to your response, these will be added automatically.", + }, + } + + if ctx.GetChannel().IsThread() { + // get all messages in thread + msgs, err := ctx.GetSession().ChannelMessages(ctx.GetChannel().ID, 100, "", "", "") + if err != nil { + return err + } + + for _, msg := range msgs { + messages = append(messages, openai.ChatCompletionMessage{ + Role: openai.ChatMessageRoleUser, + Content: fmt.Sprintf("%s: %s", msg.Author.Username, msg.Content), + }) + } + } else { + msg := ctx.GetMessage() + + messages = append(messages, openai.ChatCompletionMessage{ + Role: openai.ChatMessageRoleUser, + Content: fmt.Sprintf("%s: %s", msg.Author.Username, msg.Content), + }) + } + + ctx.GetSession().ChannelTyping(ctx.GetChannel().ID) + + // get response from GPT-3.5 + resp, err := client.CreateChatCompletion(context.Background(), openai.ChatCompletionRequest{ + Model: openai.GPT3Dot5Turbo, + Messages: messages, + }) + + if err != nil { + ctx.ReplyEmbed(embed.NewErrorEmbed(ctx).SetDescription("An error occured while asking GPT-3.5.").AddField("Error", err.Error(), false).MessageEmbed) + return err + } + + // send response + ctx.GetSession().ChannelMessageSend(ctx.GetChannel().ID, fmt.Sprintf(":robot: **AI:** %s", resp.Choices[0].Message.Content)) + + return nil +} diff --git a/internal/commands/studentlife/laundry.go b/internal/commands/studentlife/laundry.go new file mode 100644 index 0000000..81480e8 --- /dev/null +++ b/internal/commands/studentlife/laundry.go @@ -0,0 +1,122 @@ +package studentlife + +import ( + "encoding/json" + "fmt" + "net/http" + "time" + + "github.com/bwmarrin/discordgo" + "github.com/jackmerrill/hampbot/internal/utils/config" + "github.com/jackmerrill/hampbot/internal/utils/embed" + "github.com/zekroTJA/shireikan" +) + +type MachineType string + +const ( + Washer MachineType = "Washer" + Dryer MachineType = "Dryer" +) + +type Machine struct { + Name string `json:"name"` + Type MachineType `json:"type"` + Status string `json:"status"` + Time *time.Time `json:"time"` + EstimatedTime *time.Time `json:"estimatedTime"` +} + +type LaundryRoom struct { + ID string `json:"id"` + Name string `json:"name"` + NextUpdate time.Time `json:"nextUpdate"` + LastUpdate time.Time `json:"lastUpdate"` + + Machines []Machine `json:"machines"` + + updateChan chan bool +} + +type Laundry struct { +} + +func (c *Laundry) GetInvokes() []string { + return []string{"laundry"} +} + +func (c *Laundry) GetDescription() string { + return "Get the laundry status of a residence hall" +} + +func (c *Laundry) GetHelp() string { + return "`laundry [dakin/merrill/prescott/enfield]` - Laundry help" +} + +func (c *Laundry) GetGroup() string { + return config.GroupUtil +} + +func (c *Laundry) GetDomainName() string { + return "hamp.util.laundry" +} + +func (c *Laundry) GetSubPermissionRules() []shireikan.SubPermission { + return nil +} +func (c *Laundry) IsExecutableInDMChannels() bool { + return true +} + +func (c *Laundry) Exec(ctx shireikan.Context) error { + building := ctx.GetArgs().Get(0) + + if building == "" { + ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{ + Embed: embed.NewErrorEmbed(ctx).SetDescription("Please specify a building to get the laundry status of.\n\n**Usage:** `laundry [dakin/merrill/prescott/enfield]`").MessageEmbed, + Reference: ctx.GetMessage().Reference(), + }) + return nil + } + + if building != "dakin" && building != "merrill" && building != "prescott" && building != "enfield" { + ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{ + Embed: embed.NewErrorEmbed(ctx).SetDescription("Please specify a valid building to get the laundry status of.\n\n**Usage:** `laundry [dakin/merrill/prescott/enfield]`").MessageEmbed, + Reference: ctx.GetMessage().Reference(), + }) + return nil + } + + res, err := http.Get(fmt.Sprintf("%s/api/utilities/laundry/%s", config.HampAPI, ctx.GetArgs().Get(0))) + + if err != nil { + return err + } + + var room LaundryRoom + + err = json.NewDecoder(res.Body).Decode(&room) + + if err != nil { + return err + } + + embed := embed.NewSuccessEmbed(ctx).SetTitle(fmt.Sprintf("%s Laundry Status :shirt:", room.Name)).SetDescription(fmt.Sprintf("Last update: %s", config.ConvertTimestampToDiscordTimestampWithFormat(room.LastUpdate, "T"))) + + for _, machine := range room.Machines { + if machine.Status == "Available" { + embed.AddField(fmt.Sprintf("%s - %s", machine.Name, machine.Type), fmt.Sprintf(":green_circle: **Status:** %s", machine.Status), false) + } else if machine.Status == "Not online" { + embed.AddField(fmt.Sprintf("%s - %s", machine.Name, machine.Type), fmt.Sprintf(":red_circle: **Status:** %s", machine.Status), false) + } else { + embed.AddField(fmt.Sprintf("%s - %s", machine.Name, machine.Type), fmt.Sprintf(":yellow_circle: **Status:** %s\n:alarm_clock: **Time Remaining:** %s", machine.Status, config.ConvertTimestampToDiscordTimestampWithFormat(*machine.EstimatedTime, "R")), false) + } + } + + ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{ + Embed: embed.MessageEmbed, + Reference: ctx.GetMessage().Reference(), + }) + + return nil +} diff --git a/internal/commands/util/ping.go b/internal/commands/util/ping.go new file mode 100644 index 0000000..229aa89 --- /dev/null +++ b/internal/commands/util/ping.go @@ -0,0 +1,79 @@ +package commands + +import ( + "fmt" + "time" + + "github.com/bwmarrin/discordgo" + "github.com/charmbracelet/log" + "github.com/jackmerrill/hampbot/internal/utils/config" + "github.com/jackmerrill/hampbot/internal/utils/embed" + "github.com/zekroTJA/shireikan" +) + +// Ping is a command responding with a ping +// message in the commands channel. +type Ping struct { +} + +// GetInvoke returns the command invokes. +func (c *Ping) GetInvokes() []string { + return []string{"ping", "p"} +} + +// GetDescription returns the commands description. +func (c *Ping) GetDescription() string { + return "ping pong" +} + +// GetHelp returns the commands help text. +func (c *Ping) GetHelp() string { + return "`ping` - ping" +} + +// GetGroup returns the commands group. +func (c *Ping) GetGroup() string { + return config.GroupUtil +} + +// GetDomainName returns the commands domain name. +func (c *Ping) GetDomainName() string { + return "hamp.util.ping" +} + +// GetSubPermissionRules returns the commands sub +// permissions array. +func (c *Ping) GetSubPermissionRules() []shireikan.SubPermission { + return nil +} + +// IsExecutableInDMChannels returns whether +// the command is executable in DM channels. +func (c *Ping) IsExecutableInDMChannels() bool { + return true +} + +// Exec is the commands execution handler. +func (c *Ping) Exec(ctx shireikan.Context) error { + start := time.Now() + + m, err := ctx.GetSession().ChannelMessageSendComplex(ctx.GetChannel().ID, &discordgo.MessageSend{ + Reference: ctx.GetMessage().Reference(), + Embed: embed.NewWarningEmbed(ctx).SetTitle("Pinging... ").MessageEmbed, + }) + + end := time.Now() + diff := end.Sub(start) + + if err != nil { + log.Error("Failed sending message: ", err) + } + + _, err = ctx.GetSession().ChannelMessageEditComplex(&discordgo.MessageEdit{ + Channel: ctx.GetChannel().ID, + ID: m.ID, + Embed: embed.NewSuccessEmbed(ctx).SetTitle("Pong! :ping_pong:").SetDescription(fmt.Sprintf(":ping_pong: Gateway Ping: `%dms`\n:desktop: API Ping: `%dms`", ctx.GetSession().HeartbeatLatency().Milliseconds(), diff.Milliseconds())).MessageEmbed, + }) + + return err +} diff --git a/internal/commands/util/steal.go b/internal/commands/util/steal.go new file mode 100644 index 0000000..d0c4f8d --- /dev/null +++ b/internal/commands/util/steal.go @@ -0,0 +1,98 @@ +package commands + +import ( + "encoding/base64" + "fmt" + "io/ioutil" + "net/http" + "strings" + + "github.com/bwmarrin/discordgo" + "github.com/jackmerrill/hampbot/internal/utils/config" + "github.com/jackmerrill/hampbot/internal/utils/embed" + "github.com/zekroTJA/shireikan" +) + +type Steal struct { +} + +func (c *Steal) GetInvokes() []string { + return []string{"steal", "emote"} +} + +func (c *Steal) GetDescription() string { + return "Steal an emote from another server" +} + +func (c *Steal) GetHelp() string { + return "`steal [emote(s)]` - steal an emote from another server\n`steal [emote(s)]` - steal an emote (or two) from another server" +} + +func (c *Steal) GetGroup() string { + return config.GroupUtil +} + +func (c *Steal) GetDomainName() string { + return "hamp.util.steal" +} + +func (c *Steal) GetSubPermissionRules() []shireikan.SubPermission { + return nil +} +func (c *Steal) IsExecutableInDMChannels() bool { + return false +} + +func (c *Steal) Exec(ctx shireikan.Context) error { + msg := ctx.GetMessage() + + emotes := msg.GetCustomEmojis() + + if len(emotes) == 0 { + ctx.GetSession().ChannelMessageSendEmbed(msg.ChannelID, embed.NewErrorEmbed(ctx).SetTitle("No emotes found").SetDescription("No emotes were found in your message.").MessageEmbed) + return nil + } + + var newEmojis []string + + for _, emote := range emotes { + // download the emote + res, err := http.Get(fmt.Sprintf("https://cdn.discordapp.com/emojis/%s", emote.ID)) + + if err != nil { + ctx.GetSession().ChannelMessageSendEmbed(msg.ChannelID, embed.NewErrorEmbed(ctx).SetTitle("Error").SetDescription("An error occured while downloading your emote.").AddField("Error", err.Error(), false).MessageEmbed) + return err + } + + defer res.Body.Close() + + data, err := ioutil.ReadAll(res.Body) + + if err != nil { + ctx.GetSession().ChannelMessageSendEmbed(msg.ChannelID, embed.NewErrorEmbed(ctx).SetTitle("Error").SetDescription("An error occured while downloading your emote.").AddField("Error", err.Error(), false).MessageEmbed) + return err + } + + // convert to base64 + base64Emote := base64.StdEncoding.EncodeToString(data) + + emoji := discordgo.EmojiParams{ + Name: emote.Name, + Image: "data:image/png;base64," + base64Emote, + Roles: nil, + } + + _, err = ctx.GetSession().GuildEmojiCreate(config.BotGuild, &emoji) + + if err != nil { + ctx.GetSession().ChannelMessageSendEmbed(msg.ChannelID, embed.NewErrorEmbed(ctx).SetTitle("Error").SetDescription("An error occured while uploading your emote.").AddField("Error", err.Error(), false).MessageEmbed) + return err + } + + newEmojis = append(newEmojis, emote.Name) + } + + ctx.GetSession().ChannelMessageSendEmbed(msg.ChannelID, embed.NewSuccessEmbed(ctx).SetTitle("Emote(s) stolen!").SetDescription(fmt.Sprintf("The following emotes were stolen: `%s`", strings.Join(newEmojis, ", "))).MessageEmbed) + + return nil +} diff --git a/internal/utils/config/config.go b/internal/utils/config/config.go new file mode 100644 index 0000000..3b3b5cd --- /dev/null +++ b/internal/utils/config/config.go @@ -0,0 +1,47 @@ +package config + +import ( + "fmt" + "time" +) + +var ( + Version string + Build string +) + +var ( + // BotPrefix is the prefix used for bot commands. + BotPrefix = ">" + + // BotGuild is the ID of the guild the bot is running on. + BotGuild = "936651575684915201" + + // HampAPI is the URL to the hamp API. + // HampAPI = "https://api.hamp.sh" + HampAPI = "http://localhost:1323" +) + +var ( + GroupUtil = "Util" + GroupFun = "Fun" + GroupInfo = "Info" + GroupMod = "Moderation" + GroupDev = "Dev" +) + +func ConvertTimestampToDiscordTimestamp(t time.Time) string { + // format: where 1234567890 is the unix timestamp + + u := t.Unix() + + return "" +} + +func ConvertTimestampToDiscordTimestampWithFormat(t time.Time, format string) string { + // format: where 1234567890 is the unix timestamp and R is the format + + u := t.Unix() + + return "" +} diff --git a/internal/utils/embed/embed.go b/internal/utils/embed/embed.go new file mode 100644 index 0000000..2f2d85a --- /dev/null +++ b/internal/utils/embed/embed.go @@ -0,0 +1,103 @@ +package embed + +import ( + "fmt" + "time" + + "github.com/bwmarrin/discordgo" +) + +type Embed struct { + *discordgo.MessageEmbed +} + +func NewEmbed() *Embed { + return &Embed{&discordgo.MessageEmbed{}} +} + +func (e *Embed) SetTitle(title string) *Embed { + e.MessageEmbed.Title = title + return e +} + +func (e *Embed) SetDescription(description string) *Embed { + e.MessageEmbed.Description = description + return e +} + +func (e *Embed) SetColor(color int) *Embed { + e.MessageEmbed.Color = color + return e +} + +func (e *Embed) SetColorRGB(r, g, b int) *Embed { + return e.SetColor((r << 16) + (g << 8) + b) +} + +func (e *Embed) SetColorHex(hex string) *Embed { + var color int + fmt.Sscanf(hex, "#%06x", &color) + return e.SetColor(color) +} + +func (e *Embed) SetURL(url string) *Embed { + e.MessageEmbed.URL = url + return e +} + +func (e *Embed) SetTimestamp(timestamp time.Time) *Embed { + e.MessageEmbed.Timestamp = timestamp.Format(time.RFC3339) + return e +} + +func (e *Embed) SetFooter(text, iconURL string) *Embed { + e.MessageEmbed.Footer = &discordgo.MessageEmbedFooter{ + Text: text, + IconURL: iconURL, + } + return e +} + +func (e *Embed) SetImage(url string) *Embed { + e.MessageEmbed.Image = &discordgo.MessageEmbedImage{ + URL: url, + } + return e +} + +func (e *Embed) SetThumbnail(url string) *Embed { + e.MessageEmbed.Thumbnail = &discordgo.MessageEmbedThumbnail{ + URL: url, + } + return e +} + +func (e *Embed) SetAuthor(name, url, iconURL string) *Embed { + e.MessageEmbed.Author = &discordgo.MessageEmbedAuthor{ + Name: name, + URL: url, + IconURL: iconURL, + } + return e +} + +func (e *Embed) AddField(name, value string, inline bool) *Embed { + e.MessageEmbed.Fields = append(e.MessageEmbed.Fields, &discordgo.MessageEmbedField{ + Name: name, + Value: value, + Inline: inline, + }) + return e +} + +func (e *Embed) AddFields(fields ...*discordgo.MessageEmbedField) *Embed { + e.MessageEmbed.Fields = append(e.MessageEmbed.Fields, fields...) + return e +} + +func (e *Embed) AddFieldsFromMap(fields map[string]string, inline bool) *Embed { + for name, value := range fields { + e.AddField(name, value, inline) + } + return e +} diff --git a/internal/utils/embed/prebuild_embeds.go b/internal/utils/embed/prebuild_embeds.go new file mode 100644 index 0000000..2fa66e3 --- /dev/null +++ b/internal/utils/embed/prebuild_embeds.go @@ -0,0 +1,24 @@ +package embed + +import ( + "fmt" + + "github.com/jackmerrill/hampbot/internal/utils/config" + "github.com/zekroTJA/shireikan" +) + +func NewGenericEmbed(ctx shireikan.Context) *Embed { + return NewEmbed().SetFooter(fmt.Sprintf("HampBot %s", config.Version), ctx.GetSession().State.User.AvatarURL("256")) +} + +func NewSuccessEmbed(ctx shireikan.Context) *Embed { + return NewGenericEmbed(ctx).SetColor(0x00ff00) +} + +func NewWarningEmbed(ctx shireikan.Context) *Embed { + return NewGenericEmbed(ctx).SetColor(0xffff00) +} + +func NewErrorEmbed(ctx shireikan.Context) *Embed { + return NewGenericEmbed(ctx).SetColor(0xff0000).SetTitle("Error") +} diff --git a/main.go b/main.go new file mode 100644 index 0000000..77cdb4e --- /dev/null +++ b/main.go @@ -0,0 +1,67 @@ +package main + +import ( + "os" + "os/signal" + "syscall" + + "github.com/bwmarrin/discordgo" + "github.com/jackmerrill/hampbot/internal/commands/fun" + studentlife "github.com/jackmerrill/hampbot/internal/commands/studentlife" + util "github.com/jackmerrill/hampbot/internal/commands/util" + "github.com/zekroTJA/shireikan" + + "github.com/charmbracelet/log" +) + +func main() { + token := os.Getenv("TOKEN") + + session, err := discordgo.New("Bot " + token) + if err != nil { + panic(err) + } + + log.Info("Starting bot...") + + err = session.Open() + if err != nil { + panic(err) + } + + defer func() { + sc := make(chan os.Signal, 1) + signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill) + <-sc + }() + + handler := shireikan.New(&shireikan.Config{ + GeneralPrefix: ">", + AllowBots: false, + AllowDM: true, + ExecuteOnEdit: true, + InvokeToLower: true, + UseDefaultHelpCommand: true, + OnError: func(ctx shireikan.Context, typ shireikan.ErrorType, err error) { + log.Error(err) + }, + }) + + log.Info("Registering commands...") + + handler.Register(&util.Ping{}) + log.Debug("Registered ping command") + + handler.Register(&studentlife.Laundry{}) + log.Debug("Registered laundry command") + + handler.Register(&util.Steal{}) + log.Debug("Registered steal command") + + handler.Register(&fun.AI{}) + log.Debug("Registered ai command") + + handler.Setup(session) + + log.Info("Bot is now running. Press CTRL-C to exit.") +}