tests: allow admin interface to create an arknote

Signed-off-by: Vincenzo Palazzo <vincenzopalazzodev@gmail.com>

Author: vincenzopalazzo

ark-client/src/lib.rs

@@ -556,6 +556,15 @@ where
         self.redeem_notes(rng, vec![arknote]).await
     }
 
+    pub async fn create_arknote(&self, amount: Amount) -> Result<ArkNote, Error> {
+        let note = self
+            .network_client()
+            .create_arknote(amount.to_sat() as u32, 1)
+            .await?;
+        let note = ArkNote::from_string(&note[0])?;
+        Ok(note)
+    }
+
     pub async fn transaction_history(&self) -> Result<Vec<history::Transaction>, Error> {
         let mut boarding_transactions = Vec::new();
         let mut boarding_commitment_transactions = Vec::new();

ark-core/src/arknote.rs

@@ -33,7 +33,6 @@ pub struct Status {
     pub confirmed: bool,
 }
 
-
 impl fmt::Display for ArkNote {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         let value = self.to_encoded_string();
@@ -319,7 +318,6 @@ impl ArkNote {
     }
 }
 
-
 #[cfg(test)]
 mod tests {
     use super::*;

ark-grpc/build.rs

@@ -19,6 +19,7 @@ fn generate_protos() -> std::io::Result<()> {
                 "proto/ark/v1/wallet.proto",
                 "proto/ark/v1/indexer.proto",
                 "proto/ark/v1/types.proto",
+                "proto/ark/v1/admin.proto",
             ],
             &["proto"],
         )?;

ark-grpc/proto/ark/v1/admin.proto

@@ -0,0 +1,155 @@
+syntax = "proto3";
+
+package ark.v1;
+
+import "google/api/annotations.proto";
+import "ark/v1/types.proto";
+
+service AdminService {
+  rpc GetScheduledSweep(GetScheduledSweepRequest) returns (GetScheduledSweepResponse) {
+    option (google.api.http) = {
+      get: "/v1/admin/sweeps"
+    };
+  }
+  rpc GetRoundDetails(GetRoundDetailsRequest) returns (GetRoundDetailsResponse) {
+    option (google.api.http) = {
+      get: "/v1/admin/round/{round_id}"
+    };
+  }
+  rpc GetRounds(GetRoundsRequest) returns (GetRoundsResponse) {
+    option (google.api.http) = {
+      get: "/v1/admin/rounds"
+    };
+  }
+  rpc CreateNote(CreateNoteRequest) returns (CreateNoteResponse) {
+    option (google.api.http) = {
+      post: "/v1/admin/note"
+      body: "*"
+    };
+  }
+  rpc GetMarketHourConfig(GetMarketHourConfigRequest) returns (GetMarketHourConfigResponse) {
+    option (google.api.http) = {
+      get: "/v1/admin/marketHour"
+    };
+  }
+  rpc UpdateMarketHourConfig(UpdateMarketHourConfigRequest) returns (UpdateMarketHourConfigResponse) {
+    option (google.api.http) = {
+      post: "/v1/admin/marketHour"
+      body: "*"
+    };
+  }
+  rpc ListIntents(ListIntentsRequest) returns (ListIntentsResponse) {
+    option (google.api.http) = {
+      get: "/v1/admin/intents"
+    };
+  }
+  rpc DeleteIntents(DeleteIntentsRequest) returns (DeleteIntentsResponse) {
+    option (google.api.http) = {
+      post: "/v1/admin/intents/delete"
+      body: "*"
+    };
+  }
+}
+
+message GetScheduledSweepRequest {}
+message GetScheduledSweepResponse {
+  repeated ScheduledSweep sweeps = 1;
+}
+
+message GetRoundDetailsRequest {
+  string round_id = 1;
+}
+message GetRoundDetailsResponse {
+  string round_id = 1;
+  int64 started_at = 2;
+  int64 ended_at = 3;
+  string commitment_txid = 4;
+  string forfeited_amount = 5;
+  string total_vtxos_amount = 6;
+  string total_exit_amount = 7;
+  string total_fee_amount = 8;
+  repeated string inputs_vtxos = 9;
+  repeated string outputs_vtxos = 10;
+  repeated string exit_addresses = 11;
+}
+
+message GetRoundsRequest {
+  int64 after = 1;
+  int64 before = 2;
+}
+message GetRoundsResponse {
+  repeated string rounds = 1;
+}
+
+message CreateNoteRequest {
+  uint32 amount = 1;
+  uint32 quantity = 2;
+}
+message CreateNoteResponse {
+  repeated string notes = 1;
+}
+
+message GetMarketHourConfigRequest {}
+message GetMarketHourConfigResponse {
+  MarketHourConfig config = 1;
+}
+
+message UpdateMarketHourConfigRequest {
+  MarketHourConfig config = 1;
+}
+message UpdateMarketHourConfigResponse {}
+
+message ListIntentsRequest {
+  repeated string intent_ids = 1;
+}
+message ListIntentsResponse {
+  repeated IntentInfo intents = 1;
+}
+
+message DeleteIntentsRequest {
+  repeated string intent_ids = 1;
+}
+message DeleteIntentsResponse {}
+
+message SweepableOutput {
+  string txid = 1;
+  uint32 vout = 2;
+  string amount = 3;
+  int64 scheduled_at = 4;
+}
+
+message ScheduledSweep {
+  string round_id = 1;
+  repeated SweepableOutput outputs = 2;
+}
+
+message MarketHourConfig {
+  int64 start_time = 1;
+  int64 end_time = 2;
+  int64 period = 3;
+  int64 round_interval = 4;
+}
+
+message IntentInput {
+  string txid = 1;
+  uint32 vout = 2;
+  uint64 amount = 3;
+}
+
+message IntentInfo {
+  string id = 1;
+  int64 created_at = 2;
+  repeated Output receivers = 3;
+  repeated IntentInput inputs = 4;
+  repeated IntentInput boarding_inputs = 5;
+  repeated string cosigners_public_keys = 6;
+  Bip322Signature proof = 7;
+}
+
+message Output {
+  oneof destination {
+    string vtxo_script = 1;
+    string onchain_address = 2;
+  }; // onchain or off-chain
+  uint64 amount = 3; // Amount to send in satoshis.
+}

ark-grpc/src/client.rs

@@ -1,9 +1,11 @@
 use crate::generated;
+use crate::generated::ark::v1::admin_service_client::AdminServiceClient;
 use crate::generated::ark::v1::ark_service_client::ArkServiceClient;
 use crate::generated::ark::v1::indexer_service_client::IndexerServiceClient;
 use crate::generated::ark::v1::indexer_tx_history_record::Key;
 use crate::generated::ark::v1::Bip322Signature;
 use crate::generated::ark::v1::ConfirmRegistrationRequest;
+use crate::generated::ark::v1::CreateNoteRequest;
 use crate::generated::ark::v1::GetEventStreamRequest;
 use crate::generated::ark::v1::GetInfoRequest;
 use crate::generated::ark::v1::GetTransactionsStreamRequest;
@@ -63,6 +65,7 @@ pub struct Client {
     url: String,
     ark_client: Option<ArkServiceClient<tonic::transport::Channel>>,
     indexer_client: Option<IndexerServiceClient<tonic::transport::Channel>>,
+    admin_client: Option<AdminServiceClient<tonic::transport::Channel>>,
 }
 
 impl Client {
@@ -71,6 +74,7 @@ impl Client {
             url,
             ark_client: None,
             indexer_client: None,
+            admin_client: None,
         }
     }
 
@@ -81,9 +85,13 @@ impl Client {
         let indexer_client = IndexerServiceClient::connect(self.url.clone())
             .await
             .map_err(Error::connect)?;
+        let admin_client = AdminServiceClient::connect(self.url.clone())
+            .await
+            .map_err(Error::connect)?;
 
         self.ark_client = Some(ark_service_client);
         self.indexer_client = Some(indexer_client);
+        self.admin_client = Some(admin_client);
         Ok(())
     }
 
@@ -454,13 +462,26 @@ impl Client {
         Ok(result)
     }
 
+    pub async fn create_arknote(&self, amount: u32, quantity: u32) -> Result<Vec<String>, Error> {
+        let mut client = self.admin_client()?;
+
+        let request = CreateNoteRequest { amount, quantity };
+
+        let response = client.create_note(request).await.map_err(Error::request)?;
+
+        Ok(response.into_inner().notes)
+    }
+
     fn ark_client(&self) -> Result<ArkServiceClient<tonic::transport::Channel>, Error> {
         // Cloning an `ArkServiceClient<Channel>` is cheap.
         self.ark_client.clone().ok_or(Error::not_connected())
     }
     fn indexer_client(&self) -> Result<IndexerServiceClient<tonic::transport::Channel>, Error> {
         self.indexer_client.clone().ok_or(Error::not_connected())
     }
+    fn admin_client(&self) -> Result<AdminServiceClient<tonic::transport::Channel>, Error> {
+        self.admin_client.clone().ok_or(Error::not_connected())
+    }
 }
 
 impl TryFrom<generated::ark::v1::BatchStartedEvent> for BatchStartedEvent {