feature: add cli_dmsg

Author: zensh

Cargo.lock

@@ -2,6 +2,7 @@
 members = [
   "src/cli_airdrop",
   "src/cli_cryptogram",
+  "src/cli_dmsg",
   "src/ic_message",
   "src/ic_message_channel",
   "src/ic_message_profile",
@@ -55,3 +56,5 @@ ic-certification = "2.6"
 ic-oss-types = "0.9"
 getrandom = { version = "0.2", features = ["custom"] }
 rand = { version = "0.8", features = ["getrandom"] }
+chrono = { version = "0.4", features = ["serde"] }
+clap = { version = "=4.5", features = ["derive"] }

Cargo.toml

@@ -0,0 +1,25 @@
+#!/usr/bin/env bash
+
+# Load the environment variables
+source "$(pwd)"/proposals/env.sh
+
+# build and get batch_id, evidence:
+# dfx deploy ic_message_frontend --ic --by-proposal
+
+export BLOB="$(didc encode --format blob '(record {batch_id=29:nat; evidence=blob "\dd\f7\32\7f\0a\6a\82\dc\5e\56\8c\f0\57\eb\39\a3\97\f9\66\45\69\1e\64\73\41\3f\45\57\68\7d\b2\c7"})')"
+
+quill sns make-proposal --canister-ids-file ./sns_canister_ids.json --pem-file $PROPOSAL_PEM_FILE $PROPOSAL_NEURON_ID --proposal "(
+    record {
+        title = \"Execute commit_proposed_batch() to revert ic_message_frontend from v2.6.7 to v2.6.6\";
+        url = \"https://dMsg.net/\";
+        summary = \"This proposal executes commit_proposed_batch() on 2fvu6-tqaaa-aaaap-akksa-cai to release ic_message_frontend v2.6.6.\n\n1. chore: update dependencies for Svelte5.\";
+        action = opt variant {
+            ExecuteGenericNervousSystemFunction = record {
+                function_id = 1100 : nat64;
+                payload = ${BLOB};
+            }
+        };
+    }
+)" > proposal-message.json
+
+# quill send proposal-message.json

proposals/proposal-219.sh

@@ -16,7 +16,7 @@ sha2 = { workspace = true }
 tokio = { workspace = true }
 ic-agent = { workspace = true }
 ic-certification = { workspace = true }
-clap = { version = "=4.5", features = ["derive"] }
+clap = { workspace = true }
 ciborium = { workspace = true }
 num-traits = { workspace = true }
 icrc-ledger-types = { git = "https://github.com/dfinity/ic/", rev = "d19fa446ab35780b2c6d8b82ea32d808cca558d5" }

src/cli_airdrop/Cargo.toml

@@ -12,4 +12,4 @@ lib_panda = { path = "../lib_panda", version = "0.2" }
 candid = { workspace = true }
 base64 = { workspace = true }
 serde = { workspace = true }
-clap = { version = "=4.5", features = ["derive"] }
+clap = { workspace = true }

src/cli_cryptogram/Cargo.toml

@@ -0,0 +1,27 @@
+[package]
+name = "cli_dmsg"
+version = "0.1.0"
+edition = "2021"
+
+[[bin]]
+name = "cli_dmsg"
+path = "src/main.rs"
+
+[dependencies]
+candid = { workspace = true, features = ["value", "printer"] }
+hex = { workspace = true }
+serde = { workspace = true }
+serde_bytes = { workspace = true }
+sha2 = { workspace = true }
+tokio = { workspace = true }
+ic-agent = { workspace = true }
+ic-certification = { workspace = true }
+ciborium = { workspace = true }
+num-traits = { workspace = true }
+chrono = { workspace = true }
+clap = { workspace = true }
+icrc-ledger-types = { git = "https://github.com/dfinity/ic/", rev = "d19fa446ab35780b2c6d8b82ea32d808cca558d5" }
+ic-icrc1 = { git = "https://github.com/dfinity/ic/", rev = "d19fa446ab35780b2c6d8b82ea32d808cca558d5" }
+ic-ledger-core = { git = "https://github.com/dfinity/ic/", rev = "d19fa446ab35780b2c6d8b82ea32d808cca558d5" }
+ic-icrc1-tokens-u64 = { git = "https://github.com/dfinity/ic/", rev = "d19fa446ab35780b2c6d8b82ea32d808cca558d5" }
+ic_message_types = { path = "../ic_message_types", version = "2" }

