fix get user sql query

channel.go

@@ -14,4 +14,5 @@ type ChannelService interface {
 	DeleteChannel(c Channel) error
 	GuildChannels(g Guild) ([]Channel, error)
 	UpdateChannel(c Channel) (Channel, error)
+	GetOrCreateChannel(id string, guild_id string) (Channel, error)
 }

cmd/geeksbot/main.go

@@ -72,7 +72,11 @@ func main() {
 }
 
 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) {

go.mod

@@ -9,4 +9,5 @@ require (
 	github.com/golang-migrate/migrate v3.5.4+incompatible // indirect
 	github.com/golang-migrate/migrate/v4 v4.14.1
 	github.com/hashicorp/errwrap v1.1.0 // indirect
+	github.com/lib/pq v1.8.0
 )

guild.go

@@ -24,4 +24,5 @@ type GuildService interface {
 	Role(id string) (Role, error)
 	UpdateRole(r Role) (Role, error)
 	DeleteRole(r Role) error
+	GetOrCreateGuild(id string) (Guild, error)
 }

internal/database/channel.go

@@ -66,3 +66,21 @@ func (s channelService) GuildChannels(g geeksbot.Guild) ([]geeksbot.Channel, err
 	}
 	return channels, nil
 }
+
+func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot.Channel, error) {
+	channel, err := s.Channel(id)
+	if err.Error() == sql.ErrNoRows.Error() {
+		guild, err := GuildService.GetOrCreateGuild(guild_id)
+		if err != nil {
+			return geeksbot.Channel{}, err
+		}
+		channel, err = s.CreateChannel(geeksbot.Channel{
+			ID:        id,
+			Guild:     guild,
+			Admin:     false,
+			Default:   false,
+			NewPatron: false,
+		})
+	}
+	return channel, err
+}

internal/database/guild.go

@@ -5,6 +5,7 @@ import (
 	"log"
 
 	"github.com/dustinpianalto/geeksbot"
+	"github.com/lib/pq"
 )
 
 type guildService struct {
@@ -15,7 +16,7 @@ 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, &g.Prefixes)
+	err := row.Scan(&g.ID, &g.NewPatronMessage, pq.Array(&g.Prefixes))
 	if err != nil {
 		return geeksbot.Guild{}, err
 	}
@@ -24,7 +25,7 @@ func (s guildService) Guild(id string) (geeksbot.Guild, error) {
 
 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, g.Prefixes)
+	_, err := s.db.Exec(queryString, g.ID, g.NewPatronMessage, pq.Array(g.Prefixes))
 	return g, err
 }
 
@@ -36,7 +37,7 @@ func (s guildService) DeleteGuild(g geeksbot.Guild) error {
 
 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, g.Prefixes)
+	_, err := s.db.Exec(queryString, g.ID, g.NewPatronMessage, pq.Array(g.Prefixes))
 	return g, err
 }
 
@@ -98,3 +99,15 @@ func (s guildService) DeleteRole(r geeksbot.Role) error {
 	_, 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
+}

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

@@ -12,19 +12,22 @@ BEGIN;
         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 COLUMN author_id DROP NOT NULL;
+    ALTER TABLE messages
         ADD COLUMN embed json,
         ADD COLUMN previous_embeds json[];
     ALTER TABLE patreon_creators
-        RENAME TO patreon_creator,
         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
-        RENAME TO patreon_tier,
         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,

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

@@ -12,19 +12,22 @@ BEGIN;
         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 COLUMN author_id SET NOT NULL;
+    ALTER TABLE messages
         DROP COLUMN embed,
         DROP COLUMN previous_embeds;
     ALTER TABLE patreon_creator
-        RENAME TO patreon_creators,
         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
-        RENAME TO patreon_tiers,
         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,

internal/database/user.go

@@ -12,7 +12,7 @@ type userService struct {
 
 func (s userService) User(id string) (geeksbot.User, error) {
 	var user geeksbot.User
-	queryString := "SELECT id, steam_id, active, staff, admin WHERE id = $1"
+	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
@@ -36,3 +36,16 @@ func (s userService) UpdateUser(u geeksbot.User) (geeksbot.User, error) {
 	_, 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
+}

internal/exts/guild/guild.go

@@ -7,6 +7,7 @@ import (
 	"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{
@@ -38,6 +39,7 @@ func addPrefixCommandFunc(ctx disgoman.Context, args []string) {
 		}
 	} 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)
@@ -80,7 +82,7 @@ func removePrefixCommandFunc(ctx disgoman.Context, args []string) {
 		}
 	} else {
 		for _, a := range args {
-			l := len(guild.Prefixes) - 1
+			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:]...)
@@ -90,6 +92,8 @@ func removePrefixCommandFunc(ctx disgoman.Context, args []string) {
 				}
 			}
 		}
+		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)

internal/exts/init.go

@@ -3,6 +3,7 @@ package exts
 import (
 	"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"
 )
 
@@ -23,4 +24,5 @@ func AddCommandHandlers(g *disgoman.CommandManager) {
 	_ = g.AddCommand(utils.PingCommand)
 	_ = g.AddCommand(guild.AddPrefixCommand)
 	_ = g.AddCommand(guild.RemovePrefixCommand)
+	_ = g.AddCommand(requests.RequestCommand)
 }

internal/exts/requests/requests.go

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

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,10 +1,13 @@
 package geeksbot
 
-import "database/sql"
+import (
+	"database/sql"
+	"time"
+)
 
 type Message struct {
 	ID              string
-	CreatedAt       int64
+	CreatedAt       time.Time
 	ModifiedAt      sql.NullTime
 	Content         string
 	PreviousContent []string

user.go

@@ -15,4 +15,5 @@ type UserService interface {
 	CreateUser(u User) (User, error)
 	DeleteUser(u User) error
 	UpdateUser(u User) (User, error)
+	GetOrCreateUser(id string) (User, error)
 }