chug_cli/db/

models.rs

use std::path::Path;

use anyhow::Context;
use diesel::{prelude::*, sqlite::Sqlite};

use crate::db::{
    connection,
    schema::{dependencies, downloaded_bottles, linked_files},
};

#[derive(Clone, Debug, Queryable, Selectable)]
#[diesel(table_name = downloaded_bottles)]
#[diesel(check_for_backend(Sqlite))]
pub struct DownloadedBottle {
    id: i32,
    name: String,
    version: String,
    path: String,
}

#[derive(Debug, Insertable)]
#[diesel(table_name = downloaded_bottles)]
#[diesel(check_for_backend(Sqlite))]
struct NewDownloadedBottle<'a> {
    name: &'a str,
    version: &'a str,
    path: &'a str,
}

#[derive(Debug, Queryable, Selectable)]
#[diesel(table_name = linked_files)]
#[diesel(check_for_backend(Sqlite))]
pub struct LinkedFile {
    id: i32,
    path: String,
}

#[derive(Debug, Insertable)]
#[diesel(table_name = linked_files)]
#[diesel(check_for_backend(Sqlite))]
struct NewLinkedFile<'a> {
    path: &'a str,
    bottle_id: i32,
}

#[derive(Debug, Queryable, Selectable, Insertable)]
#[diesel(table_name = dependencies)]
#[diesel(check_for_backend(Sqlite))]
pub struct Dependency {
    dependent_id: Option<i32>,
    dependency_id: i32,
}

impl DownloadedBottle {
    pub fn create(name: &str, version: &str, path: &Path) -> anyhow::Result<DownloadedBottle> {
        let mut db = connection()?;

        let result = diesel::insert_into(downloaded_bottles::table)
            .values(NewDownloadedBottle {
                name,
                version,
                path: path.to_str().context("Installed bottle path is non-utf8")?,
            })
            .returning(DownloadedBottle::as_returning())
            .get_result(&mut *db)?;

        Ok(result)
    }

    pub fn get(name: &str, version: &str) -> anyhow::Result<Option<DownloadedBottle>> {
        use downloaded_bottles::dsl;

        let mut db = connection()?;
        let name_value = name;
        let version_value = version;

        let mut results = dsl::downloaded_bottles
            .filter(dsl::name.eq(name_value))
            .filter(dsl::version.eq(version_value))
            .select(DownloadedBottle::as_select())
            .load_iter(&mut *db)?;

        if let Some(bottle) = results.next() {
            Ok(Some(bottle?))
        } else {
            Ok(None)
        }
    }

    fn get_by_id(id: i32) -> anyhow::Result<DownloadedBottle> {
        use downloaded_bottles::dsl;

        let mut db = connection()?;

        let result = dsl::downloaded_bottles
            .filter(dsl::id.eq(id))
            .select(Self::as_select())
            .first(&mut *db)?;

        Ok(result)
    }

    pub fn get_all() -> anyhow::Result<Vec<DownloadedBottle>> {
        use downloaded_bottles::dsl;

        let mut db = connection()?;

        let results = dsl::downloaded_bottles
            .order((dsl::name, dsl::version))
            .select(DownloadedBottle::as_select())
            .load(&mut *db)?;

        Ok(results)
    }

    pub fn delete(&self) -> anyhow::Result<()> {
        use downloaded_bottles::dsl;

        let mut db = connection()?;

        diesel::delete(downloaded_bottles::table)
            .filter(dsl::id.eq(self.id))
            .execute(&mut *db)?;

        Ok(())
    }

    pub fn id(&self) -> i32 {
        self.id
    }

    pub fn name(&self) -> &str {
        &self.name
    }

    pub fn version(&self) -> &str {
        &self.version
    }

    pub fn path(&self) -> &Path {
        self.path.as_ref()
    }

    pub fn linked_files(&self) -> anyhow::Result<Vec<LinkedFile>> {
        use linked_files::dsl;

        let mut db = connection()?;

        let results = dsl::linked_files
            .filter(dsl::bottle_id.eq(self.id))
            .select(LinkedFile::as_select())
            .load(&mut *db)?;

        Ok(results)
    }
}

impl LinkedFile {
    pub fn create(path: &Path, bottle: &DownloadedBottle) -> anyhow::Result<()> {
        let mut db = connection()?;

        diesel::insert_into(linked_files::table)
            .values(NewLinkedFile {
                path: path.to_str().context("Linked file path is non-utf8")?,
                bottle_id: bottle.id,
            })
            .on_conflict(linked_files::dsl::path)
            .do_nothing()
            .execute(&mut *db)?;

        Ok(())
    }

    pub fn delete(&self) -> anyhow::Result<()> {
        use linked_files::dsl;

        let mut db = connection()?;

        diesel::delete(linked_files::table)
            .filter(dsl::id.eq(self.id))
            .execute(&mut *db)?;

        Ok(())
    }

    pub fn path(&self) -> &Path {
        self.path.as_ref()
    }
}

impl Dependency {
    pub fn get_all() -> anyhow::Result<Vec<Dependency>> {
        use dependencies::dsl;

        let mut db = connection()?;

        let result = dsl::dependencies
            .select(Dependency::as_select())
            .load(&mut *db)?;

        Ok(result)
    }

    pub fn replace_all<'a>(
        dependencies: impl Iterator<Item = (Option<&'a DownloadedBottle>, &'a DownloadedBottle)>,
    ) -> anyhow::Result<()> {
        use dependencies::dsl;

        let mut db = connection()?;

        db.transaction::<(), anyhow::Error, _>(|db| {
            diesel::delete(dsl::dependencies).execute(db)?;

            let new_dependencies = dependencies
                .map(|(dependent, dependency)| Dependency {
                    dependent_id: dependent.map(|b| b.id),
                    dependency_id: dependency.id,
                })
                .collect::<Vec<_>>();
            diesel::insert_into(dsl::dependencies)
                .values(&new_dependencies)
                .execute(db)?;

            Ok(())
        })?;

        Ok(())
    }

    /// Returns `None` if the bottle was manually added, i.e. if it is a root.
    pub fn dependent(&self) -> anyhow::Result<Option<DownloadedBottle>> {
        if let Some(id) = self.dependent_id {
            Ok(Some(DownloadedBottle::get_by_id(id)?))
        } else {
            Ok(None)
        }
    }

    pub fn dependent_id(&self) -> Option<i32> {
        self.dependent_id
    }

    pub fn dependency(&self) -> anyhow::Result<DownloadedBottle> {
        DownloadedBottle::get_by_id(self.dependency_id)
    }

    pub fn dependency_id(&self) -> i32 {
        self.dependency_id
    }
}