From 9350346b4cc7679af39357e36b32a9a49d820df5 Mon Sep 17 00:00:00 2001
From: Riley Smith <riley@rileysmith.me>
Date: Mon, 31 Jan 2022 00:30:29 -0800
Subject: [PATCH] state stuff in backend

---
 backend/.idea/workspace.xml             |  7 +---
 backend/internal/ws/handlers.go         | 18 +++++++---
 backend/internal/ws/hub.go              |  4 +++
 backend/internal/ws/state.go            | 46 +++++++++++++++++++++++++
 frontend/src/components/Player.tsx      | 20 +++++------
 frontend/src/interfaces/SocketEvents.ts |  1 +
 frontend/src/pages/player.tsx           |  8 +----
 7 files changed, 75 insertions(+), 29 deletions(-)
 create mode 100644 backend/internal/ws/state.go

diff --git a/backend/.idea/workspace.xml b/backend/.idea/workspace.xml
index ab20c62..8f3dc7f 100644
--- a/backend/.idea/workspace.xml
+++ b/backend/.idea/workspace.xml
@@ -5,16 +5,11 @@
   </component>
   <component name="ChangeListManager">
     <list default="true" id="8a64704d-5500-41a6-aa4c-e275933fc58c" name="Changes" comment="">
-      <change afterPath="$PROJECT_DIR$/cmd/watchtogether/build.sh" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/../frontend/src/interfaces/SocketEvents.ts" afterDir="false" />
+      <change afterPath="$PROJECT_DIR$/internal/ws/state.go" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/cmd/watchtogether/main.go" beforeDir="false" afterPath="$PROJECT_DIR$/cmd/watchtogether/main.go" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/internal/ws/handlers.go" beforeDir="false" afterPath="$PROJECT_DIR$/internal/ws/handlers.go" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/internal/ws/hub.go" beforeDir="false" afterPath="$PROJECT_DIR$/internal/ws/hub.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/internal/ws/message.go" beforeDir="false" afterPath="$PROJECT_DIR$/internal/ws/message.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/../frontend/src/components/Container.tsx" beforeDir="false" afterPath="$PROJECT_DIR$/../frontend/src/components/Container.tsx" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/../frontend/src/components/Player.tsx" beforeDir="false" afterPath="$PROJECT_DIR$/../frontend/src/components/Player.tsx" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/../frontend/src/interfaces/Identity.ts" beforeDir="false" afterPath="$PROJECT_DIR$/../frontend/src/interfaces/Identity.ts" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/../frontend/src/pages/player.tsx" beforeDir="false" afterPath="$PROJECT_DIR$/../frontend/src/pages/player.tsx" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
diff --git a/backend/internal/ws/handlers.go b/backend/internal/ws/handlers.go
index de42b2b..d018a8d 100644
--- a/backend/internal/ws/handlers.go
+++ b/backend/internal/ws/handlers.go
@@ -12,15 +12,19 @@ func handleIdentifyEvent(message *Message) {
 	if id, ok := d["clientID"]; ok {
 		log.Infof("Client %s has sent identify event", id.(string))
 	}
+	user := d["user"].(map[string]interface{})
+	userId := user["id"].(string)
+	playhead := message.hub.State.playhead
+
 	m := Message{
 		MessageData: MessageData{
 			Type: Identify,
 			Data: map[string]interface{}{
-				"admin":      true,
-				"playlist":   "http://localhost:8081/BelleOpening.m3u8",
-				"controller": true,
-				"playhead":   0,
-				"user":       d["user"],
+				"admin":         message.hub.State.IsAdmin(userId),
+				"playlist":      "http://localhost:8081/BelleOpening.m3u8",
+				"hasController": message.hub.State.IsController(userId),
+				"playhead":      playhead,
+				"user":          d["user"],
 			},
 		},
 	}
@@ -54,6 +58,10 @@ func handleSetPlayhead(message *Message) {
 		},
 	}
 	log.Infof("Received SetPlayhead event. playhead is at %s", d["playhead"])
