Yoda/src/profile/identity.rs

mod database;
mod error;
mod gemini;

use database::Database;
pub use error::Error;
use gemini::Gemini;

use sqlite::{Connection, Transaction};
use std::{rc::Rc, sync::RwLock};

/// Authorization wrapper for different protocols
pub struct Identity {
    // database: Rc<Database>,
    pub gemini: Rc<Gemini>,
}

impl Identity {
    // Constructors

    /// Create new `Self`
    pub fn new(connection: Rc<RwLock<Connection>>, profile_id: Rc<i64>) -> Result<Self, Error> {
        // Init identity database
        let database = Rc::new(Database::new(connection.clone()));

        // Get active identity set for profile or create new one
        let profile_identity_id = Rc::new(match database.active() {
            Some(identity) => identity.id,
            None => match database.add(profile_id, true) {
                Ok(id) => id,
                Err(_) => return Err(Error::Database),
            },
        });

        // Init gemini component
        let gemini = Rc::new(match Gemini::new(connection, profile_identity_id) {
            Ok(result) => result,
            Err(reason) => return Err(Error::GeminiInit(reason)),
        });

        // Done
        Ok(Self {
            // database,
            gemini,
        })
    }

    /// Get `pem` record match `request`
    /// * [Client certificates specification](https://geminiprotocol.net/docs/protocol-specification.gmi#client-certificates)
    /// * this function work with memory cache collected (not database)
    pub fn gemini(&self, request: &str) -> Option<String> {
        if let Some(id) = self.gemini.auth.memory.match_priority(request) {
            match self.gemini.memory.get(id) {
                Ok(pem) => return Some(pem),
                Err(reason) => todo!("{:?}", reason),
            }
        }
        None
    }
}

// Tools

pub fn migrate(tx: &Transaction) -> Result<(), String> {
    // Migrate self components
    if let Err(e) = database::init(tx) {
        return Err(e.to_string());
    }

    // Delegate migration to childs
    gemini::migrate(tx)?;

    // Success
    Ok(())
}