Add comment command

cmd/geeksbot/main.go

@@ -28,6 +28,7 @@ func main() {
 	}
 
 	database.ConnectDatabase(os.Getenv("DATABASE_URL"))
+	database.RunMigrations()
 	services.InitializeServices()
 
 	owners := []string{

internal/discord_utils/channel_utils.go

@@ -0,0 +1,11 @@
+package discord_utils
+
+import "github.com/dustinpianalto/disgoman"
+
+func GetChannelName(ctx disgoman.Context, id string) (string, error) {
+	channel, err := ctx.Session.Channel(id)
+	if err != nil {
+		return "", err
+	}
+	return channel.Name, nil
+}

internal/discord_utils/user_utils.go

@@ -0,0 +1,14 @@
+package discord_utils
+
+import "github.com/dustinpianalto/disgoman"
+
+func GetDisplayName(ctx disgoman.Context, id string) (string, error) {
+	member, err := ctx.Session.GuildMember(ctx.Guild.ID, id)
+	if err != nil {
+		return "", err
+	}
+	if member.Nick != "" {
+		return member.Nick, nil
+	}
+	return member.User.Username, nil
+}

internal/exts/requests/requests.go

@@ -2,6 +2,7 @@ package requests
 
 import (
 	"fmt"
+	"log"
 	"strconv"
 	"strings"
 	"time"
@@ -251,3 +252,96 @@ func listCommandFunc(ctx disgoman.Context, args []string) {
 
 	_, _ = ctx.Send(fmt.Sprintf("```There %s currently %d open %s```", utils.PluralizeString("is", len(requests)), len(requests), utils.PluralizeString("request", len(requests))))
 }
+
+var CommentCommand = &disgoman.Command{
+	Name:                "comment",
+	Aliases:             []string{"update", "add_comment"},
+	Description:         "Add a comment to an existing request.",
+	OwnerOnly:           false,
+	Hidden:              false,
+	RequiredPermissions: 0,
+	Invoke:              commentCommandFunc,
+}
+
+func commentCommandFunc(ctx disgoman.Context, args []string) {
+	guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
+		return
+	}
+	id, err := strconv.ParseInt(args[0], 10, 64)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Please include the ID of the request to update.", err)
+		return
+	}
+	message := strings.Join(args[1:len(args)-1], " ")
+	author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "Sorry, there was an issue finding your user account", err)
+		return
+	}
+	request, err := services.RequestService.Request(id)
+	if err != nil || request.Guild.ID != guild.ID {
+		discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%d is not a valid request for this guild", id), err)
+		return
+	}
+	int64ID, _ := strconv.ParseInt(ctx.Message.ID, 10, 64)
+	s := discord_utils.ParseSnowflake(int64ID)
+	comment := geeksbot.Comment{
+		Author:    author,
+		Request:   request,
+		CommentAt: s.CreationTime,
+		Content:   message,
+	}
+	comment, err = services.RequestService.CreateComment(comment)
+	if err != nil {
+		discord_utils.SendErrorMessage(ctx, "There was a problem adding your comment", err)
+		return
+	}
+	channels, err := services.ChannelService.GuildChannels(guild)
+	if err == nil {
+		comments, _ := services.RequestService.RequestComments(request)
+		var commentString string
+		var commentStrings []string
+		displayName, _ := discord_utils.GetDisplayName(ctx, request.Author.ID)
+		channelName, _ := discord_utils.GetChannelName(ctx, request.Channel.ID)
+		commentString = fmt.Sprintf("Comment added:\n```md\n"+
+			"< Request ID            Requested By >\n"+
+			"< %-11d %23s >\n"+
+			"%s\n\n"+
+			"Comments: Not Implemented Yet\n"+
+			"Requested At: %s\n"+
+			"In: %s\n"+
+			"```",
+			request.ID,
+			displayName,
+			request.Content,
+			request.RequestedAt.Format("2006-01-02 15:04:05 MST"),
+			channelName,
+		)
+		for _, c := range comments {
+			cAuthorName, err := discord_utils.GetDisplayName(ctx, c.Author.ID)
+			if err != nil {
+				log.Println(err)
+				continue
+			}
+			cs := fmt.Sprintf("```md\n%s\n- %s At %s\n```\n",
+				c.Content,
+				cAuthorName,
+				c.CommentAt.Format("2006-01-02 15:04:05 MST"),
+			)
+			if len(commentString+cs) >= 2000 {
+				commentStrings = append(commentStrings, commentString)
+				commentString = ""
+			}
+		}
+		for _, c := range channels {
+			if c.Admin {
+				for _, s := range commentStrings {
+					_, _ = ctx.Session.ChannelMessageSend(c.ID, s)
+				}
+			}
+		}
+	}
+	_, err = ctx.Send(fmt.Sprintf("%s your comment has been added.", ctx.Message.Author.Mention()))
+}

