code-moniker-workspace 0.4.0

use std::collections::BTreeMap;

use crate::snapshot::{SourceFileRecord, SourceId, SymbolRecord, WorkspaceSnapshot};

use super::model::{Note, NoteResolution, ResolvedNote};

pub fn resolve_notes(notes: &[Note], snapshot: &WorkspaceSnapshot) -> Vec<ResolvedNote> {
	let sources = source_by_id(snapshot);
	let sources_by_uri = source_by_uri(snapshot);
	let symbols = symbol_by_identity(snapshot);
	notes
		.iter()
		.cloned()
		.map(|note| {
			let resolution = resolve_note(&note, &symbols, &sources, &sources_by_uri);
			ResolvedNote { note, resolution }
		})
		.collect()
}

fn resolve_note(
	note: &Note,
	symbols: &BTreeMap<&str, &SymbolRecord>,
	sources: &BTreeMap<SourceId, &SourceFileRecord>,
	sources_by_uri: &BTreeMap<&str, &SourceFileRecord>,
) -> NoteResolution {
	if let Some(target_label) = navigation_target_label(note.moniker.as_str()) {
		return NoteResolution::Resolved {
			target_label,
			target_file: String::new(),
			target_slice: None,
		};
	}
	if let Some(target_label) = view_target_label(note.moniker.as_str()) {
		return NoteResolution::Resolved {
			target_label,
			target_file: String::new(),
			target_slice: None,
		};
	}
	if let Some(source) = sources_by_uri.get(note.moniker.as_str()).copied() {
		return NoteResolution::Resolved {
			target_label: format!("file {}", source.rel_path),
			target_file: source.rel_path.clone(),
			target_slice: None,
		};
	}
	let Some(symbol) = symbols.get(note.moniker.as_str()).copied() else {
		return NoteResolution::Orphan;
	};
	let Some(source) = sources.get(&symbol.source) else {
		return NoteResolution::Orphan;
	};
	NoteResolution::Resolved {
		target_label: format!("{} {}", symbol.kind, symbol.name),
		target_file: source.rel_path.clone(),
		target_slice: symbol.line_range,
	}
}

fn navigation_target_label(moniker: &str) -> Option<String> {
	workspace_uri_suffix(moniker, "navigation/")
		.map(|target| format!("navigation {}", decode_navigation_id(target)))
}

fn view_target_label(moniker: &str) -> Option<String> {
	workspace_uri_suffix(moniker, "views/").map(|target| format!("view {target}"))
}

fn workspace_uri_suffix<'a>(moniker: &'a str, path: &str) -> Option<&'a str> {
	let workspace_path = format!("workspace/{path}");
	moniker.strip_prefix(&workspace_path).or_else(|| {
		moniker
			.split_once(&workspace_path)
			.map(|(_, suffix)| suffix)
	})
}

fn decode_navigation_id(value: &str) -> String {
	let mut decoded = Vec::new();
	let bytes = value.as_bytes();
	let mut idx = 0;
	while idx < bytes.len() {
		if bytes[idx] == b'%' && idx + 2 < bytes.len() {
			if let Ok(hex) = std::str::from_utf8(&bytes[idx + 1..idx + 3]) {
				if let Ok(byte) = u8::from_str_radix(hex, 16) {
					decoded.push(byte);
					idx += 3;
					continue;
				}
			}
		}
		decoded.push(bytes[idx]);
		idx += 1;
	}
	String::from_utf8(decoded).unwrap_or_else(|_| value.to_string())
}

fn source_by_id(snapshot: &WorkspaceSnapshot) -> BTreeMap<SourceId, &SourceFileRecord> {
	snapshot
		.index
		.sources
		.iter()
		.map(|source| (source.id, source))
		.collect()
}

fn source_by_uri(snapshot: &WorkspaceSnapshot) -> BTreeMap<&str, &SourceFileRecord> {
	snapshot
		.index
		.sources
		.iter()
		.map(|source| (source.uri.as_str(), source))
		.collect()
}

fn symbol_by_identity(snapshot: &WorkspaceSnapshot) -> BTreeMap<&str, &SymbolRecord> {
	snapshot
		.index
		.symbols
		.iter()
		.map(|symbol| (symbol.identity.as_ref(), symbol))
		.collect()
}

#[cfg(test)]
mod tests {
	use crate::snapshot::{
		ChangeOverlay, CodeIndex, CodeIndexTimings, LinkageSnapshot, RecordTable,
		ResourceGeneration, SourceCatalog, SourceFileRecord, SourceId, SymbolId, SymbolRecord,
		WorkspaceSnapshot, WorkspaceTimings,
	};

	use super::*;
	use crate::notes::{NoteAuthor, NoteId, NoteKind, NoteStatus};

	#[test]
	fn resolves_matching_moniker_and_keeps_orphans() {
		let moniker = "code+moniker://./lang:rs/module:example/fn:run()";
		let notes = vec![
			sample_note("note_1", moniker),
			sample_note("note_2", "code+moniker://./lang:rs/module:missing"),
		];
		let snapshot = snapshot_with_symbol(moniker);

		let resolved = resolve_notes(&notes, &snapshot);

		assert_eq!(resolved.len(), 2);
		assert_eq!(
			resolved[0].resolution,
			NoteResolution::Resolved {
				target_label: "fn run()".to_string(),
				target_file: "src/lib.rs".to_string(),
				target_slice: Some((3, 7)),
			}
		);
		assert!(resolved[1].resolution.is_orphan());
	}

