Compare commits

..
32 Commits
Author SHA1 Message Date
Dustin Pianalto a6dc6f5e56 fix get user sql query
CI / build (push) Has been cancelled
2021-01-25 01:52:14 -09:00
Dustin Pianalto 203d1bbf69 fix get user sql query
CI / build (push) Has been cancelled
2021-01-25 01:43:31 -09:00
Dustin Pianalto c504f7b165 fix get user sql query
CI / build (push) Has been cancelled
2021-01-25 01:38:22 -09:00
Dustin Pianalto 34564140f5 add request command
CI / build (push) Has been cancelled
2021-01-25 01:33:46 -09:00
Dustin Pianalto e3ccc56dba add request command
CI / build (push) Has been cancelled
2021-01-25 01:21:49 -09:00
Dustin Pianalto e19f987c4e add request command
CI / build (push) Has been cancelled
2021-01-24 22:57:55 -09:00
Dustin Pianalto 552664268b use custom prefixes
CI / build (push) Has been cancelled
2021-01-24 01:11:22 -09:00
Dustin Pianalto 414757188e remove duplicate prefixes
CI / build (push) Has been cancelled
2021-01-24 01:05:16 -09:00
Dustin Pianalto 888d356b67 debug removing prefix
CI / build (push) Has been cancelled
2021-01-24 00:57:00 -09:00
Dustin Pianalto c776b25899 debug removing prefix
CI / build (push) Has been cancelled
2021-01-24 00:53:04 -09:00
Dustin Pianalto 81f6e23487 Fix array in guild service
CI / build (push) Has been cancelled
2021-01-24 00:46:41 -09:00
Dustin Pianalto 6cf574b22d Add prefix commands
CI / build (push) Has been cancelled
2021-01-24 00:40:02 -09:00
Dustin Pianalto 623df1d55a Add prefix commands
CI / build (push) Has been cancelled
2021-01-24 00:26:34 -09:00
Dustin Pianalto 489d96594c Add prefix commands
CI / build (push) Has been cancelled
2021-01-24 00:24:07 -09:00
Dustin Pianalto 580eea3dfa Fix nullability of columns 2021-01-23 20:10:48 -09:00
Dustin Pianalto 7a8e7bf213 use nullable string in user
CI / build (push) Has been cancelled
2021-01-23 16:05:02 -09:00
Dustin Pianalto 96e8388562 fix bug in user database
CI / build (push) Has been cancelled
2021-01-23 15:10:04 -09:00
Dustin Pianalto 6369f37ce5 fix bug in user service
CI / build (push) Has been cancelled
2021-01-23 15:03:15 -09:00
Dustin Pianalto ab2d51c582 fix bug in user service
CI / build (push) Has been cancelled
2021-01-23 14:59:35 -09:00
Dustin Pianalto 5400d6acd9 fix bug in user service
CI / build (push) Has been cancelled
2021-01-23 14:48:59 -09:00
Dustin Pianalto 74edd574b3 fix bug in user service
CI / build (push) Has been cancelled
2021-01-23 14:46:25 -09:00
Dustin Pianalto 266cde6468 fix bug in running migrations
CI / build (push) Has been cancelled
2021-01-23 04:27:34 -09:00
Dustin Pianalto 8b646731ee change test command name
CI / build (push) Has been cancelled
2021-01-23 04:20:17 -09:00
Dustin Pianalto c54f8450b1 fix sql typo
CI / build (push) Has been cancelled
2021-01-23 04:14:29 -09:00
Dustin Pianalto 12f5b43f35 rework services and add test command
CI / build (push) Has been cancelled
2021-01-23 03:53:34 -09:00
Dustin Pianalto 9724f77c8e Initialize services 2021-01-23 03:11:12 -09:00
Dustin Pianalto 780613d7b9 initial database crud 2021-01-23 01:25:06 -09:00
Dustin Pianalto 94dcb738bc debug errors
CI / build (push) Has been cancelled
2021-01-21 00:19:52 -09:00
Dustin Pianalto 84e2bdea5e debug errors
CI / build (push) Has been cancelled
2021-01-20 23:58:25 -09:00
Dustin Pianalto 63e6212a0b debug errors
CI / build (push) Has been cancelled
2021-01-20 23:28:15 -09:00
Dustin Pianalto cbd899ff8c debug errors
CI / build (push) Has been cancelled
2021-01-20 23:22:23 -09:00
Dustin Pianalto c9ac3235db Fix syntax errors
CI / build (push) Has been cancelled
2021-01-20 23:18:30 -09:00
32 changed files with 1450 additions and 113 deletions
+1 -6
View File
@@ -21,17 +21,12 @@ jobs:
id: get_version id: get_version
uses: battila7/get-version-action@v2.0.0 uses: battila7/get-version-action@v2.0.0
- name: Setup Golang 1.15
uses: actions/setup-go@v1
with:
go-version: 1.15
- name: Install go-bindata and build migrations - name: Install go-bindata and build migrations
env: env:
GOPATH: /home/runner/work/Geeksbot/ GOPATH: /home/runner/work/Geeksbot/
run: | run: |
go get -u github.com/go-bindata/go-bindata/... go get -u github.com/go-bindata/go-bindata/...
/home/runner/work/Geeksbot/bin/go-bindata -pkg migrations -o $GITHUB_WORKSPACE/internal/database/migrations/bindata.go $GITHUB_WORKSPACE/internal/database/migrations/ /home/runner/work/Geeksbot/bin/go-bindata -pkg migrations -prefix $GITHUB_WORKSPACE/internal/database/migrations/ -o $GITHUB_WORKSPACE/internal/database/migrations/bindata.go $GITHUB_WORKSPACE/internal/database/migrations/
head $GITHUB_WORKSPACE/internal/database/migrations/bindata.go head $GITHUB_WORKSPACE/internal/database/migrations/bindata.go
- name: Build container image - name: Build container image
+6 -5
View File
@@ -9,9 +9,10 @@ type Channel struct {
} }
type ChannelService interface { type ChannelService interface {
Channel(id string) (*Channel, error) Channel(id string) (Channel, error)
CreateChannel(c *Channel) (*Channel, error) CreateChannel(c Channel) (Channel, error)
DeleteChannel(c *Channel) error DeleteChannel(c Channel) error
GuildChannels(g *Guild) ([]*Channel, error) GuildChannels(g Guild) ([]Channel, error)
UpdateChannel(c *Channel) (*Channel, error) UpdateChannel(c Channel) (Channel, error)
GetOrCreateChannel(id string, guild_id string) (Channel, error)
} }
+7 -18
View File
@@ -10,6 +10,7 @@ import (
"github.com/dustinpianalto/disgoman" "github.com/dustinpianalto/disgoman"
"github.com/dustinpianalto/geeksbot/internal/database" "github.com/dustinpianalto/geeksbot/internal/database"
"github.com/dustinpianalto/geeksbot/internal/exts" "github.com/dustinpianalto/geeksbot/internal/exts"
"github.com/dustinpianalto/geeksbot/internal/services"
) )
func main() { func main() {
@@ -27,11 +28,7 @@ func main() {
} }
database.ConnectDatabase(os.Getenv("DATABASE_URL")) database.ConnectDatabase(os.Getenv("DATABASE_URL"))
//postgres.InitializeDatabase() services.InitializeServices()
//utils.LoadTestData()
//us := &postgres.UserService{DB: postgres.DB}
//gs := &postgres.GuildService{DB: postgres.DB}
owners := []string{ owners := []string{
"351794468870946827", "351794468870946827",
@@ -49,18 +46,9 @@ func main() {
// Add Command Handlers // Add Command Handlers
exts.AddCommandHandlers(&manager) exts.AddCommandHandlers(&manager)
//services.InitalizeServices(us, gs)
//if _, ok := handler.Commands["help"]; !ok {
// handler.AddDefaultHelpCommand()
//}
dg.AddHandler(manager.OnMessage) dg.AddHandler(manager.OnMessage)
dg.AddHandler(manager.StatusManager.OnReady) dg.AddHandler(manager.StatusManager.OnReady)
//dg.AddHandler(guild_management.OnMessageUpdate)
//dg.AddHandler(guild_management.OnMessageDelete)
//dg.AddHandler(user_management.OnGuildMemberAddLogging)
//dg.AddHandler(user_management.OnGuildMemberRemoveLogging)
err = dg.Open() err = dg.Open()
if err != nil { if err != nil {
@@ -71,9 +59,6 @@ func main() {
// Start the Error handler in a goroutine // Start the Error handler in a goroutine
go ErrorHandler(manager.ErrorChannel) go ErrorHandler(manager.ErrorChannel)
// Start the Logging handler in a goroutine
//go logging.LoggingHandler(logging.LoggingChannel)
log.Println("The Bot is now running.") log.Println("The Bot is now running.")
sc := make(chan os.Signal, 1) sc := make(chan os.Signal, 1)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill) signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
@@ -87,7 +72,11 @@ func main() {
} }
func getPrefixes(guildID string) []string { func getPrefixes(guildID string) []string {
return []string{"G.", "g."} guild, err := services.GuildService.Guild(guildID)
if err != nil || len(guild.Prefixes) == 0 {
return []string{"G$", "g$"}
}
return guild.Prefixes
} }
func ErrorHandler(ErrorChan chan disgoman.CommandError) { func ErrorHandler(ErrorChan chan disgoman.CommandError) {
+1
View File
@@ -9,4 +9,5 @@ require (
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/hashicorp/errwrap v1.1.0 // indirect github.com/hashicorp/errwrap v1.1.0 // indirect
github.com/lib/pq v1.8.0
) )
+1
View File
@@ -294,6 +294,7 @@ golang.org/x/crypto v0.0.0-20190820162420-60c769a6c586/go.mod h1:yigFU9vqHzYiE8U
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
golang.org/x/crypto v0.0.0-20200510223506-06a226fb4e37/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= golang.org/x/crypto v0.0.0-20200510223506-06a226fb4e37/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/crypto v0.0.0-20200709230013-948cd5f35899 h1:DZhuSZLsGlFL4CmhA8BcRA0mnthyA/nZ00AqCUo7vHg=
golang.org/x/crypto v0.0.0-20200709230013-948cd5f35899/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= golang.org/x/crypto v0.0.0-20200709230013-948cd5f35899/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
golang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= golang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
+14 -11
View File
@@ -1,25 +1,28 @@
package geeksbot package geeksbot
import "database/sql"
type Guild struct { type Guild struct {
ID string ID string
NewPatronMessage string NewPatronMessage sql.NullString
Prefixes []string Prefixes []string
} }
type Role struct { type Role struct {
ID string ID string
RoleType int RoleType string
Guild Guild Guild Guild
} }
type GuildService interface { type GuildService interface {
Guild(id string) (*Guild, error) Guild(id string) (Guild, error)
CreateGuild(g *Guild) (*Guild, error) CreateGuild(g Guild) (Guild, error)
DeleteGuild(g *Guild) error DeleteGuild(g Guild) error
UpdateGuild(g *Guild) (*Guild, error) UpdateGuild(g Guild) (Guild, error)
GuildRoles(g *Guild) ([]*Role, error) GuildRoles(g Guild) ([]Role, error)
CreateRole(r *Role) (*Role, error) CreateRole(r Role) (Role, error)
Role(id string) (*Role, error) Role(id string) (Role, error)
UpdateRole(r *Role) (*Role, error) UpdateRole(r Role) (Role, error)
DeleteRole(r *Role) error DeleteRole(r Role) error
GetOrCreateGuild(id string) (Guild, error)
} }
+92
View File
@@ -0,0 +1,92 @@
package database
import (
"database/sql"
"log"
"github.com/dustinpianalto/geeksbot"
)
type channelService struct {
db *sql.DB
}
func (s channelService) Channel(id string) (geeksbot.Channel, error) {
var channel geeksbot.Channel
var guild_id string
queryString := "SELECT id, guild_id, admin, default_channel, new_patron FROM channels WHERE id = $1"
row := s.db.QueryRow(queryString, id)
err := row.Scan(&channel.ID, &guild_id, &channel.Admin, &channel.Default, &channel.NewPatron)
if err != nil {
return geeksbot.Channel{}, err
}
guild, err := GuildService.Guild(guild_id)
if err != nil {
return geeksbot.Channel{}, err
}
channel.Guild = guild
return channel, nil
}
func (s channelService) CreateChannel(c geeksbot.Channel) (geeksbot.Channel, error) {
queryString := "INSERT INTO channels (id, guild_id, admin, default_channel, new_patron VALUES ($1, $2, $3, $4, $5))"
_, err := s.db.Exec(queryString, c.ID, c.Guild.ID, c.Admin, c.Default, c.NewPatron)
return c, err
}
func (s channelService) DeleteChannel(c geeksbot.Channel) error {
queryString := "DELETE FROM channels WHERE id = $1"
_, err := s.db.Exec(queryString, c.ID)
return err
}
func (s channelService) UpdateChannel(c geeksbot.Channel) (geeksbot.Channel, error) {
queryString := "UPDATE channels SET admin = $2, default_channel = $3, new_patron = $4 WHERE id = $1"
_, err := s.db.Exec(queryString, c.ID, c.Admin, c.Default, c.NewPatron)
return c, err
}
func (s channelService) GuildChannels(g geeksbot.Guild) ([]geeksbot.Channel, error) {
var channels []geeksbot.Channel
queryString := "SELECT id FROM channels WHERE guild_id = $1"
rows, err := s.db.Query(queryString, g.ID)
for rows.Next() {
var id string
err = rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
channel, err := s.Channel(id)
if err != nil {
log.Println(err)
continue
}
channels = append(channels, channel)
}
return channels, nil
}
func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot.Channel, error) {
channel, err := s.Channel(id)
log.Println(err)
log.Println(1)
if err.Error() == sql.ErrNoRows.Error() {
guild, err := GuildService.GetOrCreateGuild(guild_id)
log.Println(err)
log.Println(2)
if err != nil {
return geeksbot.Channel{}, err
}
channel, err = s.CreateChannel(geeksbot.Channel{
ID: id,
Guild: guild,
Admin: false,
Default: false,
NewPatron: false,
})
}
log.Println(err)
log.Println(3)
return channel, err
}
+28 -4
View File
@@ -12,7 +12,14 @@ import (
) )
var ( var (
db *sql.DB db *sql.DB
GuildService guildService
UserService userService
ChannelService channelService
MessageService messageService
RequestService requestService
PatreonService patreonService
ServerService serverService
) )
func ConnectDatabase(dbConnString string) { func ConnectDatabase(dbConnString string) {
@@ -20,7 +27,8 @@ func ConnectDatabase(dbConnString string) {
func(name string) ([]byte, error) { func(name string) ([]byte, error) {
return migrations.Asset(name) return migrations.Asset(name)
}) })
db, err := sql.Open("postgres", dbConnString) var err error
db, err = sql.Open("postgres", dbConnString)
if err != nil { if err != nil {
log.Fatal(fmt.Errorf("Can't connect to the database: %w", err)) log.Fatal(fmt.Errorf("Can't connect to the database: %w", err))
} }
@@ -36,13 +44,29 @@ func ConnectDatabase(dbConnString string) {
if err != nil { if err != nil {
log.Fatal(fmt.Errorf("cannot create db driver: %w", err)) log.Fatal(fmt.Errorf("cannot create db driver: %w", err))
} }
m, err := migrate.NewWithInstance("go-bindata", d, "postgres", instance) m, err := migrate.NewWithInstance("go-bindatafoo", d, "postgres", instance)
if err != nil { if err != nil {
log.Fatal(fmt.Errorf("cannot create migration instance: %w", err)) log.Fatal(fmt.Errorf("cannot create migration instance: %w", err))
} }
err = m.Up() err = m.Up()
if err != nil { if err != nil {
log.Fatal(fmt.Errorf("error running migrations: %w", err)) if err.Error() == "no change" {
log.Println(err)
} else {
log.Fatal(err)
}
} }
log.Println("Migrations Run") log.Println("Migrations Run")
initServices()
log.Println("Services Initialized")
}
func initServices() {
GuildService = guildService{db: db}
UserService = userService{db: db}
ChannelService = channelService{db: db}
MessageService = messageService{db: db}
PatreonService = patreonService{db: db}
RequestService = requestService{db: db}
ServerService = serverService{db: db}
} }
+113
View File
@@ -0,0 +1,113 @@
package database
import (
"database/sql"
"log"
"github.com/dustinpianalto/geeksbot"
"github.com/lib/pq"
)
type guildService struct {
db *sql.DB
}
func (s guildService) Guild(id string) (geeksbot.Guild, error) {
var g geeksbot.Guild
queryString := "SELECT id, new_patron_message, prefixes FROM guilds WHERE id = $1"
row := s.db.QueryRow(queryString, id)
err := row.Scan(&g.ID, &g.NewPatronMessage, pq.Array(&g.Prefixes))
if err != nil {
return geeksbot.Guild{}, err
}
return g, nil
}
func (s guildService) CreateGuild(g geeksbot.Guild) (geeksbot.Guild, error) {
queryString := "INSERT INTO guilds (id, new_patron_message, prefixes) VALUES ($1, $2, $3)"
_, err := s.db.Exec(queryString, g.ID, g.NewPatronMessage, pq.Array(g.Prefixes))
return g, err
}
func (s guildService) DeleteGuild(g geeksbot.Guild) error {
queryString := "DELETE FROM guilds WHERE id = $1"
_, err := s.db.Exec(queryString, g.ID)
return err
}
func (s guildService) UpdateGuild(g geeksbot.Guild) (geeksbot.Guild, error) {
queryString := "UPDATE guilds SET new_patron_message = $2, prefixes = $3 WHERE id = $1"
_, err := s.db.Exec(queryString, g.ID, g.NewPatronMessage, pq.Array(g.Prefixes))
return g, err
}
func (s guildService) GuildRoles(g geeksbot.Guild) ([]geeksbot.Role, error) {
var roles []geeksbot.Role
queryString := "SELECT id FROM roles WHERE guild_id = $1"
rows, err := s.db.Query(queryString, g.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id string
err = rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
role, err := s.Role(id)
if err != nil {
log.Println(err)
continue
}
roles = append(roles, role)
}
return roles, nil
}
func (s guildService) CreateRole(r geeksbot.Role) (geeksbot.Role, error) {
queryString := "INSERT INTO roles (id, role_type, guild_id) VALUES ($1, $2, $3)"
_, err := s.db.Exec(queryString, r.ID, r.RoleType, r.Guild.ID)
return r, err
}
func (s guildService) Role(id string) (geeksbot.Role, error) {
var role geeksbot.Role
var guild_id string
queryString := "SELECT id, role_type, guild_id FROM roles WHERE id = $1"
row := s.db.QueryRow(queryString, id)
err := row.Scan(&role.ID, &role.RoleType, &guild_id)
if err != nil {
return geeksbot.Role{}, err
}
guild, err := s.Guild(guild_id)
if err != nil {
return geeksbot.Role{}, err
}
role.Guild = guild
return role, nil
}
func (s guildService) UpdateRole(r geeksbot.Role) (geeksbot.Role, error) {
queryString := "UPDATE roles SET role_type = $2 WHERE id = $1"
_, err := s.db.Exec(queryString, r.ID, r.RoleType)
return r, err
}
func (s guildService) DeleteRole(r geeksbot.Role) error {
queryString := "DELETE FROM roles WHERE id = $1"
_, err := s.db.Exec(queryString, r.ID)
return err
}
func (s guildService) GetOrCreateGuild(id string) (geeksbot.Guild, error) {
guild, err := s.Guild(id)
if err == sql.ErrNoRows {
guild = geeksbot.Guild{
ID: id,
Prefixes: []string{},
}
guild, err = s.CreateGuild(guild)
}
return guild, err
}
+94
View File
@@ -0,0 +1,94 @@
package database
import (
"database/sql"
"log"
"github.com/dustinpianalto/geeksbot"
)
type messageService struct {
db *sql.DB
}
func (s messageService) Message(id string) (geeksbot.Message, error) {
var m geeksbot.Message
var channel_id string
var author_id string
queryString := `SELECT m.id, m.created_at, m.modified_at, m.content,
m.previous_content, m.channel_id, m.author_id
FROM messages as m
WHERE m.id = $1`
row := s.db.QueryRow(queryString, id)
err := row.Scan(&m.ID, &m.CreatedAt, &m.ModifiedAt, &m.Content,
&m.PreviousContent, &channel_id, &author_id)
if err != nil {
return geeksbot.Message{}, err
}
author, err := UserService.User(author_id)
if err != nil {
return geeksbot.Message{}, err
}
m.Author = author
channel, err := ChannelService.Channel(channel_id)
if err != nil {
return geeksbot.Message{}, err
}
m.Channel = channel
return m, nil
}
func (s messageService) CreateMessage(m geeksbot.Message) (geeksbot.Message, error) {
queryString := `INSERT INTO messages (id, created_at, content, channel_id, author_id)
VALUES ($1, $2, $3, $4, $5)`
_, err := s.db.Exec(queryString, m.ID, m.CreatedAt, m.Content, m.Channel.ID, m.Author.ID)
return m, err
}
func (s messageService) UpdateMessage(m geeksbot.Message) (geeksbot.Message, error) {
var content string
var previousContent []string
queryString := "SELECT content, previous_content FROM messages WHERE id = $1"
row := s.db.QueryRow(queryString, m.ID)
err := row.Scan(&content, &previousContent)
if err != nil {
return geeksbot.Message{}, err
}
if m.Content != content {
previousContent = append(previousContent, content)
}
queryString = "UPDATE messages SET modified_at = $2, content = $3, previous_content = $4 WHERE id = $1"
_, err = s.db.Exec(queryString, m.ID, m.ModifiedAt, m.Content, previousContent)
m.PreviousContent = previousContent
return m, nil
}
func (s messageService) DeleteMessage(m geeksbot.Message) error {
queryString := "DELETE FROM messages WHERE id = $1"
_, err := s.db.Exec(queryString, m.ID)
return err
}
func (s messageService) ChannelMessages(c geeksbot.Channel) ([]geeksbot.Message, error) {
var messages []geeksbot.Message
queryString := `SELECT id FROM messages WHERE channel_id = $1`
rows, err := s.db.Query(queryString, c.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id string
err = rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
m, err := s.Message(id)
if err != nil {
log.Println(err)
continue
}
messages = append(messages, m)
}
return messages, nil
}
@@ -5,7 +5,7 @@ BEGIN;
prefixes varchar(10)[], prefixes varchar(10)[],
PRIMARY KEY(id) PRIMARY KEY(id)
); );
CREATE TYPE IF NOT EXISTS role_type as ENUM ( CREATE TYPE role_type as ENUM (
'normal', 'normal',
'moderator', 'moderator',
'admin', 'admin',
@@ -25,7 +25,7 @@ BEGIN;
id varchar(30), id varchar(30),
guild_id varchar(30), guild_id varchar(30),
admin boolean, admin boolean,
default boolean, default_channel boolean,
new_patron boolean, new_patron boolean,
PRIMARY KEY(id), PRIMARY KEY(id),
CONSTRAINT fk_guild CONSTRAINT fk_guild
@@ -45,7 +45,7 @@ BEGIN;
id varchar(30), id varchar(30),
created_at timestamp, created_at timestamp,
modified_at timestamp, modified_at timestamp,
content varchar(2000) content varchar(2000),
previous_content varchar(2000)[], previous_content varchar(2000)[],
channel_id varchar(30), channel_id varchar(30),
author_id varchar(30), author_id varchar(30),
@@ -89,7 +89,7 @@ BEGIN;
REFERENCES roles(id) REFERENCES roles(id)
ON DELETE SET NULL, ON DELETE SET NULL,
CONSTRAINT fk_tier CONSTRAINT fk_tier
FOREIGN_KEY(next_tier) FOREIGN KEY(next_tier)
REFERENCES patreon_tier(id) REFERENCES patreon_tier(id)
ON DELETE SET NULL ON DELETE SET NULL
); );
@@ -0,0 +1 @@
ALTER TABLE requests DROP COLUMN guild_id;
@@ -0,0 +1,2 @@
ALTER TABLE requests
ADD COLUMN guild_id varchar(30) CONSTRAINT fk_guild REFERENCES guilds(id) ON DELETE CASCADE;
@@ -0,0 +1,48 @@
BEGIN;
ALTER TABLE users
ALTER COLUMN active DROP NOT NULL,
ALTER COLUMN staff DROP NOT NULL,
ALTER COLUMN admin DROP NOT NULL;
ALTER TABLE channels
ALTER COLUMN guild_id DROP NOT NULL,
ALTER COLUMN admin DROP NOT NULL,
ALTER COLUMN default_channel DROP NOT NULL,
ALTER COLUMN new_patron DROP NOT NULL;
ALTER TABLE messages
ALTER COLUMN created_at DROP NOT NULL,
ALTER COLUMN content DROP NOT NULL,
ALTER COLUMN channel_id DROP NOT NULL,
ALTER COLUMN author_id DROP NOT NULL;
ALTER TABLE messages
ADD COLUMN embed json,
ADD COLUMN previous_embeds json[];
ALTER TABLE patreon_creators
ALTER COLUMN creator DROP NOT NULL,
ALTER COLUMN link DROP NOT NULL,
ALTER COLUMN guild_id DROP NOT NULL;
ALTER TABLE patreon_creators
RENAME TO patreon_creator;
ALTER TABLE patreon_tiers
ALTER COLUMN name DROP NOT NULL,
ALTER COLUMN creator DROP NOT NULL,
ALTER COLUMN role DROP NOT NULL;
ALTER TABLE patreon_tiers
RENAME TO patreon_tier;
ALTER TABLE requests
ALTER COLUMN author_id DROP NOT NULL,
ALTER COLUMN channel_id DROP NOT NULL,
ALTER COLUMN content DROP NOT NULL,
ALTER COLUMN requested_at DROP NOT NULL,
ALTER COLUMN completed DROP NOT NULL,
ALTER COLUMN message_id DROP NOT NULL,
ALTER COLUMN guild_id DROP NOT NULL;
ALTER TABLE roles
ALTER COLUMN role_type DROP NOT NULL,
ALTER COLUMN guild_id DROP NOT NULL;
ALTER TABLE servers
ALTER COLUMN name DROP NOT NULL,
ALTER COLUMN ip_address DROP NOT NULL,
ALTER COLUMN port DROP NOT NULL,
ALTER COLUMN password DROP NOT NULL,
ALTER COLUMN guild_id DROP NOT NULL;
COMMIT;
@@ -0,0 +1,48 @@
BEGIN;
ALTER TABLE users
ALTER COLUMN active SET NOT NULL,
ALTER COLUMN staff SET NOT NULL,
ALTER COLUMN admin SET NOT NULL;
ALTER TABLE channels
ALTER COLUMN guild_id SET NOT NULL,
ALTER COLUMN admin SET NOT NULL,
ALTER COLUMN default_channel SET NOT NULL,
ALTER COLUMN new_patron SET NOT NULL;
ALTER TABLE messages
ALTER COLUMN created_at SET NOT NULL,
ALTER COLUMN content SET NOT NULL,
ALTER COLUMN channel_id SET NOT NULL,
ALTER COLUMN author_id SET NOT NULL;
ALTER TABLE messages
DROP COLUMN embed,
DROP COLUMN previous_embeds;
ALTER TABLE patreon_creator
ALTER COLUMN creator SET NOT NULL,
ALTER COLUMN link SET NOT NULL,
ALTER COLUMN guild_id SET NOT NULL;
ALTER TABLE patreon_creator
RENAME TO patreon_creators;
ALTER TABLE patreon_tier
ALTER COLUMN name SET NOT NULL,
ALTER COLUMN creator SET NOT NULL,
ALTER COLUMN role SET NOT NULL;
ALTER TABLE patreon_tier
RENAME TO patreon_tiers;
ALTER TABLE requests
ALTER COLUMN author_id SET NOT NULL,
ALTER COLUMN channel_id SET NOT NULL,
ALTER COLUMN content SET NOT NULL,
ALTER COLUMN requested_at SET NOT NULL,
ALTER COLUMN completed SET NOT NULL,
ALTER COLUMN message_id SET NOT NULL,
ALTER COLUMN guild_id SET NOT NULL;
ALTER TABLE roles
ALTER COLUMN role_type SET NOT NULL,
ALTER COLUMN guild_id SET NOT NULL;
ALTER TABLE servers
ALTER COLUMN name SET NOT NULL,
ALTER COLUMN ip_address SET NOT NULL,
ALTER COLUMN port SET NOT NULL,
ALTER COLUMN password SET NOT NULL,
ALTER COLUMN guild_id SET NOT NULL;
COMMIT;
+2 -2
View File
@@ -98,7 +98,7 @@ func _000001_init_schemaDownSql() (*asset, error) {
return a, nil return a, nil
} }
var __000001_init_schemaUpSql = []byte("\x1f\x8b\x08\x00\x00\x00\x00\x00\x00\xff\xd4\x58\x4d\x6f\xab\x38\x14\xdd\xf7\x57\x78\x97\x44\xea\xa2\x33\xa3\x59\x75\x45\x13\xb7\x42\x93\x90\x8a\x50\xcd\x44\xa3\x11\x72\xc3\x4d\xea\x29\xd8\x3c\xdb\x49\xdb\x7f\xff\x04\xe5\xc3\x01\x0c\x7e\x2a\xaf\x7a\x65\x15\xf9\x5e\xec\xe3\x73\xcf\xfd\x08\x37\xf8\xce\xf5\xae\x2f\x10\x42\x68\xee\x63\x27\xc0\x28\x70\x6e\x96\x18\xb9\xb7\xc8\x5b\x07\x08\xff\xe3\x6e\x82\x0d\x3a\x1c\x69\x1c\x49\x34\xcd\xfd\xb2\x87\x46\xe8\x44\xc4\xee\x89\x88\xe9\x1f\x57\xb3\xcb\x6a\x9d\xc1\x4b\x98\x12\x25\x38\x0b\x13\x90\x92\x1c\xa0\xf2\xfb\xed\xea\x4a\xf7\x4c\x05\xec\xe9\x2b\x48\xcd\x3e\xfb\xf7\xbf\xda\x7e\xef\xbb\x2b\xc7\xdf\xa2\xbf\xf0\x76\x4a\xa3\x59\xbe\x3e\x3b\x07\xba\xbd\x6f\xe2\x14\x3c\x86\x50\xbd\xa5\x80\x88\x44\xd8\x7b\x58\x69\x90\x27\x8c\x8b\x84\xc4\x93\xfa\x88\x49\xc2\x23\x10\x44\x71\xa1\x2f\x92\x28\xa1\x4c\x5f\xc8\xee\x03\x9c\x4d\xba\x20\x74\x70\x95\x61\xb0\xa1\xaa\xc6\x5a\xfd\xaa\x8d\x39\xe1\xa1\xe9\xd5\x06\x37\xb5\x61\xbe\xf6\x36\x81\xef\xb8\x5e\x80\xf6\xcf\x61\xbe\x49\x65\xcb\x9e\xdb\xb5\x8f\xdd\x3b\x2f\x7f\xb1\x3c\x62\x76\xe6\x91\x3d\x3e\xbe\xc5\x3e\xf6\xe6\xb8\x0c\xfc\xb4\xcb\x6b\xed\xa1\x05\x5e\xe2\x00\xa3\xb9\xb3\x99\x3b\x0b\x6c\xc9\xcf\xee\x89\x30\x06\xb1\x0d\x45\xfd\x2c\xe4\x81\x42\x8f\x9c\xc7\x40\x58\xbd\x1c\xc1\x9e\x1c\x63\xd5\x36\xd4\xda\x6c\xdb\xbe\x34\xa3\x47\x09\xc2\x86\x4e\xa9\x80\x24\x66\x3a\x77\x8a\x9e\xa0\x4d\x8d\x54\x64\xbf\x6f\x2f\x1b\xd8\xb7\x48\xdb\x8e\x1b\x14\xb5\xc2\xe6\x12\x3b\x01\x44\x41\x14\x12\x85\x14\x4d\x40\x2a\x92\xa4\xb5\x35\xe1\x11\xdd\x53\xa3\x79\xc7\x99\x02\xa6\xaa\x9d\x7f\xcf\x6a\x92\x5e\x92\x4e\x94\x1f\x65\xd8\xe9\xa6\x17\xa7\x42\xc3\x66\x2e\x8f\xea\x89\x0b\xa3\x19\x92\x47\x88\xd0\xff\x92\xb3\xcb\xf6\xe1\xb9\x51\xe6\xd6\x9e\x7a\x68\x52\x68\x81\xcc\xa8\xd1\x1a\x79\xaf\x4a\xcb\x24\xb5\xd3\xa9\x09\x4c\x26\x4c\x23\x92\x8a\xa3\x5e\x20\xb9\xb6\x07\x50\x6c\x70\x80\xbc\x87\xe5\xd2\x52\x6c\x45\x3d\x0f\x73\x25\x71\x71\xae\x39\xca\x14\x1c\x40\xa0\x3b\xec\x61\xdf\x09\xf0\x02\x39\xcb\xbf\x9d\xed\x06\x39\x1b\xe4\x2e\xb0\x17\xb8\xc1\xb6\x21\x46\x2e\xf4\x1e\xa7\x45\x26\xa6\xec\x59\xd7\x90\x75\x65\xfb\xd2\xd5\xa8\xa4\x57\x51\x68\x70\xfb\x83\xe4\x32\x92\x80\x81\xd9\x08\xe4\x4e\xd0\x54\x51\xce\x4c\xf3\x45\x19\x9a\xe2\xcc\xf3\xbe\x6b\x1a\x5e\x5e\xd5\x3b\xec\xd6\x4b\xb6\xd9\xf7\x7e\xa8\x39\xfb\xde\xed\xbd\x21\x69\xc8\xf3\x63\x19\x98\x5d\xd6\x88\x26\x33\xf6\x42\xc9\x47\x19\xcb\xe4\x33\x21\xc8\xf8\xec\x42\x10\x66\x08\x2a\xc6\xad\x18\xc9\x1c\xc7\x2d\x05\x02\xbe\x1d\x41\x2a\xf9\x91\x1a\x30\x50\xeb\x87\x5a\x45\xb3\xd7\x34\x64\x5c\x20\xec\xe9\x68\x49\x1a\x83\x85\xbd\xdd\xaa\xeb\x57\x1f\xdf\xba\xb1\x15\x6d\xb9\x07\x7b\xb9\xc3\xc0\xb0\x6f\x99\x3e\x9f\xd4\x2f\x06\x72\xe6\x97\x6a\xa1\x7a\x90\xcc\x88\x34\xa7\xd1\xba\xa9\x09\x51\x11\x6a\x23\x98\x5a\x34\xbd\x50\xca\x91\x6f\xdc\xff\x16\x3c\x49\x80\xfd\xd4\x7c\x2e\x12\x32\xac\xf7\x3d\xcb\x87\xec\x78\xcb\xe1\xf3\x2b\xe7\x48\xc1\x82\xb9\xb5\x54\x2c\xf5\x37\x98\xa2\xfe\x8e\x2a\x02\x09\xe2\xd4\xfa\x43\x34\xda\xe8\x41\xd3\x90\x44\x91\x00\xa9\x7d\xb9\xf8\x53\xff\xae\xc1\x85\x6a\x2b\x23\x25\x52\xbe\x70\x11\x19\xc6\x41\x12\x83\x50\x32\x1c\xea\x15\xfd\x53\x23\x65\x7b\x3e\xb8\x45\xee\x34\x54\xd7\x25\x28\x45\xd9\x41\x0e\x3a\x5a\x6a\x36\xbf\xde\x60\x5d\x6d\x91\x30\x46\x79\x1d\xaa\x66\x9f\x3c\x43\x9b\x60\xe8\xb1\x33\xa2\x69\x04\xf8\x33\xe8\xd1\xe5\xd2\x8f\x6b\xdc\xb2\x3f\x84\xab\xa9\x50\x23\xb6\x0e\x29\x8f\x89\xaf\x2c\x49\xf3\xf5\x6a\xe5\x06\xd7\x17\xdf\x03\x00\x00\xff\xff\x77\x2d\xda\x08\x4f\x15\x00\x00") var __000001_init_schemaUpSql = []byte("\x1f\x8b\x08\x00\x00\x00\x00\x00\x00\xff\xd4\x58\x5d\x6f\xa3\x38\x14\x7d\xef\xaf\xf0\x5b\x12\x69\x1f\xba\xbb\xda\xa7\x3e\xd1\xc4\xad\xd0\x26\xa4\x22\x54\xbb\xd1\x68\x84\xdc\x70\x93\x7a\x0a\x36\x63\x3b\x69\xfb\xef\x47\x50\x3e\x1c\xc0\xe0\x51\x99\x6a\xca\x53\xe4\x7b\xb1\x8f\xcf\x3d\xf7\x83\x5c\xe3\x5b\xd7\xbb\xba\x40\x08\xa1\xb9\x8f\x9d\x00\xa3\xc0\xb9\x5e\x62\xe4\xde\x20\x6f\x1d\x20\xfc\xbf\xbb\x09\x36\xe8\x70\xa4\x71\x24\xd1\x34\xf7\xcb\x1e\x1a\xa1\x13\x11\xbb\x47\x22\xa6\x7f\x5f\xce\xfe\xa8\xd6\x19\x3c\x87\x29\x51\x82\xb3\x30\x01\x29\xc9\x01\x2a\xbf\x3f\x2f\x2f\x75\xcf\x54\xc0\x9e\xbe\x80\xd4\xec\xb3\x2f\x5f\x6b\xfb\x9d\xef\xae\x1c\x7f\x8b\xfe\xc5\xdb\x29\x8d\x66\xf9\xfa\xec\x1c\xe8\xf6\x0e\x23\xc1\x63\x08\xd5\x6b\x0a\x88\x48\x84\xbd\xfb\x95\x06\x72\xc2\xb8\x48\x48\x3c\xa9\x37\x9d\x24\x3c\x02\x41\x14\x17\xfa\x22\x89\x12\xca\xf4\x85\xec\x06\xc0\xd9\xa4\xeb\xd0\x0e\x76\x32\x0c\x36\xe4\xd4\x58\xab\x5f\xb5\x31\xa7\x38\x34\xbd\xda\x60\xa3\x36\xcc\xd7\xde\x26\xf0\x1d\xd7\x0b\xd0\xfe\x29\xcc\x37\xa9\x6c\xd9\x73\xb3\xf6\xb1\x7b\xeb\xe5\x2f\x96\x47\xcc\xce\x3c\xb2\xc7\xc7\x37\xd8\xc7\xde\x1c\x97\xa1\x9e\x76\x79\xad\x3d\xb4\xc0\x4b\x1c\x60\x34\x77\x36\x73\x67\x81\x2d\xf9\xd9\x3d\x12\xc6\x20\xb6\xa1\xa8\x9f\x85\x3c\x50\xe8\x81\xf3\x18\x08\xab\x97\x23\xd8\x93\x63\xac\xc2\xe2\x9c\xb6\x43\xad\xca\xb6\xed\x53\x33\x7b\x94\x20\x6c\x68\x95\x0a\x48\x62\xa6\x75\xa7\xe8\x09\xda\xd4\x48\x45\xf6\xfb\xf6\xb2\x21\x0a\x16\x09\xdb\x71\x83\xa2\x4a\xd8\x5c\x62\x27\x80\x28\x88\x42\xa2\x90\xa2\x09\x48\x45\x92\xb4\xb6\x26\x3c\xa2\x7b\x6a\x34\xef\x38\x53\xc0\x54\xb5\xf3\x5f\xad\x6a\x74\xa2\xfc\x28\xc3\x4e\x3f\xbd\x2e\x15\x22\x33\x93\x79\x54\x8f\x5c\x18\xcd\x90\x3c\x40\x84\xbe\x49\xce\x3a\x0e\xcf\x8d\x32\xb7\xf6\x94\x42\x93\x44\x0b\x64\x46\x91\xd6\xc8\x7b\x65\x5a\x66\xab\x9d\x50\x4d\x60\x32\x65\x1a\x91\x54\x1c\xf5\x02\xc9\xc5\x3d\x80\x62\x83\x03\xe4\xdd\x2f\x97\x96\x6a\x2b\x0a\x7b\x98\x4b\x89\x8b\x73\xd1\x51\xa6\xe0\x00\x02\xdd\x62\x0f\xfb\x4e\x80\x17\xc8\x59\xfe\xe7\x6c\x37\xc8\xd9\x20\x77\x81\xbd\xc0\x0d\xb6\x0d\x35\x72\xa1\xb7\x37\x2d\x32\x31\x65\x4f\xba\x86\xac\x4b\xdc\xa7\x2e\x47\x25\xbd\x8a\x42\x83\xdb\x9f\x24\x97\x91\x04\x0c\xcc\x46\x20\x77\x82\xa6\x8a\x72\x66\x1a\x2d\xca\xd0\x14\x67\x9e\x37\x60\xd3\xdc\xf2\xa2\xde\x60\xb7\x5e\xb2\xcd\xbe\xb7\x43\xcd\xd9\xf7\x66\xef\x0d\x49\x43\x9e\xef\xcb\xc0\xec\xb2\x46\x34\x99\xb1\x17\x4a\x3e\xd3\x58\x26\x9f\x09\x41\xc6\xa7\x11\x41\xc5\xb8\x15\x23\x99\xe3\xb8\xa5\x40\xc0\xf7\x23\x48\x25\xdf\x53\x03\x06\x6a\xfd\x50\xab\x68\xf6\x9a\x86\x8c\x0b\x84\x3d\x2d\x2d\x49\x63\xb0\xb0\xb7\x7b\x75\xfd\xea\xc3\x6b\x37\xb6\xa2\x2f\xf7\x60\x2f\x77\x18\x98\xf3\x2d\xd3\xe7\x83\xfa\xc5\x40\xce\xfc\x56\x2d\x54\x0f\x92\x19\x91\xe6\x34\x5a\x37\x35\x21\x2a\x42\x6d\x04\x53\x8b\xa6\x17\x4a\x39\xf3\x8d\xfb\x91\xc1\x93\x04\xd8\x2f\xcd\xe7\x22\x21\xc3\x7a\xdf\xb3\x7c\xc8\x8e\xb7\x9c\x3e\x3f\x73\x8e\x14\x2c\x98\x5b\x4b\xc5\x52\x7f\x83\x29\xea\xef\xa8\x22\x90\x20\x4e\xad\x2f\xa2\xd1\x46\x0f\x9a\x86\x24\x8a\x04\x48\xed\x4f\x8b\x7f\xf4\x8f\x08\x2e\x54\x5b\x19\x29\x91\xf2\x99\x8b\xc8\x30\x0e\x92\x18\x84\x92\xe1\x50\xaf\xe8\x9f\x1a\x29\xdb\xf3\xc1\x2d\x72\xa7\xa1\xba\x2e\x41\x29\xca\x0e\x72\xd0\xd1\x52\xb3\xf9\xf5\x06\xeb\x6a\x8b\x84\x31\xca\xeb\x50\x35\xfb\xe0\x19\xda\x04\x43\x8f\x9d\x11\x4d\x23\xc0\x1f\x41\x8f\x2e\x97\x7e\x5c\xe3\x96\xfd\x21\x5c\x4d\x85\x1a\xb1\x75\x48\x79\x4c\x7c\x65\x49\x9a\xaf\x57\x2b\x37\xb8\xba\xf8\x11\x00\x00\xff\xff\xb3\xc0\x11\x7f\x4a\x15\x00\x00")
func _000001_init_schemaUpSqlBytes() ([]byte, error) { func _000001_init_schemaUpSqlBytes() ([]byte, error) {
return bindataRead( return bindataRead(
@@ -113,7 +113,7 @@ func _000001_init_schemaUpSql() (*asset, error) {
return nil, err return nil, err
} }
info := bindataFileInfo{name: "000001_init_schema.up.sql", size: 5455, mode: os.FileMode(420), modTime: time.Unix(1611206743, 0)} info := bindataFileInfo{name: "000001_init_schema.up.sql", size: 5450, mode: os.FileMode(420), modTime: time.Unix(1611216889, 0)}
a := &asset{bytes: bytes, info: info} a := &asset{bytes: bytes, info: info}
return a, nil return a, nil
} }
+178
View File
@@ -0,0 +1,178 @@
package database
import (
"database/sql"
"log"
"github.com/dustinpianalto/geeksbot"
)
type patreonService struct {
db *sql.DB
}
func (s patreonService) PatreonCreatorByID(id int) (geeksbot.PatreonCreator, error) {
var creator geeksbot.PatreonCreator
var gID string
queryString := "SELECT id, creator, link, guild_id FROM patreon_creator WHERE id = $1"
row := s.db.QueryRow(queryString, id)
err := row.Scan(&creator.ID, &creator.Creator, &creator.Link, &gID)
if err != nil {
return geeksbot.PatreonCreator{}, err
}
guild, err := GuildService.Guild(gID)
if err != nil {
return geeksbot.PatreonCreator{}, err
}
creator.Guild = guild
return creator, nil
}
func (s patreonService) PatreonCreatorByName(name string, g geeksbot.Guild) (geeksbot.PatreonCreator, error) {
var id int
queryString := "SELECT id FROM patreon_creator WHERE creator = $1 AND guild_id = $2"
err := s.db.QueryRow(queryString, name, g.ID).Scan(&id)
if err != nil {
return geeksbot.PatreonCreator{}, nil
}
creator, err := s.PatreonCreatorByID(id)
return creator, err
}
func (s patreonService) CreatePatreonCreator(c geeksbot.PatreonCreator) (geeksbot.PatreonCreator, error) {
var id int
queryString := `INSERT INTO patreon_creator (creator, link, guild_id) VALUES ($1, $2, $3) RETURNING id`
err := s.db.QueryRow(queryString, c.Creator, c.Link, c.Guild.ID).Scan(&id)
if err != nil {
return geeksbot.PatreonCreator{}, err
}
c.ID = id
return c, nil
}
func (s patreonService) UpdatePatreonCreator(c geeksbot.PatreonCreator) (geeksbot.PatreonCreator, error) {
queryString := `UPDATE patreon_creator SET creator = $2, link = $3 WHERE id = $1`
_, err := s.db.Exec(queryString, c.ID, c.Creator, c.Link)
return c, err
}
func (s patreonService) DeletePatreonCreator(c geeksbot.PatreonCreator) error {
queryString := `DELETE FROM patreon_creator WHERE id = $1`
_, err := s.db.Exec(queryString, c.ID)
return err
}
func (s patreonService) PatreonTierByID(id int) (geeksbot.PatreonTier, error) {
var tier geeksbot.PatreonTier
var cID int
var rID string
var next int
queryString := `SELECT id, name, description, creator, role, next_tier FROM patreon_tier WHERE id = id`
err := s.db.QueryRow(queryString, id).Scan(&tier.ID, &tier.Name, &tier.Description, &cID, &rID, &next)
if err != nil {
return geeksbot.PatreonTier{}, err
}
creator, err := s.PatreonCreatorByID(cID)
if err != nil {
return geeksbot.PatreonTier{}, err
}
tier.Creator = creator
role, err := GuildService.Role(rID)
if err != nil {
return geeksbot.PatreonTier{}, err
}
tier.Role = role
if next == -1 {
tier.NextTier = nil
return tier, nil
}
nextTier, err := s.PatreonTierByID(next)
if err != nil {
return geeksbot.PatreonTier{}, err
}
tier.NextTier = &nextTier
return tier, nil
}
func (s patreonService) PatreonTierByName(name string, creator string) (geeksbot.PatreonTier, error) {
var id int
queryString := `SELECT id FROM patreon_tier WHERE name = $1 AND creator = $2`
err := s.db.QueryRow(queryString, name, creator).Scan(&id)
if err != nil {
return geeksbot.PatreonTier{}, err
}
tier, err := s.PatreonTierByID(id)
return tier, err
}
func (s patreonService) CreatePatreonTier(t geeksbot.PatreonTier) (geeksbot.PatreonTier, error) {
var id int
queryString := `INSERT INTO patreon_tier (name, description, creator, role, next_tier)
VALUES ($1, $2, $3, $4, $5) RETURNING id`
err := s.db.QueryRow(queryString, t.Name, t.Description, t.Creator.ID, t.Role.ID, t.NextTier.ID).Scan(&id)
if err != nil {
return geeksbot.PatreonTier{}, err
}
t.ID = id
return t, nil
}
func (s patreonService) UpdatePatreonTier(t geeksbot.PatreonTier) (geeksbot.PatreonTier, error) {
queryString := `UPDATE patreon_tier SET name = $2, description = $3, role = $4, next_tier = $5 WHERE id = $1`
_, err := s.db.Exec(queryString, t.ID, t.Name, t.Description, t.Role.ID, t.NextTier.ID)
return t, err
}
func (s patreonService) DeletePatreonTier(t geeksbot.PatreonTier) error {
queryString := `DELETE FROM patreon_tier WHERE id = $1`
_, err := s.db.Exec(queryString, t.ID)
return err
}
func (s patreonService) GuildPatreonCreators(g geeksbot.Guild) ([]geeksbot.PatreonCreator, error) {
var creators []geeksbot.PatreonCreator
queryString := `SELECT id FROM patreon_creator WHERE guild_id = $1`
rows, err := s.db.Query(queryString, g.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id int
err := rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
creator, err := s.PatreonCreatorByID(id)
if err != nil {
log.Println(err)
continue
}
creators = append(creators, creator)
}
return creators, nil
}
func (s patreonService) CreatorPatreonTiers(c geeksbot.PatreonCreator) ([]geeksbot.PatreonTier, error) {
var tiers []geeksbot.PatreonTier
queryString := `SELECT id FROM patreon_tier WHERE creator = $1`
rows, err := s.db.Query(queryString, c.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id int
err := rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
tier, err := s.PatreonTierByID(id)
if err != nil {
log.Println(err)
continue
}
tiers = append(tiers, tier)
}
return tiers, nil
}
+230
View File
@@ -0,0 +1,230 @@
package database
import (
"database/sql"
"log"
"github.com/dustinpianalto/geeksbot"
)
type requestService struct {
db *sql.DB
}
func (s requestService) Request(id int64) (geeksbot.Request, error) {
var r geeksbot.Request
var aID string
var cID string
var gID string
var uID sql.NullString
var mID string
queryString := `SELECT id, author_id, channel_id, guild_id, content, requested_at, completed,
completed_at, completed_by, message_id, completed_message
WHERE id = $1`
row := s.db.QueryRow(queryString, id)
err := row.Scan(&r.ID, &aID, &cID, &gID, &r.Content, &r.RequestedAt, &r.Completed,
&r.CompletedAt, &uID, &mID, &r.CompletedMessage)
if err != nil {
return geeksbot.Request{}, err
}
author, err := UserService.User(aID)
if err != nil {
return geeksbot.Request{}, err
}
guild, err := GuildService.Guild(gID)
if err != nil {
return geeksbot.Request{}, err
}
channel, err := ChannelService.Channel(cID)
if err != nil {
return geeksbot.Request{}, err
}
if !uID.Valid {
r.CompletedBy = nil
} else {
completedBy, err := UserService.User(uID.String)
if err != nil {
return geeksbot.Request{}, err
}
r.CompletedBy = &completedBy
}
message, err := MessageService.Message(mID)
if err != nil {
return geeksbot.Request{}, err
}
r.Author = author
r.Guild = guild
r.Channel = channel
r.Message = message
return r, nil
}
func (s requestService) UserRequests(u geeksbot.User, completed bool) ([]geeksbot.Request, error) {
var requests []geeksbot.Request
var queryString string
if completed {
queryString = "SELECT id FROM requests WHERE author_id = $1"
} else {
queryString = "SELECT id FROM requests WHERE author_id = $1 AND completed = False"
}
rows, err := s.db.Query(queryString, u.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id int64
err = rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
request, err := s.Request(id)
if err != nil {
log.Println(err)
continue
}
requests = append(requests, request)
}
return requests, nil
}
func (s requestService) GuildRequests(g geeksbot.Guild, completed bool) ([]geeksbot.Request, error) {
var requests []geeksbot.Request
var queryString string
if completed {
queryString = "SELECT id FROM requests WHERE guild_id = $1"
} else {
queryString = "SELECT id FROM requests WHERE guild_id = $1 AND completed = False"
}
rows, err := s.db.Query(queryString, g.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id int64
err = rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
request, err := s.Request(id)
if err != nil {
log.Println(err)
continue
}
requests = append(requests, request)
}
return requests, nil
}
func (s requestService) CreateRequest(r geeksbot.Request) (geeksbot.Request, error) {
queryString := `INSERT INTO requests
(author_id, channel_id, guild_id, content, requested_at,
completed, completed_at, completed_by, message_id, completed_message)
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10) RETURNING id`
var id int64
err := s.db.QueryRow(queryString,
r.Author.ID,
r.Channel.ID,
r.Guild.ID,
r.Content,
r.RequestedAt,
r.Completed,
r.CompletedAt,
r.CompletedBy.ID,
r.Message.ID,
r.CompletedMessage).Scan(&id)
if err != nil {
return geeksbot.Request{}, err
}
r.ID = id
return r, nil
}
func (s requestService) UpdateRequest(r geeksbot.Request) (geeksbot.Request, error) {
queryString := `UPDATE requests SET
completed = $2, completed_at = $3, completed_by = $4, completed_message = $5
WHERE id = $1`
_, err := s.db.Exec(queryString, r.ID, r.Completed, r.CompletedAt, r.CompletedBy, r.CompletedMessage)
return r, err
}
func (s requestService) DeleteRequest(r geeksbot.Request) error {
queryString := "DELETE FROM requests WHERE id = $1"
_, err := s.db.Exec(queryString, r.ID)
return err
}
func (s requestService) Comment(id int64) (geeksbot.Comment, error) {
var c geeksbot.Comment
var aID string
var rID int64
queryString := "SELECT id, author_id, request_id, comment_at, content WHERE id = $1"
row := s.db.QueryRow(queryString, id)
err := row.Scan(&c.ID, &aID, &rID, &c.CommentAt, &c.Content)
if err != nil {
return geeksbot.Comment{}, err
}
author, err := UserService.User(aID)
if err != nil {
return geeksbot.Comment{}, err
}
c.Author = author
request, err := s.Request(rID)
if err != nil {
return geeksbot.Comment{}, err
}
c.Request = request
return c, nil
}
func (s requestService) RequestComments(r geeksbot.Request) ([]geeksbot.Comment, error) {
var comments []geeksbot.Comment
queryString := "SELECT id FROM comments WHERE request_id = $1"
rows, err := s.db.Query(queryString, r.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id int64
err := rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
comment, err := s.Comment(id)
if err != nil {
log.Println(err)
continue
}
comments = append(comments, comment)
}
return comments, nil
}
func (s requestService) RequestCommentCount(r geeksbot.Request) (int, error) {
var count int
queryString := "SELECT COUNT(id) FROM comments WHERE request_id = $1"
row := s.db.QueryRow(queryString, r.ID)
err := row.Scan(&count)
return count, err
}
func (s requestService) CreateComment(c geeksbot.Comment) (geeksbot.Comment, error) {
queryString := `INSERT INTO comments (author_id, request_id, comment_at, content)
VALUES ($1, $2, $3, $4) RETURNING id`
var id int64
err := s.db.QueryRow(queryString, c.Author.ID, c.Request.ID, c.CommentAt, c.Content).Scan(&id)
if err != nil {
return geeksbot.Comment{}, err
}
c.ID = id
return c, nil
}
func (s requestService) DeleteComment(c geeksbot.Comment) error {
queryString := "DELETE FROM comments WHERE id = $1"
_, err := s.db.Exec(queryString, c.ID)
return err
}
+154
View File
@@ -0,0 +1,154 @@
package database
import (
"database/sql"
"log"
"github.com/dustinpianalto/geeksbot"
)
type serverService struct {
db *sql.DB
}
func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
var server geeksbot.Server
var guildID string
var aChanID sql.NullString
var iChanID sql.NullString
var iMsgID sql.NullString
var sMsgID sql.NullString
queryString := `SELECT id, name, ip_address, port, password, alerts_channel_id,
guild_id, info_channel_id, info_message_id, settings_message_id
FROM servers WHERE id = $1`
row := s.db.QueryRow(queryString, id)
err := row.Scan(&server.ID, &server.Name, &server.IPAddr, &server.Port, &server.Password,
&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID)
if err != nil {
return geeksbot.Server{}, err
}
guild, err := GuildService.Guild(guildID)
if err != nil {
return geeksbot.Server{}, err
}
if !aChanID.Valid {
server.AlertsChannel = nil
} else {
alertChannel, err := ChannelService.Channel(aChanID.String)
if err != nil {
return geeksbot.Server{}, err
}
server.AlertsChannel = &alertChannel
}
if !iChanID.Valid {
server.InfoChannel = nil
} else {
infoChannel, err := ChannelService.Channel(iChanID.String)
if err != nil {
return geeksbot.Server{}, err
}
server.InfoChannel = &infoChannel
}
if !iMsgID.Valid {
server.InfoMessage = nil
} else {
infoMessage, err := MessageService.Message(iMsgID.String)
if err != nil {
return geeksbot.Server{}, err
}
server.InfoMessage = &infoMessage
}
if !sMsgID.Valid {
server.SettingsMessage = nil
} else {
settingsMessage, err := MessageService.Message(sMsgID.String)
if err != nil {
return geeksbot.Server{}, err
}
server.SettingsMessage = &settingsMessage
}
server.Guild = guild
return server, nil
}
func (s serverService) ServerByName(name string, guild geeksbot.Guild) (geeksbot.Server, error) {
var id int
queryString := "SELECT id FROM servers WHERE name = $1 AND guild_id = $2"
row := s.db.QueryRow(queryString, name, guild.ID)
err := row.Scan(&id)
if err != nil {
return geeksbot.Server{}, err
}
server, err := s.ServerByID(id)
return server, err
}
func (s serverService) CreateServer(server geeksbot.Server) (geeksbot.Server, error) {
var id int
queryString := `INSERT INTO servers (name, ip_address, port, password, alerts_channel_id,
guild_id, info_channel_id, info_message_id, settings_message_id)
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9) RETURNING id`
err := s.db.QueryRow(queryString,
server.Name,
server.IPAddr,
server.Port,
server.Password,
server.AlertsChannel,
server.Guild,
server.InfoChannel,
server.InfoMessage,
server.SettingsMessage,
).Scan(&id)
if err != nil {
return geeksbot.Server{}, err
}
server.ID = id
return server, nil
}
func (s serverService) DeleteServer(server geeksbot.Server) error {
queryString := `DELETE FROM servers WHERE id = $1`
_, err := s.db.Exec(queryString, server.ID)
return err
}
func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, error) {
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,
settings_message_id = $9 WHERE id = $1`
_, err := s.db.Exec(queryString,
server.Name,
server.IPAddr,
server.Port,
server.Password,
server.AlertsChannel.ID,
server.InfoChannel.ID,
server.InfoMessage.ID,
server.SettingsMessage.ID,
)
return server, err
}
func (s serverService) GuildServers(g geeksbot.Guild) ([]geeksbot.Server, error) {
var servers []geeksbot.Server
queryString := `SELECT id FROM servers WHERE guild_id = $1`
rows, err := s.db.Query(queryString, g.ID)
if err != nil {
return nil, err
}
for rows.Next() {
var id int
err = rows.Scan(&id)
if err != nil {
log.Println(err)
continue
}
server, err := s.ServerByID(id)
if err != nil {
log.Println(err)
continue
}
servers = append(servers, server)
}
return servers, nil
}
+51
View File
@@ -0,0 +1,51 @@
package database
import (
"database/sql"
"github.com/dustinpianalto/geeksbot"
)
type userService struct {
db *sql.DB
}
func (s userService) User(id string) (geeksbot.User, error) {
var user geeksbot.User
queryString := "SELECT id, steam_id, active, staff, admin FROM users WHERE id = $1"
row := s.db.QueryRow(queryString, id)
err := row.Scan(&user.ID, &user.SteamID, &user.IsActive, &user.IsStaff, &user.IsAdmin)
return user, err
}
func (s userService) CreateUser(u geeksbot.User) (geeksbot.User, error) {
queryString := "INSERT INTO users (id, steam_id, active, staff, admin) VALUES ($1, $2, $3, $4, $5)"
var err error
_, err = s.db.Exec(queryString, u.ID, u.SteamID, u.IsActive, u.IsStaff, u.IsAdmin)
return u, err
}
func (s userService) DeleteUser(u geeksbot.User) error {
queryString := "DELETE FROM users WHERE id = $1"
_, err := s.db.Exec(queryString, u.ID)
return err
}
func (s userService) UpdateUser(u geeksbot.User) (geeksbot.User, error) {
queryString := "UPDATE users SET steam_id = $2, active = $3, staff = $4, admin = $5 WHERE id = $1"
_, err := s.db.Exec(queryString, u.ID, u.SteamID, u.IsActive, u.IsStaff, u.IsAdmin)
return u, err
}
func (s userService) GetOrCreateUser(id string) (geeksbot.User, error) {
user, err := s.User(id)
if err == sql.ErrNoRows {
user, err = s.CreateUser(geeksbot.User{
ID: id,
IsActive: true,
IsAdmin: false,
IsStaff: false,
})
}
return user, err
}
+11
View File
@@ -0,0 +1,11 @@
package discord_utils
import "github.com/dustinpianalto/disgoman"
func SendErrorMessage(ctx disgoman.Context, msg string, err error) {
ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
Context: ctx,
Message: msg,
Error: err,
}
}
+109
View File
@@ -0,0 +1,109 @@
package guild
import (
"fmt"
"github.com/dustinpianalto/disgoman"
"github.com/dustinpianalto/geeksbot"
"github.com/dustinpianalto/geeksbot/internal/discord_utils"
"github.com/dustinpianalto/geeksbot/internal/services"
"github.com/dustinpianalto/geeksbot/internal/utils"
)
var AddPrefixCommand = &disgoman.Command{
Name: "addPrefix",
Aliases: []string{"ap"},
Description: "Add a prefix for use in this guild.",
OwnerOnly: false,
Hidden: false,
RequiredPermissions: disgoman.PermissionManageServer,
Invoke: addPrefixCommandFunc,
}
func addPrefixCommandFunc(ctx disgoman.Context, args []string) {
if len(args) == 0 {
discord_utils.SendErrorMessage(ctx, "Please include at least one prefix to add", nil)
return
}
guild, err := services.GuildService.Guild(ctx.Guild.ID)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Guild not configured, adding new guild to the database", nil)
guild = geeksbot.Guild{
ID: ctx.Guild.ID,
Prefixes: args,
}
guild, err = services.GuildService.CreateGuild(guild)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error adding guild to database.", err)
return
}
} else {
guild.Prefixes = append(guild.Prefixes, args...)
guild.Prefixes = utils.RemoveDuplicateStrings(guild.Prefixes)
guild, err = services.GuildService.UpdateGuild(guild)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error adding prefixes to guild.", err)
return
}
}
_, err = ctx.Send(fmt.Sprintf("Prefixes Updates.\nThe Prefixes for this guild are currently %#v", guild.Prefixes))
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error sending update message", err)
}
}
var RemovePrefixCommand = &disgoman.Command{
Name: "removePrefix",
Aliases: []string{"rp"},
Description: "Remove a prefix so it can't be used in this guild",
OwnerOnly: false,
Hidden: false,
RequiredPermissions: disgoman.PermissionManageServer,
Invoke: removePrefixCommandFunc,
}
func removePrefixCommandFunc(ctx disgoman.Context, args []string) {
if len(args) == 0 {
discord_utils.SendErrorMessage(ctx, "Please include at least one prefix to remove", nil)
return
}
guild, err := services.GuildService.Guild(ctx.Guild.ID)
var removed []string
if err != nil {
discord_utils.SendErrorMessage(ctx, "Guild not configured, adding new guild to the database", nil)
guild = geeksbot.Guild{
ID: ctx.Guild.ID,
Prefixes: []string{},
}
guild, err = services.GuildService.CreateGuild(guild)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error adding guild to database.", err)
return
}
} else {
for _, a := range args {
l := len(guild.Prefixes)
for i := 0; i < l; i++ {
if a == guild.Prefixes[i] {
guild.Prefixes = append(guild.Prefixes[:i], guild.Prefixes[i+1:]...)
l--
i--
removed = append(removed, a)
}
}
}
removed = utils.RemoveDuplicateStrings(removed)
guild.Prefixes = utils.RemoveDuplicateStrings(guild.Prefixes)
guild, err = services.GuildService.UpdateGuild(guild)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error removing prefixes from guild.", err)
return
}
}
_, err = ctx.Send(fmt.Sprintf("Prefixes Updates.\n"+
"The Prefixes for this guild are currently %#v\n"+
"Removed: %#v", guild.Prefixes, removed))
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error sending update message", err)
}
}
+12 -6
View File
@@ -2,10 +2,12 @@ package exts
import ( import (
"github.com/dustinpianalto/disgoman" "github.com/dustinpianalto/disgoman"
"github.com/dustinpianalto/geeksbot/internal/exts/guild"
"github.com/dustinpianalto/geeksbot/internal/exts/requests"
"github.com/dustinpianalto/geeksbot/internal/exts/utils" "github.com/dustinpianalto/geeksbot/internal/exts/utils"
) )
func AddCommandHandlers(h *disgoman.CommandManager) { func AddCommandHandlers(g *disgoman.CommandManager) {
// Arguments: // Arguments:
// name - command name - string // name - command name - string
// desc - command description - string // desc - command description - string
@@ -14,9 +16,13 @@ func AddCommandHandlers(h *disgoman.CommandManager) {
// perms - permissisions required - anpan.Permission (int) // perms - permissisions required - anpan.Permission (int)
// type - command type, sets where the command is available // type - command type, sets where the command is available
// run - function to run - func(anpan.Context, []string) / CommandRunFunc // run - function to run - func(anpan.Context, []string) / CommandRunFunc
_ = h.AddCommand(utils.UserCommand) _ = g.AddCommand(utils.UserCommand)
_ = h.AddCommand(utils.SayCommand) _ = g.AddCommand(utils.AddUserCommand)
_ = h.AddCommand(utils.GitCommand) _ = g.AddCommand(utils.SayCommand)
_ = h.AddCommand(utils.InviteCommand) _ = g.AddCommand(utils.GitCommand)
_ = h.AddCommand(utils.PingCommand) _ = g.AddCommand(utils.InviteCommand)
_ = g.AddCommand(utils.PingCommand)
_ = g.AddCommand(guild.AddPrefixCommand)
_ = g.AddCommand(guild.RemovePrefixCommand)
_ = g.AddCommand(requests.RequestCommand)
} }
+103
View File
@@ -0,0 +1,103 @@
package requests
import (
"fmt"
"strconv"
"strings"
"time"
"github.com/dustinpianalto/disgoman"
"github.com/dustinpianalto/geeksbot"
"github.com/dustinpianalto/geeksbot/internal/discord_utils"
"github.com/dustinpianalto/geeksbot/internal/services"
)
var RequestCommand = &disgoman.Command{
Name: "request",
Aliases: nil,
Description: "Submit a request for the guild staff",
OwnerOnly: false,
Hidden: false,
RequiredPermissions: 0,
Invoke: requestCommandFunc,
}
func requestCommandFunc(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
}
requestMsg := strings.Join(args, " ")
author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error creating the request. Could not get user.", err)
return
}
channel, err := services.ChannelService.GetOrCreateChannel(ctx.Message.ChannelID, ctx.Guild.ID)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error creating the request. Could not get channel.", err)
return
}
int64ID, _ := strconv.ParseInt(ctx.Message.ID, 10, 64)
s := discord_utils.ParseSnowflake(int64ID)
message, err := services.MessageService.CreateMessage(geeksbot.Message{
ID: ctx.Message.ID,
CreatedAt: s.CreationTime,
Content: ctx.Message.Content,
Channel: channel,
Author: author,
})
request := geeksbot.Request{
Author: author,
Channel: channel,
Guild: guild,
Content: requestMsg,
RequestedAt: s.CreationTime,
Completed: false,
Message: message,
}
request, err = services.RequestService.CreateRequest(request)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error creating the request", err)
return
}
channels, err := services.ChannelService.GuildChannels(guild)
if err == nil {
var mentionRolesString string
roles, err := services.GuildService.GuildRoles(guild)
if err == nil {
for _, r := range roles {
if r.RoleType == "admin" || r.RoleType == "moderator" {
mentionRolesString += fmt.Sprintf("<@&%s> ", r.ID)
}
}
}
for _, c := range channels {
if c.Admin {
_, _ = ctx.Session.ChannelMessageSend(c.ID,
fmt.Sprintf("%s\n"+
"New Request ID %d "+
"%s has requested assistance: \n"+
"```\n%s\n```\n"+
"Requested At: %s\n"+
"In: %s",
mentionRolesString,
request.ID,
ctx.Message.Author.Mention,
request.Content,
request.RequestedAt.UTC().Format(time.UnixDate),
ctx.Channel.Mention(),
),
)
}
}
}
_, err = ctx.Send(fmt.Sprintf("%s The admin have recieved your request.\n "+
"If you would like to close or add a comment to this request please reference ID `%v`",
ctx.Message.Author.Mention, request.ID,
))
if err != nil {
discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was created.", err)
}
}
+33
View File
@@ -9,7 +9,9 @@ import (
"github.com/bwmarrin/discordgo" "github.com/bwmarrin/discordgo"
"github.com/dustinpianalto/disgoman" "github.com/dustinpianalto/disgoman"
"github.com/dustinpianalto/geeksbot"
"github.com/dustinpianalto/geeksbot/internal/discord_utils" "github.com/dustinpianalto/geeksbot/internal/discord_utils"
"github.com/dustinpianalto/geeksbot/internal/services"
) )
var PingCommand = &disgoman.Command{ var PingCommand = &disgoman.Command{
@@ -246,3 +248,34 @@ func userCommandFunc(ctx disgoman.Context, args []string) {
} }
} }
} }
var AddUserCommand = &disgoman.Command{
Name: "adduser",
Aliases: nil,
Description: "Get user info",
OwnerOnly: false,
Hidden: false,
RequiredPermissions: 0,
Invoke: addUserCommandFunc,
}
func addUserCommandFunc(ctx disgoman.Context, args []string) {
if ctx.Message.Author.ID == ctx.CommandManager.Owners[0] {
user := geeksbot.User{
ID: ctx.Message.Author.ID,
IsActive: true,
IsStaff: true,
IsAdmin: true,
}
user, err := services.UserService.CreateUser(user)
if err != nil {
ctx.CommandManager.ErrorChannel <- disgoman.CommandError{
Context: ctx,
Message: "Error with adding user",
Error: err,
}
return
}
ctx.Session.MessageReactionAdd(ctx.Channel.ID, ctx.Message.ID, "✅")
}
}
+26
View File
@@ -0,0 +1,26 @@
package services
import (
"github.com/dustinpianalto/geeksbot"
"github.com/dustinpianalto/geeksbot/internal/database"
)
var (
GuildService geeksbot.GuildService
UserService geeksbot.UserService
ChannelService geeksbot.ChannelService
MessageService geeksbot.MessageService
PatreonService geeksbot.PatreonService
RequestService geeksbot.RequestService
ServerService geeksbot.ServerService
)
func InitializeServices() {
GuildService = database.GuildService
UserService = database.UserService
ChannelService = database.ChannelService
MessageService = database.MessageService
PatreonService = database.PatreonService
RequestService = database.RequestService
ServerService = database.ServerService
}
+14
View File
@@ -0,0 +1,14 @@
package utils
func RemoveDuplicateStrings(s []string) []string {
keys := make(map[string]bool)
o := []string{}
for _, e := range s {
if _, v := keys[e]; !v {
keys[e] = true
o = append(o, e)
}
}
return o
}
+13 -12
View File
@@ -1,23 +1,24 @@
package geeksbot package geeksbot
import "github.com/bwmarrin/discordgo" import (
"database/sql"
"time"
)
type Message struct { type Message struct {
ID string ID string
CreatedAt int64 CreatedAt time.Time
ModifiedAt int64 ModifiedAt sql.NullTime
Content string Content string
PreviousContent []string PreviousContent []string
Channel *Channel Channel Channel
Author *User Author User
Embed *discordgo.MessageEmbed
PreviousEmbeds []*discordgo.MessageEmbed
} }
type MessageService interface { type MessageService interface {
Message(id string) (*Message, error) Message(id string) (Message, error)
CreateMessage(m *Message) (*Message, error) CreateMessage(m Message) (Message, error)
DeleteMessage(m *Message) error DeleteMessage(m Message) error
ChannelMessages(c *Channel) ([]*Message, error) ChannelMessages(c Channel) ([]Message, error)
UpdateMessage(m *Message) (*Message, error) UpdateMessage(m Message) (Message, error)
} }
+18 -15
View File
@@ -1,31 +1,34 @@
package geeksbot package geeksbot
import "database/sql"
type PatreonCreator struct { type PatreonCreator struct {
ID int ID int
Creator string Creator string
Link string Link string
Guild *Guild Guild Guild
} }
type PatreonTier struct { type PatreonTier struct {
ID int ID int
Name string Name string
Description string Description sql.NullString
Creator *PatreonCreator Creator PatreonCreator
Role *Role Role Role
NextTier *PatreonTier NextTier *PatreonTier
} }
type PatreonService interface { type PatreonService interface {
PatreonCreatorByID(id int) (*PatreonCreator, error) PatreonCreatorByID(id int) (PatreonCreator, error)
PatreonCreatorByName(name string) (*PatreonCreator, error) PatreonCreatorByName(name string, guild Guild) (PatreonCreator, error)
CreatePatreonCreator(c *PatreonCreator) (*PatreonCreator, error) CreatePatreonCreator(c PatreonCreator) (PatreonCreator, error)
UpdatePatreonCreator(c *PatreonCreator) (*PatreonCreator, error) UpdatePatreonCreator(c PatreonCreator) (PatreonCreator, error)
DeletePatreonCreator(c *PatreonCreator) error DeletePatreonCreator(c PatreonCreator) error
PatreonTierByID(id int) (*PatreonTier, error) PatreonTierByID(id int) (PatreonTier, error)
PatreonTierByName(name string) (*PatreonTier, error) PatreonTierByName(name string, creator string) (PatreonTier, error)
CreatePatreonTier(t *PatreonTier) (*PatreonTier, error) CreatePatreonTier(t PatreonTier) (PatreonTier, error)
UpdatePatreonTier(t *PatreonTier) (*PatreonTier, error) UpdatePatreonTier(t PatreonTier) (PatreonTier, error)
DeletePatreonTier(t *PatreonTier) error DeletePatreonTier(t PatreonTier) error
GuildPatreonCreators(g *Guild) ([]*PatreonCreator, error) GuildPatreonCreators(g Guild) ([]PatreonCreator, error)
CreatorPatreonTiers(c PatreonCreator) ([]PatreonTier, error)
} }
+24 -19
View File
@@ -1,37 +1,42 @@
package geeksbot package geeksbot
import "time" import (
"database/sql"
"time"
)
type Request struct { type Request struct {
ID int64 ID int64
Author *User Author User
Channel *Channel Channel Channel
Guild Guild
Content string Content string
RequestedAt time.Time RequestedAt time.Time
CompletedAt time.Time Completed bool
CompletedAt sql.NullTime
CompletedBy *User CompletedBy *User
Message *Message Message Message
CompletedMessage string CompletedMessage sql.NullString
} }
type Comment struct { type Comment struct {
ID int64 ID int64
Author *User Author User
Request *Request Request Request
CommentAt time.Time CommentAt time.Time
Content string Content string
} }
type RequestService interface { type RequestService interface {
Request(id int64) (*Request, error) Request(id int64) (Request, error)
UserRequests(u *User, completed bool) ([]*Request, error) UserRequests(u User, completed bool) ([]Request, error)
GuildRequests(g *Guild, completed bool) ([]*Request, error) GuildRequests(g Guild, completed bool) ([]Request, error)
CreateRequest(r *Request) (*Request, error) CreateRequest(r Request) (Request, error)
UpdateRequest(r *Request) (*Request, error) UpdateRequest(r Request) (Request, error)
DeleteRequest(r *Request) error DeleteRequest(r Request) error
Comment(id int64) (*Comment, error) Comment(id int64) (Comment, error)
RequestComments(r *Request) ([]*Comment, error) RequestComments(r Request) ([]Comment, error)
RequestCommentCount(r *Request) (int, error) RequestCommentCount(r Request) (int, error)
CreateComment(c *Comment) (*Comment, error) CreateComment(c Comment) (Comment, error)
DeleteComment(c *Comment) error DeleteComment(c Comment) error
} }
+7 -7
View File
@@ -7,17 +7,17 @@ type Server struct {
Port int Port int
Password string Password string
AlertsChannel *Channel AlertsChannel *Channel
Guild *Guild Guild Guild
InfoChannel *Channel InfoChannel *Channel
InfoMessage *Message InfoMessage *Message
SettingsMessage *Message SettingsMessage *Message
} }
type ServerService interface { type ServerService interface {
ServerByID(id int) (*Server, error) ServerByID(id int) (Server, error)
ServerByName(name string) (*Server, error) ServerByName(name string, guild Guild) (Server, error)
CreateServer(s *Server) (*Server, error) CreateServer(s Server) (Server, error)
DeleteServer(s *Server) error DeleteServer(s Server) error
UpdateServer(s *Server) (*Server, error) UpdateServer(s Server) (Server, error)
GuildServers(g *Guild) ([]*Server, error) GuildServers(g Guild) ([]Server, error)
} }
+5 -4
View File
@@ -11,8 +11,9 @@ type User struct {
} }
type UserService interface { type UserService interface {
User(id string) (*User, error) User(id string) (User, error)
CreateUser(u *User) (*User, error) CreateUser(u User) (User, error)
DeleteUser(u *User) error DeleteUser(u User) error
UpdateUser(u *User) (*User, error) UpdateUser(u User) (User, error)
GetOrCreateUser(id string) (User, error)
} }