use anyhow::Result;
use clap::Args;
use gen_models::errors::OperationError;
use crate::{
commands::{cli_context::CliContext, get_default_collection},
imports::library::{LibraryImportError, import_library},
};
#[derive(Debug, Args)]
pub struct Command {
#[clap(index = 1)]
library_name: String,
#[clap(index = 2)]
parts: Option<String>,
#[clap(index = 3)]
library: Option<String>,
#[arg(short, long)]
name: Option<String>,
#[arg(short, long)]
sample: Option<String>,
}
pub fn execute(cli_context: &CliContext, cmd: Command) -> Result<()> {
println!("Library import called");
let context = cli_context.context;
let operation_conn = context.operations().conn();
let conn = context.graph().conn();
conn.execute("BEGIN TRANSACTION", []).unwrap();
operation_conn.execute("BEGIN TRANSACTION", []).unwrap();
let name = &cmd
.name
.clone()
.unwrap_or_else(|| get_default_collection(operation_conn));
match import_library(
context,
name,
cmd.sample.as_deref(),
cmd.parts.as_deref().unwrap(),
cmd.library.as_deref().unwrap(),
&cmd.library_name,
) {
Ok(_) => {
println!("Library imported.");
conn.execute("END TRANSACTION;", []).unwrap();
operation_conn.execute("END TRANSACTION;", []).unwrap();
Ok(())
}
Err(LibraryImportError::OperationError(OperationError::NoChanges)) => {
conn.execute("ROLLBACK TRANSACTION;", []).unwrap();
operation_conn.execute("ROLLBACK TRANSACTION;", []).unwrap();
println!("Library already exists.");
Ok(())
}
Err(e) => {
println!("Library import failed: {}", e);
Err(e.into())
}
}
}