feat(unseal): add unseal protocol support for user agents

server/crates/arbiter-server/src/context/bootstrap.rs

@@ -1,104 +0,0 @@
-use arbiter_proto::{BOOTSTRAP_TOKEN_PATH, home_path};
-use diesel::{ExpressionMethods, QueryDsl};
-use diesel_async::RunQueryDsl;
-use kameo::{Actor, messages};
-use memsafe::MemSafe;
-use miette::Diagnostic;
-use rand::{RngExt, distr::StandardUniform, make_rng, rngs::StdRng};
-use secrecy::SecretString;
-use thiserror::Error;
-use tracing::info;
-use zeroize::{Zeroize, Zeroizing};
-
-use crate::{
-    context::{self, ServerContext},
-    db::{self, DatabasePool, schema},
-};
-
-const TOKEN_LENGTH: usize = 64;
-
-pub async fn generate_token() -> Result<String, std::io::Error> {
-    let rng: StdRng = make_rng();
-
-    let token: String = rng
-        .sample_iter::<char, _>(StandardUniform)
-        .take(TOKEN_LENGTH)
-        .fold(Default::default(), |mut accum, char| {
-            accum += char.to_string().as_str();
-            accum
-        });
-
-    tokio::fs::write(home_path()?.join(BOOTSTRAP_TOKEN_PATH), token.as_str()).await?;
-
-    Ok(token)
-}
-
-#[derive(Error, Debug, Diagnostic)]
-pub enum BootstrapError {
-    #[error("Database error: {0}")]
-    #[diagnostic(code(arbiter_server::bootstrap::database))]
-    Database(#[from] db::PoolError),
-
-    #[error("Database query error: {0}")]
-    #[diagnostic(code(arbiter_server::bootstrap::database_query))]
-    Query(#[from] diesel::result::Error),
-
-    #[error("I/O error: {0}")]
-    #[diagnostic(code(arbiter_server::bootstrap::io))]
-    Io(#[from] std::io::Error),
-}
-
-#[derive(Actor)]
-pub struct BootstrapActor {
-    token: Option<String>,
-}
-
-impl BootstrapActor {
-    pub async fn new(db: &DatabasePool) -> Result<Self, BootstrapError> {
-        let mut conn = db.get().await?;
-
-        let row_count: i64 = schema::useragent_client::table
-            .count()
-            .get_result(&mut conn)
-            .await?;
-
-        drop(conn);
-
-        let token = if row_count == 0 {
-            let token = generate_token().await?;
-            info!(%token, "Generated bootstrap token");
-            tokio::fs::write(home_path()?.join(BOOTSTRAP_TOKEN_PATH), token.as_str()).await?;
-            Some(token)
-        } else {
-            None
-        };
-
-        Ok(Self { token })
-    }
-
-    #[cfg(test)]
-    pub fn get_token(&self) -> Option<String> {
-        self.token.clone()
-    }
-}
-
-#[messages]
-impl BootstrapActor {
-    #[message]
-    pub fn is_correct_token(&self, token: String) -> bool {
-        match &self.token {
-            Some(expected) => *expected == token,
-            None => false,
-        }
-    }
-
-    #[message]
-    pub fn consume_token(&mut self, token: String) -> bool {
-        if self.is_correct_token(token) {
-            self.token = None;
-            true
-        } else {
-            false
-        }
-    }
-}

server/crates/arbiter-server/src/context/lease.rs

@@ -1,41 +0,0 @@
-use std::sync::Arc;
-
-use dashmap::DashSet;
-
-#[derive(Clone, Default)]
-struct LeaseStorage<T: Eq + std::hash::Hash>(Arc<DashSet<T>>);
-
-// A lease that automatically releases the item when dropped
-pub struct Lease<T: Clone + std::hash::Hash + Eq> {
-    item: T,
-    storage: LeaseStorage<T>,
-}
-impl<T: Clone + std::hash::Hash + Eq> Drop for Lease<T> {
-    fn drop(&mut self) {
-        self.storage.0.remove(&self.item);
-    }
-}
-
-#[derive(Clone, Default)]
-pub struct LeaseHandler<T: Clone + std::hash::Hash + Eq> {
-    storage: LeaseStorage<T>,
-}
-
-impl<T: Clone + std::hash::Hash + Eq> LeaseHandler<T> {
-    pub fn new() -> Self {
-        Self {
-            storage: LeaseStorage(Arc::new(DashSet::new())),
-        }
-    }
-
-    pub fn acquire(&self, item: T) -> Result<Lease<T>, ()> {
-        if self.storage.0.insert(item.clone()) {
-            Ok(Lease {
-                item,
-                storage: self.storage.clone(),
-            })
-        } else {
-            Err(())
-        }
-    }
-}