From b4515ad957606fe91968cfabdcef470b58e501a1 Mon Sep 17 00:00:00 2001 From: Riley Date: Sat, 21 Aug 2021 19:16:39 -0700 Subject: [PATCH] privated some variables, minor clean up --- core.go | 4 ++-- databases/fs/fs.go | 14 ++++++++------ guilds.go | 12 ++++++------ 3 files changed, 16 insertions(+), 14 deletions(-) diff --git a/core.go b/core.go index 296997d..c75886a 100644 --- a/core.go +++ b/core.go @@ -117,8 +117,8 @@ func Start() { if initProvider == nil { log.Fatalf("You have not chosen a database provider. Please refer to the docs") } - CurrentProvider = initProvider() - loadGuilds() + currentProvider = initProvider() + Guilds = loadGuilds() // We need a token if botToken == "" { diff --git a/databases/fs/fs.go b/databases/fs/fs.go index 2ca88be..a9ee6e6 100644 --- a/databases/fs/fs.go +++ b/databases/fs/fs.go @@ -32,7 +32,7 @@ var saveLock = make(map[string]*sync.Mutex) // loadGuilds // Load all known guilds from the filesystem, from inside GuildsDir -func loadGuilds() { +func loadGuilds() (guilds map[string]*framework.Guild) { // Check if the configured guild directory exists, and create it if otherwise if _, existErr := os.Stat(GuildsDir); os.IsNotExist(existErr) { mkErr := os.MkdirAll(GuildsDir, 0755) @@ -42,10 +42,11 @@ func loadGuilds() { log.Warningf("There are no Guilds to load; data for new Guilds will be saved to: %s", GuildsDir) // There are no guilds to load, so we can return early - return + return guilds } // Get a list of files in the directory + guilds = make(map[string]*framework.Guild) files, rdErr := ioutil.ReadDir(GuildsDir) if rdErr != nil { log.Fatalf("Failed to read guild directory: %s", rdErr) @@ -98,15 +99,15 @@ func loadGuilds() { } // Add the loaded guild to the map - framework.Guilds[guildId] = &framework.Guild{ + guilds[guildId] = &framework.Guild{ ID: guildId, Info: gInfo, } } - if len(framework.Guilds) == 0 { + if len(guilds) == 0 { log.Warningf("There are no guilds to load; data for new guilds will be saved to \"%s\"", GuildsDir) - return + return guilds } // :) @@ -115,7 +116,8 @@ func loadGuilds() { plural = "s" } - log.Infof("Loaded %d guild%s", len(framework.Guilds), plural) + log.Infof("Loaded %d guild%s", len(guilds), plural) + return guilds } // save diff --git a/guilds.go b/guilds.go index 6d02155..9bf4a2e 100644 --- a/guilds.go +++ b/guilds.go @@ -33,7 +33,7 @@ type GuildInfo struct { // of storage types type GuildProvider struct { Save func(guild *Guild) - Load func() + Load func() map[string]*Guild } // Guild @@ -49,10 +49,10 @@ type Guild struct { // Otherwise, there will be information desync var Guilds = make(map[string]*Guild) -// CurrentProvider +// currentProvider // A reference to a function that provides the guild info system with a database // Or similar system to save guild data. -var CurrentProvider GuildProvider +var currentProvider GuildProvider // getGuild // Return a Guild object corresponding to the given guildId @@ -116,14 +116,14 @@ func getGuild(guildId string) *Guild { // loadGuilds // Load all known guilds from the database -func loadGuilds() { - CurrentProvider.Load() +func loadGuilds() map[string]*Guild { + return currentProvider.Load() } // save // saves guild data to the database func (g *Guild) save() { - CurrentProvider.Save(g) + currentProvider.Save(g) } // GetMember