hitomi 0.8.1

A CLI application that builds and updates playlists on a Plex server using json-based profiles.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

//! Loading and saving data to sqlite database

use std::env;
use std::str::FromStr;

use anyhow::{anyhow, Result};
use simplelog::warn;
use sqlx::sqlite::{SqliteConnectOptions, SqliteJournalMode};
use sqlx::SqlitePool;
use tokio::sync::OnceCell;

pub mod config;
pub mod profiles;

static POOL: OnceCell<SqlitePool> = OnceCell::const_new();

fn get_pool() -> Result<&'static SqlitePool> {
    match POOL.get() {
        None => Err(anyhow!("Could not acquire Sqlite Pool")),
        Some(pool) => Ok(pool),
    }
}

pub async fn initialize_pool(database_url: Option<&str>) -> Result<()> {
    let database_url = if let Ok(database_url) = env::var("DATABASE_URL") {
        database_url
    } else if let Some(database_url) = database_url {
        database_url.to_string()
    } else {
        warn!("Environment variable `DATABASE_URL` not set and --db flag not provided. Using default URL.");
        String::from("sqlite:./data/hitomi.db")
    };

    let database_url = if database_url.contains("sqlite:") {
        database_url
    } else {
        format!("sqlite:{database_url}")
    };

    let options =
        SqliteConnectOptions::from_str(&database_url)?.journal_mode(SqliteJournalMode::Wal);

    let pool = SqlitePool::connect_with(options).await?;

    POOL.get_or_init(|| async { pool }).await;

    Ok(())
}