Minor refactor for fake server players

rc_core/src/persist/user/lobby.rs

@@ -0,0 +1,102 @@
+use super::account_json::UserData;
+
+#[async_trait::async_trait]
+impl super::LobbyUser for UserData {
+    fn user_id(&self) -> i32 {
+        self.account.id
+    }
+
+    async fn player_data(&self, cpu_counter: &crate::cubes::CpuListParser) -> Result<crate::data::player_data::PlayerData, polariton_server::operations::SimpleOpError> {
+        self.user_player_data(cpu_counter).await.map_err(|e| {
+            if let Some(msg) = e.error_msg() {
+                polariton_server::operations::SimpleOpError::with_message(crate::data::error_codes::LobbyReasonCode::from_service_error(e.error_code()) as i16, msg.to_owned())
+            } else {
+                polariton_server::operations::SimpleOpError::with_code(crate::data::error_codes::LobbyReasonCode::from_service_error(e.error_code()) as i16)
+            }
+
+        })
+    }
+
+    async fn start_game(&self, game: super::GameDescriptor, players: Vec<super::PlayerLobbyDescriptor>) -> Result<super::FakePlayers, polariton_server::operations::SimpleOpError> {
+        let now = chrono::Utc::now().timestamp();
+        let guid = crate::persist::user::str_to_i64(&game.guid)
+            .ok_or_else(|| polariton_server::operations::SimpleOpError::with_message(
+                crate::data::error_codes::LobbyReasonCode::UnexpectedError as i16, "Invalid GUID".to_owned()
+            )
+        )?;
+        let variant = if game.is_ranked {
+            oj_rc_database::schema::multiplayer_game::GameType::Ranked
+        } else if game.is_custom {
+            oj_rc_database::schema::multiplayer_game::GameType::Custom
+        } else {
+            oj_rc_database::schema::multiplayer_game::GameType::Standard
+        };
+
+        let fake_players = self.generate_fake_players_data(guid).await;
+
+        let game_dbo = oj_rc_database::schema::multiplayer_game::ActiveModel {
+            id: oj_rc_database::sea_orm::ActiveValue::NotSet,
+            creation_time: oj_rc_database::sea_orm::ActiveValue::Set(now),
+            guid: oj_rc_database::sea_orm::ActiveValue::Set(guid),
+            map: oj_rc_database::sea_orm::ActiveValue::Set(game.map),
+            mode: oj_rc_database::sea_orm::ActiveValue::Set(game.mode.to_db()),
+            visibility: oj_rc_database::sea_orm::ActiveValue::Set(game.visibility.to_db()),
+            auto_heal: oj_rc_database::sea_orm::ActiveValue::Set(game.auto_heal),
+            variant: oj_rc_database::sea_orm::ActiveValue::Set(variant),
+            is_complete: oj_rc_database::sea_orm::ActiveValue::Set(false),
+        };
+        let game_dbo = self.db.insert_game(game_dbo).await.map_err(|e| {
+            log::error!("Failed to create game {} through user_id {}: {}", game.guid, self.account.id, e);
+            polariton_server::operations::SimpleOpError::with_message(
+                crate::data::error_codes::LobbyReasonCode::UnexpectedError as i16,
+                format!("Failed to create game {}: {}", game.guid, e),
+            )
+        })?;
+
+        let players_len = players.len();
+
+        let players: Vec<oj_rc_database::schema::multiplayer_game_player::ActiveModel> = players.into_iter()
+            .enumerate()
+            .map(|(i, player)| {
+                oj_rc_database::schema::multiplayer_game_player::ActiveModel {
+                    id: oj_rc_database::sea_orm::ActiveValue::NotSet,
+                    user_id: oj_rc_database::sea_orm::ActiveValue::Set(Some(player.user_id)),
+                    game_id: oj_rc_database::sea_orm::ActiveValue::Set(game_dbo.id),
+                    creation_time: oj_rc_database::sea_orm::ActiveValue::Set(now),
+                    player_id: oj_rc_database::sea_orm::ActiveValue::Set((i as u8) as _),
+                    team: oj_rc_database::sea_orm::ActiveValue::Set(player.team),
+                    group: oj_rc_database::sea_orm::ActiveValue::Set(player.group),
+                    is_claimed: oj_rc_database::sea_orm::ActiveValue::Set(false),
+                    public_id: oj_rc_database::sea_orm::ActiveValue::Set(player.public_id),
+                    display_name: oj_rc_database::sea_orm::ActiveValue::Set(player.display_name),
+                }
+            })
+            .chain(fake_players.iter()
+                .enumerate()
+                .map(|(i, fake)| {
+                    oj_rc_database::schema::multiplayer_game_player::ActiveModel {
+                        id: oj_rc_database::sea_orm::ActiveValue::NotSet,
+                        user_id: oj_rc_database::sea_orm::ActiveValue::Set(None),
+                        game_id: oj_rc_database::sea_orm::ActiveValue::Set(game_dbo.id),
+                        creation_time: oj_rc_database::sea_orm::ActiveValue::Set(now),
+                        player_id: oj_rc_database::sea_orm::ActiveValue::Set(((i + players_len) as u8) as _),
+                        team: oj_rc_database::sea_orm::ActiveValue::Set(fake.team),
+                        group: oj_rc_database::sea_orm::ActiveValue::Set(None),
+                        is_claimed: oj_rc_database::sea_orm::ActiveValue::Set(true),
+                        public_id: oj_rc_database::sea_orm::ActiveValue::Set(fake.name.clone()),
+                        display_name: oj_rc_database::sea_orm::ActiveValue::Set(fake.display_name.clone()),
+                    }
+                })
+            )
+            .collect();
+        self.db.insert_players(players).await.map_err(|e| {
+            log::error!("Failed to create game players for {} through user_id {}: {}", game.guid, self.account.id, e);
+            polariton_server::operations::SimpleOpError::with_message(
+                crate::data::error_codes::LobbyReasonCode::UnexpectedError as i16,
+                format!("Failed to create game players for {}: {}", game.guid, e),
+            )
+        })?;
+
+        Ok(super::FakePlayers { players: fake_players })
+    }
+}