stash: move import logic into stash::commmands

Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: I6a6a6964f8fb8c9b14049ba3a343bb453ca59004
This commit is contained in:
raf 2025-08-20 10:35:10 +03:00
commit 1ed518a3b6
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
3 changed files with 47 additions and 15 deletions

View file

@ -1,45 +0,0 @@
use std::io::{self, BufRead};
use log::{error, info};
use crate::db::{Entry, SqliteClipboardDb, detect_mime};
pub trait ImportCommand {
fn import_tsv(&self, input: impl io::Read);
}
impl ImportCommand for SqliteClipboardDb {
fn import_tsv(&self, input: impl io::Read) {
let reader = io::BufReader::new(input);
let mut imported = 0;
for line in reader.lines().map_while(Result::ok) {
let mut parts = line.splitn(2, '\t');
let (Some(id_str), Some(val)) = (parts.next(), parts.next()) else {
error!("Malformed TSV line: {line:?}");
continue;
};
let Ok(_id) = id_str.parse::<u64>() else {
error!("Failed to parse id from line: {id_str}");
continue;
};
let entry = Entry {
contents: val.as_bytes().to_vec(),
mime: detect_mime(val.as_bytes()),
};
match self.conn.execute(
"INSERT INTO clipboard (contents, mime) VALUES (?1, ?2)",
rusqlite::params![entry.contents, entry.mime],
) {
Ok(_) => {
imported += 1;
info!("Imported entry from TSV");
},
Err(e) => error!("Failed to insert entry: {e}"),
}
}
info!("Imported {imported} records from TSV into SQLite database.");
}
}