remove duplicate prefixes

channel.go

@@ -9,9 +9,9 @@ type Channel struct {
 }
 
 type ChannelService interface {
-	Channel(id string) (*Channel, error)
-	CreateChannel(c *Channel) (*Channel, error)
-	DeleteChannel(c *Channel) error
-	GuildChannels(g *Guild) ([]*Channel, error)
-	UpdateChannel(c *Channel) (*Channel, error)
+	Channel(id string) (Channel, error)
+	CreateChannel(c Channel) (Channel, error)
+	DeleteChannel(c Channel) error
+	GuildChannels(g Guild) ([]Channel, error)
+	UpdateChannel(c Channel) (Channel, error)
 }

cmd/geeksbot/main.go

@@ -10,6 +10,7 @@ import (
 	"github.com/dustinpianalto/disgoman"
 	"github.com/dustinpianalto/geeksbot/internal/database"
 	"github.com/dustinpianalto/geeksbot/internal/exts"
+	"github.com/dustinpianalto/geeksbot/internal/services"
 )
 
 func main() {
@@ -27,11 +28,7 @@ func main() {
 	}
 
 	database.ConnectDatabase(os.Getenv("DATABASE_URL"))
-	//postgres.InitializeDatabase()
-	//utils.LoadTestData()
-
-	//us := &postgres.UserService{DB: postgres.DB}
-	//gs := &postgres.GuildService{DB: postgres.DB}
+	services.InitializeServices()
 
 	owners := []string{
 		"351794468870946827",
@@ -49,18 +46,9 @@ func main() {
 
 	// Add Command Handlers
 	exts.AddCommandHandlers(&manager)
-	//services.InitalizeServices(us, gs)
-
-	//if _, ok := handler.Commands["help"]; !ok {
-	//	handler.AddDefaultHelpCommand()
-	//}
 
 	dg.AddHandler(manager.OnMessage)
 	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()
 	if err != nil {
@@ -71,9 +59,6 @@ func main() {
 	// Start the Error handler in a goroutine
 	go ErrorHandler(manager.ErrorChannel)
 
-	// Start the Logging handler in a goroutine
-	//go logging.LoggingHandler(logging.LoggingChannel)
-
 	log.Println("The Bot is now running.")
 	sc := make(chan os.Signal, 1)
 	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)

go.sum

@@ -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-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-20200709230013-948cd5f35899 h1:DZhuSZLsGlFL4CmhA8BcRA0mnthyA/nZ00AqCUo7vHg=
 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-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=

guild.go

@@ -1,25 +1,27 @@
 package geeksbot
 
+import "database/sql"
+
 type Guild struct {
 	ID               string
-	NewPatronMessage string
+	NewPatronMessage sql.NullString
 	Prefixes         []string
 }
 
 type Role struct {
 	ID       string
-	RoleType int
+	RoleType string
 	Guild    Guild
 }
 
 type GuildService interface {
-	Guild(id string) (*Guild, error)
-	CreateGuild(g *Guild) (*Guild, error)
-	DeleteGuild(g *Guild) error
-	UpdateGuild(g *Guild) (*Guild, error)
-	GuildRoles(g *Guild) ([]*Role, error)
-	CreateRole(r *Role) (*Role, error)
-	Role(id string) (*Role, error)
-	UpdateRole(r *Role) (*Role, error)
-	DeleteRole(r *Role) error
+	Guild(id string) (Guild, error)
+	CreateGuild(g Guild) (Guild, error)
+	DeleteGuild(g Guild) error
+	UpdateGuild(g Guild) (Guild, error)
+	GuildRoles(g Guild) ([]Role, error)
+	CreateRole(r Role) (Role, error)
+	Role(id string) (Role, error)
+	UpdateRole(r Role) (Role, error)
+	DeleteRole(r Role) error
 }

internal/database/channel.go

@@ -0,0 +1,68 @@
+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
+}

internal/database/database.go

@@ -13,6 +13,13 @@ import (
 
 var (
 	db             *sql.DB
+	GuildService   guildService
+	UserService    userService
+	ChannelService channelService
+	MessageService messageService
+	RequestService requestService
+	PatreonService patreonService
+	ServerService  serverService
 )
 
 func ConnectDatabase(dbConnString string) {
@@ -20,7 +27,8 @@ func ConnectDatabase(dbConnString string) {
 		func(name string) ([]byte, error) {
 			return migrations.Asset(name)
 		})
-	db, err := sql.Open("postgres", dbConnString)
+	var err error
+	db, err = sql.Open("postgres", dbConnString)
 	if err != nil {
 		log.Fatal(fmt.Errorf("Can't connect to the database: %w", err))
 	}
@@ -42,7 +50,23 @@ func ConnectDatabase(dbConnString string) {
 	}
 	err = m.Up()
 	if err != nil {
+		if err.Error() == "no change" {
+			log.Println(err)
+		} else {
 			log.Fatal(err)
 		}
+	}
 	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}
 }

internal/database/guild.go

@@ -0,0 +1,101 @@
+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
+}

internal/database/message.go

@@ -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
+}

internal/database/migrations/000002_add_guild_to_request.down.sql

@@ -0,0 +1 @@
+ALTER TABLE requests DROP COLUMN guild_id;

internal/database/migrations/000002_add_guild_to_request.up.sql

@@ -0,0 +1,2 @@
+ALTER TABLE requests 
+    ADD COLUMN guild_id varchar(30) CONSTRAINT fk_guild REFERENCES guilds(id) ON DELETE CASCADE;

internal/database/migrations/000003_set_column_not_null.down.sql

@@ -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;

internal/database/migrations/000003_set_column_not_null.up.sql

@@ -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;

internal/database/patreon.go

@@ -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
+}

