Implement clan chat channels

Author: Swan

chat/channel.go

@@ -29,6 +29,7 @@ const (
 	ChannelNormal ChannelType = iota
 	ChannelTypeMultiplayer
 	ChannelTypeSpectator
+	ChannelTypeClan
 )
 
 // NewChannel Creates a new chat channel instance

chat/chat.go

@@ -34,6 +34,16 @@ func Initialize() {
 		addChannel(NewChannel(ChannelNormal, channel.Name, channel.Description, channel.AdminOnly, channel.AutoJoin, channel.LimitedChat, channel.DiscordWebhook))
 	}
 
+	clans, err := db.GetAllClans()
+
+	if err != nil {
+		panic(err)
+	}
+
+	for _, clan := range clans {
+		AddClanChannel(clan.Id)
+	}
+
 	_ = sessions.AddUser(Bot)
 	addBotChatHandlers()
 	addSpectatorHandlers()
@@ -47,7 +57,10 @@ func GetAvailableChannels(userGroups common.UserGroups) []*Channel {
 	var availableChannels []*Channel
 
 	for _, channel := range channels {
-		if (channel.Type != ChannelTypeMultiplayer && channel.Type != ChannelTypeSpectator && !channel.AdminOnly) || (channel.AdminOnly && isChatModerator(userGroups)) {
+		if (channel.Type != ChannelNormal &&
+			channel.Type != ChannelTypeMultiplayer &&
+			channel.Type != ChannelTypeSpectator && !channel.AdminOnly) ||
+			(channel.AdminOnly && isChatModerator(userGroups)) {
 			availableChannels = append(availableChannels, channel)
 		}
 	}
@@ -243,6 +256,18 @@ func addChannel(channel *Channel) {
 	log.Printf("Initialized chat channel: %v\n", channel.Name)
 }
 
+// AddClanChannel Adds a new clan channel
+func AddClanChannel(clanId int) *Channel {
+	name := fmt.Sprintf("#clan_%v", clanId)
+
+	channel := NewChannel(ChannelTypeClan, name, "Communicate with your fellow clan members",
+		false, false, false, "")
+
+	addChannel(channel)
+
+	return channel
+}
+
 // Removes a channel from channels
 func removeChannel(channel *Channel) {
 	chatMutex.Lock()

db/clans.go

@@ -0,0 +1,24 @@
+package db
+
+import "database/sql"
+
+type Clan struct {
+	Id           int    `db:"id"`
+	OwnerId      int    `db:"owner_id"`
+	Name         string `db:"name"`
+	Tag          string `db:"tag"`
+	Customizable bool   `db:"customizable"`
+}
+
+// GetAllClans Retrieves all of the clans in the db
+func GetAllClans() ([]*Clan, error) {
+	result := make([]*Clan, 0)
+
+	err := SQL.Select(&result, "SELECT id, owner_id, name, tag, customizable FROM clans")
+
+	if err != nil && err != sql.ErrNoRows {
+		return nil, err
+	}
+
+	return result, nil
+}

db/users.go

@@ -21,6 +21,7 @@ type User struct {
 	Country        string            `db:"country"`
 	AvatarUrl      sql.NullString    `db:"avatar_url"`
 	TwitchUsername sql.NullString    `db:"twitch_username"`
+	ClanId         sql.NullInt32     `db:"clan_id"`
 }
 
 // GetProfileUrl Returns the full profile url for the user
@@ -30,7 +31,7 @@ func (u *User) GetProfileUrl() string {
 
 // GetUserById Retrieves a user from the database by their id
 func GetUserById(id int) (*User, error) {
-	query := "SELECT id, steam_id, username, allowed, privileges, usergroups, mute_endtime, country, avatar_url, twitch_username FROM users WHERE id = ? LIMIT 1"
+	query := "SELECT id, steam_id, username, allowed, privileges, usergroups, mute_endtime, country, avatar_url, twitch_username, clan_id FROM users WHERE id = ? LIMIT 1"
 
 	var user User
 	err := SQL.Get(&user, query, id)
@@ -44,7 +45,7 @@ func GetUserById(id int) (*User, error) {
 
 // GetUserBySteamId Retrieves a user from the database by their Steam id
 func GetUserBySteamId(steamId string) (*User, error) {
-	query := "SELECT id, steam_id, username, allowed, privileges, usergroups, mute_endtime, country, avatar_url, twitch_username FROM users WHERE steam_id = ? LIMIT 1"
+	query := "SELECT id, steam_id, username, allowed, privileges, usergroups, mute_endtime, country, avatar_url, twitch_username, clan_id FROM users WHERE steam_id = ? LIMIT 1"
 
 	var user User
 	err := SQL.Get(&user, query, steamId)
@@ -58,7 +59,7 @@ func GetUserBySteamId(steamId string) (*User, error) {
 
 // GetUserByUsername Rerieves a user from the database by their username
 func GetUserByUsername(username string) (*User, error) {
-	query := "SELECT id, steam_id, username, allowed, privileges, usergroups, mute_endtime, country, avatar_url, twitch_username FROM users WHERE username = ? LIMIT 1"
+	query := "SELECT id, steam_id, username, allowed, privileges, usergroups, mute_endtime, country, avatar_url, twitch_username, clan_id FROM users WHERE username = ? LIMIT 1"
 
 	var user User
 	err := SQL.Get(&user, query, username)

handlers/login.go

@@ -422,6 +422,21 @@ func joinChatChannels(user *sessions.User) {
 			channel.AddUser(user)
 		}
 	}
+
+	// Add user to clan channel (creates the new clan channel if necessary)
+	if user.Info.ClanId.Valid {
+		name := fmt.Sprintf("#clan_%v", user.Info.ClanId.Int32)
+
+		channel := chat.GetChannelByName(name)
+
+		if channel != nil {
+			channel.AddUser(user)
+			return
+		}
+
+		channel = chat.AddClanChannel(int(user.Info.ClanId.Int32))
+		channel.AddUser(user)
+	}
 }
 
 // Logs a generic login failure