fix(server::user_agent): useragents now self-sign themselves on bootstrap

server/crates/arbiter-server/src/grpc/client.rs

@@ -10,8 +10,8 @@ use tonic::Status;
 use tracing::{info, warn};
 
 use crate::{
-    peers::client::{ClientConnection, session::ClientSession},
     grpc::request_tracker::RequestTracker,
+    peers::client::{ClientConnection, session::ClientSession},
 };
 
 mod auth;

server/crates/arbiter-server/src/grpc/client/auth.rs

@@ -22,8 +22,8 @@ use tonic::Status;
 use tracing::warn;
 
 use crate::{
-    peers::client::{self, ClientConnection, auth},
     grpc::request_tracker::RequestTracker,
+    peers::client::{self, ClientConnection, auth},
 };
 
 pub struct AuthTransportAdapter<'a> {

server/crates/arbiter-server/src/grpc/client/evm.rs

@@ -16,11 +16,11 @@ use tonic::Status;
 use tracing::warn;
 
 use crate::{
-    peers::client::session::{ClientSession, HandleSignTransaction, SignTransactionRpcError},
     grpc::{
         Convert, TryConvert,
         common::inbound::{RawEvmAddress, RawEvmTransaction},
     },
+    peers::client::session::{ClientSession, HandleSignTransaction, SignTransactionRpcError},
 };
 
 fn wrap_response(payload: EvmResponsePayload) -> ClientResponsePayload {

server/crates/arbiter-server/src/grpc/client/vault.rs

@@ -13,8 +13,8 @@ use tonic::Status;
 use tracing::warn;
 
 use crate::{
-    peers::client::session::{ClientSession, Error, HandleQueryVaultState},
     actors::vault::VaultState,
+    peers::client::session::{ClientSession, Error, HandleQueryVaultState},
 };
 
 pub(super) async fn dispatch(

server/crates/arbiter-server/src/grpc/mod.rs

@@ -10,8 +10,8 @@ use tonic::{Request, Response, Status, async_trait};
 use tracing::info;
 
 use crate::{
-    peers::{client::ClientConnection, user_agent::UserAgentConnection},
     grpc::user_agent::start,
+    peers::{client::ClientConnection, user_agent::UserAgentConnection},
 };
 
 mod request_tracker;

server/crates/arbiter-server/src/grpc/user_agent.rs

@@ -14,8 +14,8 @@ use tonic::Status;
 use tracing::{error, info, warn};
 
 use crate::{
-    peers::user_agent::{OutOfBand, UserAgentConnection, UserAgentSession},
     grpc::request_tracker::RequestTracker,
+    peers::user_agent::{OutOfBand, UserAgentConnection, UserAgentSession},
 };
 
 mod auth;
@@ -124,7 +124,7 @@ pub async fn start(
 ) {
     let mut request_tracker = RequestTracker::default();
 
-    let pubkey = match auth::start(&mut conn, &mut bi, &mut request_tracker).await {
+    let (id, pubkey) = match auth::start(&mut conn, &mut bi, &mut request_tracker).await {
         Ok(pubkey) => pubkey,
         Err(e) => {
             warn!(error = ?e, "Authentication failed");
@@ -132,13 +132,19 @@ pub async fn start(
         }
     };
 
+    info!(?pubkey, "User authenticated successfully");
+
     let (oob_sender, oob_receiver) = mpsc::channel(16);
     let oob_adapter = OutOfBandAdapter(oob_sender);
 
-    let actor = UserAgentSession::spawn(UserAgentSession::new(conn, Box::new(oob_adapter)));
+    let actor = UserAgentSession::spawn(UserAgentSession::new(
+        conn,
+        id,
+        pubkey,
+        Box::new(oob_adapter),
+    ));
     let actor_for_cleanup = actor.clone();
 
-    info!(?pubkey, "User authenticated successfully");
     dispatch_loop(bi, actor, oob_receiver, request_tracker).await;
     actor_for_cleanup.kill();
 }

server/crates/arbiter-server/src/grpc/user_agent/auth.rs

@@ -18,8 +18,8 @@ use tonic::Status;
 use tracing::warn;
 
 use crate::{
-    peers::user_agent::{UserAgentConnection, auth},
     grpc::request_tracker::RequestTracker,
+    peers::user_agent::{UserAgentConnection, auth},
 };
 
 pub struct AuthTransportAdapter<'a> {
@@ -140,7 +140,6 @@ impl Receiver<auth::Inbound> for AuthTransportAdapter<'_> {
             AuthRequestPayload::ChallengeRequest(ProtoAuthChallengeRequest {
                 pubkey,
                 bootstrap_token,
-                key_type: _,
             }) => {
                 let Ok(pubkey) = authn::PublicKey::try_from(pubkey.as_slice()) else {
                     warn!(
@@ -168,7 +167,7 @@ pub async fn start(
     conn: &mut UserAgentConnection,
     bi: &mut GrpcBi<UserAgentRequest, UserAgentResponse>,
     request_tracker: &mut RequestTracker,
-) -> Result<authn::PublicKey, auth::Error> {
+) -> Result<(i32, authn::PublicKey), auth::Error> {
     let transport = AuthTransportAdapter::new(bi, request_tracker);
     auth::authenticate(conn, transport).await
 }

server/crates/arbiter-server/src/grpc/user_agent/evm.rs

@@ -23,6 +23,10 @@ use tonic::Status;
 use tracing::warn;
 
 use crate::{
+    grpc::{
+        Convert, TryConvert,
+        common::inbound::{RawEvmAddress, RawEvmTransaction},
+    },
     peers::user_agent::{
         UserAgentSession,
         session::connection::{
@@ -31,10 +35,6 @@ use crate::{
             SignTransactionError as SessionSignTransactionError,
         },
     },
-    grpc::{
-        Convert, TryConvert,
-        common::inbound::{RawEvmAddress, RawEvmTransaction},
-    },
 };
 
 fn wrap_evm_response(payload: EvmResponsePayload) -> UserAgentResponsePayload {

server/crates/arbiter-server/src/grpc/user_agent/sdk_client.rs

@@ -21,6 +21,8 @@ use tonic::Status;
 use tracing::{info, warn};
 
 use crate::{
+    db::models::NewEvmWalletAccess,
+    grpc::Convert,
     peers::user_agent::{
         OutOfBand, UserAgentSession,
         session::connection::{
@@ -28,8 +30,6 @@ use crate::{
             HandleRevokeEvmWalletAccess, HandleSdkClientList,
         },
     },
-    db::models::NewEvmWalletAccess,
-    grpc::Convert,
 };
 
 fn wrap_sdk_client_response(payload: SdkClientResponsePayload) -> UserAgentResponsePayload {