	#[test]
	fn treats_symbol_without_source_as_orphan() {
		let moniker = "code+moniker://./lang:rs/module:example/fn:run()";
		let notes = vec![sample_note("note_1", moniker)];
		let snapshot = snapshot_with_orphaned_symbol(moniker);

		let resolved = resolve_notes(&notes, &snapshot);

		assert!(resolved[0].resolution.is_orphan());
	}

	#[test]
	fn resolves_navigation_note_targets() {
		let notes = vec![sample_note(
			"note_nav",
			"code+moniker://workspace/navigation/root%3Aexplorer",
		)];
		let snapshot = snapshot_with_symbol("code+moniker://./lang:rs/module:example/fn:run()");

		let resolved = resolve_notes(&notes, &snapshot);

		assert_eq!(
			resolved[0].resolution,
			NoteResolution::Resolved {
				target_label: "navigation root:explorer".to_string(),
				target_file: String::new(),
				target_slice: None,
			}
		);
	}

	fn sample_note(id: &str, moniker: &str) -> Note {
		Note {
			id: NoteId::new(id),
			moniker: moniker.to_string(),
			kind: NoteKind::Todo,
			status: NoteStatus::Pending,
			title: "Title".to_string(),
			body: "Body".to_string(),
			created_by: NoteAuthor::User,
			created_at: "2026-06-02T00:00:00Z".to_string(),
			updated_at: "2026-06-02T00:00:00Z".to_string(),
		}
	}

	fn snapshot_with_symbol(moniker: &str) -> WorkspaceSnapshot {
		let source = SourceFileRecord {
			id: SourceId::at(1),
			uri: "code+moniker://./lang:rs/path:src/path:lib.rs".to_string(),
			source_root: 0,
			path: "/tmp/src/lib.rs".to_string(),
			rel_path: "src/lib.rs".to_string(),
			anchor: ".".to_string(),
			language: "rs".to_string(),
			text: "pub fn run() {}\n".to_string(),
		};
		let symbol = SymbolRecord {
			id: SymbolId::at(0, 1),
			source: source.id,
			identity: std::sync::Arc::from(moniker.to_string()),
			name: "run()".to_string(),
			kind: "fn".to_string(),
			visibility: "public".to_string(),
			signature: "run()".to_string(),
			navigable: true,
			line_range: Some((3, 7)),
			parent: None,
		};
		WorkspaceSnapshot {
			generation: ResourceGeneration::new(1),
			catalog: SourceCatalog::new(ResourceGeneration::new(1), Vec::new()),
			index: CodeIndex {
				generation: ResourceGeneration::new(1),
				catalog_generation: ResourceGeneration::new(1),
				identity_scheme: "code+moniker://".to_string(),
				sources: vec![source],
				symbols: RecordTable::from_records(vec![symbol]),
				references: RecordTable::from_records(Vec::new()),
				timings: CodeIndexTimings::default(),
			},
			linkage: LinkageSnapshot::new(
				ResourceGeneration::new(1),
				ResourceGeneration::new(1),
				0,
				0,
			),
			changes: ChangeOverlay::new(
				ResourceGeneration::new(1),
				ResourceGeneration::new(1),
				ResourceGeneration::new(1),
				Vec::new(),
			),
			timings: WorkspaceTimings::default(),
		}
	}

	fn snapshot_with_orphaned_symbol(moniker: &str) -> WorkspaceSnapshot {
		let symbol = SymbolRecord {
			id: SymbolId::at(0, 1),
			source: SourceId::at(9999),
			identity: std::sync::Arc::from(moniker.to_string()),
			name: "run()".to_string(),
			kind: "fn".to_string(),
			visibility: "public".to_string(),
			signature: "run()".to_string(),
			navigable: true,
			line_range: Some((3, 7)),
			parent: None,
		};
		WorkspaceSnapshot {
			generation: ResourceGeneration::new(1),
			catalog: SourceCatalog::new(ResourceGeneration::new(1), Vec::new()),
			index: CodeIndex {
				generation: ResourceGeneration::new(1),
				catalog_generation: ResourceGeneration::new(1),
				identity_scheme: "code+moniker://".to_string(),
				sources: Vec::new(),
				symbols: RecordTable::from_records(vec![symbol]),
				references: RecordTable::from_records(Vec::new()),
				timings: CodeIndexTimings::default(),
			},
			linkage: LinkageSnapshot::new(
				ResourceGeneration::new(1),
				ResourceGeneration::new(1),
				0,
				0,
			),
			changes: ChangeOverlay::new(
				ResourceGeneration::new(1),
				ResourceGeneration::new(1),
				ResourceGeneration::new(1),
				Vec::new(),
			),
			timings: WorkspaceTimings::default(),
		}
	}
}