Kordophone/kordophone/src/api/mod.rs

use async_trait::async_trait;
pub use crate::model::{
    Conversation, Message, ConversationID, MessageID,
};
use crate::model::JwtToken;

pub mod http_client;
pub use http_client::HTTPAPIClient;

use self::http_client::Credentials;

#[async_trait]
pub trait APIInterface {
    type Error;

    // (GET) /version
    async fn get_version(&mut self) -> Result<String, Self::Error>;
    
    // (GET) /conversations
    async fn get_conversations(&mut self) -> Result<Vec<Conversation>, Self::Error>;

    // (GET) /messages
    async fn get_messages(
        &mut self, 
        conversation_id: &ConversationID,
        limit: Option<u32>,
        before: Option<MessageID>,
        after: Option<MessageID>,
    ) -> Result<Vec<Message>, Self::Error>;

    // (POST) /authenticate
    async fn authenticate(&mut self, credentials: Credentials) -> Result<JwtToken, Self::Error>;
}

#[async_trait]
pub trait AuthenticationStore {
    async fn get_credentials(&mut self) -> Option<Credentials>;
    async fn get_token(&mut self) -> Option<JwtToken>;
    async fn set_token(&mut self, token: JwtToken);
}

pub struct InMemoryAuthenticationStore {
    credentials: Option<Credentials>,
    token: Option<JwtToken>,
}

impl Default for InMemoryAuthenticationStore {
    fn default() -> Self {
        Self::new(None)
    }
}

impl InMemoryAuthenticationStore {
    pub fn new(credentials: Option<Credentials>) -> Self {
        Self { 
            credentials,
            token: None,
        }
    }
}

#[async_trait]
impl AuthenticationStore for InMemoryAuthenticationStore {
    async fn get_credentials(&mut self) -> Option<Credentials> {
        self.credentials.clone()
    }

    async fn get_token(&mut self) -> Option<JwtToken> {
        self.token.clone()
    }

    async fn set_token(&mut self, token: JwtToken) {
        self.token = Some(token);
    }
}