init with react+axum+spacetimedb

server-rs/crates/spacetime-module/src/big_fish/runtime.rs

@@ -0,0 +1,190 @@
+use crate::big_fish::tables::{big_fish_creation_session, big_fish_runtime_run};
+use crate::*;
+
+#[spacetimedb::procedure]
+pub fn start_big_fish_run(
+    ctx: &mut ProcedureContext,
+    input: BigFishRunStartInput,
+) -> BigFishRunProcedureResult {
+    match ctx.try_with_tx(|tx| start_big_fish_run_tx(tx, input.clone())) {
+        Ok(run) => BigFishRunProcedureResult {
+            ok: true,
+            run: Some(run),
+            error_message: None,
+        },
+        Err(message) => BigFishRunProcedureResult {
+            ok: false,
+            run: None,
+            error_message: Some(message),
+        },
+    }
+}
+
+#[spacetimedb::procedure]
+pub fn submit_big_fish_input(
+    ctx: &mut ProcedureContext,
+    input: BigFishRunInputSubmitInput,
+) -> BigFishRunProcedureResult {
+    match ctx.try_with_tx(|tx| submit_big_fish_input_tx(tx, input.clone())) {
+        Ok(run) => BigFishRunProcedureResult {
+            ok: true,
+            run: Some(run),
+            error_message: None,
+        },
+        Err(message) => BigFishRunProcedureResult {
+            ok: false,
+            run: None,
+            error_message: Some(message),
+        },
+    }
+}
+
+#[spacetimedb::procedure]
+pub fn get_big_fish_run(
+    ctx: &mut ProcedureContext,
+    input: BigFishRunGetInput,
+) -> BigFishRunProcedureResult {
+    match ctx.try_with_tx(|tx| get_big_fish_run_tx(tx, input.clone())) {
+        Ok(run) => BigFishRunProcedureResult {
+            ok: true,
+            run: Some(run),
+            error_message: None,
+        },
+        Err(message) => BigFishRunProcedureResult {
+            ok: false,
+            run: None,
+            error_message: Some(message),
+        },
+    }
+}
+
+fn start_big_fish_run_tx(
+    ctx: &ReducerContext,
+    input: BigFishRunStartInput,
+) -> Result<BigFishRuntimeSnapshot, String> {
+    validate_run_start_input(&input).map_err(|error| error.to_string())?;
+    if ctx
+        .db
+        .big_fish_runtime_run()
+        .run_id()
+        .find(&input.run_id)
+        .is_some()
+    {
+        return Err("big_fish_runtime_run.run_id 已存在".to_string());
+    }
+    let session = ctx
+        .db
+        .big_fish_creation_session()
+        .session_id()
+        .find(&input.session_id)
+        .filter(|row| row.owner_user_id == input.owner_user_id)
+        .ok_or_else(|| "big_fish_creation_session 不存在".to_string())?;
+    let draft = session
+        .draft_json
+        .as_deref()
+        .ok_or_else(|| "big_fish.draft 尚未编译".to_string())
+        .and_then(|value| deserialize_draft(value).map_err(|error| error.to_string()))?;
+    let snapshot = build_initial_runtime_snapshot(
+        input.run_id.clone(),
+        input.session_id.clone(),
+        &draft,
+        input.started_at_micros,
+    );
+    let now = Timestamp::from_micros_since_unix_epoch(input.started_at_micros);
+    ctx.db.big_fish_runtime_run().insert(BigFishRuntimeRun {
+        run_id: input.run_id,
+        session_id: input.session_id,
+        owner_user_id: input.owner_user_id,
+        status: snapshot.status,
+        snapshot_json: serialize_runtime_snapshot(&snapshot).map_err(|error| error.to_string())?,
+        last_input_x: 0.0,
+        last_input_y: 0.0,
+        tick: snapshot.tick,
+        created_at: now,
+        updated_at: now,
+    });
+
+    Ok(snapshot)
+}
+
+fn submit_big_fish_input_tx(
+    ctx: &ReducerContext,
+    input: BigFishRunInputSubmitInput,
+) -> Result<BigFishRuntimeSnapshot, String> {
+    validate_run_input_submit_input(&input).map_err(|error| error.to_string())?;
+    let run = ctx
+        .db
+        .big_fish_runtime_run()
+        .run_id()
+        .find(&input.run_id)
+        .filter(|row| row.owner_user_id == input.owner_user_id)
+        .ok_or_else(|| "big_fish_runtime_run 不存在".to_string())?;
+    let session = ctx
+        .db
+        .big_fish_creation_session()
+        .session_id()
+        .find(&run.session_id)
+        .filter(|row| row.owner_user_id == input.owner_user_id)
+        .ok_or_else(|| "big_fish_creation_session 不存在".to_string())?;
+    let draft = session
+        .draft_json
+        .as_deref()
+        .ok_or_else(|| "big_fish.draft 尚未编译".to_string())
+        .and_then(|value| deserialize_draft(value).map_err(|error| error.to_string()))?;
+    let current_snapshot =
+        deserialize_runtime_snapshot(&run.snapshot_json).map_err(|error| error.to_string())?;
+    let next_snapshot = advance_runtime_snapshot(
+        current_snapshot,
+        &draft.runtime_params,
+        input.input_x,
+        input.input_y,
+        input.submitted_at_micros,
+    );
+    replace_big_fish_run(
+        ctx,
+        &run,
+        BigFishRuntimeRun {
+            run_id: run.run_id.clone(),
+            session_id: run.session_id.clone(),
+            owner_user_id: run.owner_user_id.clone(),
+            status: next_snapshot.status,
+            snapshot_json: serialize_runtime_snapshot(&next_snapshot)
+                .map_err(|error| error.to_string())?,
+            last_input_x: input.input_x,
+            last_input_y: input.input_y,
+            tick: next_snapshot.tick,
+            created_at: run.created_at,
+            updated_at: Timestamp::from_micros_since_unix_epoch(input.submitted_at_micros),
+        },
+    );
+
+    Ok(next_snapshot)
+}
+
+fn get_big_fish_run_tx(
+    ctx: &ReducerContext,
+    input: BigFishRunGetInput,
+) -> Result<BigFishRuntimeSnapshot, String> {
+    validate_run_get_input(&input).map_err(|error| error.to_string())?;
+    let run = ctx
+        .db
+        .big_fish_runtime_run()
+        .run_id()
+        .find(&input.run_id)
+        .filter(|row| row.owner_user_id == input.owner_user_id)
+        .ok_or_else(|| "big_fish_runtime_run 不存在".to_string())?;
+
+    deserialize_runtime_snapshot(&run.snapshot_json).map_err(|error| error.to_string())
+}
+
+fn replace_big_fish_run(
+    ctx: &ReducerContext,
+    current: &BigFishRuntimeRun,
+    next: BigFishRuntimeRun,
+) {
+    ctx.db
+        .big_fish_runtime_run()
+        .run_id()
+        .delete(&current.run_id);
+    ctx.db.big_fish_runtime_run().insert(next);
+}