3 files changed, 47 insertions, 3 deletions
diff --git a/mumd/src/state.rs b/mumd/src/state.rs
index 81b6c98..def5c03 100644
--- a/mumd/src/state.rs
+++ b/mumd/src/state.rs
@@ -217,6 +217,47 @@ impl State {
                 self.reload_config();
                 now!(Ok(None))
             }
+            Command::DeafenSelf => {
+                if !matches!(*self.phase_receiver().borrow(), StatePhase::Connected) {
+                    return now!(Err(Error::DisconnectedError));
+                }
+
+                let mut msg = msgs::UserState::new();
+                msg.set_self_deaf(true);
+                self.packet_sender.send(msg.into()).unwrap();
+
+                now!(Ok(None))
+            }
+            Command::MuteSelf => {
+                if !matches!(*self.phase_receiver().borrow(), StatePhase::Connected) {
+                    return now!(Err(Error::DisconnectedError));
+                }
+
+                let mut msg = msgs::UserState::new();
+                msg.set_self_mute(true);
+                self.packet_sender.send(msg.into()).unwrap();
+
+                now!(Ok(None))
+            }
+            Command::MuteOther(string) => {
+                if !matches!(*self.phase_receiver().borrow(), StatePhase::Connected) {
+                    return now!(Err(Error::DisconnectedError));
+                }
+
+                let id = self.server().unwrap().users().iter().find(|(_, user)| user.name() == &string);
+
+                let id = match id {
+                    Some(id) => *id.0,
+                    None => return now!(Err(Error::InvalidUserIdentifierError(string))),
+                };
+
+                let mut msg = msgs::UserState::new();
+                msg.set_mute(true);
+                msg.set_session(id);
+                self.packet_sender.send(msg.into()).unwrap();
+
+                return now!(Ok(None));
+            }
         }
     }
 
diff --git a/mumlib/src/command.rs b/mumlib/src/command.rs
index e404056..63dd5f9 100644
--- a/mumlib/src/command.rs
+++ b/mumlib/src/command.rs
@@ -18,6 +18,9 @@ pub enum Command {
     },
     ServerDisconnect,
     Status,
+    DeafenSelf,
+    MuteSelf,
+    MuteOther(String),
 }
 
 #[derive(Debug, Deserialize, Serialize)]
diff --git a/mumlib/src/error.rs b/mumlib/src/error.rs
index c6d9255..e8b0323 100644
--- a/mumlib/src/error.rs
+++ b/mumlib/src/error.rs
@@ -9,6 +9,7 @@ pub enum Error {
     DisconnectedError,
     AlreadyConnectedError,
     ChannelIdentifierError(String, ChannelIdentifierError),
+    InvalidUserIdentifierError(String),
     InvalidServerAddrError(String, u16),
 }
 
@@ -18,9 +19,8 @@ impl Display for Error {
             Error::DisconnectedError => write!(f, "Not connected to a server"),
             Error::AlreadyConnectedError => write!(f, "Already connected to a server"),
             Error::ChannelIdentifierError(id, kind) => write!(f, "{}: {}", kind, id),
-            Error::InvalidServerAddrError(addr, port) => {
-                write!(f, "Invalid server address: {}: {}", addr, port)
-            }
+            Error::InvalidServerAddrError(addr, port) => write!(f, "Invalid server address: {}: {}", addr, port),
+            Error::InvalidUserIdentifierError(name) => write!(f, "Invalid username: {}", name),
         }
     }
 }