Compare commits

...
6 Commits
Author SHA1 Message Date
Dustin Pianalto 005dfc815f Move migrations to separate func
CI / build (push) Has been cancelled
2021-02-22 23:59:13 -09:00
Dustin Pianalto 986d16b346 Add ftp credentials to domain types
CI / build (push) Has been cancelled
2021-02-22 23:24:19 -09:00
Dustin Pianalto 83c41e2788 Add ftp credentials to db schema
CI / build (push) Has been cancelled
2021-02-22 23:10:57 -09:00
Dustin Pianalto 24aff1be72 Fix request message
CI / build (push) Has been cancelled
2021-02-22 22:19:25 -09:00
Dustin Pianalto b28beaba0c Fix request message 2021-02-22 22:18:59 -09:00
Dustin Pianalto 4f8aaff94a Fix request message 2021-02-22 22:17:18 -09:00
8 changed files with 39 additions and 11 deletions
+1
View File
@@ -28,6 +28,7 @@ func main() {
} }
database.ConnectDatabase(os.Getenv("DATABASE_URL")) database.ConnectDatabase(os.Getenv("DATABASE_URL"))
database.RunMigrations()
services.InitializeServices() services.InitializeServices()
owners := []string{ owners := []string{
+2 -2
View File
@@ -235,7 +235,7 @@ func listCommandFunc(ctx disgoman.Context, args []string) {
_, _ = ctx.Send(fmt.Sprintf("```md\n"+ _, _ = ctx.Send(fmt.Sprintf("```md\n"+
"< Request ID Requested By >\n"+ "< Request ID Requested By >\n"+
"< %11s %-23s >\n"+ "< %-11d %23s >\n"+
"%s\n\n"+ "%s\n\n"+
"Comments: Not Implemented Yet\n"+ "Comments: Not Implemented Yet\n"+
"Requested At: %s\n"+ "Requested At: %s\n"+
@@ -249,5 +249,5 @@ func listCommandFunc(ctx disgoman.Context, args []string) {
)) ))
} }
_, _ = ctx.Send(fmt.Sprintf("There are currently %d open %s", len(requests), utils.PluralizeString("request", len(requests)))) _, _ = ctx.Send(fmt.Sprintf("```There %s currently %d open %s```", utils.PluralizeString("is", len(requests)), len(requests), utils.PluralizeString("request", len(requests))))
} }
+3
View File
@@ -17,5 +17,8 @@ func PluralizeString(s string, i int) string {
if i == 1 { if i == 1 {
return s return s
} }
if s == "is" {
return "are"
}
return s + "s" return s + "s"
} }
+10 -6
View File
@@ -23,10 +23,6 @@ var (
) )
func ConnectDatabase(dbConnString string) { func ConnectDatabase(dbConnString string) {
s := bindata.Resource(migrations.AssetNames(),
func(name string) ([]byte, error) {
return migrations.Asset(name)
})
var err error var err error
db, err = sql.Open("postgres", dbConnString) db, err = sql.Open("postgres", dbConnString)
if err != nil { if err != nil {
@@ -36,6 +32,15 @@ func ConnectDatabase(dbConnString string) {
db.SetMaxOpenConns(75) db.SetMaxOpenConns(75)
db.SetMaxIdleConns(5) db.SetMaxIdleConns(5)
db.SetConnMaxLifetime(300) db.SetConnMaxLifetime(300)
initServices()
log.Println("Services Initialized")
}
func RunMigrations() {
s := bindata.Resource(migrations.AssetNames(),
func(name string) ([]byte, error) {
return migrations.Asset(name)
})
d, err := bindata.WithInstance(s) d, err := bindata.WithInstance(s)
if err != nil { if err != nil {
log.Fatal(fmt.Errorf("cannot load migrations: %w", err)) log.Fatal(fmt.Errorf("cannot load migrations: %w", err))
@@ -57,8 +62,7 @@ func ConnectDatabase(dbConnString string) {
} }
} }
log.Println("Migrations Run") log.Println("Migrations Run")
initServices()
log.Println("Services Initialized")
} }
func initServices() { func initServices() {
@@ -0,0 +1,6 @@
BEGIN;
ALTER TABLE servers
DROP COLUMN ftp_port,
DROP COLUMN ftp_username,
DROP COLUMN ftp_password;
COMMIT;
@@ -0,0 +1,6 @@
BEGIN;
ALTER TABLE servers
ADD COLUMN ftp_port int4,
ADD COLUMN ftp_username varchar(200),
ADD COLUMN ftp_password varchar(200);
COMMIT;
+8 -3
View File
@@ -19,11 +19,12 @@ func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
var iMsgID sql.NullString var iMsgID sql.NullString
var sMsgID sql.NullString var sMsgID sql.NullString
queryString := `SELECT id, name, ip_address, port, password, alerts_channel_id, queryString := `SELECT id, name, ip_address, port, password, alerts_channel_id,
guild_id, info_channel_id, info_message_id, settings_message_id guild_id, info_channel_id, info_message_id, settings_message_id,
ftp_port, ftp_username, ftp_password
FROM servers WHERE id = $1` FROM servers WHERE id = $1`
row := s.db.QueryRow(queryString, id) row := s.db.QueryRow(queryString, id)
err := row.Scan(&server.ID, &server.Name, &server.IPAddr, &server.Port, &server.Password, err := row.Scan(&server.ID, &server.Name, &server.IPAddr, &server.Port, &server.Password,
&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID) &aChanID, &guildID, &iChanID, &iMsgID, &server.FTPPort, &server.FTPUser, &server.FTPPass, &sMsgID)
if err != nil { if err != nil {
return geeksbot.Server{}, err 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) { func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, error) {
queryString := `UPDATE servers SET name = $2, ip_address = $3, port = $4, password = $5, queryString := `UPDATE servers SET name = $2, ip_address = $3, port = $4, password = $5,
alerts_channel_id = $6, info_channel_id = $7, info_message_id = $8, alerts_channel_id = $6, info_channel_id = $7, info_message_id = $8,
settings_message_id = $9 WHERE id = $1` settings_message_id = $9, ftp_port = $10, ftp_username = $11,
ftp_password = $12 WHERE id = $1`
_, err := s.db.Exec(queryString, _, err := s.db.Exec(queryString,
server.Name, server.Name,
server.IPAddr, server.IPAddr,
@@ -125,6 +127,9 @@ func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, er
server.InfoChannel.ID, server.InfoChannel.ID,
server.InfoMessage.ID, server.InfoMessage.ID,
server.SettingsMessage.ID, server.SettingsMessage.ID,
server.FTPPort,
server.FTPUser,
server.FTPPass,
) )
return server, err return server, err
} }
+3
View File
@@ -11,6 +11,9 @@ type Server struct {
InfoChannel *Channel InfoChannel *Channel
InfoMessage *Message InfoMessage *Message
SettingsMessage *Message SettingsMessage *Message
FTPPort int
FTPUser string
FTPPass string
} }
type ServerService interface { type ServerService interface {