internal/database/request.go

@@ -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
+}

internal/database/server.go

@@ -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
+}

internal/database/user.go

@@ -0,0 +1,38 @@
+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 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
+}

internal/discord_utils/errors.go

@@ -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,
+	}
+}

internal/exts/guild/guild.go

@@ -0,0 +1,111 @@
+package guild
+
+import (
+	"fmt"
+	"log"
+
+	"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)
+					log.Println(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)
+	}
+}

internal/exts/init.go

@@ -2,10 +2,11 @@ package exts
 
 import (
 	"github.com/dustinpianalto/disgoman"
+	"github.com/dustinpianalto/geeksbot/internal/exts/guild"
 	"github.com/dustinpianalto/geeksbot/internal/exts/utils"
 )
 
-func AddCommandHandlers(h *disgoman.CommandManager) {
+func AddCommandHandlers(g *disgoman.CommandManager) {
 	// Arguments:
 	// name - command name - string
 	// desc - command description - string
@@ -14,9 +15,12 @@ func AddCommandHandlers(h *disgoman.CommandManager) {
 	// perms - permissisions required - anpan.Permission (int)
 	// type - command type, sets where the command is available
 	// run - function to run - func(anpan.Context, []string) / CommandRunFunc
-	_ = h.AddCommand(utils.UserCommand)
-	_ = h.AddCommand(utils.SayCommand)
-	_ = h.AddCommand(utils.GitCommand)
-	_ = h.AddCommand(utils.InviteCommand)
-	_ = h.AddCommand(utils.PingCommand)
+	_ = g.AddCommand(utils.UserCommand)
+	_ = g.AddCommand(utils.AddUserCommand)
+	_ = g.AddCommand(utils.SayCommand)
+	_ = g.AddCommand(utils.GitCommand)
+	_ = g.AddCommand(utils.InviteCommand)
+	_ = g.AddCommand(utils.PingCommand)
+	_ = g.AddCommand(guild.AddPrefixCommand)
+	_ = g.AddCommand(guild.RemovePrefixCommand)
 }

internal/exts/utils/utils.go

@@ -9,7 +9,9 @@ import (
 
 	"github.com/bwmarrin/discordgo"
 	"github.com/dustinpianalto/disgoman"
+	"github.com/dustinpianalto/geeksbot"
 	"github.com/dustinpianalto/geeksbot/internal/discord_utils"
+	"github.com/dustinpianalto/geeksbot/internal/services"
 )
 
 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, "✅")
+	}
+}

internal/services/services.go

@@ -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
+}

internal/utils/utils.go

@@ -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
+}

message.go

@@ -1,23 +1,21 @@
 package geeksbot
 
-import "github.com/bwmarrin/discordgo"
+import "database/sql"
 
 type Message struct {
 	ID              string
 	CreatedAt       int64
-	ModifiedAt      int64
+	ModifiedAt      sql.NullTime
 	Content         string
 	PreviousContent []string
-	Channel         *Channel
-	Author          *User
-	Embed           *discordgo.MessageEmbed
-	PreviousEmbeds  []*discordgo.MessageEmbed
+	Channel         Channel
+	Author          User
 }
 
 type MessageService interface {
-	Message(id string) (*Message, error)
-	CreateMessage(m *Message) (*Message, error)
-	DeleteMessage(m *Message) error
-	ChannelMessages(c *Channel) ([]*Message, error)
-	UpdateMessage(m *Message) (*Message, error)
+	Message(id string) (Message, error)
+	CreateMessage(m Message) (Message, error)
+	DeleteMessage(m Message) error
+	ChannelMessages(c Channel) ([]Message, error)
+	UpdateMessage(m Message) (Message, error)
 }

patreon.go

@@ -1,31 +1,34 @@
 package geeksbot
 
