start on RPP proxy

Author: TheEnbyperor

Cargo.toml

@@ -28,11 +28,9 @@ clap = { version = "4", features = ["env"] }
 base64 = "0.22"
 xml_serde = "1"
 rust-keycloak = { git = "https://github.com/TheEnbyperor/rust-keycloak", rev = "1f81e41407a1fa123131c2f4d213609165ac8b34" }
-#rust-keycloak = { path = "../rust-keycloak" }
 dotenv = "0.15.0"
 tower-service = "0.3"
 sys-info = "0.9"
-sentry = { version = "0.37", features = ["log"] }
 foreign-types-shared = "0.1"
 lazy_static = "1"
 paste = "1"
@@ -45,13 +43,14 @@ warp = "0.3.6"
 uuid = { version = "1.16.0", features = ["v4"] }
 rustls = { version = "0.23.25", default-features = false, features = ["ring", "std"] }
 once_cell = "1.21.3"
+rocket = { version = "0.5.1", features = ["json"] }
 
 [target.'cfg(target_os = "linux")'.dependencies]
 systemd-journal-logger = "2"
 
 [build-dependencies]
 tonic-build = "0.13"
-built = { version = "0.7", features = ["git2"] }
+built = { version = "0.8", features = ["git2"] }
 
 [dev-dependencies]
 test-log = "0.2"
@@ -67,7 +66,7 @@ path = "src/lib.rs"
 
 [[bin]]
 name = "epp-proxy"
-path = "src/main.rs"
+path = "src/bin/main.rs"
 
 [[bin]]
 name = "verisign-com-net-test"

src/auth.rs

@@ -0,0 +1,62 @@
+#[derive(Copy, Clone)]
+pub enum AuthMethod {
+    OAuth,
+    StaticKey,
+}
+
+impl clap::ValueEnum for AuthMethod {
+    fn value_variants<'a>() -> &'a [Self] {
+        &[Self::OAuth, Self::StaticKey]
+    }
+
+    fn to_possible_value(&self) -> Option<clap::builder::PossibleValue> {
+        Some(match self {
+            Self::OAuth => clap::builder::PossibleValue::new("oauth"),
+            Self::StaticKey => clap::builder::PossibleValue::new("static"),
+        })
+    }
+}
+
+
+#[tonic::async_trait]
+pub trait Auth {
+    async fn auth(&self, token: &str) -> bool;
+}
+
+#[tonic::async_trait]
+impl Auth for rust_keycloak::oauth::OAuthClient {
+    async fn auth(&self, token: &str) -> bool {
+        self.verify_token(token, "access-epp").await.is_ok()
+    }
+}
+
+#[derive(Clone)]
+struct StaticAuth {
+    token: String,
+}
+
+impl StaticAuth {
+    fn new() -> Self {
+        dotenv::dotenv().ok();
+
+        let token = std::env::var("AUTH_TOKEN").expect("AUTH_TOKEN must be set");
+
+        Self {
+            token: token.trim().to_string(),
+        }
+    }
+}
+
+#[tonic::async_trait]
+impl Auth for StaticAuth {
+    async fn auth(&self, token: &str) -> bool {
+        token == self.token
+    }
+}
+
+pub fn create_auth_method(method: AuthMethod) -> Box<dyn Auth + Send + Sync> {
+    match method {
+        AuthMethod::OAuth => Box::new(crate::oauth_client()),
+        AuthMethod::StaticKey => Box::new(StaticAuth::new()),
+    }
+}

src/bin/donuts-test.rs

