-
Notifications
You must be signed in to change notification settings - Fork 18
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
858 additions
and
83 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,13 +1,26 @@ | ||
[package] | ||
name = "world-id-bridge" | ||
version = "0.1.0" | ||
license = "MIT" | ||
edition = "2021" | ||
version = "0.1.0" | ||
name = "world-id-bridge" | ||
authors = ["Miguel Piedrafita <[email protected]>"] | ||
repository = "https://github.com/worldcoin/wallet-bridge" | ||
description = "A bridge between the World ID SDK and the World App" | ||
|
||
[dependencies] | ||
axum = "0.6.20" | ||
tower = "0.4.13" | ||
serde = "1.0.183" | ||
dotenvy = "0.15.7" | ||
tokio = { version = "1", features = ["full"] } | ||
lambda_http = "0.8.1" | ||
axum-aws-lambda = "0.5.0" | ||
tracing = { version = "0.1", features = ["log"] } | ||
tower = { version = "0.4.13", features = ["make"] } | ||
hyper = { version = "0.14.26", features = ["http1", "server", "tcp"] } | ||
tokio = { version = "1.31.0", features = ["full"] } | ||
tower-http = { version = "0.4.3", features = ["cors"] } | ||
redis = { version = "0.23.0", features = ["tokio-comp", "connection-manager"] } | ||
tracing-subscriber = { version = "0.3", default-features = false, features = ["fmt"] } | ||
tracing-subscriber = { version = "0.3", default-features = false, features = [ | ||
"fmt", | ||
] } | ||
|
||
[build-dependencies] | ||
chrono = "0.4.26" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1 @@ | ||
# wallet-bridge | ||
Establishes a very simple zero-knowledge bridge to pass World ID ZKPs from wallets to verifying apps. | ||
# A bridge between the World ID SDK and the World App |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,29 @@ | ||
use chrono::prelude::{DateTime, Utc}; | ||
use std::process::Command; | ||
|
||
fn get_git_rev() -> Option<String> { | ||
let output = Command::new("git") | ||
.args(["rev-parse", "--short", "HEAD"]) | ||
.output() | ||
.ok()?; | ||
|
||
if output.status.success() { | ||
String::from_utf8(output.stdout).ok() | ||
} else { | ||
None | ||
} | ||
} | ||
|
||
fn get_compile_date() -> String { | ||
let system_time = std::time::SystemTime::now(); | ||
let date_time: DateTime<Utc> = system_time.into(); | ||
format!("{}", date_time.format("%+")) | ||
} | ||
|
||
fn main() { | ||
println!("cargo:rustc-env=STATIC_BUILD_DATE={}", get_compile_date()); | ||
|
||
if let Some(rev) = get_git_rev() { | ||
println!("cargo:rustc-env=GIT_REV={}", rev); | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
use axum::Router; | ||
|
||
mod request; | ||
mod response; | ||
mod system; | ||
|
||
pub fn handler() -> Router { | ||
Router::new() | ||
.merge(system::handler()) | ||
.merge(request::handler()) | ||
.merge(response::handler()) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,78 @@ | ||
use axum::{ | ||
body::Bytes, | ||
extract::Path, | ||
http::{Method, StatusCode}, | ||
routing::head, | ||
Extension, Router, | ||
}; | ||
use redis::{aio::ConnectionManager, AsyncCommands}; | ||
use tower_http::cors::{Any, CorsLayer}; | ||
|
||
use crate::EXPIRE_AFTER_SECONDS; | ||
|
||
const REQ_PREFIX: &str = "req:"; | ||
|
||
pub fn handler() -> Router { | ||
let cors = CorsLayer::new() | ||
.allow_methods([Method::PUT, Method::HEAD]) | ||
.allow_origin(Any); | ||
|
||
Router::new().route( | ||
"/request/:request_id", | ||
head(has_request) | ||
.get(get_request) | ||
.put(insert_request) | ||
.route_layer(cors), | ||
) | ||
} | ||
|
||
async fn has_request( | ||
Path(request_id): Path<String>, | ||
Extension(mut redis): Extension<ConnectionManager>, | ||
) -> StatusCode { | ||
let Ok(exists) = redis | ||
.exists::<_, bool>(format!("{REQ_PREFIX}{request_id}")) | ||
.await | ||
else { | ||
return StatusCode::INTERNAL_SERVER_ERROR; | ||
}; | ||
|
||
if exists { | ||
StatusCode::OK | ||
} else { | ||
StatusCode::NOT_FOUND | ||
} | ||
} | ||
|
||
async fn get_request( | ||
Path(request_id): Path<String>, | ||
Extension(mut redis): Extension<ConnectionManager>, | ||
) -> Result<Vec<u8>, StatusCode> { | ||
let value = redis | ||
.get_del::<_, Option<Vec<u8>>>(format!("{REQ_PREFIX}{request_id}")) | ||
.await | ||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; | ||
|
||
value.ok_or(StatusCode::NOT_FOUND) | ||
} | ||
|
||
async fn insert_request( | ||
Path(request_id): Path<String>, | ||
Extension(mut redis): Extension<ConnectionManager>, | ||
body: Bytes, | ||
) -> Result<StatusCode, StatusCode> { | ||
if !redis | ||
.set_nx::<_, _, bool>(format!("{REQ_PREFIX}{request_id}"), body.to_vec()) | ||
.await | ||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)? | ||
{ | ||
return Ok(StatusCode::CONFLICT); | ||
} | ||
|
||
redis | ||
.expire::<_, ()>(format!("{REQ_PREFIX}{request_id}"), EXPIRE_AFTER_SECONDS) | ||
.await | ||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; | ||
|
||
Ok(StatusCode::CREATED) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,57 @@ | ||
use axum::{ | ||
body::Bytes, | ||
extract::Path, | ||
http::{Method, StatusCode}, | ||
routing::get, | ||
Extension, Router, | ||
}; | ||
use redis::{aio::ConnectionManager, AsyncCommands}; | ||
use tower_http::cors::{Any, CorsLayer}; | ||
|
||
use crate::EXPIRE_AFTER_SECONDS; | ||
|
||
const RES_PREFIX: &str = "res:"; | ||
|
||
pub fn handler() -> Router { | ||
let cors = CorsLayer::new() | ||
.allow_methods([Method::GET]) | ||
.allow_origin(Any); | ||
|
||
Router::new().route( | ||
"/response/:request_id", | ||
get(get_response).put(insert_response).route_layer(cors), | ||
) | ||
} | ||
|
||
async fn get_response( | ||
Path(request_id): Path<String>, | ||
Extension(mut redis): Extension<ConnectionManager>, | ||
) -> Result<Vec<u8>, StatusCode> { | ||
let value = redis | ||
.get_del::<_, Option<Vec<u8>>>(format!("{RES_PREFIX}{request_id}")) | ||
.await | ||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; | ||
|
||
value.ok_or(StatusCode::NOT_FOUND) | ||
} | ||
|
||
async fn insert_response( | ||
Path(request_id): Path<String>, | ||
Extension(mut redis): Extension<ConnectionManager>, | ||
body: Bytes, | ||
) -> Result<StatusCode, StatusCode> { | ||
if !redis | ||
.set_nx::<_, _, bool>(format!("{RES_PREFIX}{request_id}"), body.to_vec()) | ||
.await | ||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)? | ||
{ | ||
return Ok(StatusCode::CONFLICT); | ||
} | ||
|
||
redis | ||
.expire::<_, ()>(&request_id, EXPIRE_AFTER_SECONDS) | ||
.await | ||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; | ||
|
||
Ok(StatusCode::CREATED) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,32 @@ | ||
use axum::{routing::get, Json, Router}; | ||
|
||
pub fn handler() -> Router { | ||
Router::new().route("/", get(get_info)) | ||
} | ||
|
||
#[derive(Debug, serde::Serialize)] | ||
pub struct AppVersion { | ||
semver: String, | ||
rev: Option<String>, | ||
compile_time: String, | ||
} | ||
|
||
#[derive(Debug, serde::Serialize)] | ||
pub struct RootResponse { | ||
/// Repository URL | ||
pub repo_url: String, | ||
/// Application version | ||
pub version: AppVersion, | ||
} | ||
|
||
#[allow(clippy::unused_async)] | ||
async fn get_info() -> Json<RootResponse> { | ||
Json(RootResponse { | ||
repo_url: "https://github.com/worldcoin/wallet-bridge".to_string(), | ||
version: AppVersion { | ||
semver: env!("CARGO_PKG_VERSION").to_string(), | ||
compile_time: env!("STATIC_BUILD_DATE").to_string(), | ||
rev: option_env!("GIT_REV").map(ToString::to_string), | ||
}, | ||
}) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
use axum::Extension; | ||
use axum_aws_lambda::LambdaLayer; | ||
use redis::aio::ConnectionManager; | ||
use tower::ServiceBuilder; | ||
|
||
use crate::routes; | ||
|
||
pub async fn start(redis: ConnectionManager) { | ||
let router = routes::handler().layer(Extension(redis)); | ||
|
||
let app = ServiceBuilder::new() | ||
.layer(LambdaLayer::default()) | ||
.service(router); | ||
|
||
lambda_http::run(app).await.unwrap(); | ||
} |