bestool-psql 1.7.2

use miette::{IntoDiagnostic, Result};
use rustyline::{
	Config, Editor,
	history::{History as HistoryTrait, MemHistory},
};
use tracing::debug;

use crate::audit::Audit;

/// Prompt for OTS value with rustyline and history from previous OTS values
pub fn prompt_for_ots(audit: &Audit) -> Result<String> {
	let ots_history = load_ots_history(audit)?;

	let mut rl: Editor<(), MemHistory> = Editor::with_history(
		Config::builder()
			.auto_add_history(false)
			.history_ignore_dups(true)
			.unwrap()
			.build(),
		MemHistory::new(),
	)
	.into_diagnostic()?;

	for ots_value in ots_history.iter().rev() {
		let _ = rl.history_mut().add(ots_value);
	}

	loop {
		match rl.readline("OTS? ") {
			Ok(line) => {
				let trimmed = line.trim();
				if trimmed.is_empty() {
					eprintln!("OTS is required for write mode");
					continue;
				}
				return Ok(trimmed.to_string());
			}
			Err(rustyline::error::ReadlineError::Interrupted) => {
				return Err(miette::miette!("OTS prompt interrupted"));
			}
			Err(rustyline::error::ReadlineError::Eof) => {
				return Err(miette::miette!("OTS is required for write mode"));
			}
			Err(err) => {
				return Err(err).into_diagnostic();
			}
		}
	}
}

/// Load unique OTS values from the audit database
fn load_ots_history(audit: &Audit) -> Result<Vec<String>> {
	let entries = audit.list()?;

	let mut ots_values = Vec::new();
	let mut seen = std::collections::HashSet::new();
	for (_timestamp, entry) in entries.into_iter().rev() {
		if let Some(ots) = entry.ots
			&& !ots.is_empty()
			&& seen.insert(ots.clone())
		{
			ots_values.push(ots);
		}
	}

	debug!(count = ots_values.len(), "loaded OTS history");
	Ok(ots_values)
}