change payg-price with exchange price ratio

Author: getong

apps/indexer-proxy/proxy/src/graphql.rs

@@ -40,6 +40,23 @@ pub const PROJECT_QUERY: &str =
 
 pub const PAYG_QUERY: &str = "query { getAlivePaygs { id price token expiration overflow } }";
 
+pub fn exchange_rate_query(exchange_rate: &str) -> String {
+    format!(
+        r#"
+query {{
+  getAlivePaygs(exchangeRate: "{}") {{
+    id
+    price
+    token
+    expiration
+    overflow
+  }}
+}}
+"#,
+        exchange_rate
+    )
+}
+
 pub const CHANNEL_QUERY: &str =
     "query { getAliveChannels { id consumer deploymentId agent total spent remote price lastFinal expiredAt } }";
 

apps/indexer-proxy/proxy/src/payg.rs

@@ -40,7 +40,7 @@ use subql_indexer_utils::{
         QueryState, MULTIPLE_RANGE_MAX,
     },
     request::{graphql_request, GraphQLQuery},
-    tools::{cid_deployment, deployment_cid},
+    tools::{cid_deployment, deployment_cid, string_u256},
     types::Result,
 };
 
@@ -53,6 +53,7 @@ use crate::metrics::{MetricsNetwork, MetricsQuery};
 use crate::p2p::report_conflict;
 use crate::project::{get_project, list_projects, Project};
 use crate::sentry_log::make_sentry_message;
+use crate::server::RawPaygPriceItem;
 const CURRENT_VERSION: u8 = 3;
 
 #[derive(Debug)]
@@ -237,6 +238,55 @@ async fn build_project_price(
     )))
 }
 
