chug-cli 0.1.1

The fastest way to consume Homebrew bottles
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

use std::sync::{Mutex, MutexGuard};

use anyhow::{Context, anyhow};
use diesel::{prelude::*, sql_query};
use diesel_migrations::{EmbeddedMigrations, MigrationHarness, embed_migrations};

use crate::dirs::db_file;

pub mod models;
pub mod schema;

const MIGRATIONS: EmbeddedMigrations = embed_migrations!();

fn connection() -> anyhow::Result<MutexGuard<'static, SqliteConnection>> {
    let mutex = cache!(Mutex<SqliteConnection>).get_or_init(|| {
        let path = db_file()?.to_str().context("DB file path is non-utf8")?;
        let mut db = SqliteConnection::establish(path)?;

        sql_query("PRAGMA foreign_keys = ON;").execute(&mut db)?;

        db.run_pending_migrations(MIGRATIONS)
            .map_err(|e| anyhow!(e))?;

        Ok(Mutex::new(db))
    })?;

    Ok(mutex.lock().unwrap())
}