+	err := message.hub.State.setPlayhead(d["playhead"].(float64))
+	if err != nil {
+		log.Errorf("unable to set playhead. %s", err)
+	}
 	for client := range message.Client.hub.Clients {
 		if client == message.Client {
 			continue
diff --git a/backend/internal/ws/hub.go b/backend/internal/ws/hub.go
index 0ca7fc5..efec173 100644
--- a/backend/internal/ws/hub.go
+++ b/backend/internal/ws/hub.go
@@ -4,6 +4,9 @@ type Hub struct {
 	// Registered Clients
 	Clients map[*Client]bool
 
+	// State
+	State *State
+
 	// Inbound messages from the Clients
 	broadcast chan RawMessage
 
@@ -20,6 +23,7 @@ func NewHub() *Hub {
 		register:   make(chan *Client),
 		unregister: make(chan *Client),
 		Clients:    make(map[*Client]bool),
+		State:      NewState(),
 	}
 }
 
diff --git a/backend/internal/ws/state.go b/backend/internal/ws/state.go
new file mode 100644
index 0000000..8e97abe
--- /dev/null
+++ b/backend/internal/ws/state.go
@@ -0,0 +1,46 @@
+package ws
+
+import (
+	"errors"
+	"sync"
+)
+
+type State struct {
+	sync.RWMutex
+
+	playhead         float64
+	controllerUserId string
+	adminUserId      string
+	paused           bool
+}
+
+func NewState() *State {
+	return &State{
+		playhead:         0,
+		controllerUserId: "218072060923084802",
+		adminUserId:      "218072060923084802",
+	}
+}
+
+func (s *State) setPlayhead(playhead float64) error {
+	if s == nil {
+		return errors.New("unable to find state")
+	}
+	s.Lock()
+	defer s.Lock()
+	s.playhead = playhead
+	return nil
+}
+
+func (s *State) IsController(userID string) bool {
+	if userID == s.controllerUserId {
+		return true
+	}
+	return false
+}
+func (s *State) IsAdmin(userID string) bool {
+	if userID == s.adminUserId {
+		return true
+	}
+	return false
+}
diff --git a/frontend/src/components/Player.tsx b/frontend/src/components/Player.tsx
index d4af009..ebd43d2 100644
--- a/frontend/src/components/Player.tsx
+++ b/frontend/src/components/Player.tsx
@@ -2,6 +2,7 @@ import { Box, css } from "@chakra-ui/react";
 import React, { FC, useRef, useState } from "react";
 import ReactPlayer, { Config, ReactPlayerProps } from "react-player";
 import { MessageTypes } from "../interfaces/IMessage";
+import SocketEvents from "../interfaces/SocketEvents";
 import Message from "../util/Message";
 import MessageUtil from "../util/MessageUtil";
 import PlayerSocket from "../ws/websocket";
@@ -11,13 +12,6 @@ type PlayerProps = {
   socket: PlayerSocket;
 } & ReactPlayerProps;
 
-interface ProgressProps {
-  played: number;
-  playedSeconds: number;
-  loaded: number;
-  loadedSeconds: number;
-}
-
 const Player: FC<PlayerProps> = (props) => {
   const playerRef = useRef<ReactPlayer>(null);
   const [paused, setPaused] = useState<boolean>(false);
@@ -27,21 +21,25 @@ const Player: FC<PlayerProps> = (props) => {
       forceHLS: true,
     },
   };
-  const onProgress = (state: ProgressProps) => {
+  socket.emitter.on(SocketEvents.GetPlayhead, (e) => {
+    playerRef.current.seekTo(e.playhead);
+  });
+  const onSeek = (playedSeconds: number) => {
     if (paused) {
       socket?.send(
         MessageUtil.encode(
           new Message(MessageTypes.SetPlayhead, {
-            playhead: state.playedSeconds,
+            playhead: playedSeconds,
             paused: true,
           })
         )
       );
+      return;
     }
     socket?.send(
       MessageUtil.encode(
         new Message(MessageTypes.SetPlayhead, {
-          playhead: state.playedSeconds,
+          playhead: playedSeconds,
           paused: false,
         })
       )
@@ -79,7 +77,7 @@ const Player: FC<PlayerProps> = (props) => {
         controls
         onPlay={onPlay}
         onPause={onPause}
-        onProgress={onProgress}
+        onSeek={onSeek}
         ref={playerRef}
         {...props}
       />
diff --git a/frontend/src/interfaces/SocketEvents.ts b/frontend/src/interfaces/SocketEvents.ts
index 74d37b7..9b25b0a 100644
--- a/frontend/src/interfaces/SocketEvents.ts
+++ b/frontend/src/interfaces/SocketEvents.ts
@@ -1,5 +1,6 @@
 enum SocketEvents {
   Identify = "Identify",
+  GetPlayhead = "GetPlayhead",
 }
 
 export default SocketEvents;
diff --git a/frontend/src/pages/player.tsx b/frontend/src/pages/player.tsx
index cf69902..608fb27 100644
--- a/frontend/src/pages/player.tsx
+++ b/frontend/src/pages/player.tsx
@@ -1,20 +1,14 @@
-import { Socket } from "dgram";
 import { GetServerSideProps, NextPage } from "next";
 import { User } from "next-auth";
 import { getSession } from "next-auth/react";
 import dynamic from "next/dynamic";
 import Head from "next/head";
-import React, { useRef, useState } from "react";
-import ReactPlayer from "react-player";
-import { BaseReactPlayerProps } from "react-player/base";
+import React, { useState } from "react";
 import { Container } from "../components/Container";
 import useWS from "../hooks/useWS";
 import IdentityData from "../interfaces/Identity";
-import { MessageTypes } from "../interfaces/IMessage";
 import SocketEvents from "../interfaces/SocketEvents";
 import isBrowser from "../util/isBrowser";
-import Message from "../util/Message";
-import MessageUtil from "../util/MessageUtil";
 
 const Player = dynamic(() => import("../components/Player"), { ssr: false });