src/cli_dmsg/Cargo.toml

@@ -0,0 +1,63 @@
+use candid::{
+    utils::{encode_args, ArgumentEncoder},
+    CandidType, Decode, Principal,
+};
+use ic_agent::{Agent, Identity};
+
+use super::format_error;
+
+pub async fn build_agent(host: &str, identity: Box<dyn Identity>) -> Result<Agent, String> {
+    let agent = Agent::builder()
+        .with_url(host)
+        .with_boxed_identity(identity)
+        .with_verify_query_signatures(true)
+        .build()
+        .map_err(format_error)?;
+    if host.starts_with("http://") {
+        agent.fetch_root_key().await.map_err(format_error)?;
+    }
+
+    Ok(agent)
+}
+
+pub async fn update_call<In, Out>(
+    agent: &Agent,
+    canister_id: &Principal,
+    method_name: &str,
+    args: In,
+) -> Result<Out, String>
+where
+    In: ArgumentEncoder + Send,
+    Out: CandidType + for<'a> candid::Deserialize<'a>,
+{
+    let input = encode_args(args).map_err(format_error)?;
+    let res = agent
+        .update(canister_id, method_name)
+        .with_arg(input)
+        .call_and_wait()
+        .await
+        .map_err(format_error)?;
+    let output = Decode!(res.as_slice(), Out).map_err(format_error)?;
+    Ok(output)
+}
+
+pub async fn query_call<In, Out>(
+    agent: &Agent,
+    canister_id: &Principal,
+    method_name: &str,
+    args: In,
+) -> Result<Out, String>
+where
+    In: ArgumentEncoder + Send,
+    Out: CandidType + for<'a> candid::Deserialize<'a>,
+{
+    let input = encode_args(args).map_err(format_error)?;
+    let res = agent
+        .query(canister_id, method_name)
+        .with_arg(input)
+        .call()
+        .await
+        .map_err(format_error)?;
+    let output = Decode!(res.as_slice(), Out).map_err(format_error)?;
+    Ok(output)
+}

src/cli_dmsg/src/agent.rs

@@ -0,0 +1,58 @@
+use candid::{Nat, Principal};
+use ic_agent::Agent;
+use ic_message_types::profile::UserInfo;
+use icrc_ledger_types::icrc1::{
+    account::Account,
+    transfer::{TransferArg, TransferError},
+};
+
+use super::{
+    agent::{query_call, update_call},
+    format_error,
+};
+
+#[derive(Clone)]
+pub struct DMsgAgent {
+    pub agent: Agent,
+    pub canister_id: Principal,
+}
+
+impl DMsgAgent {
+    pub async fn get_user_by_username(&self, username: &str) -> Result<UserInfo, String> {
+        let output: Result<UserInfo, String> = query_call(
+            &self.agent,
+            &self.canister_id,
+            "get_by_username",
+            (username,),
+        )
+        .await?;
+
+        output
+    }
+
+    pub async fn send_token_to(
+        &self,
+        token_canister: &Principal,
+        to: Principal,
+        amount_e8s: u64,
+    ) -> Result<Nat, String> {
+        let output: Result<Nat, TransferError> = update_call(
+            &self.agent,
+            &token_canister,
+            "icrc1_transfer",
+            (TransferArg {
+                from_subaccount: None,
+                to: Account {
+                    owner: to,
+                    subaccount: None,
+                },
+                fee: None,
+                created_at_time: None,
+                memo: None,
+                amount: amount_e8s.into(),
+            },),
+        )
+        .await?;
+        output.map_err(format_error)
+    }
+}