Implement non-multiplayer parts of custom games

rc_services_room/src/data/custom_games.rs

@@ -1,6 +1,5 @@
 #![allow(dead_code)]
-
-pub use oj_rc_core::data::game_mode::GameMode;
+use polariton::operation::Typed;
 
 #[repr(u8)]
 #[derive(Copy, Clone)]
@@ -8,3 +7,192 @@ pub enum CustomGameInviteCode {
     NoInvite = 0,
     PendingInvite = 1,
 }
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum SessionCreateResponseCode {
+    SessionCreated = 0,
+    AlreadyInSession = 1,
+    SessionCreateError = 2,
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum AdjustCustomGameConfigResponseCode {
+    Success = 0,
+    NotInSession = 1,
+    AdjustmentRejected = 2,
+    // 3, 4, 5 also exist but seem equivalent to 2
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum SessionLeaveResponseCode {
+    Success = 0,
+    NotInSession = 1,
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum KickResponseCode {
+    SessionNoLongerExists = 0,
+    KickTargetIsNotInsession = 1,
+    UserIsNotSessionLeader = 2,
+    UserRemovedFromSession = 3,
+    ErrorKickingFromSession = 4
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum SessionRetrieveResponse {
+    UserNotInAnySession = 0,
+    SessionRetrieved = 1,
+    PlayerIsInvitedOnly = 2,
+    ErrorRetrievingSession = 3,
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum InviteToCustomGameResponseCode {
+    UserIsNotInSession = 0,
+    UserIsNotSessionLeader = 1,
+    InviteeHasAlreadyBeenInvited = 2,
+    UserIsNotOnline = 3,
+    UserInvited = 4,
+    ErrorDispatchingMessage = 5,
+    InviteeIsInAnotherCustomGame = 6,
+    InviteeIsAlreadyInvitedToAnotherCustomGame = 7,
+    UserDoesNotExist = 8,
+    UserOnlyAcceptsInvitesFromFriendsAndClanmates = 9,
+    UserBlockedYou = 10,
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum InviteReplyCustomGameResponseCode {
+    Success = 1,
+    Failure2 = 2,
+    UserIsNotInSession = 4,
+    UserIsNoLongerInvited = 5,
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum CheckCanJoinQueueResponseCode {
+    UserNotInSession0 = 0,
+    OnlyOneAllowed = 1,
+    Unbalanced = 2,
+    AlreadyInBattle = 3,
+    Ok = 4,
+    UserNotInSession5 = 5,
+}
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum PlayerSessionStatus {
+    Unknown = 0,
+    Ready = 1,
+    Queuing = 2,
+    InBattle = 3,
+}
+
+impl PlayerSessionStatus {
+    #[inline]
+    pub fn from_u8(num: u8) -> Option<Self> {
+        match num {
+            0 => Some(Self::Unknown),
+            1 => Some(Self::Ready),
+            2 => Some(Self::Queuing),
+            3 => Some(Self::InBattle),
+            _ => None,
+        }
+    }
+
+    #[inline]
+    pub fn to_u8(self) -> u8 {
+        self as u8
+    }
+}
+
+pub struct Session {
+    pub leader: String,
+    pub session: String,
+    pub members: Vec<String>,
+    pub members_display_name: Vec<String>,
+    pub invited: Vec<String>,
+    pub team_b_members: Vec<String>,
+    pub config: std::collections::HashMap<String, String>,
+    pub avatar_info: std::collections::HashMap<String, oj_rc_core::data::player_data::AvatarInfo>,
+    pub player_session_state: std::collections::HashMap<String, PlayerSessionStatus>,
+}
+
+impl Session {
+    pub fn as_transmissible<C>(&self) -> Typed<C> {
+        Typed::HashMap(vec![
+            (Typed::Str("Leader".into()), Typed::Str(self.leader.clone().into())),
+            (Typed::Str("SessionID".into()), Typed::Str(self.session.clone().into())),
+            (Typed::Str("Members".into()), Typed::HashMap(self.members.iter()
+                .enumerate()
+                .map(|(i, pub_id)| (Typed::Int(i as _), Typed::Str(pub_id.into())))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+            (Typed::Str("MembersDisplayName".into()), Typed::HashMap(self.members_display_name.iter()
+                .enumerate()
+                .map(|(i, display_name)| (Typed::Int(i as _), Typed::Str(display_name.into())))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+            (Typed::Str("Invited".into()), Typed::HashMap(self.invited.iter()
+                .enumerate()
+                .map(|(i, pub_id)| (Typed::Int(i as _), Typed::Str(pub_id.into())))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+            (Typed::Str("TeamBMembers".into()), Typed::HashMap(self.team_b_members.iter()
+                .enumerate()
+                .map(|(i, pub_id)| (Typed::Int(i as _), Typed::Str(pub_id.into())))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+            (Typed::Str("Config".into()), Typed::HashMap(self.config.iter()
+                .map(|(key, val)| (Typed::Str(key.into()), Typed::Str(val.into())))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+            (Typed::Str("AvatarInfo".into()), Typed::HashMap(self.avatar_info.iter()
+                .map(|(key, val)| (Typed::Str(key.into()), val.as_transmissible()))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+            (Typed::Str("PlayerSessionState".into()), Typed::HashMap(self.player_session_state.iter()
+                .map(|(key, val)| (Typed::Str(key.into()), Typed::Int(*val as _)))
+                .collect::<Vec<(_, _)>>()
+                .into()
+            )),
+        ].into())
+    }
+}
+
+/// not to be confused with the event which contains the same data
+/// ... just serialized differently (to be difficult?)
+pub struct CustomGameInvite {
+    pub inviter_public_id: String,
+    pub inviter_display_name: String,
+    pub session: String,
+    pub avatar_id: Option<i32>,
+    pub invited_to_team_b: bool,
+}
+
+impl CustomGameInvite {
+    pub fn as_transmissible<C>(&self) -> Typed<C> {
+        Typed::HashMap(vec![
+            (polariton::operation::Typed::Str("SenderName".into()), polariton::operation::Typed::Str(self.inviter_public_id.clone().into())),
+            (polariton::operation::Typed::Str("SenderDisplayName".into()), polariton::operation::Typed::Str(self.inviter_display_name.clone().into())),
+            (polariton::operation::Typed::Str("SessionGUID".into()), polariton::operation::Typed::Str(self.session.clone().into())),
+            (polariton::operation::Typed::Str("UseCustomAvatar".into()), polariton::operation::Typed::Bool(self.avatar_id.is_none())),
+            (polariton::operation::Typed::Str("AvatarID".into()), polariton::operation::Typed::Int(self.avatar_id.unwrap_or(0))),
+            (polariton::operation::Typed::Str("IsInvitedToTeamB".into()), polariton::operation::Typed::Bool(self.invited_to_team_b)),
+        ].into())
+    }
+}