Merge pull request #14 from Neptune-Crypto/pow-puzzle-endpoint

Pow puzzle endpoint

src/main.rs

@@ -1,5 +1,6 @@
 use anyhow::Context;
 use axum::routing::get;
+use axum::routing::post;
 use axum::routing::Router;
 use neptune_explorer::alert_email;
 use neptune_explorer::html::page::block::block_page;
@@ -11,6 +12,8 @@ use neptune_explorer::model::app_state::AppState;
 use neptune_explorer::neptune_rpc;
 use neptune_explorer::rpc::block_digest::block_digest;
 use neptune_explorer::rpc::block_info::block_info;
+use neptune_explorer::rpc::pow_puzzle::pow_puzzle;
+use neptune_explorer::rpc::provide_pow_solution::provide_pow_solution;
 use neptune_explorer::rpc::utxo_digest::utxo_digest;
 use tower_http::services::ServeDir;
 use tracing::info;
@@ -53,6 +56,8 @@ pub fn setup_routes(app_state: AppState) -> Router {
         .route("/rpc/block_info/*selector", get(block_info))
         .route("/rpc/block_digest/*selector", get(block_digest))
         .route("/rpc/utxo_digest/:index", get(utxo_digest))
+        .route("/rpc/pow_puzzle/*address", get(pow_puzzle))
+        .route("/rpc/provide_pow_solution", post(provide_pow_solution))
         // -- Dynamic HTML pages --
         .route("/", get(root))
         .route("/block/*selector", get(block_page))

src/rpc/mod.rs

@@ -1,3 +1,5 @@
 pub mod block_digest;
 pub mod block_info;
+pub mod pow_puzzle;
+pub mod provide_pow_solution;
 pub mod utxo_digest;

src/rpc/pow_puzzle.rs

@@ -0,0 +1,36 @@
+use axum::extract::Path;
+use axum::extract::State;
+use axum::response::IntoResponse;
+use axum::response::Json;
+use neptune_cash::models::state::wallet::address::generation_address::GenerationReceivingAddress;
+use neptune_cash::rpc_server::error::RpcError;
+use neptune_cash::rpc_server::ProofOfWorkPuzzle;
+use std::sync::Arc;
+use tarpc::context;
+
+use crate::http_util::not_found_err;
+use crate::http_util::rpc_err;
+use crate::http_util::rpc_method_err;
+use crate::model::app_state::AppState;
+
+#[axum::debug_handler]
+pub async fn pow_puzzle(
+    Path(address): Path<String>,
+    State(state): State<Arc<AppState>>,
+) -> Result<Json<ProofOfWorkPuzzle>, impl IntoResponse> {
+    let s = state.load();
+    let Ok(receiving_address) = GenerationReceivingAddress::from_bech32m(&address, s.network)
+    else {
+        return Err(rpc_method_err(RpcError::Failed(address)));
+    };
+    match s
+        .rpc_client
+        .pow_puzzle_external_key(context::current(), s.token(), receiving_address.into())
+        .await
+        .map_err(rpc_err)?
+        .map_err(rpc_method_err)?
+    {
+        Some(pow_puzzle) => Ok(Json(pow_puzzle)),
+        None => Err(not_found_err()),
+    }
+}

src/rpc/provide_pow_solution.rs

@@ -0,0 +1,39 @@
+use crate::http_util::rpc_err;
+use crate::http_util::rpc_method_err;
+use crate::model::app_state::AppState;
+use axum::extract::State;
+use axum::response::Json;
+use axum::response::Response;
+use neptune_cash::models::blockchain::block::block_header::BlockPow;
+use neptune_cash::prelude::twenty_first::tip5::Digest;
+use serde::Deserialize;
+use serde::Serialize;
+use std::sync::Arc;
+use tarpc::context;
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct PowSolution {
+    pow: BlockPow,
+    proposal_id: Digest,
+}
+
+#[axum::debug_handler]
+pub async fn provide_pow_solution(
+    State(state): State<Arc<AppState>>,
+    Json(payload): Json<PowSolution>,
+) -> Result<Json<bool>, Response> {
+    let s = state.load();
+    let result = s
+        .rpc_client
+        .provide_pow_solution(
+            context::current(),
+            s.token(),
+            payload.pow,
+            payload.proposal_id,
+        )
+        .await
+        .map_err(rpc_err)?
+        .map_err(rpc_method_err)?;
+
+    Ok(Json(result))
+}