use cloud_database::SqliteDBContext;
use jwst_storage::{BlobAutoStorage, DocAutoStorage};
use std::{fs, path::Path};
use tauri::api::path::document_dir;
use tokio::sync::Mutex;

pub struct AppStateRaw {
  pub doc_db: DocAutoStorage,
  pub blob_storage: BlobAutoStorage,
  pub metadata_db: SqliteDBContext,
}

impl AppStateRaw {
  pub async fn new() -> Option<AppStateRaw> {
    let affine_document_path = Path::new(&document_dir()?.into_os_string()).join("affine");
    let metadata_db_env = format!(
      "sqlite://{}?mode=rwc",
      affine_document_path
        .join("metadata")
        .with_extension("db")
        .display()
    );
    let blob_db_env = format!(
      "sqlite://{}?mode=rwc",
      affine_document_path
        .join("blob")
        .with_extension("db")
        .display()
    );
    let doc_db_env = format!(
      "sqlite://{}?mode=rwc",
      affine_document_path
        .join("doc")
        .with_extension("db")
        .display()
    );
    fs::create_dir_all(affine_document_path.clone()).unwrap();

    Some(Self {
      doc_db: DocAutoStorage::init_pool(&doc_db_env).await.unwrap(),
      blob_storage: BlobAutoStorage::init_pool(&blob_db_env).await.unwrap(),
      metadata_db: SqliteDBContext::new(metadata_db_env).await,
    })
  }
}

pub struct AppState(pub Mutex<AppStateRaw>); // need pub, otherwise will be "field `0` of struct `types::state::AppState` is private"