+async fn build_exchange_rate_price(
+    price_item: &RawPaygPriceItem,
+    expired: i64,
+    controller: &LocalWallet,
+) -> Result<Value> {
+    // sign the price
+    let price = price_item.price.clone();
+    let token = price_item
+        .token
+        .parse()
+        .map_err(|_e| Error::Serialize(1137))?;
+
+    // price + price_token + price_expired
+    let sign = price_sign(string_u256(&price), token, expired, controller).await?;
+
+    Ok(json!((
+        price_item.id.clone(),
+        price.to_string(),
+        price_item.expiration.to_string(),
+        format!("{:?}", token),
+        expired,
+        convert_sign_to_string(&sign)
+    )))
+}
+
+pub async fn merket_price_with_exchange_rate_price(
+    price_list: &Vec<RawPaygPriceItem>,
+) -> Result<Value> {
+    let account = ACCOUNT.read().await;
+    let indexer = account.indexer.clone();
+    let controller = account.controller.clone();
+    let controller_address = account.controller_address();
+    drop(account);
+    let mut values = vec![];
+    let expired = Utc::now().timestamp() + 86400;
+    for price_item in price_list {
+        if string_u256(&price_item.price) > U256::zero() && price_item.expiration > 0 {
+            let v = build_exchange_rate_price(&price_item, expired, &controller).await?;
+            values.push(v);
+        }
+    }
+
+    Ok(json!({
+        "indexer": format!("{:?}", indexer),
+        "controller": format!("{:?}", controller_address),
+        "deployments": values,
+    }))
+}
+
 pub async fn merket_price(project_id: Option<String>) -> Result<Value> {
     let account = ACCOUNT.read().await;
     let indexer = account.indexer.clone();

apps/indexer-proxy/proxy/src/server.rs

@@ -17,9 +17,26 @@
 // along with this program. If not, see <https://www.gnu.org/licenses/>.
 
 #![deny(warnings)]
+use crate::ai::api_stream;
+use crate::auth::{create_jwt, AuthQuery, AuthQueryLimit, Payload};
+use crate::cli::COMMAND;
+use crate::contracts::check_agreement_and_consumer;
+use crate::graphql::{exchange_rate_query, PAYG_QUERY};
+use crate::metrics::{get_owner_metrics, MetricsNetwork, MetricsQuery};
+use crate::payg::{
+    extend_channel, fetch_channel_cache, merket_price, merket_price_with_exchange_rate_price,
+    open_state, pay_channel, query_multiple_state, query_single_state, AuthPayg,
+};
+use crate::project::get_project;
+use crate::sentry_log::make_sentry_message;
+use crate::websocket::{connect_to_project_ws, handle_websocket, validate_project, QueryType};
+use crate::{
+    account::{get_indexer, indexer_healthy},
+    auth::AuthWhitelistQuery,
+};
 use axum::extract::ws::WebSocket;
 use axum::{
-    extract::{ConnectInfo, Path, WebSocketUpgrade},
+    extract::{ConnectInfo, Path, Query, WebSocketUpgrade},
     http::{
         header::{self, HeaderMap, HeaderValue},
         Method, Response, StatusCode,
@@ -34,7 +51,10 @@ use base64::{engine::general_purpose, Engine as _};
 use ethers::prelude::U256;
 use serde::{Deserialize, Serialize};
 use serde_json::{json, Value};
-use std::net::{IpAddr, Ipv4Addr, SocketAddr};
+use std::{
+    collections::HashMap,
+    net::{IpAddr, Ipv4Addr, SocketAddr},
+};
 use subql_indexer_utils::{
     eip712::{recover_consumer_token_payload, recover_indexer_token_payload},
     error::Error,
@@ -44,23 +64,6 @@ use subql_indexer_utils::{
 };
 use tower_http::cors::{Any, CorsLayer};
 
-use crate::ai::api_stream;
-use crate::auth::{create_jwt, AuthQuery, AuthQueryLimit, Payload};
-use crate::cli::COMMAND;
-use crate::contracts::check_agreement_and_consumer;
-use crate::metrics::{get_owner_metrics, MetricsNetwork, MetricsQuery};
-use crate::payg::{
-    extend_channel, fetch_channel_cache, merket_price, open_state, pay_channel,
-    query_multiple_state, query_single_state, AuthPayg,
-};
-use crate::project::get_project;
-use crate::sentry_log::make_sentry_message;
-use crate::websocket::{connect_to_project_ws, handle_websocket, validate_project, QueryType};
-use crate::{
-    account::{get_indexer, indexer_healthy},
-    auth::AuthWhitelistQuery,
-};
-
 #[derive(Serialize)]
 pub struct QueryUri {
     /// the url refer to specific project
@@ -413,7 +416,36 @@ async fn query_limit_handler(
     })))
 }
 
-async fn payg_price() -> Result<Json<Value>, Error> {
+#[derive(Serialize, Deserialize, Debug)]
+pub struct RawPaygPriceItem {
+    pub id: String,
+    pub price: String,
+    pub token: String,
+    pub expiration: u64,
+    pub overflow: u64,
+}
+
+// http://localhost/payg-price?exchange_price=19000000000000000000
+async fn payg_price(
+    Query(query_map): Query<HashMap<String, String>>,
+) -> Result<Json<Value>, Error> {
+    let query = if let Some(exchange_price) = query_map.get("exchange_price") {
+        GraphQLQuery::query(&exchange_rate_query(exchange_price))
+    } else {
+        GraphQLQuery::query(PAYG_QUERY)
+    };
+    let url = COMMAND.graphql_url();
+    if let Ok(value) = graphql_request(&url, &query).await {
+        if let Some(item) = value.pointer("/data/getAlivePaygs") {
+            let price_items: Vec<RawPaygPriceItem> =
+                serde_json::from_str(item.to_string().as_str())
+                    .map_err(|_e| Error::Serialize(1120))?;
+            // warn!("price_items: {:?}", price_items);
+            let projects = merket_price_with_exchange_rate_price(&price_items).await?;
+            return Ok(Json(projects));
+        }
+    }
+
     let projects = merket_price(None).await?;
     Ok(Json(projects))
 }