58 lines
1.8 KiB
Rust
58 lines
1.8 KiB
Rust
// src/services/auth.rs
|
|
use tonic::transport::Channel;
|
|
use common::proto::komp_ac::auth::{
|
|
auth_service_client::AuthServiceClient,
|
|
LoginRequest, LoginResponse,
|
|
RegisterRequest, AuthResponse,
|
|
};
|
|
use anyhow::{Context, Result};
|
|
|
|
#[derive(Clone)]
|
|
pub struct AuthClient {
|
|
client: AuthServiceClient<Channel>,
|
|
}
|
|
|
|
impl AuthClient {
|
|
pub async fn new() -> Result<Self> {
|
|
// Kept for backward compatibility; opens a new connection.
|
|
let client = AuthServiceClient::connect("http://[::1]:50051")
|
|
.await
|
|
.context("Failed to connect to auth service")?;
|
|
Ok(Self { client })
|
|
}
|
|
|
|
/// Preferred: reuse an existing Channel (from GrpcClient).
|
|
pub async fn with_channel(channel: Channel) -> Result<Self> {
|
|
Ok(Self {
|
|
client: AuthServiceClient::new(channel),
|
|
})
|
|
}
|
|
|
|
/// Login user via gRPC.
|
|
pub async fn login(&mut self, identifier: String, password: String) -> Result<LoginResponse> {
|
|
let request = tonic::Request::new(LoginRequest { identifier, password });
|
|
let response = self.client.login(request).await?.into_inner();
|
|
Ok(response)
|
|
}
|
|
|
|
/// Registers a new user via gRPC.
|
|
pub async fn register(
|
|
&mut self,
|
|
username: String,
|
|
email: String,
|
|
password: Option<String>,
|
|
password_confirmation: Option<String>,
|
|
role: Option<String>,
|
|
) -> Result<AuthResponse> {
|
|
let request = tonic::Request::new(RegisterRequest {
|
|
username,
|
|
email,
|
|
password: password.unwrap_or_default(),
|
|
password_confirmation: password_confirmation.unwrap_or_default(),
|
|
role: role.unwrap_or_default(),
|
|
});
|
|
let response = self.client.register(request).await?.into_inner();
|
|
Ok(response)
|
|
}
|
|
}
|