privated some variables, minor clean up
This commit is contained in:
parent
e84930df6d
commit
b4515ad957
4
core.go
4
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 == "" {
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
12
guilds.go
12
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
|
||||
|
|
|
|||
Loading…
Reference in New Issue