1 files changed, 53 insertions, 8 deletions

diff --git a/src/commands.cpp b/src/commands.cpp
index 1db5d43..4e67962 100644
--- a/src/commands.cpp
+++ b/src/commands.cpp
@@ -7,11 +7,16 @@ struct SKickArgs {
 	std::string reason;
 };
 
+struct SMuteArgs {
+	std::string target;
+	std::string reason;
+};
+
 void registerCommands() {
 	g_pCommandHandler->registerCommand({.name = "raw", .parser = [](const std::string &args) -> CCommandHandler::SParseResult { return {.data = args, .good = true}; }, .exe = [](const std::any &parsed) -> CCommandHandler::SResult {
-      const std::string &data = std::any_cast<const std::string &>(parsed);
-      g_pSessionManager->broadcast(data);
-      return {.result = "Broadcasted: " + data, .good = true}; }});
+			const std::string &data = std::any_cast<const std::string &>(parsed);
+			g_pSessionManager->broadcast(data);
+			return {.result = "Broadcasted: " + data, .good = true}; }});
 
 	g_pCommandHandler->registerCommand(CCommandHandler::makeCommand<SKickArgs>({.name = "kick", .parser = [](const std::string &args) -> std::pair<SKickArgs, bool> {
       size_t pos = args.find(' ');
@@ -37,10 +42,50 @@ void registerCommands() {
           g_pSessionManager->kick(session, true, data.reason);
           return {.result = "Kicked '" + data.target + "': " + data.reason, .good = true}; }}));
 
-	g_pCommandHandler->registerCommand({.name = "list", .parser = [](const std::string &args) -> CCommandHandler::SParseResult { return {.good = true}; }, .exe = [](const std::any &parsed) -> CCommandHandler::SResult {
-      std::string result = "Clients:\n";
-      for (const auto &session : g_pSessionManager->getSessions())
-        result += session->getName() + "\t" + session->getIp() + "\n";
+	g_pCommandHandler->registerCommand({.name = "list", .parser = [](const std::string &args) -> CCommandHandler::SParseResult {
+			(void)args;
+			return {.data = std::any(), .good = true}; }, .exe = [](const std::any &parsed) -> CCommandHandler::SResult {
+			(void)parsed;
+			std::string result = "Clients:\n";
+			for (const auto &session : g_pSessionManager->getSessions())
+				result += session->getName() + "\t" + session->getIp() + "\t" + (session->isMuted() ? "[MUTED]" : "[UNMUTED]") + "\n";
+
+			return {.result = result, .good = true}; }});
+
+	g_pCommandHandler->registerCommand(CCommandHandler::makeCommand<SMuteArgs>({.name = "mute", .parser = [](const std::string &args) -> std::pair<SMuteArgs, bool> {
+      size_t pos = args.find(' ');
+      std::string target = args;
+      std::string reason = "No reason specified";
+
+      if (pos != std::string::npos) {
+        target = args.substr(0, pos);
+        reason = args.substr(pos + 1);
+        if (reason.empty())
+          reason = "Muted: No reason specified";
+      }
+
+      if (target.empty())
+        return {SMuteArgs{}, false};
+
+      return {SMuteArgs{.target=target, .reason=reason}, true}; }, .exe = [](const SMuteArgs &data) -> CCommandHandler::SResult {
+      const auto session = g_pSessionManager->getByName(data.target);
+      if (!session)
+        return {.result = "User '" + data.target + "' not found", .good = false};
+
+      session->setMuted(true);
+      session->write("You have been muted");
+      return {.result = "Muted '" + data.target + "': " + data.reason, .good = true}; }}));
+
+	g_pCommandHandler->registerCommand(CCommandHandler::makeCommand<SMuteArgs>({.name = "unmute", .parser = [](const std::string &args) -> std::pair<SMuteArgs, bool> {
+      if (args.empty())
+        return {SMuteArgs{}, false};
+
+      return {SMuteArgs{.target=args, .reason="Unmuted"}, true}; }, .exe = [](const SMuteArgs &data) -> CCommandHandler::SResult {
+      const auto session = g_pSessionManager->getByName(data.target);
+      if (!session)
+        return {.result = "User '" + data.target + "' not found", .good = false};
 
-      return {.result = result, .good = true}; }});
+      session->setMuted(false);
+      session->write("You have been unmuted");
+      return {.result = "Unmuted '" + data.target + "'", .good = true}; }}));
 }