layla/birdbot
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Implement Persentance with SQLite3 (#6)

Browse Source
This commit is contained in:
Layla
2023-04-01 03:48:01 -04:00
committed by GitHub
parent e1038a15cd
commit a40ec4dd78
9 changed files with 904 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
modules/role_selection.go
View File

@ -7,19 +7,23 @@ import (
"github.com/yeslayla/birdbot/common"
"github.com/yeslayla/birdbot/core"
"github.com/yeslayla/birdbot/discord"
"github.com/yeslayla/birdbot/persistence"
)
type roleSelectionModule struct {
session *discord.Discord
session *discord.Discord
db persistence.Database
cfg core.RoleSelectionConfig
exlusive bool
}
// NewRoleSelectionComponent creates a new component
func NewRoleSelectionComponent(discord *discord.Discord, cfg core.RoleSelectionConfig) common.Module {
func NewRoleSelectionComponent(discord *discord.Discord, db persistence.Database, cfg core.RoleSelectionConfig) common.Module {
return &roleSelectionModule{
session: discord,
cfg: cfg,
db: db,
exlusive: true,
}
}
@ -84,7 +88,16 @@ func (c *roleSelectionModule) Initialize(birdbot common.ModuleManager) error {
actionRow.AddComponent(btn)
}
c.session.CreateMessageComponent(c.cfg.SelectionChannel, fmt.Sprintf("**%s**\n%s", c.cfg.Title, c.cfg.Description), components)
localID := fmt.Sprint("role_selection:", c.cfg.Title)
messageID, err := c.db.GetDiscordMessage(localID)
if err != nil {
return err
}
if messageID == "" {
messageID = c.session.CreateMessageComponent(c.cfg.SelectionChannel, fmt.Sprintf("**%s**\n%s", c.cfg.Title, c.cfg.Description), components)
return c.db.SetDiscordMessage(localID, messageID)
}
return nil
}