@@ -225,7 +225,7 @@ async fn main() {
     let mut contact_id_i = 1;
     let contact_id = loop {
         let contact_id = format!("STACLAR-{}", contact_id_i);
-        let res = epp_proxy::client::contact::check(&contact_id, &mut cmd_tx_1)
+        let res = epp_proxy::client::contact::check(&contact_id, None, &mut cmd_tx_1)
             .await
             .unwrap();
         if res.response.avail {
@@ -267,6 +267,7 @@ async fn main() {
             qualified_lawyer: None,
             keysys: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -279,6 +280,7 @@ async fn main() {
         None,
         None,
         None,
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -332,6 +334,7 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -384,6 +387,7 @@ async fn main() {
                 nominet_ext: None,
                 ttl: None,
             },
+            None,
             &mut cmd_tx_1,
         )
         .await
@@ -402,12 +406,13 @@ async fn main() {
             None,
             None,
             None,
+            None,
             &mut cmd_tx_2,
         )
         .await
         .unwrap();
 
-        epp_proxy::client::domain::transfer_accept(domain.as_str(), None, &mut cmd_tx_1)
+        epp_proxy::client::domain::transfer_accept(domain.as_str(), None, None, &mut cmd_tx_1)
             .await
             .unwrap();
     }
@@ -423,6 +428,7 @@ async fn main() {
                 phase_name: None,
             },
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -492,6 +498,7 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -545,6 +552,7 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -598,6 +606,7 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -606,7 +615,7 @@ async fn main() {
     // Register an Early Access domain
     info!("Create early access domain");
     info!("EAP domain: {}", eap_domain);
-    let eap_check = epp_proxy::client::domain::check(&eap_domain, None, None, None, &mut cmd_tx_1)
+    let eap_check = epp_proxy::client::domain::check(&eap_domain, None, None, None, None, &mut cmd_tx_1)
         .await
         .unwrap();
     let eap_fee = eap_check
@@ -668,6 +677,7 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -676,7 +686,7 @@ async fn main() {
     // Register a Premium domain name for 3 separate premium price points
     info!("Create 3 premium domains");
     let premium_check_1 =
-        epp_proxy::client::domain::check(premium_domain_1, None, None, None, &mut cmd_tx_1)
+        epp_proxy::client::domain::check(premium_domain_1, None, None, None, None, &mut cmd_tx_1)
             .await
             .unwrap();
     let premium_fee_1 = premium_check_1
@@ -743,13 +753,14 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
     .unwrap();
 
     let premium_check_2 =
-        epp_proxy::client::domain::check(premium_domain_2, None, None, None, &mut cmd_tx_1)
+        epp_proxy::client::domain::check(premium_domain_2, None, None, None, None, &mut cmd_tx_1)
             .await
             .unwrap();
     let premium_fee_2 = premium_check_2
@@ -816,13 +827,14 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
     .unwrap();
 
     let premium_check_3 =
-        epp_proxy::client::domain::check(premium_domain_3, None, None, None, &mut cmd_tx_1)
+        epp_proxy::client::domain::check(premium_domain_3, None, None, None, None, &mut cmd_tx_1)
             .await
             .unwrap();
     let premium_fee_3 = premium_check_3
@@ -884,6 +896,7 @@ async fn main() {
             nominet_ext: None,
             ttl: None,
         },
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -909,12 +922,13 @@ async fn main() {
         }),
         None,
         None,
+        None,
         &mut cmd_tx_2,
     )
     .await
     .unwrap();
 
-    epp_proxy::client::domain::transfer_accept(premium_domain_1, None, &mut cmd_tx_1)
+    epp_proxy::client::domain::transfer_accept(premium_domain_1, None, None, &mut cmd_tx_1)
         .await
         .unwrap();
 
@@ -945,6 +959,7 @@ async fn main() {
         }),
         None,
         None,
+        None,
         &mut cmd_tx_1,
     )
     .await
@@ -953,7 +968,7 @@ async fn main() {
     // Delete and Restore a Premium name
     info!("Deleting and restoring premiun name");
     let premium_check_dr =
-        epp_proxy::client::domain::check(premium_domain_dr, None, None, None, &mut cmd_tx_1)
+        epp_proxy::client::domain::check(premium_domain_dr, None, None, None, None, &mut cmd_tx_1)
             .await
             .unwrap();
     let premium_fee_dr = premium_check_dr
@@ -973,7 +988,7 @@ async fn main() {
         })
         .unwrap();
 
-    epp_proxy::client::domain::delete(premium_domain_dr, None, None, None, None, &mut cmd_tx_1)
+    epp_proxy::client::domain::delete(premium_domain_dr, None, None, None, None, None, &mut cmd_tx_1)
         .await
         .unwrap();
 
@@ -990,14 +1005,15 @@ async fn main() {
                 }],
             }],
         }),
+        None,
         &mut cmd_tx_1,
     )
     .await
     .unwrap();
 
     info!("Logging out of accounts");
-    let final_cmd_1 = epp_proxy::client::logout(cmd_tx_1).await.unwrap();
-    let final_cmd_2 = epp_proxy::client::logout(cmd_tx_2).await.unwrap();
+    let final_cmd_1 = epp_proxy::client::logout(None, cmd_tx_1).await.unwrap();
+    let final_cmd_2 = epp_proxy::client::logout(None, cmd_tx_2).await.unwrap();
 
     println!(
         "Final command transaction: {:#?}",