Allow hosts to set rankings during enrollment

Arena-DAO · Dec 26, 2024 · ec5bbc9 · ec5bbc9
1 parent 3871efa
commit ec5bbc9
Show file tree

Hide file tree

Showing 5 changed files with 83 additions and 3 deletions.
diff --git a/contracts/arena-competition-enrollment/schema/arena-competition-enrollment.json b/contracts/arena-competition-enrollment/schema/arena-competition-enrollment.json
@@ -201,6 +201,34 @@
         },
         "additionalProperties": false
       },
+      {
+        "type": "object",
+        "required": [
+          "set_rankings"
+        ],
+        "properties": {
+          "set_rankings": {
+            "type": "object",
+            "required": [
+              "id",
+              "rankings"
+            ],
+            "properties": {
+              "id": {
+                "$ref": "#/definitions/Uint128"
+              },
+              "rankings": {
+                "type": "array",
+                "items": {
+                  "$ref": "#/definitions/MemberMsg_for_String"
+                }
+              }
+            },
+            "additionalProperties": false
+          }
+        },
+        "additionalProperties": false
+      },
       {
         "description": "Update the contract's ownership. The `action` to be provided can be either to propose transferring ownership to an account, accept a pending ownership transfer, or renounce the ownership permanently.",
         "type": "object",
@@ -579,6 +607,22 @@
         },
         "additionalProperties": false
       },
+      "MemberMsg_for_String": {
+        "type": "object",
+        "required": [
+          "addr",
+          "seed"
+        ],
+        "properties": {
+          "addr": {
+            "type": "string"
+          },
+          "seed": {
+            "$ref": "#/definitions/Uint64"
+          }
+        },
+        "additionalProperties": false
+      },
       "ModuleInstantiateInfo": {
         "description": "Information needed to instantiate a module.",
         "type": "object",

diff --git a/contracts/arena-competition-enrollment/src/contract.rs b/contracts/arena-competition-enrollment/src/contract.rs
@@ -74,6 +74,9 @@ pub fn execute(
             group_contract_info,
             require_team_size,
         ),
+        ExecuteMsg::SetRankings { id, rankings } => {
+            execute::set_rankings(deps, env, info, id, rankings)
+        }
         ExecuteMsg::TriggerExpiration { id, escrow_id } => {
             execute::trigger_expiration(deps, env, info, id, escrow_id)
         }

diff --git a/contracts/arena-competition-enrollment/src/execute.rs b/contracts/arena-competition-enrollment/src/execute.rs
@@ -1,7 +1,7 @@
 use arena_interface::{
     competition::msg::EscrowInstantiateInfo,
     core::{CompetitionModuleQuery, CompetitionModuleResponse},
-    group::{self, GroupContractInfo},
+    group::{self, GroupContractInfo, MemberMsg},
 };
 use arena_league_module::msg::LeagueInstantiateExt;
 use arena_tournament_module::{msg::TournamentInstantiateExt, state::EliminationType};
@@ -611,3 +611,32 @@ fn get_min_min_members(competition_type: &CompetitionType) -> Uint64 {
         },
     }
 }
+
+pub fn set_rankings(
+    deps: DepsMut,
+    _env: Env,
+    info: MessageInfo,
+    id: Uint128,
+    rankings: Vec<MemberMsg<String>>,
+) -> Result<Response, ContractError> {
+    let enrollment = enrollment_entries().load(deps.storage, id.u128())?;
+
+    ensure!(
+        enrollment.host == info.sender,
+        ContractError::Unauthorized {}
+    );
+
+    let msg = WasmMsg::Execute {
+        contract_addr: enrollment.group_contract.to_string(),
+        msg: to_json_binary(&group::ExecuteMsg::UpdateMembers {
+            to_add: None,
+            to_update: Some(rankings),
+            to_remove: None,
+        })?,
+        funds: vec![],
+    };
+
+    Ok(Response::new()
+        .add_attribute("action", "set_rankings")
+        .add_message(msg))
+}
diff --git a/contracts/arena-competition-enrollment/src/msg.rs b/contracts/arena-competition-enrollment/src/msg.rs
@@ -1,4 +1,4 @@
-use arena_interface::fees::FeeInformation;
+use arena_interface::{fees::FeeInformation, group::MemberMsg};
 use cosmwasm_schema::{cw_serde, QueryResponses};
 use cosmwasm_std::{Coin, Uint128, Uint64};
 use cw_utils::Expiration;
@@ -48,6 +48,10 @@ pub enum ExecuteMsg {
         id: Uint128,
         members: Vec<String>,
     },
+    SetRankings {
+        id: Uint128,
+        rankings: Vec<MemberMsg<String>>,
+    },
 }
 
 #[cw_serde]

diff --git a/scripts/state.json b/scripts/state.json
@@ -2,7 +2,7 @@
   "neutron": {
     "neutron-1": {
       "code_ids": {
-        "arena_competition_enrollment": 2876,
+        "arena_competition_enrollment": 2878,
         "arena_core": 2217,
         "arena_discord_identity": 2874,
         "arena_escrow": 2875,