AddHUD + RmHUD forwarding

2 files changed, 11 insertions, 2 deletions

diff --git a/connect.go b/connect.go
index b79f5dd..72ec9ae 100644
--- a/connect.go
+++ b/connect.go
@@ -14,7 +14,8 @@ func connect(conn net.Conn, name string, cc *clientConn) *serverConn {
 		name:             name,
 		aos:              make(map[mt.AOID]struct{}),
 		particleSpawners: make(map[mt.ParticleSpawnerID]struct{}),
-		sounds: make(map[mt.SoundID]struct{}),
+		sounds:           make(map[mt.SoundID]struct{}),
+		huds:             make(map[mt.HUDID]struct{}),
 	}
 	sc.log("-->", "connect")
 	cc.srv = sc
diff --git a/server_conn.go b/server_conn.go
index 3df6274..fc353cc 100644
--- a/server_conn.go
+++ b/server_conn.go
@@ -33,6 +33,8 @@ type serverConn struct {
 	particleSpawners map[mt.ParticleSpawnerID]struct{}
 
 	sounds map[mt.SoundID]struct{}
+
+	huds map[mt.HUDID]struct{}
 }
 
 func (sc *serverConn) client() *clientConn { return sc.clt }
@@ -391,7 +393,7 @@ func handleSrv(sc *serverConn) {
 			prepend(sc.name, &cmd.Name)
 			sc.swapAOID(&cmd.SrcAOID)
 			if cmd.Loop {
-				sc.sounds[sc.ID] = struct{}{}
+				sc.sounds[cmd.ID] = struct{}{}
 			}
 
 			sc.client().SendCmd(cmd)
@@ -401,6 +403,12 @@ func handleSrv(sc *serverConn) {
 		case *mt.ToCltStopSound:
 			delete(sc.sounds, cmd.ID)
 			sc.client().SendCmd(cmd)
+		case *mt.ToCltAddHUD:
+			sc.huds[cmd.ID] = struct{}{}
+			sc.client().SendCmd(cmd)
+		case *mt.ToCltRmHUD:
+			delete(sc.huds, cmd.ID)
+			sc.client().SendCmd(cmd)
 		}
 	}
 }