pkg/database/database.go

@@ -23,10 +23,6 @@ var (
 )
 
 func ConnectDatabase(dbConnString string) {
-	s := bindata.Resource(migrations.AssetNames(),
-		func(name string) ([]byte, error) {
-			return migrations.Asset(name)
-		})
 	var err error
 	db, err = sql.Open("postgres", dbConnString)
 	if err != nil {
@@ -36,6 +32,15 @@ func ConnectDatabase(dbConnString string) {
 	db.SetMaxOpenConns(75)
 	db.SetMaxIdleConns(5)
 	db.SetConnMaxLifetime(300)
+	initServices()
+	log.Println("Services Initialized")
+}
+
+func RunMigrations() {
+	s := bindata.Resource(migrations.AssetNames(),
+		func(name string) ([]byte, error) {
+			return migrations.Asset(name)
+		})
 	d, err := bindata.WithInstance(s)
 	if err != nil {
 		log.Fatal(fmt.Errorf("cannot load migrations: %w", err))
@@ -57,8 +62,7 @@ func ConnectDatabase(dbConnString string) {
 		}
 	}
 	log.Println("Migrations Run")
-	initServices()
+
-	log.Println("Services Initialized")
 }
 
 func initServices() {

pkg/database/migrations/000006_add_ftp_credentials.down.sql

@@ -0,0 +1,6 @@
+BEGIN;
+    ALTER TABLE servers
+        DROP COLUMN ftp_port,
+        DROP COLUMN ftp_username,
+        DROP COLUMN ftp_password;
+COMMIT;

pkg/database/migrations/000006_add_ftp_credentials.up.sql

@@ -0,0 +1,6 @@
+BEGIN;
+    ALTER TABLE servers
+        ADD COLUMN ftp_port int4,
+        ADD COLUMN ftp_username varchar(200),
+        ADD COLUMN ftp_password varchar(200);
+COMMIT;

pkg/database/request.go

@@ -154,7 +154,7 @@ func (s requestService) UpdateRequest(r geeksbot.Request) (geeksbot.Request, err
 	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)
+	_, err := s.db.Exec(queryString, r.ID, r.Completed, r.CompletedAt, r.CompletedBy.ID, r.CompletedMessage)
 	return r, err
 }
 

pkg/database/server.go

@@ -19,11 +19,12 @@ func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
 	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
+						guild_id, info_channel_id, info_message_id, settings_message_id,
+						ftp_port, ftp_username, ftp_password
 						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)
+		&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID, &server.FTPPort, &server.FTPUser, &server.FTPPass)
 	if err != nil {
 		return geeksbot.Server{}, err
 	}
@@ -115,7 +116,8 @@ func (s serverService) DeleteServer(server geeksbot.Server) error {
 func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, error) {
 	queryString := `UPDATE servers SET name = $2, ip_address = $3, port = $4, password = $5,
 						alerts_channel_id = $6, info_channel_id = $7, info_message_id = $8,
-						settings_message_id = $9 WHERE id = $1`
+						settings_message_id = $9, ftp_port = $10, ftp_username = $11,
+						ftp_password = $12 WHERE id = $1`
 	_, err := s.db.Exec(queryString,
 		server.Name,
 		server.IPAddr,
@@ -125,6 +127,9 @@ func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, er
 		server.InfoChannel.ID,
 		server.InfoMessage.ID,
 		server.SettingsMessage.ID,
+		server.FTPPort,
+		server.FTPUser,
+		server.FTPPass,
 	)
 	return server, err
 }

server.go

@@ -11,6 +11,9 @@ type Server struct {
 	InfoChannel     *Channel
 	InfoMessage     *Message
 	SettingsMessage *Message
+	FTPPort         int
+	FTPUser         string
+	FTPPass         string
 }
 
 type ServerService interface {