Compare commits
44 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
940781cb83 | ||
|
|
3a7bf68ba9 | ||
|
|
243fa19e36 | ||
|
|
fcbf803ae2 | ||
|
|
47ecefb048 | ||
|
|
7bffb96431 | ||
|
|
14390c5c21 | ||
|
|
1e88425853 | ||
|
|
eacdd455f9 | ||
|
|
1e184c97a2 | ||
|
|
8c2bb50637 | ||
|
|
5f20f1b129 | ||
|
|
8a9373637f | ||
|
|
b1f705746e | ||
|
|
8893cc62f4 | ||
|
|
0bc5953c65 | ||
|
|
821ede69cf | ||
|
|
fe2e28357a | ||
|
|
b8ca9506c9 | ||
|
|
1d8de473c4 | ||
|
|
6a08f06101 | ||
|
|
dddcc8c316 | ||
|
|
5a384305b0 | ||
|
|
fb5f508fdf | ||
|
|
1fbe5ca7bf | ||
|
|
2e0762ee0f | ||
|
|
ccb450e543 | ||
|
|
c34f475084 | ||
|
|
1a7d9b824f | ||
|
|
0bf31a3e5d | ||
|
|
128011c79a | ||
|
|
53dc44abda | ||
|
|
6262ebfeb4 | ||
|
|
005dfc815f | ||
|
|
986d16b346 | ||
|
|
83c41e2788 | ||
|
|
24aff1be72 | ||
|
|
b28beaba0c | ||
|
|
4f8aaff94a | ||
|
|
a21e884bee | ||
|
|
faf64a8bf9 | ||
|
|
2abe574977 | ||
|
|
04c32955c7 | ||
|
|
1f70318d26 |
54
.github/workflows/main.yml
vendored
54
.github/workflows/main.yml
vendored
@ -21,40 +21,44 @@ jobs:
|
|||||||
id: get_version
|
id: get_version
|
||||||
uses: battila7/get-version-action@v2.0.0
|
uses: battila7/get-version-action@v2.0.0
|
||||||
|
|
||||||
- name: Install go-bindata and build migrations
|
- name: install buildx
|
||||||
env:
|
id: buildx
|
||||||
GOPATH: /home/runner/work/Geeksbot/
|
uses: crazy-max/ghaction-docker-buildx@v1
|
||||||
run: |
|
|
||||||
go get -u github.com/go-bindata/go-bindata/...
|
|
||||||
/home/runner/work/Geeksbot/bin/go-bindata -pkg migrations -prefix $GITHUB_WORKSPACE/pkg/database/migrations/ -o $GITHUB_WORKSPACE/pkg/database/migrations/bindata.go $GITHUB_WORKSPACE/pkg/database/migrations/
|
|
||||||
head $GITHUB_WORKSPACE/pkg/database/migrations/bindata.go
|
|
||||||
|
|
||||||
- name: Build container image
|
|
||||||
env:
|
|
||||||
IMAGE_TAG: ${{ steps.get_version.outputs.version-without-v }}
|
|
||||||
run: docker build -t registry.digitalocean.com/djpianalto/geeksbot:$IMAGE_TAG .
|
|
||||||
|
|
||||||
- name: Install doctl
|
|
||||||
uses: digitalocean/action-doctl@v2
|
|
||||||
with:
|
with:
|
||||||
token: ${{ secrets.DIGITALOCEAN_ACCESS_TOKEN }}
|
version: latest
|
||||||
|
|
||||||
- name: Login to DigitalOcean Container Registry with short-lived credentials
|
- name: Docker Login
|
||||||
run: doctl registry login --expiry-seconds 600
|
# You may pin to the exact commit or the version.
|
||||||
|
# uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
|
||||||
|
uses: docker/login-action@v1.10.0
|
||||||
|
with:
|
||||||
|
registry: ${{ secrets.DR_URL }}
|
||||||
|
# Username used to log against the Docker registry
|
||||||
|
username: ${{ secrets.DH_USERNAME }}
|
||||||
|
# Password or personal access token used to log against the Docker registry
|
||||||
|
password: ${{ secrets.DH_PASSWORD }}
|
||||||
|
# Log out from the Docker registry at the end of a job
|
||||||
|
logout: true
|
||||||
|
|
||||||
- name: Push image to DigitalOcean Container Registry
|
- name: Docker Build & Push
|
||||||
run: docker push registry.digitalocean.com/djpianalto/geeksbot:$IMAGE_TAG
|
|
||||||
env:
|
env:
|
||||||
IMAGE_TAG: ${{ steps.get_version.outputs.version-without-v }}
|
IMAGE_TAG: ${{ steps.get_version.outputs.version-without-v }}
|
||||||
|
run: |
|
||||||
|
docker buildx build --push \
|
||||||
|
--tag ${{ secrets.DR_URL }}/geeksbot:$IMAGE_TAG \
|
||||||
|
--platform linux/amd64,linux/arm/v7,linux/arm64 .
|
||||||
|
|
||||||
- name: Update deployment file
|
- name: Update deployment file
|
||||||
run: TAG=${{ steps.get_version.outputs.version-without-v }} && sed -i 's|<IMAGE>|registry.digitalocean.com/djpianalto/geeksbot:'${TAG}'|' $GITHUB_WORKSPACE/deployment.yml
|
run: TAG=${{ steps.get_version.outputs.version-without-v }} && sed -i 's|<IMAGE>|${{ secrets.DR_URL }}/geeksbot:'${TAG}'|' $GITHUB_WORKSPACE/deployment.yml
|
||||||
|
|
||||||
- name: Save DigitalOcean kubeconfig with short-lived credentials
|
- uses: azure/k8s-set-context@v1
|
||||||
run: doctl kubernetes cluster kubeconfig save --expiry-seconds 600 discord-bots
|
with:
|
||||||
|
method: kubeconfig
|
||||||
|
kubeconfig: ${{ secrets.KUBE_CONFIG }}
|
||||||
|
id: setcontext
|
||||||
|
|
||||||
- name: Deploy to DigitalOcean Kubernetes
|
- name: Deploy to Kubernetes
|
||||||
run: kubectl apply -f $GITHUB_WORKSPACE/deployment.yml
|
run: kubectl apply -f $GITHUB_WORKSPACE/deployment.yml
|
||||||
|
|
||||||
- name: Verify deployment
|
- name: Verify deployment
|
||||||
run: kubectl rollout status deployment/geeksbot
|
run: kubectl rollout status -n discord-bots deployment/geeksbot
|
||||||
|
|||||||
@ -28,6 +28,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
database.ConnectDatabase(os.Getenv("DATABASE_URL"))
|
database.ConnectDatabase(os.Getenv("DATABASE_URL"))
|
||||||
|
//database.RunMigrations()
|
||||||
services.InitializeServices()
|
services.InitializeServices()
|
||||||
|
|
||||||
owners := []string{
|
owners := []string{
|
||||||
|
|||||||
@ -2,7 +2,7 @@ apiVersion: apps/v1
|
|||||||
kind: Deployment
|
kind: Deployment
|
||||||
metadata:
|
metadata:
|
||||||
name: geeksbot
|
name: geeksbot
|
||||||
namespace: default
|
namespace: discord-bots
|
||||||
labels:
|
labels:
|
||||||
app: geeksbot
|
app: geeksbot
|
||||||
spec:
|
spec:
|
||||||
@ -21,8 +21,42 @@ spec:
|
|||||||
app: geeksbot
|
app: geeksbot
|
||||||
spec:
|
spec:
|
||||||
containers:
|
containers:
|
||||||
|
- name: pgbouncer
|
||||||
|
image: timoha/pgbouncer:1.15.0
|
||||||
|
resources:
|
||||||
|
requests:
|
||||||
|
memory: "256Mi"
|
||||||
|
cpu: "0.5"
|
||||||
|
limits:
|
||||||
|
memory: "512Mi"
|
||||||
|
cpu: "1"
|
||||||
|
env:
|
||||||
|
- name: DATABASE_URL
|
||||||
|
valueFrom:
|
||||||
|
secretKeyRef:
|
||||||
|
name: geeksbot
|
||||||
|
key: pgbouncer_url
|
||||||
|
- name: SERVER_TLS_SSLMODE
|
||||||
|
valueFrom:
|
||||||
|
secretKeyRef:
|
||||||
|
name: geeksbot
|
||||||
|
key: pgbouncer_ssl
|
||||||
|
- name: AUTH_TYPE
|
||||||
|
valueFrom:
|
||||||
|
secretKeyRef:
|
||||||
|
name: geeksbot
|
||||||
|
key: pgbouncer_auth
|
||||||
|
ports:
|
||||||
|
- containerPort: 5432
|
||||||
- name: geeksbot
|
- name: geeksbot
|
||||||
image: <IMAGE>
|
image: <IMAGE>
|
||||||
|
resources:
|
||||||
|
requests:
|
||||||
|
memory: "512Mi"
|
||||||
|
cpu: "1"
|
||||||
|
limits:
|
||||||
|
memory: "1Gi"
|
||||||
|
cpu: "2"
|
||||||
env:
|
env:
|
||||||
- name: DATABASE_URL
|
- name: DATABASE_URL
|
||||||
valueFrom:
|
valueFrom:
|
||||||
@ -34,3 +68,5 @@ spec:
|
|||||||
secretKeyRef:
|
secretKeyRef:
|
||||||
name: geeksbot
|
name: geeksbot
|
||||||
key: discord_token
|
key: discord_token
|
||||||
|
imagePullSecrets:
|
||||||
|
- name: registry-1
|
||||||
|
|||||||
1
go.mod
1
go.mod
@ -8,6 +8,7 @@ require (
|
|||||||
github.com/go-bindata/go-bindata v3.1.2+incompatible // indirect
|
github.com/go-bindata/go-bindata v3.1.2+incompatible // indirect
|
||||||
github.com/golang-migrate/migrate v3.5.4+incompatible // indirect
|
github.com/golang-migrate/migrate v3.5.4+incompatible // indirect
|
||||||
github.com/golang-migrate/migrate/v4 v4.14.1
|
github.com/golang-migrate/migrate/v4 v4.14.1
|
||||||
|
github.com/gorcon/rcon v1.3.1
|
||||||
github.com/hashicorp/errwrap v1.1.0 // indirect
|
github.com/hashicorp/errwrap v1.1.0 // indirect
|
||||||
github.com/lib/pq v1.8.0
|
github.com/lib/pq v1.8.0
|
||||||
)
|
)
|
||||||
|
|||||||
2
go.sum
2
go.sum
@ -152,6 +152,8 @@ github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+
|
|||||||
github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
|
github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
|
||||||
github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
|
github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
|
||||||
|
github.com/gorcon/rcon v1.3.1 h1:z6a5iOlojfdkvA1qaKEng7QfCJuCzYlC9BUDs6/M+74=
|
||||||
|
github.com/gorcon/rcon v1.3.1/go.mod h1:2gztBPSV2WxkPkqV4jiJkdHs+NT46mNSGb8JxbPesx4=
|
||||||
github.com/gorilla/context v1.1.1/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
|
github.com/gorilla/context v1.1.1/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
|
||||||
github.com/gorilla/handlers v1.4.2/go.mod h1:Qkdc/uu4tH4g6mTK6auzZ766c4CA0Ng8+o/OAirnOIQ=
|
github.com/gorilla/handlers v1.4.2/go.mod h1:Qkdc/uu4tH4g6mTK6auzZ766c4CA0Ng8+o/OAirnOIQ=
|
||||||
github.com/gorilla/mux v1.6.2/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
|
github.com/gorilla/mux v1.6.2/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
|
||||||
|
|||||||
11
internal/discord_utils/channel_utils.go
Normal file
11
internal/discord_utils/channel_utils.go
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
package discord_utils
|
||||||
|
|
||||||
|
import "github.com/dustinpianalto/disgoman"
|
||||||
|
|
||||||
|
func GetChannelName(ctx disgoman.Context, id string) string {
|
||||||
|
channel, err := ctx.Session.Channel(id)
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return channel.Name
|
||||||
|
}
|
||||||
14
internal/discord_utils/user_utils.go
Normal file
14
internal/discord_utils/user_utils.go
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
package discord_utils
|
||||||
|
|
||||||
|
import "github.com/dustinpianalto/disgoman"
|
||||||
|
|
||||||
|
func GetDisplayName(ctx disgoman.Context, id string) string {
|
||||||
|
member, err := ctx.Session.GuildMember(ctx.Guild.ID, id)
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
if member.Nick != "" {
|
||||||
|
return member.Nick
|
||||||
|
}
|
||||||
|
return member.User.Username
|
||||||
|
}
|
||||||
184
internal/exts/arcon/arcon.go
Normal file
184
internal/exts/arcon/arcon.go
Normal file
@ -0,0 +1,184 @@
|
|||||||
|
package arcon
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/dustinpianalto/disgoman"
|
||||||
|
"github.com/dustinpianalto/geeksbot"
|
||||||
|
"github.com/dustinpianalto/geeksbot/internal/discord_utils"
|
||||||
|
"github.com/dustinpianalto/geeksbot/pkg/services"
|
||||||
|
"github.com/gorcon/rcon"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ListplayersCommand = &disgoman.Command{
|
||||||
|
Name: "listplayers",
|
||||||
|
Aliases: nil,
|
||||||
|
Description: "List the players currently connected to a ARK server.",
|
||||||
|
OwnerOnly: false,
|
||||||
|
Hidden: false,
|
||||||
|
RequiredPermissions: 0,
|
||||||
|
Invoke: listplayersCommandFunc,
|
||||||
|
}
|
||||||
|
|
||||||
|
func listplayersCommandFunc(ctx disgoman.Context, args []string) {
|
||||||
|
guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Sorry, there was a problem getting your user.", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !discord_utils.IsGuildAdmin(ctx, author) && !discord_utils.IsGuildMod(ctx, author) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(args) == 0 {
|
||||||
|
servers, err := services.ServerService.GuildServers(guild)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Could not find any servers for this guild", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, server := range servers {
|
||||||
|
go listplayers(ctx, server)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
serverName := strings.Join(args, " ")
|
||||||
|
server, err := services.ServerService.ServerByName(serverName, guild)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx,
|
||||||
|
fmt.Sprintf("Could not find **%s** in this guild.", serverName),
|
||||||
|
err,
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
listplayers(ctx, server)
|
||||||
|
}
|
||||||
|
|
||||||
|
func listplayers(ctx disgoman.Context, server geeksbot.Server) {
|
||||||
|
msg, err := ctx.Send(fmt.Sprintf("**Getting data for %s**", server.Name))
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "There was an error getting the player list", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
conn, err := rcon.Dial(fmt.Sprintf("%s:%d", server.IPAddr, server.Port), server.Password)
|
||||||
|
if err != nil {
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**Could not open connection to %s**", server.Name),
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
response, err := conn.Execute("listplayers")
|
||||||
|
if err != nil {
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**There was a problem getting a response from %s**", server.Name),
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(response, "No Players") {
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**%s: %s**", server.Name, response),
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
players := strings.Split(response, "\n")
|
||||||
|
for i, player := range players {
|
||||||
|
parts := strings.Split(player, ", ")
|
||||||
|
steamID := parts[len(parts)-1]
|
||||||
|
user, err := services.UserService.GetBySteamID(steamID)
|
||||||
|
if err == nil {
|
||||||
|
duser, err := ctx.Session.GuildMember(ctx.Guild.ID, user.ID)
|
||||||
|
if err == nil {
|
||||||
|
players[i] = fmt.Sprintf("%s (%s)", player, duser.Mention())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**%s:**%s", server.Name, strings.Join(players, "\n")),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
var BroadcastCommand = &disgoman.Command{
|
||||||
|
Name: "broadcast",
|
||||||
|
Aliases: nil,
|
||||||
|
Description: "Broadcast a message to ARK servers.",
|
||||||
|
OwnerOnly: false,
|
||||||
|
Hidden: false,
|
||||||
|
RequiredPermissions: 0,
|
||||||
|
Invoke: broadcastCommandFunc,
|
||||||
|
}
|
||||||
|
|
||||||
|
func broadcastCommandFunc(ctx disgoman.Context, args []string) {
|
||||||
|
guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Sorry, there was a problem getting your user.", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !discord_utils.IsGuildAdmin(ctx, author) && !discord_utils.IsGuildMod(ctx, author) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
message := strings.Join(args[1:len(args)], " ")
|
||||||
|
if strings.ToLower(args[0]) == "all" {
|
||||||
|
servers, err := services.ServerService.GuildServers(guild)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Could not find any servers for this guild", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, server := range servers {
|
||||||
|
go broadcast(ctx, server, message)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
serverName := strings.Title(strings.ReplaceAll(args[0], "_", " "))
|
||||||
|
server, err := services.ServerService.ServerByName(serverName, guild)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, fmt.Sprintf("Could not find %s", serverName), err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
broadcast(ctx, server, message)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func broadcast(ctx disgoman.Context, server geeksbot.Server, message string) {
|
||||||
|
msg, err := ctx.Send(fmt.Sprintf("**Broadcasting to: %s**", server.Name))
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "There was an error getting the player list", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
conn, err := rcon.Dial(fmt.Sprintf("%s:%d", server.IPAddr, server.Port), server.Password)
|
||||||
|
if err != nil {
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**Could not open connection to %s**", server.Name),
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
userName := discord_utils.GetDisplayName(ctx, ctx.Message.Author.ID)
|
||||||
|
response, err := conn.Execute(fmt.Sprintf("broadcast %s: %s", userName, message))
|
||||||
|
if err != nil {
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**There was a problem getting a response from %s**", server.Name),
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Printf("%T - %#v", response, response)
|
||||||
|
if strings.Contains(response, "Server received, But no response!!") {
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**%s Broadcast Successful**", server.Name),
|
||||||
|
)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_, _ = ctx.Session.ChannelMessageEdit(ctx.Channel.ID, msg.ID,
|
||||||
|
fmt.Sprintf("**Broadcasting to %s Failed!**", server.Name),
|
||||||
|
)
|
||||||
|
}
|
||||||
@ -2,6 +2,7 @@ package exts
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/dustinpianalto/disgoman"
|
"github.com/dustinpianalto/disgoman"
|
||||||
|
"github.com/dustinpianalto/geeksbot/internal/exts/arcon"
|
||||||
"github.com/dustinpianalto/geeksbot/internal/exts/guild"
|
"github.com/dustinpianalto/geeksbot/internal/exts/guild"
|
||||||
"github.com/dustinpianalto/geeksbot/internal/exts/requests"
|
"github.com/dustinpianalto/geeksbot/internal/exts/requests"
|
||||||
"github.com/dustinpianalto/geeksbot/internal/exts/utils"
|
"github.com/dustinpianalto/geeksbot/internal/exts/utils"
|
||||||
@ -34,4 +35,8 @@ func AddCommandHandlers(g *disgoman.CommandManager) {
|
|||||||
_ = g.AddCommand(requests.RequestCommand)
|
_ = g.AddCommand(requests.RequestCommand)
|
||||||
_ = g.AddCommand(requests.CloseCommand)
|
_ = g.AddCommand(requests.CloseCommand)
|
||||||
_ = g.AddCommand(requests.ListCommand)
|
_ = g.AddCommand(requests.ListCommand)
|
||||||
|
_ = g.AddCommand(requests.ViewCommand)
|
||||||
|
_ = g.AddCommand(requests.CommentCommand)
|
||||||
|
_ = g.AddCommand(arcon.ListplayersCommand)
|
||||||
|
_ = g.AddCommand(arcon.BroadcastCommand)
|
||||||
}
|
}
|
||||||
|
|||||||
@ -2,6 +2,7 @@ package requests
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
@ -9,6 +10,7 @@ import (
|
|||||||
"github.com/dustinpianalto/disgoman"
|
"github.com/dustinpianalto/disgoman"
|
||||||
"github.com/dustinpianalto/geeksbot"
|
"github.com/dustinpianalto/geeksbot"
|
||||||
"github.com/dustinpianalto/geeksbot/internal/discord_utils"
|
"github.com/dustinpianalto/geeksbot/internal/discord_utils"
|
||||||
|
"github.com/dustinpianalto/geeksbot/internal/utils"
|
||||||
"github.com/dustinpianalto/geeksbot/pkg/services"
|
"github.com/dustinpianalto/geeksbot/pkg/services"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -77,7 +79,7 @@ func requestCommandFunc(ctx disgoman.Context, args []string) {
|
|||||||
if c.Admin {
|
if c.Admin {
|
||||||
_, _ = ctx.Session.ChannelMessageSend(c.ID,
|
_, _ = ctx.Session.ChannelMessageSend(c.ID,
|
||||||
fmt.Sprintf("%s\n"+
|
fmt.Sprintf("%s\n"+
|
||||||
"New Request ID %d "+
|
"New Request ID %d\n"+
|
||||||
"%s has requested assistance: \n"+
|
"%s has requested assistance: \n"+
|
||||||
"```\n%s\n```\n"+
|
"```\n%s\n```\n"+
|
||||||
"Requested At: %s\n"+
|
"Requested At: %s\n"+
|
||||||
@ -152,7 +154,7 @@ func closeCommandFunc(ctx disgoman.Context, args []string) {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if request.Author != closer && !closer.IsStaff && !closer.IsAdmin {
|
if request.Author != closer && !closer.IsStaff && !closer.IsAdmin {
|
||||||
if !discord_utils.IsGuildMod(ctx, closer) || !discord_utils.IsGuildAdmin(ctx, closer) {
|
if !discord_utils.IsGuildMod(ctx, closer) && !discord_utils.IsGuildAdmin(ctx, closer) {
|
||||||
discord_utils.SendErrorMessage(ctx, fmt.Sprintf("You are not authorized to close %d", id), nil)
|
discord_utils.SendErrorMessage(ctx, fmt.Sprintf("You are not authorized to close %d", id), nil)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@ -178,7 +180,17 @@ func closeCommandFunc(ctx disgoman.Context, args []string) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was closed.", err)
|
discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was closed.", err)
|
||||||
}
|
}
|
||||||
|
dmChannel, err := ctx.Session.UserChannelCreate(request.Author.ID)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_, _ = ctx.Session.ChannelMessageSend(dmChannel.ID,
|
||||||
|
fmt.Sprintf("%s has closed request %d which you opened in the %s channel.\n```%s```\n",
|
||||||
|
discord_utils.GetDisplayName(ctx, request.CompletedBy.ID),
|
||||||
|
request.ID,
|
||||||
|
discord_utils.GetChannelName(ctx, request.Channel.ID),
|
||||||
|
request.Content,
|
||||||
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -231,20 +243,207 @@ func listCommandFunc(ctx disgoman.Context, args []string) {
|
|||||||
} else {
|
} else {
|
||||||
channelName = channel.Name
|
channelName = channel.Name
|
||||||
}
|
}
|
||||||
|
commentCount, err := services.RequestService.RequestCommentCount(request)
|
||||||
|
if err != nil {
|
||||||
|
commentCount = 0
|
||||||
|
}
|
||||||
_, _ = ctx.Send(fmt.Sprintf("```md\n"+
|
_, _ = ctx.Send(fmt.Sprintf("```md\n"+
|
||||||
"< Request ID Requested By >\n"+
|
"< Request ID Requested By >\n"+
|
||||||
"< %11s %-23s >\n"+
|
"< %-11d %23s >\n"+
|
||||||
"%s\n\n"+
|
"%s\n\n"+
|
||||||
"Comments: Not Implemented Yet\n"+
|
"Comments: %d\n"+
|
||||||
"Requested At: %s\n"+
|
"Requested At: %s\n"+
|
||||||
"In: %s\n"+
|
"In: %s\n"+
|
||||||
"```",
|
"```",
|
||||||
request.ID,
|
request.ID,
|
||||||
authorName,
|
authorName,
|
||||||
request.Content,
|
request.Content,
|
||||||
|
commentCount,
|
||||||
request.RequestedAt.Format("2006-01-02 15:04:05 MST"),
|
request.RequestedAt.Format("2006-01-02 15:04:05 MST"),
|
||||||
channelName,
|
channelName,
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
_, _ = ctx.Send(fmt.Sprintf("```There %s currently %d open %s```", utils.PluralizeString("is", len(requests)), len(requests), utils.PluralizeString("request", len(requests))))
|
||||||
|
}
|
||||||
|
|
||||||
|
var CommentCommand = &disgoman.Command{
|
||||||
|
Name: "comment",
|
||||||
|
Aliases: []string{"update", "add_comment"},
|
||||||
|
Description: "Add a comment to an existing request.",
|
||||||
|
OwnerOnly: false,
|
||||||
|
Hidden: false,
|
||||||
|
RequiredPermissions: 0,
|
||||||
|
Invoke: commentCommandFunc,
|
||||||
|
}
|
||||||
|
|
||||||
|
func commentCommandFunc(ctx disgoman.Context, args []string) {
|
||||||
|
guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id, err := strconv.ParseInt(args[0], 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Please include the ID of the request to update.", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
message := strings.Join(args[1:len(args)], " ")
|
||||||
|
author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Sorry, there was an issue finding your user account", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
request, err := services.RequestService.Request(id)
|
||||||
|
if err != nil || request.Guild.ID != guild.ID {
|
||||||
|
discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%d is not a valid request for this guild", id), err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
int64ID, _ := strconv.ParseInt(ctx.Message.ID, 10, 64)
|
||||||
|
s := discord_utils.ParseSnowflake(int64ID)
|
||||||
|
comment := geeksbot.Comment{
|
||||||
|
Author: author,
|
||||||
|
Request: request,
|
||||||
|
CommentAt: s.CreationTime,
|
||||||
|
Content: message,
|
||||||
|
}
|
||||||
|
comment, err = services.RequestService.CreateComment(comment)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "There was a problem adding your comment", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
channels, err := services.ChannelService.GuildChannels(guild)
|
||||||
|
if err == nil {
|
||||||
|
comments, _ := services.RequestService.RequestComments(request)
|
||||||
|
var commentString string
|
||||||
|
var commentStrings []string
|
||||||
|
commentString = fmt.Sprintf("Comment added:\n```md\n"+
|
||||||
|
"< Request ID Requested By >\n"+
|
||||||
|
"< %-11d %23s >\n"+
|
||||||
|
"%s\n\n"+
|
||||||
|
"Comments: Not Implemented Yet\n"+
|
||||||
|
"Requested At: %s\n"+
|
||||||
|
"In: %s\n"+
|
||||||
|
"```",
|
||||||
|
request.ID,
|
||||||
|
discord_utils.GetDisplayName(ctx, request.Author.ID),
|
||||||
|
request.Content,
|
||||||
|
request.RequestedAt.Format("2006-01-02 15:04:05"),
|
||||||
|
discord_utils.GetChannelName(ctx, request.Channel.ID),
|
||||||
|
)
|
||||||
|
for _, c := range comments {
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
cs := fmt.Sprintf("```md\n%s\n- %s At %s\n```\n",
|
||||||
|
c.Content,
|
||||||
|
discord_utils.GetDisplayName(ctx, c.Author.ID),
|
||||||
|
c.CommentAt.Format("2006-01-02 15:04:05"),
|
||||||
|
)
|
||||||
|
if len(commentString+cs) >= 2000 {
|
||||||
|
commentStrings = append(commentStrings, commentString)
|
||||||
|
commentString = ""
|
||||||
|
}
|
||||||
|
commentString += cs
|
||||||
|
}
|
||||||
|
commentStrings = append(commentStrings, commentString)
|
||||||
|
for _, c := range channels {
|
||||||
|
if c.Admin {
|
||||||
|
for _, s := range commentStrings {
|
||||||
|
_, _ = ctx.Session.ChannelMessageSend(c.ID, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
log.Println(err)
|
||||||
|
}
|
||||||
|
_, err = ctx.Send(fmt.Sprintf("%s your comment has been added.", ctx.Message.Author.Mention()))
|
||||||
|
dmChannel, err := ctx.Session.UserChannelCreate(request.Author.ID)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_, _ = ctx.Session.ChannelMessageSend(dmChannel.ID,
|
||||||
|
fmt.Sprintf("%s has add a comment to request %d which you opened in the %s channel.\n```%s```\n```%s```",
|
||||||
|
discord_utils.GetDisplayName(ctx, author.ID),
|
||||||
|
request.ID,
|
||||||
|
discord_utils.GetChannelName(ctx, ctx.Channel.ID),
|
||||||
|
request.Content,
|
||||||
|
message,
|
||||||
|
))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
var ViewCommand = &disgoman.Command{
|
||||||
|
Name: "view",
|
||||||
|
Aliases: nil,
|
||||||
|
Description: "View the details about a request.",
|
||||||
|
OwnerOnly: false,
|
||||||
|
Hidden: false,
|
||||||
|
RequiredPermissions: 0,
|
||||||
|
Invoke: viewCommandFunc,
|
||||||
|
}
|
||||||
|
|
||||||
|
func viewCommandFunc(ctx disgoman.Context, args []string) {
|
||||||
|
guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id, err := strconv.ParseInt(args[0], 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Please include the ID of the request to view.", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
request, err := services.RequestService.Request(id)
|
||||||
|
if err != nil || request.Guild.ID != guild.ID {
|
||||||
|
discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%d is not a valid request in this guild", id), err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
requestor, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "Sorry, there was an issue finding your user account", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if request.Author.ID != ctx.Message.Author.ID &&
|
||||||
|
!discord_utils.IsGuildMod(ctx, requestor) &&
|
||||||
|
!discord_utils.IsGuildAdmin(ctx, requestor) {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "You are not authorized to view that request", nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
comments, err := services.RequestService.RequestComments(request)
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "There was an error getting the comments.", err)
|
||||||
|
}
|
||||||
|
var commentString string
|
||||||
|
var commentStrings []string
|
||||||
|
commentString = fmt.Sprintf("```md\n"+
|
||||||
|
"< Request ID Requested By >\n"+
|
||||||
|
"< %-11d %23s >\n"+
|
||||||
|
"%s\n\n"+
|
||||||
|
"Requested At: %s\n"+
|
||||||
|
"In: %s\n"+
|
||||||
|
"```",
|
||||||
|
request.ID,
|
||||||
|
discord_utils.GetDisplayName(ctx, request.Author.ID),
|
||||||
|
request.Content,
|
||||||
|
request.RequestedAt.Format("2006-01-02 15:04:05"),
|
||||||
|
discord_utils.GetChannelName(ctx, request.Channel.ID),
|
||||||
|
)
|
||||||
|
for _, c := range comments {
|
||||||
|
cs := fmt.Sprintf("```md\n%s\n- %s At %s\n```\n",
|
||||||
|
c.Content,
|
||||||
|
discord_utils.GetDisplayName(ctx, c.Author.ID),
|
||||||
|
c.CommentAt.Format("2006-01-02 15:04:05"),
|
||||||
|
)
|
||||||
|
if len(commentString+cs) >= 2000 {
|
||||||
|
commentStrings = append(commentStrings, commentString)
|
||||||
|
commentString = ""
|
||||||
|
}
|
||||||
|
commentString += cs
|
||||||
|
}
|
||||||
|
commentStrings = append(commentStrings, commentString)
|
||||||
|
for _, c := range commentStrings {
|
||||||
|
_, _ = ctx.Send(c)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -17,5 +17,8 @@ func PluralizeString(s string, i int) string {
|
|||||||
if i == 1 {
|
if i == 1 {
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
if s == "is" {
|
||||||
|
return "are"
|
||||||
|
}
|
||||||
return s + "s"
|
return s + "s"
|
||||||
}
|
}
|
||||||
|
|||||||
@ -23,10 +23,6 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func ConnectDatabase(dbConnString string) {
|
func ConnectDatabase(dbConnString string) {
|
||||||
s := bindata.Resource(migrations.AssetNames(),
|
|
||||||
func(name string) ([]byte, error) {
|
|
||||||
return migrations.Asset(name)
|
|
||||||
})
|
|
||||||
var err error
|
var err error
|
||||||
db, err = sql.Open("postgres", dbConnString)
|
db, err = sql.Open("postgres", dbConnString)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -36,6 +32,15 @@ func ConnectDatabase(dbConnString string) {
|
|||||||
db.SetMaxOpenConns(75)
|
db.SetMaxOpenConns(75)
|
||||||
db.SetMaxIdleConns(5)
|
db.SetMaxIdleConns(5)
|
||||||
db.SetConnMaxLifetime(300)
|
db.SetConnMaxLifetime(300)
|
||||||
|
initServices()
|
||||||
|
log.Println("Services Initialized")
|
||||||
|
}
|
||||||
|
|
||||||
|
func RunMigrations() {
|
||||||
|
s := bindata.Resource(migrations.AssetNames(),
|
||||||
|
func(name string) ([]byte, error) {
|
||||||
|
return migrations.Asset(name)
|
||||||
|
})
|
||||||
d, err := bindata.WithInstance(s)
|
d, err := bindata.WithInstance(s)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(fmt.Errorf("cannot load migrations: %w", err))
|
log.Fatal(fmt.Errorf("cannot load migrations: %w", err))
|
||||||
@ -57,8 +62,7 @@ func ConnectDatabase(dbConnString string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
log.Println("Migrations Run")
|
log.Println("Migrations Run")
|
||||||
initServices()
|
|
||||||
log.Println("Services Initialized")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func initServices() {
|
func initServices() {
|
||||||
|
|||||||
@ -5,6 +5,7 @@ import (
|
|||||||
"log"
|
"log"
|
||||||
|
|
||||||
"github.com/dustinpianalto/geeksbot"
|
"github.com/dustinpianalto/geeksbot"
|
||||||
|
"github.com/lib/pq"
|
||||||
)
|
)
|
||||||
|
|
||||||
type messageService struct {
|
type messageService struct {
|
||||||
@ -21,7 +22,7 @@ func (s messageService) Message(id string) (geeksbot.Message, error) {
|
|||||||
WHERE m.id = $1`
|
WHERE m.id = $1`
|
||||||
row := s.db.QueryRow(queryString, id)
|
row := s.db.QueryRow(queryString, id)
|
||||||
err := row.Scan(&m.ID, &m.CreatedAt, &m.ModifiedAt, &m.Content,
|
err := row.Scan(&m.ID, &m.CreatedAt, &m.ModifiedAt, &m.Content,
|
||||||
&m.PreviousContent, &channel_id, &author_id)
|
pq.Array(&m.PreviousContent), &channel_id, &author_id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return geeksbot.Message{}, err
|
return geeksbot.Message{}, err
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1,7 +1,6 @@
|
|||||||
BEGIN;
|
BEGIN;
|
||||||
UPDATE messages SET previous_content = array[]::varchar[];
|
|
||||||
ALTER TABLE messages
|
ALTER TABLE messages
|
||||||
ALTER COLUMN previous_content SET NOT NULL;
|
ALTER COLUMN previous_content SET NOT NULL;
|
||||||
ALTER TABLE messages
|
ALTER TABLE messages
|
||||||
ALTER COLUMN prevous_content SET DEFAULT array[]::varchar[]
|
ALTER COLUMN previous_content SET DEFAULT array[]::varchar[];
|
||||||
COMMIT;
|
COMMIT;
|
||||||
|
|||||||
@ -0,0 +1,6 @@
|
|||||||
|
BEGIN;
|
||||||
|
ALTER TABLE servers
|
||||||
|
DROP COLUMN ftp_port,
|
||||||
|
DROP COLUMN ftp_username,
|
||||||
|
DROP COLUMN ftp_password;
|
||||||
|
COMMIT;
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
BEGIN;
|
||||||
|
ALTER TABLE servers
|
||||||
|
ADD COLUMN ftp_port int4,
|
||||||
|
ADD COLUMN ftp_username varchar(200),
|
||||||
|
ADD COLUMN ftp_password varchar(200);
|
||||||
|
COMMIT;
|
||||||
@ -154,7 +154,7 @@ func (s requestService) UpdateRequest(r geeksbot.Request) (geeksbot.Request, err
|
|||||||
queryString := `UPDATE requests SET
|
queryString := `UPDATE requests SET
|
||||||
completed = $2, completed_at = $3, completed_by = $4, completed_message = $5
|
completed = $2, completed_at = $3, completed_by = $4, completed_message = $5
|
||||||
WHERE id = $1`
|
WHERE id = $1`
|
||||||
_, err := s.db.Exec(queryString, r.ID, r.Completed, r.CompletedAt, r.CompletedBy, r.CompletedMessage)
|
_, err := s.db.Exec(queryString, r.ID, r.Completed, r.CompletedAt, r.CompletedBy.ID, r.CompletedMessage)
|
||||||
return r, err
|
return r, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -19,11 +19,12 @@ func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
|
|||||||
var iMsgID sql.NullString
|
var iMsgID sql.NullString
|
||||||
var sMsgID sql.NullString
|
var sMsgID sql.NullString
|
||||||
queryString := `SELECT id, name, ip_address, port, password, alerts_channel_id,
|
queryString := `SELECT id, name, ip_address, port, password, alerts_channel_id,
|
||||||
guild_id, info_channel_id, info_message_id, settings_message_id
|
guild_id, info_channel_id, info_message_id, settings_message_id,
|
||||||
|
ftp_port, ftp_username, ftp_password
|
||||||
FROM servers WHERE id = $1`
|
FROM servers WHERE id = $1`
|
||||||
row := s.db.QueryRow(queryString, id)
|
row := s.db.QueryRow(queryString, id)
|
||||||
err := row.Scan(&server.ID, &server.Name, &server.IPAddr, &server.Port, &server.Password,
|
err := row.Scan(&server.ID, &server.Name, &server.IPAddr, &server.Port, &server.Password,
|
||||||
&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID)
|
&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID, &server.FTPPort, &server.FTPUser, &server.FTPPass)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return geeksbot.Server{}, err
|
return geeksbot.Server{}, err
|
||||||
}
|
}
|
||||||
@ -73,7 +74,7 @@ func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
|
|||||||
|
|
||||||
func (s serverService) ServerByName(name string, guild geeksbot.Guild) (geeksbot.Server, error) {
|
func (s serverService) ServerByName(name string, guild geeksbot.Guild) (geeksbot.Server, error) {
|
||||||
var id int
|
var id int
|
||||||
queryString := "SELECT id FROM servers WHERE name = $1 AND guild_id = $2"
|
queryString := "SELECT id FROM servers WHERE LOWER(name) = LOWER($1) AND guild_id = $2"
|
||||||
row := s.db.QueryRow(queryString, name, guild.ID)
|
row := s.db.QueryRow(queryString, name, guild.ID)
|
||||||
err := row.Scan(&id)
|
err := row.Scan(&id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -115,7 +116,8 @@ func (s serverService) DeleteServer(server geeksbot.Server) error {
|
|||||||
func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, error) {
|
func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, error) {
|
||||||
queryString := `UPDATE servers SET name = $2, ip_address = $3, port = $4, password = $5,
|
queryString := `UPDATE servers SET name = $2, ip_address = $3, port = $4, password = $5,
|
||||||
alerts_channel_id = $6, info_channel_id = $7, info_message_id = $8,
|
alerts_channel_id = $6, info_channel_id = $7, info_message_id = $8,
|
||||||
settings_message_id = $9 WHERE id = $1`
|
settings_message_id = $9, ftp_port = $10, ftp_username = $11,
|
||||||
|
ftp_password = $12 WHERE id = $1`
|
||||||
_, err := s.db.Exec(queryString,
|
_, err := s.db.Exec(queryString,
|
||||||
server.Name,
|
server.Name,
|
||||||
server.IPAddr,
|
server.IPAddr,
|
||||||
@ -125,6 +127,9 @@ func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, er
|
|||||||
server.InfoChannel.ID,
|
server.InfoChannel.ID,
|
||||||
server.InfoMessage.ID,
|
server.InfoMessage.ID,
|
||||||
server.SettingsMessage.ID,
|
server.SettingsMessage.ID,
|
||||||
|
server.FTPPort,
|
||||||
|
server.FTPUser,
|
||||||
|
server.FTPPass,
|
||||||
)
|
)
|
||||||
return server, err
|
return server, err
|
||||||
}
|
}
|
||||||
|
|||||||
@ -49,3 +49,14 @@ func (s userService) GetOrCreateUser(id string) (geeksbot.User, error) {
|
|||||||
}
|
}
|
||||||
return user, err
|
return user, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s userService) GetBySteamID(steamID string) (geeksbot.User, error) {
|
||||||
|
var id string
|
||||||
|
queryString := "SELECT id FROM users WHERE steam_id = $1"
|
||||||
|
err := s.db.QueryRow(queryString, steamID).Scan(&id)
|
||||||
|
if err != nil {
|
||||||
|
return geeksbot.User{}, err
|
||||||
|
}
|
||||||
|
user, err := s.User(id)
|
||||||
|
return user, err
|
||||||
|
}
|
||||||
|
|||||||
@ -11,6 +11,9 @@ type Server struct {
|
|||||||
InfoChannel *Channel
|
InfoChannel *Channel
|
||||||
InfoMessage *Message
|
InfoMessage *Message
|
||||||
SettingsMessage *Message
|
SettingsMessage *Message
|
||||||
|
FTPPort int
|
||||||
|
FTPUser string
|
||||||
|
FTPPass string
|
||||||
}
|
}
|
||||||
|
|
||||||
type ServerService interface {
|
type ServerService interface {
|
||||||
|
|||||||
1
user.go
1
user.go
@ -16,4 +16,5 @@ type UserService interface {
|
|||||||
DeleteUser(u User) error
|
DeleteUser(u User) error
|
||||||
UpdateUser(u User) (User, error)
|
UpdateUser(u User) (User, error)
|
||||||
GetOrCreateUser(id string) (User, error)
|
GetOrCreateUser(id string) (User, error)
|
||||||
|
GetBySteamID(steamID string) (User, error)
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user