77 lines
2.4 KiB
Rust
77 lines
2.4 KiB
Rust
use anyhow::Result;
|
|
use std::path::PathBuf;
|
|
use std::str::FromStr;
|
|
|
|
use crate::db;
|
|
use crate::compression_engine::CompressionType;
|
|
use clap::error::ErrorKind;
|
|
use clap::Command;
|
|
use log::{debug, info};
|
|
use rusqlite::Connection;
|
|
|
|
pub fn mode_update(
|
|
cmd: &mut Command,
|
|
args: crate::Args,
|
|
ids: &mut Vec<i64>,
|
|
tags: &mut Vec<String>,
|
|
conn: &mut Connection,
|
|
data_path: PathBuf,
|
|
) -> Result<()> {
|
|
if ids.is_empty() {
|
|
cmd.error(
|
|
ErrorKind::InvalidValue,
|
|
"No ID given, you must supply exactly one ID when using --update",
|
|
)
|
|
.exit();
|
|
} else if ids.len() > 1 {
|
|
cmd.error(ErrorKind::InvalidValue, "More than one ID given, you must supply exactly one ID or atleast one tag when using --update").exit();
|
|
}
|
|
|
|
let item_id = ids.iter().next().expect("Unable to determine item id");
|
|
let item_maybe = db::get_item(conn, *item_id)?;
|
|
|
|
let mut item = item_maybe.expect("Unable to find item in database");
|
|
debug!("MAIN: Found item {:?}", item);
|
|
|
|
if !tags.is_empty() {
|
|
debug!("MAIN: Updating item tags");
|
|
db::set_item_tags(conn, item.clone(), tags)?;
|
|
}
|
|
|
|
if item.size.is_none() {
|
|
info!("Updating unknown stream size");
|
|
let mut item_path = data_path.clone();
|
|
item_path.push(item.id.unwrap().to_string());
|
|
let item_file_metadata = item_path.metadata();
|
|
|
|
if item_file_metadata.is_ok() {
|
|
debug!("MAIN: Updating stream size of {:?}", item_path);
|
|
let compression_type = CompressionType::from_str(&item.compression)?;
|
|
let compression_engine = crate::compression::get_engine(compression_type)
|
|
.expect("Unable to get compression engine");
|
|
let size = compression_engine.size(item_path)? as i64;
|
|
item.size = Some(size);
|
|
db::update_item(&conn, item.clone())?;
|
|
} else {
|
|
debug!(
|
|
"MAIN: Unable to update size of item due to missing file {:?}",
|
|
item_path
|
|
);
|
|
}
|
|
}
|
|
|
|
if args.item.meta.len() > 0 {
|
|
debug!("MAIN: Updating item meta");
|
|
for kv in args.item.meta.iter() {
|
|
let meta = db::Meta {
|
|
id: item.id.unwrap(),
|
|
name: kv.key.to_string(),
|
|
value: kv.value.to_string(),
|
|
};
|
|
db::store_meta(conn, meta)?;
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|