+import "database/sql"
+
 type PatreonCreator struct {
 	ID      int
 	Creator string
 	Link    string
-	Guild   *Guild
+	Guild   Guild
 }
 
 type PatreonTier struct {
 	ID          int
 	Name        string
-	Description string
-	Creator     *PatreonCreator
-	Role        *Role
+	Description sql.NullString
+	Creator     PatreonCreator
+	Role        Role
 	NextTier    *PatreonTier
 }
 
 type PatreonService interface {
-	PatreonCreatorByID(id int) (*PatreonCreator, error)
-	PatreonCreatorByName(name string) (*PatreonCreator, error)
-	CreatePatreonCreator(c *PatreonCreator) (*PatreonCreator, error)
-	UpdatePatreonCreator(c *PatreonCreator) (*PatreonCreator, error)
-	DeletePatreonCreator(c *PatreonCreator) error
-	PatreonTierByID(id int) (*PatreonTier, error)
-	PatreonTierByName(name string) (*PatreonTier, error)
-	CreatePatreonTier(t *PatreonTier) (*PatreonTier, error)
-	UpdatePatreonTier(t *PatreonTier) (*PatreonTier, error)
-	DeletePatreonTier(t *PatreonTier) error
-	GuildPatreonCreators(g *Guild) ([]*PatreonCreator, error)
+	PatreonCreatorByID(id int) (PatreonCreator, error)
+	PatreonCreatorByName(name string, guild Guild) (PatreonCreator, error)
+	CreatePatreonCreator(c PatreonCreator) (PatreonCreator, error)
+	UpdatePatreonCreator(c PatreonCreator) (PatreonCreator, error)
+	DeletePatreonCreator(c PatreonCreator) error
+	PatreonTierByID(id int) (PatreonTier, error)
+	PatreonTierByName(name string, creator string) (PatreonTier, error)
+	CreatePatreonTier(t PatreonTier) (PatreonTier, error)
+	UpdatePatreonTier(t PatreonTier) (PatreonTier, error)
+	DeletePatreonTier(t PatreonTier) error
+	GuildPatreonCreators(g Guild) ([]PatreonCreator, error)
+	CreatorPatreonTiers(c PatreonCreator) ([]PatreonTier, error)
 }

request.go

@@ -1,37 +1,42 @@
 package geeksbot
 
-import "time"
+import (
+	"database/sql"
+	"time"
+)
 
 type Request struct {
 	ID               int64
-	Author           *User
-	Channel          *Channel
+	Author           User
+	Channel          Channel
+	Guild            Guild
 	Content          string
 	RequestedAt      time.Time
-	CompletedAt      time.Time
+	Completed        bool
+	CompletedAt      sql.NullTime
 	CompletedBy      *User
-	Message          *Message
-	CompletedMessage string
+	Message          Message
+	CompletedMessage sql.NullString
 }
 
 type Comment struct {
 	ID        int64
-	Author    *User
-	Request   *Request
+	Author    User
+	Request   Request
 	CommentAt time.Time
 	Content   string
 }
 
 type RequestService interface {
-	Request(id int64) (*Request, error)
-	UserRequests(u *User, completed bool) ([]*Request, error)
-	GuildRequests(g *Guild, completed bool) ([]*Request, error)
-	CreateRequest(r *Request) (*Request, error)
-	UpdateRequest(r *Request) (*Request, error)
-	DeleteRequest(r *Request) error
-	Comment(id int64) (*Comment, error)
-	RequestComments(r *Request) ([]*Comment, error)
-	RequestCommentCount(r *Request) (int, error)
-	CreateComment(c *Comment) (*Comment, error)
-	DeleteComment(c *Comment) error
+	Request(id int64) (Request, error)
+	UserRequests(u User, completed bool) ([]Request, error)
+	GuildRequests(g Guild, completed bool) ([]Request, error)
+	CreateRequest(r Request) (Request, error)
+	UpdateRequest(r Request) (Request, error)
+	DeleteRequest(r Request) error
+	Comment(id int64) (Comment, error)
+	RequestComments(r Request) ([]Comment, error)
+	RequestCommentCount(r Request) (int, error)
+	CreateComment(c Comment) (Comment, error)
+	DeleteComment(c Comment) error
 }

server.go

@@ -7,17 +7,17 @@ type Server struct {
 	Port            int
 	Password        string
 	AlertsChannel   *Channel
-	Guild           *Guild
+	Guild           Guild
 	InfoChannel     *Channel
 	InfoMessage     *Message
 	SettingsMessage *Message
 }
 
 type ServerService interface {
-	ServerByID(id int) (*Server, error)
-	ServerByName(name string) (*Server, error)
-	CreateServer(s *Server) (*Server, error)
-	DeleteServer(s *Server) error
-	UpdateServer(s *Server) (*Server, error)
-	GuildServers(g *Guild) ([]*Server, error)
+	ServerByID(id int) (Server, error)
+	ServerByName(name string, guild Guild) (Server, error)
+	CreateServer(s Server) (Server, error)
+	DeleteServer(s Server) error
+	UpdateServer(s Server) (Server, error)
+	GuildServers(g Guild) ([]Server, error)
 }

user.go

@@ -11,8 +11,8 @@ type User struct {
 }
 
 type UserService interface {
-	User(id string) (*User, error)
-	CreateUser(u *User) (*User, error)
-	DeleteUser(u *User) error
-	UpdateUser(u *User) (*User, error)
+	User(id string) (User, error)
+	CreateUser(u User) (User, error)
+	DeleteUser(u User) error
+	UpdateUser(u User) (User, error)
 }