plm_rs/

lib.rs

#[macro_use]
extern crate diesel;
#[macro_use]
extern crate diesel_migrations;

#[macro_use]
extern crate prettytable;

pub mod config;
pub mod models;
pub mod prompt;
pub mod schema;
pub mod schematic;
pub mod tables;

use diesel::prelude::*;

use models::*;

use std::io::{StdinLock, Stdout};

pub struct Application<'a> {
    /// Config
    pub config: config::Config,

    /// Path to config
    pub config_path: std::path::PathBuf,

    /// Global prompt
    pub prompt: prompt::Prompt<StdinLock<'a>, Stdout>,

    /// Global conn
    pub conn: SqliteConnection,
}

// Migrate
embed_migrations!();

pub fn establish_connection(db_name: &str) -> SqliteConnection {
    // Get text version of configpath
    let mut database_url =
        config::get_default_config_path().unwrap_or_else(|_| panic!("Unable to get config path."));

    // Add database name
    database_url.push(db_name);

    // Establish the "connection" (We're using SQLite here so no connection excpet to the filesystem)
    let conn = SqliteConnection::establish(&database_url.to_string_lossy())
        .unwrap_or_else(|_| panic!("Error connecting to {}", &database_url.to_string_lossy()));

    // This will run migrations and also create a DB if it doesn't exist.
    embedded_migrations::run(&conn).expect("Unable to run migration.");

    conn
}

// Part related
pub fn create_part(
    conn: &SqliteConnection,
    part: &NewUpdatePart,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::parts;

    diesel::insert_into(parts::table).values(part).execute(conn)
}

pub fn update_part(
    conn: &SqliteConnection,
    id: &i32,
    part: &NewUpdatePart,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::parts;

    diesel::update(parts::dsl::parts.filter(parts::dsl::id.eq(id)))
        .set(part)
        .execute(conn)
}

pub fn rename_part(
    conn: &SqliteConnection,
    oldpn: &str,
    newpn: &str,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::parts::dsl::*;

    let part = find_part_by_pn(&conn, &oldpn).expect("Old part not found");

    diesel::update(parts.find(part.id))
        .set(pn.eq(newpn))
        .execute(conn)
}

pub fn delete_part(
    conn: &SqliteConnection,
    id: &i32,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::parts;

    diesel::delete(parts::dsl::parts.filter(parts::dsl::id.eq(id))).execute(conn)
}

pub fn find_part_by_pn(
    conn: &SqliteConnection,
    pn: &str,
) -> std::result::Result<Part, diesel::result::Error> {
    use schema::parts;

    parts::dsl::parts.filter(parts::dsl::pn.eq(pn)).first(conn)
}

pub fn find_part_by_mpn(
    conn: &SqliteConnection,
    mpn: &str,
) -> std::result::Result<Part, diesel::result::Error> {
    use schema::parts;

    println!("mpn: \"{}\"", mpn);

    parts::dsl::parts
        .filter(parts::dsl::mpn.eq(mpn))
        .first(conn)
}

pub fn find_part_by_pn_and_ver(
    conn: &SqliteConnection,
    pn: &str,
    ver: &i32,
) -> std::result::Result<Part, diesel::result::Error> {
    use schema::parts;

    parts::dsl::parts
        .filter(parts::dsl::pn.eq(pn))
        .filter(parts::dsl::ver.eq(ver))
        .first(conn)
}

pub fn find_part_by_id(
    conn: &SqliteConnection,
    id: &i32,
) -> std::result::Result<Part, diesel::result::Error> {
    use schema::parts;

    parts::dsl::parts.filter(parts::dsl::id.eq(id)).first(conn)
}

pub fn create_bom_line_item(
    conn: &SqliteConnection,
    part: &NewPartsParts,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::parts_parts;

    diesel::insert_into(parts_parts::table)
        .values(part)
        .execute(conn)
}

pub fn delete_bom_list_by_id_and_ver(
    conn: &SqliteConnection,
    bom_id: &i32,
    ver: &i32,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::parts_parts::dsl::*;

    // First get list of ids that match the bom_part_id
    let query = parts_parts
        .select(id)
        .filter(bom_part_id.eq(bom_id))
        .load::<i32>(conn)?;

    // Then make sure that the bom ver is equal. Match against the ids found in the first step
    let target = parts_parts.filter(bom_ver.eq(ver)).filter(id.eq_any(query));

    // Delete appropriately
    diesel::delete(target).execute(conn)
}

// Build related

pub fn create_build(
    conn: &SqliteConnection,
    build: &NewUpdateBuild,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::builds;

    diesel::insert_into(builds::table)
        .values(build)
        .execute(conn)
}

pub fn update_build_by_id(
    conn: &SqliteConnection,
    id: &i32,
    entry: &NewUpdateBuild,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::builds;

    diesel::update(builds::dsl::builds.filter(builds::dsl::id.eq(id)))
        .set(entry)
        .execute(conn)
}

pub fn find_builds_by_pn(
    conn: &SqliteConnection,
    pn: &str,
) -> std::result::Result<Vec<Build>, diesel::result::Error> {
    use schema::builds;

    let part = find_part_by_pn(&conn, &pn).expect("Unable to run parts query.");

    builds::dsl::builds
        .filter(builds::dsl::part_id.eq(part.id))
        .load::<Build>(conn)
}

