Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
53dc44abda | ||
|
|
6262ebfeb4 | ||
|
|
005dfc815f | ||
|
|
986d16b346 | ||
|
|
83c41e2788 |
@@ -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{
|
||||||
|
|||||||
@@ -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;
|
||||||
@@ -154,7 +154,7 @@ func (s requestService) UpdateRequest(r geeksbot.Request) (geeksbot.Request, err
|
|||||||
queryString := `UPDATE requests SET
|
queryString := `UPDATE requests SET
|
||||||
completed = $2, completed_at = $3, completed_by = $4, completed_message = $5
|
completed = $2, completed_at = $3, completed_by = $4, completed_message = $5
|
||||||
WHERE id = $1`
|
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
|
return r, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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, &sMsgID, &server.FTPPort, &server.FTPUser, &server.FTPPass)
|
||||||
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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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 {
|
||||||
|
|||||||
Reference in New Issue
Block a user