feat: Theoros validators stuff (#24)

* feat(theoros_validators): Theoros validators stuff * feat(theoros_validators): theoros storage * feat(theoros_validators): Som docssss * feat(theoros_validators): Hyperlane rpc stuff * feat(theoros_validators): Storage stuff * feat(theoros_validators): Stufff * feat(theoros_validators): docs * feat(theoros_validators): Storages -> Storage * feat(theoros_validators): latest toolchain * feat(theoros_validators): Docs' * feat(theoros_validators):
astraly-labs · Sep 9, 2024 · 13e5c0d · 13e5c0d
1 parent ea8ac52
commit 13e5c0d
Show file tree

Hide file tree

Showing 36 changed files with 667 additions and 297 deletions.
diff --git a/rust/.env.example b/rust/.env.example
diff --git a/rust/Cargo.lock b/rust/Cargo.lock
diff --git a/rust/Cargo.toml b/rust/Cargo.toml
@@ -73,6 +73,7 @@ axum-macros = { version = "0.4.1" }
 ya-gcp = { version = "0.11.3", features = ["storage"] }
 rusoto_s3 = "0.48.0"
 rusoto_core = "0.48.0"
+lazy_static = "1.5.0"
 
 # Apibara DNA (indexing)
 apibara-core = { git = "https://github.com/apibara/dna", rev = "9caa385" }

diff --git a/rust/rust-toolchain.toml b/rust/rust-toolchain.toml
@@ -1,5 +1,5 @@
 [toolchain]
-channel = "1.80.0"
+channel = "1.81.0"
 components = ["rustfmt", "clippy", "rust-analyzer"]
 targets = ["wasm32-unknown-unknown"]
 profile = "minimal"
diff --git a/rust/theoros/Cargo.toml b/rust/theoros/Cargo.toml
@@ -33,5 +33,6 @@ ya-gcp = { workspace = true }
 rusoto_s3 = { workspace = true }
 rusoto_core = { workspace = true }
 alloy = { workspace = true }
+lazy_static = { workspace = true }
 
 pragma-utils = { path = "../pragma-utils" }
diff --git a/rust/theoros/openapi.json b/rust/theoros/openapi.json
@@ -1 +1,141 @@
-{"openapi":"3.0.3","info":{"title":"theoros","description":"","license":{"name":""},"version":"0.1.0"},"paths":{"/v1/calldata":{"get":{"tags":["crate::handlers::get_calldata"],"operationId":"get_calldata","parameters":[],"responses":{"200":{"description":"Get the calldata","content":{"application/json":{"schema":{"type":"array","items":{"$ref":"#/components/schemas/GetCalldataResponse"}}}}}}}}},"components":{"schemas":{"GetCalldataError":{"type":"string","enum":["InternalServerError","DatabaseConnection"]},"GetCalldataQuery":{"type":"object"},"GetCalldataResponse":{"type":"object","required":["hash"],"properties":{"hash":{"type":"string"}}}},"responses":{"GetCalldataResponse":{"description":"","content":{"application/json":{"schema":{"type":"object","required":["hash"],"properties":{"hash":{"type":"string"}}}}}}}},"tags":[{"name":"theoros","description":"Theoros - the Pragma Consultant"}]}
+{
+  "openapi": "3.0.3",
+  "info": {
+    "title": "theoros",
+    "description": "",
+    "license": {
+      "name": ""
+    },
+    "version": "0.1.0"
+  },
+  "paths": {
+    "/v1/calldata": {
+      "get": {
+        "tags": [
+          "crate::handlers::get_data_feeds"
+        ],
+        "operationId": "get_data_feeds",
+        "parameters": [],
+        "responses": {
+          "200": {
+            "description": "Get all the available data feeds",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "type": "array",
+                  "items": {
+                    "$ref": "#/components/schemas/GetDataFeedsResponse"
+                  }
+                }
+              }
+            }
+          }
+        }
+      }
+    },
+    "/v1/calldata/{data_feed_id}": {
+      "get": {
+        "tags": [
+          "crate::handlers::get_calldata"
+        ],
+        "operationId": "get_calldata",
+        "parameters": [],
+        "responses": {
+          "200": {
+            "description": "Constructs the calldata used to update the data feed id specified",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "type": "array",
+                  "items": {
+                    "$ref": "#/components/schemas/GetCalldataResponse"
+                  }
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+  },
+  "components": {
+    "schemas": {
+      "GetCalldataError": {
+        "type": "string",
+        "enum": [
+          "InternalServerError",
+          "DatabaseConnection"
+        ]
+      },
+      "GetCalldataQuery": {
+        "type": "object"
+      },
+      "GetCalldataResponse": {
+        "type": "object",
+        "required": [
+          "hash"
+        ],
+        "properties": {
+          "hash": {
+            "type": "string"
+          }
+        }
+      },
+      "GetDataFeedsError": {
+        "type": "string",
+        "enum": [
+          "InternalServerError",
+          "DatabaseConnection"
+        ]
+      },
+      "GetDataFeedsQuery": {
+        "type": "object"
+      },
+      "GetDataFeedsResponse": {
+        "type": "array",
+        "items": {
+          "type": "string"
+        }
+      }
+    },
+    "responses": {
+      "GetCalldataResponse": {
+        "description": "",
+        "content": {
+          "application/json": {
+            "schema": {
+              "type": "object",
+              "required": [
+                "hash"
+              ],
+              "properties": {
+                "hash": {
+                  "type": "string"
+                }
+              }
+            }
+          }
+        }
+      },
+      "GetDataFeedsResponse": {
+        "description": "",
+        "content": {
+          "application/json": {
+            "schema": {
+              "type": "array",
+              "items": {
+                "type": "string"
+              }
+            }
+          }
+        }
+      }
+    }
+  },
+  "tags": [
+    {
+      "name": "theoros",
+      "description": "Theoros - the Pragma Consultant"
+    }
+  ]
+}
diff --git a/rust/theoros/src/config.rs b/rust/theoros/src/config.rs
diff --git a/rust/theoros/src/errors/data_feeds_error.rs b/rust/theoros/src/errors/data_feeds_error.rs
@@ -0,0 +1,25 @@
+use axum::{http::StatusCode, response::IntoResponse, Json};
+use serde_json::json;
+use utoipa::ToSchema;
+
+#[derive(Debug, thiserror::Error, ToSchema)]
+#[allow(unused)]
+pub enum GetDataFeedsError {
+    #[error("internal server error")]
+    InternalServerError,
+    #[error("could not establish a connection with the database")]
+    DatabaseConnection,
+}
+
+impl IntoResponse for GetDataFeedsError {
+    fn into_response(self) -> axum::response::Response {
+        let (status, err_msg) = match self {
+            Self::DatabaseConnection => {
+                (StatusCode::SERVICE_UNAVAILABLE, "Could not establish a connection with the Database".to_string())
+            }
+            _ => (StatusCode::INTERNAL_SERVER_ERROR, String::from("Internal server error")),
+        };
+        (status, Json(json!({"resource":"Calldata", "message": err_msg, "happened_at" : chrono::Utc::now() })))
+            .into_response()
+    }
+}
diff --git a/rust/theoros/src/errors/mod.rs b/rust/theoros/src/errors/mod.rs
@@ -1,5 +1,7 @@
 pub mod app_error;
 pub mod calldata_error;
+pub mod data_feeds_error;
 
 pub use app_error::AppError;
 pub use calldata_error::GetCalldataError;
+pub use data_feeds_error::GetDataFeedsError;
diff --git a/rust/theoros/src/handlers/get_calldata.rs b/rust/theoros/src/handlers/get_calldata.rs
@@ -4,6 +4,7 @@ use serde::{Deserialize, Serialize};
 use utoipa::{IntoParams, ToResponse, ToSchema};
 
 use crate::errors::GetCalldataError;
+use crate::extractors::PathExtractor;
 use crate::AppState;
 
 #[derive(Default, Deserialize, IntoParams, ToSchema)]
@@ -16,19 +17,23 @@ pub struct GetCalldataResponse {
 
 #[utoipa::path(
     get,
-    // TODO: path
-    path = "/v1/calldata",
+    path = "/v1/calldata/{data_feed_id}",
     responses(
-        (status = 200, description = "Get the calldata", body = [GetCalldataResponse])
+        (
+            status = 200,
+            description = "Constructs the calldata used to update the data feed id specified",
+            body = [GetCalldataResponse]
+        )
     ),
     params(
         GetCalldataQuery
     ),
 )]
 pub async fn get_calldata(
     State(_state): State<AppState>,
+    PathExtractor(data_feed_id): PathExtractor<String>,
     Query(_params): Query<GetCalldataQuery>,
 ) -> Result<Json<GetCalldataResponse>, GetCalldataError> {
-    tracing::info!("Received get calldata request");
+    tracing::info!("Received get calldata request for feed: {data_feed_id}");
     Ok(Json(GetCalldataResponse::default()))
 }
diff --git a/rust/theoros/src/handlers/get_data_feeds.rs b/rust/theoros/src/handlers/get_data_feeds.rs
@@ -0,0 +1,32 @@
+use axum::extract::{Query, State};
+use axum::Json;
+use serde::{Deserialize, Serialize};
+use utoipa::{IntoParams, ToResponse, ToSchema};
+
+use crate::errors::GetDataFeedsError;
+use crate::AppState;
+
+#[derive(Default, Deserialize, IntoParams, ToSchema)]
+pub struct GetDataFeedsQuery {}
+
+#[derive(Debug, Default, Serialize, Deserialize, ToResponse, ToSchema)]
+pub struct GetDataFeedsResponse(pub Vec<String>);
+
+#[utoipa::path(
+    get,
+    // TODO: path
+    path = "/v1/calldata",
+    responses(
+        (status = 200, description = "Get all the available data feeds", body = [GetDataFeedsResponse])
+    ),
+    params(
+        GetDataFeedsQuery
+    ),
+)]
+pub async fn get_data_feeds(
+    State(_state): State<AppState>,
+    Query(_params): Query<GetDataFeedsQuery>,
+) -> Result<Json<GetDataFeedsResponse>, GetDataFeedsError> {
+    tracing::info!("Received get calldata request");
+    Ok(Json(GetDataFeedsResponse::default()))
+}
diff --git a/rust/theoros/src/handlers/mod.rs b/rust/theoros/src/handlers/mod.rs
@@ -1,2 +1,3 @@
 pub mod get_calldata;
+pub mod get_data_feeds;
 pub mod subscribe_to_calldata;