From 1dd0c432fd3ade1637e36811f3a9c7749229d4b2 Mon Sep 17 00:00:00 2001
From: Shivaram Lingamneni <slingamn@cs.stanford.edu>
Date: Sat, 22 Aug 2020 22:43:21 -0400
Subject: [PATCH] use utils.StringSet in more places

---
 irc/accounts.go          | 2 +-
 irc/client_lookup_set.go | 2 +-
 irc/nickname.go          | 4 ++--
 irc/services.go          | 4 ++--
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/irc/accounts.go b/irc/accounts.go
index e573ed04..e1e5c54c 100644
--- a/irc/accounts.go
+++ b/irc/accounts.go
@@ -381,7 +381,7 @@ func (am *AccountManager) Register(client *Client, account string, callbackNames
 		return errAccountCreation
 	}
 
-	if restrictedCasefoldedNicks[casefoldedAccount] || restrictedSkeletons[skeleton] {
+	if restrictedCasefoldedNicks.Has(casefoldedAccount) || restrictedSkeletons.Has(skeleton) {
 		return errAccountAlreadyRegistered
 	}
 
diff --git a/irc/client_lookup_set.go b/irc/client_lookup_set.go
index 1f4fbe82..b0b82d86 100644
--- a/irc/client_lookup_set.go
+++ b/irc/client_lookup_set.go
@@ -173,7 +173,7 @@ func (clients *ClientManager) SetNick(client *Client, session *Session, newNick
 			return "", errNicknameInvalid, false
 		}
 
-		if restrictedCasefoldedNicks[newCfNick] || restrictedSkeletons[newSkeleton] {
+		if restrictedCasefoldedNicks.Has(newCfNick) || restrictedSkeletons.Has(newSkeleton) {
 			return "", errNicknameInvalid, false
 		}
 
diff --git a/irc/nickname.go b/irc/nickname.go
index c44d7635..150ccb5b 100644
--- a/irc/nickname.go
+++ b/irc/nickname.go
@@ -23,8 +23,8 @@ var (
 		"MemoServ", "BotServ", "OperServ",
 	}
 
-	restrictedCasefoldedNicks = make(map[string]bool)
-	restrictedSkeletons       = make(map[string]bool)
+	restrictedCasefoldedNicks = make(utils.StringSet)
+	restrictedSkeletons       = make(utils.StringSet)
 )
 
 func performNickChange(server *Server, client *Client, target *Client, session *Session, nickname string, rb *ResponseBuffer) error {
diff --git a/irc/services.go b/irc/services.go
index ee5712e3..9fe7dd19 100644
--- a/irc/services.go
+++ b/irc/services.go
@@ -350,11 +350,11 @@ func initializeServices() {
 		if err != nil {
 			panic(err)
 		}
-		restrictedCasefoldedNicks[cfName] = true
+		restrictedCasefoldedNicks.Add(cfName)
 		skeleton, err := Skeleton(restrictedNickname)
 		if err != nil {
 			panic(err)
 		}
-		restrictedSkeletons[skeleton] = true
+		restrictedSkeletons.Add(skeleton)
 	}
 }