pub fn find_build_by_id(
    conn: &SqliteConnection,
    id: &i32,
) -> std::result::Result<Build, diesel::result::Error> {
    use schema::builds;

    builds::dsl::builds
        .filter(builds::dsl::id.eq(id))
        .first(conn)
}

pub fn delete_build(
    conn: &SqliteConnection,
    id: &i32,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::builds;

    diesel::delete(builds::dsl::builds.filter(builds::dsl::id.eq(id))).execute(conn)
}

// Inventory related

pub fn create_inventory(
    conn: &SqliteConnection,
    entry: &NewUpdateInventoryEntry,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::inventories;

    diesel::insert_into(inventories::table)
        .values(entry)
        .execute(conn)
}

pub fn update_inventory_by_id(
    conn: &SqliteConnection,
    id: &i32,
    entry: &NewUpdateInventoryEntry,
) -> std::result::Result<usize, diesel::result::Error> {
    use schema::inventories;

    diesel::update(inventories::dsl::inventories.filter(inventories::dsl::id.eq(id)))
        .set(entry)
        .execute(conn)
}

pub fn find_inventories_by_part_id(
    conn: &SqliteConnection,
    id: &i32,
) -> std::result::Result<Vec<Inventory>, diesel::result::Error> {
    use schema::inventories;

    inventories::dsl::inventories
        .filter(inventories::dsl::part_id.eq(id))
        .load::<Inventory>(conn)
}

pub fn test_connection() -> SqliteConnection {
    // Start a connection from memory
    let conn = SqliteConnection::establish(":memory:").expect("Unable to establish db in memory!");

    // This will run the necessary migrations.
    embedded_migrations::run(&conn).expect("Unable to run test migration.");

    // Return the active connection
    conn
}

/* START: Part Related Tests */
mod part_tests {

    #[test]
    fn create_part_check_if_created() {
        use super::*;
        use models::Part;
        use schema::parts::dsl::*;

        let conn = test_connection();

        // Create NewUpdatePart instance
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402",
            mpn: "ABCD",
            descr: "CAP 0.1U 10V 0402",
            ver: &1,
            mqty: &1,
        };

        // Create the part
        create_part(&conn, &part).expect("Error creating part!");

        // Serach for it and make sure that it matches
        let found: Part = parts.find(1).first(&conn).unwrap();

        // Make sure these guys are equal
        assert_eq!(part.pn, found.pn);
        assert_eq!(part.mpn, found.mpn);
        assert_eq!(part.descr, found.descr);
        assert_eq!(*part.ver, found.ver);
    }

    #[test]
    #[should_panic]
    // This is testing the schema more than anything
    // Only one part with the same PN!
    fn create_duplicate_pn_should_panic() {
        use super::*;
        let conn = test_connection();

        // Create NewUpdatePart instance
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402",
            mpn: "ABCD",
            descr: "CAP 0.1U 10V 0402",
            ver: &1,
            mqty: &1,
        };

        // Create the part
        create_part(&conn, &part).expect("Error creating part!");

        // Create NewUpdatePart instance
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402",
            mpn: "ABCD-ND",
            descr: "CAP 0.1U 10V 0402",
            ver: &1,
            mqty: &1,
        };

        // Do it again
        create_part(&conn, &part).expect("Error creating part!");
    }

    #[test]
    #[should_panic]
    // This is testing the schema more than anything
    // Only one part with the same MPN!
    fn create_duplicate_mpn_should_panic() {
        use super::*;
        let conn = test_connection();

        // Create NewUpdatePart instance
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402",
            mpn: "ABCD",
            descr: "CAP 0.1U 10V 0402",
            ver: &1,
            mqty: &1,
        };

        // Create the part
        create_part(&conn, &part).expect("Error creating part!");

        // Create NewUpdatePart instance
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402-01",
            mpn: "ABCD",
            descr: "CAP 0.1U 10V 0402",
            ver: &1,
            mqty: &1,
        };

        // Do it again
        create_part(&conn, &part).expect("Error creating part!");
    }

    #[test]
    fn create_and_update_part() {
        use super::*;
        use models::Part;
        use schema::parts::dsl::*;

        let conn = test_connection();

        // Create NewUpdatePart instance
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402",
            mpn: "ABCD",
            descr: "CAP 0.1U 10V 0402",
            ver: &1,
            mqty: &1,
        };

        // Create the part
        create_part(&conn, &part).expect("Error creating part!");

        // Get part back
        let found = find_part_by_pn(&conn, &part.pn).expect("Error getting part back.");

        // Update the value
        let part = NewUpdatePart {
            pn: "CAP-0.1U-10V-0402",
            mpn: "ABCD",
            descr: "CAP 0.1 10V 0402 GOOD", // Only changing this guy
            ver: &1,
            mqty: &1,
        };

        // Update the part
        update_part(&conn, &found.id, &part).expect("Error creating part!");

        // Serach for it and make sure that it matches
        let found: Part = parts.find(1).first(&conn).unwrap();

        // Make sure these guys are equal
        assert_eq!(part.descr, found.descr);
    }
}

/* START: Inventory Related Tests */
mod inventory_tests {}

/* START: Build Related Tests */
mod build_tests {}