msy 0.4.6

use anyhow::{Context, Result};
use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::Path;

/// Filter rule action
#[derive(Debug, Clone, PartialEq)]
pub enum FilterAction {
	/// Include the file
	Include,
	/// Exclude the file
	Exclude,
}

/// A single filter rule
#[derive(Debug, Clone)]
pub struct FilterRule {
	/// Action to take if pattern matches
	pub action: FilterAction,
	/// Compiled glob pattern
	pub pattern: glob::Pattern,
	/// Original pattern string (for debugging)
	#[allow(dead_code)] // Used for debugging and error messages
	pub pattern_str: String,
	/// Whether pattern contains '/' (affects matching behavior)
	pub has_slash: bool,
	/// Whether pattern ends with '/' (directory-only pattern)
	pub is_dir_only: bool,
}

impl FilterRule {
	/// Create a new filter rule from a pattern string
	pub fn new(action: FilterAction, pattern: &str) -> Result<Self> {
		let pattern_str = pattern.to_string();
		let is_dir_only = pattern.ends_with('/');

		// Strip trailing slash for glob matching (we'll handle directory logic separately)
		let pattern_for_glob = if is_dir_only { pattern.trim_end_matches('/') } else { pattern };

		let has_slash = pattern_for_glob.contains('/');
		let pattern = glob::Pattern::new(pattern_for_glob).with_context(|| format!("Invalid filter pattern: {}", pattern))?;

		Ok(Self { action, pattern, pattern_str, has_slash, is_dir_only })
	}

	/// Check if this rule matches the given path
	///
	/// Implements rsync-style matching:
	/// - If pattern ends with '/', it's a directory pattern - match directory and all contents
	/// - If pattern contains '/', match against full relative path
	/// - Otherwise, match against basename only
	pub fn matches(&self, path: &Path, is_dir: bool) -> bool {
		if self.is_dir_only {
			// Pattern ends with '/' - directory-only pattern
			// Matches the directory itself AND everything inside it

			if self.has_slash {
				// Pattern with slash like "foo/bar/" - match against full path
				if let Some(path_str) = path.to_str() {
					// Check if path itself matches (if it's a directory)
					if is_dir && self.pattern.matches(path_str) {
						return true;
					}
					// Check if path is inside a matching directory
					for ancestor in path.ancestors().skip(1) {
						if let Some(ancestor_str) = ancestor.to_str()
							&& !ancestor_str.is_empty()
							&& self.pattern.matches(ancestor_str)
						{
							return true;
						}
					}
				}
				false
			} else {
				// Pattern like "*/" or "build/" - match against basename
				// Special case: "*/" only matches directories, not their contents
				// But "build/" matches the directory AND its contents
				let pattern_str = self.pattern.as_str();
				let is_wildcard_only = pattern_str == "*";

				if is_wildcard_only {
					// "*/" pattern - only match directories, not their contents
					if !is_dir {
						return false;
					}
					if let Some(basename) = path.file_name().and_then(|n| n.to_str()) {
						return self.pattern.matches(basename);
					}
					false
				} else {
					// Specific directory name like "build/" - match directory and its contents
					if let Some(basename) = path.file_name().and_then(|n| n.to_str()) {
						// Check if path itself is the matching directory
						if is_dir && self.pattern.matches(basename) {
							return true;
						}
					}
					// Check if any parent directory basename matches
					for ancestor in path.ancestors().skip(1) {
						if let Some(ancestor_basename) = ancestor.file_name().and_then(|n| n.to_str())
							&& self.pattern.matches(ancestor_basename)
						{
							return true;
						}
					}
					false
				}
			}
		} else if self.has_slash {
			// Pattern has '/' - match against full path
			if let Some(path_str) = path.to_str() { self.pattern.matches(path_str) } else { false }
		} else {
			// No '/' in pattern - match against basename only (rsync behavior)
			if let Some(basename) = path.file_name().and_then(|n| n.to_str()) { self.pattern.matches(basename) } else { false }
		}
	}
}

/// Filter engine that processes include/exclude rules
#[derive(Debug, Clone)]
pub struct FilterEngine {
	/// Ordered list of filter rules (first match wins)
	rules: Vec<FilterRule>,
}

impl FilterEngine {
	/// Create a new empty filter engine
	pub fn new() -> Self {
		Self { rules: Vec::new() }
	}

	/// Add a filter rule from rsync-style syntax
	///
	/// Rules can be:
	/// - "+ pattern" - Include rule
	/// - "- pattern" - Exclude rule
	/// - "pattern" - Defaults to exclude
	pub fn add_rule(&mut self, rule: &str) -> Result<()> {
		let rule = rule.trim();

		if rule.is_empty() || rule.starts_with('#') {
			// Skip empty lines and comments
			return Ok(());
		}

		let (action, pattern) = if let Some(pattern) = rule.strip_prefix("+ ") {
			(FilterAction::Include, pattern.trim())
		} else if let Some(pattern) = rule.strip_prefix("+") {
			(FilterAction::Include, pattern.trim())
		} else if let Some(pattern) = rule.strip_prefix("- ") {
			(FilterAction::Exclude, pattern.trim())
		} else if let Some(pattern) = rule.strip_prefix("-") {
			(FilterAction::Exclude, pattern.trim())
		} else {
			// Default to exclude if no prefix
			(FilterAction::Exclude, rule)
		};

		if pattern.is_empty() {
			anyhow::bail!("Empty filter pattern");
		}

		let rule = FilterRule::new(action, pattern)?;
		self.rules.push(rule);
		Ok(())
	}

	/// Add an include rule
	pub fn add_include(&mut self, pattern: &str) -> Result<()> {
		let rule = FilterRule::new(FilterAction::Include, pattern)?;
		self.rules.push(rule);
		Ok(())
	}

	/// Add an exclude rule
	pub fn add_exclude(&mut self, pattern: &str) -> Result<()> {
		let rule = FilterRule::new(FilterAction::Exclude, pattern)?;
		self.rules.push(rule);
		Ok(())
	}

	/// Load filter rules from a file
	pub fn add_rules_from_file(&mut self, file_path: &Path) -> Result<()> {
		let file = File::open(file_path).with_context(|| format!("Failed to open filter file: {}", file_path.display()))?;

		let reader = BufReader::new(file);

		for (line_num, line) in reader.lines().enumerate() {
			let line = line.with_context(|| format!("Failed to read line {} from {}", line_num + 1, file_path.display()))?;

			self.add_rule(&line).with_context(|| format!("Invalid rule at line {} in {}", line_num + 1, file_path.display()))?;
		}

		Ok(())
	}

	/// Load ignore template from ~/.config/sy/templates/
	///
	/// Template names are resolved to ~/.config/sy/templates/{name}.syignore
	/// Example: "rust" -> ~/.config/sy/templates/rust.syignore
	pub fn add_template(&mut self, template_name: &str) -> Result<()> {
		let config_dir = dirs::config_dir().ok_or_else(|| anyhow::anyhow!("Could not determine config directory"))?;

		let template_dir = config_dir.join("sy").join("templates");
		let template_file = template_dir.join(format!("{}.syignore", template_name));

		if !template_file.exists() {
			anyhow::bail!("Template '{}' not found at {}", template_name, template_file.display());
		}

		self.add_rules_from_file(&template_file).with_context(|| format!("Failed to load template '{}'", template_name))
	}

	/// Load .syignore file if it exists in the given directory
	///
	/// Returns Ok(true) if file was loaded, Ok(false) if file doesn't exist
	pub fn add_syignore_if_exists(&mut self, directory: &Path) -> Result<bool> {
		let syignore_path = directory.join(".syignore");

		if !syignore_path.exists() {
			return Ok(false);
		}

		self.add_rules_from_file(&syignore_path)?;
		Ok(true)
	}

	/// Check if a path should be included (not excluded)
	///
	/// Returns true if the file should be synced, false if it should be excluded.
	/// First matching rule wins. If no rules match, default is to include.
	pub fn should_include(&self, path: &Path, is_dir: bool) -> bool {
		if self.rules.is_empty() {
			// No rules = include everything
			return true;
		}

		// Find first matching rule
		for rule in &self.rules {
			if rule.matches(path, is_dir) {
				return rule.action == FilterAction::Include;
			}
		}

		// No rules matched - default is to include
		true
	}

	/// Check if a path should be excluded
	pub fn should_exclude(&self, path: &Path, is_dir: bool) -> bool {
		!self.should_include(path, is_dir)
	}

	/// Get number of rules
	#[allow(dead_code)] // Public API for filter introspection
	pub fn rule_count(&self) -> usize {
		self.rules.len()
	}

	/// Check if filter has any rules
	#[allow(dead_code)] // Public API for filter introspection
	pub fn is_empty(&self) -> bool {
		self.rules.is_empty()
	}
}

impl Default for FilterEngine {
	fn default() -> Self {
		Self::new()
	}
}

#[cfg(test)]
mod tests {
	use super::*;

	#[test]
	fn test_empty_filter_includes_all() {
		let filter = FilterEngine::new();
		assert!(filter.should_include(Path::new("foo.txt"), false));
		assert!(filter.should_include(Path::new("bar/baz.rs"), false));
	}

	#[test]
	fn test_exclude_pattern() {
		let mut filter = FilterEngine::new();
		filter.add_exclude("*.log").unwrap();

		assert!(!filter.should_include(Path::new("test.log"), false));
		assert!(filter.should_include(Path::new("test.txt"), false));
	}

	#[test]
	fn test_include_pattern() {
		let mut filter = FilterEngine::new();
		// Exclude all .log files
		filter.add_exclude("*.log").unwrap();
		// But include important.log
		filter.add_include("important.log").unwrap();

		// First rule matches (exclude)
		assert!(!filter.should_include(Path::new("test.log"), false));
		// No rules match (default include)
		assert!(filter.should_include(Path::new("test.txt"), false));
		// Second rule matches (include) - but first rule already matched!
		// This shows order matters
		assert!(!filter.should_include(Path::new("important.log"), false));
	}

	#[test]
	fn test_rule_order_matters() {
		let mut filter = FilterEngine::new();
		// Include important.log first
		filter.add_include("important.log").unwrap();
		// Then exclude all .log files
		filter.add_exclude("*.log").unwrap();

		// First rule matches (include)
		assert!(filter.should_include(Path::new("important.log"), false));
		// Second rule matches (exclude)
		assert!(!filter.should_include(Path::new("test.log"), false));
		// No rules match (default include)
		assert!(filter.should_include(Path::new("test.txt"), false));
	}

	#[test]
	fn test_rsync_style_syntax() {
		let mut filter = FilterEngine::new();
		// Test rsync-style + and - prefixes
		filter.add_rule("+ *.rs").unwrap();
		filter.add_rule("- *.log").unwrap();
		filter.add_rule("*.tmp").unwrap(); // No prefix = exclude

		assert!(filter.should_include(Path::new("foo.rs"), false));
		assert!(!filter.should_include(Path::new("bar.log"), false));
		assert!(!filter.should_include(Path::new("baz.tmp"), false));
		assert!(filter.should_include(Path::new("qux.txt"), false));
	}

	#[test]
	fn test_directory_patterns() {
		let mut filter = FilterEngine::new();
		filter.add_exclude("target/*").unwrap();
		filter.add_exclude("node_modules/*").unwrap();

		assert!(!filter.should_include(Path::new("target/debug"), false));
		assert!(!filter.should_include(Path::new("node_modules/foo"), false));
		assert!(filter.should_include(Path::new("src/main.rs"), false));
	}

	#[test]
	fn test_comments_and_empty_lines() {
		let mut filter = FilterEngine::new();
		filter.add_rule("# This is a comment").unwrap();
		filter.add_rule("").unwrap();
		filter.add_rule("   ").unwrap();
		filter.add_rule("*.log").unwrap();

		// Should only have one rule (the *.log pattern)
		assert_eq!(filter.rule_count(), 1);
		assert!(!filter.should_include(Path::new("test.log"), false));
	}

	#[test]
	fn test_invalid_pattern() {
		let mut filter = FilterEngine::new();
		// Invalid glob pattern with unbalanced brackets
		let result = filter.add_exclude("[invalid");
		assert!(result.is_err());
	}

	#[test]
	fn test_default_action() {
		let mut filter = FilterEngine::new();
		// Pattern without prefix defaults to exclude
		filter.add_rule("*.log").unwrap();

		assert!(!filter.should_include(Path::new("test.log"), false));
		assert!(filter.should_include(Path::new("test.txt"), false));
	}

	#[test]
	fn test_glob_wildcards() {
		let mut filter = FilterEngine::new();
		filter.add_exclude("**/*.log").unwrap();
		filter.add_exclude("temp/**").unwrap();

		assert!(!filter.should_include(Path::new("foo/bar/test.log"), false));
		assert!(!filter.should_include(Path::new("temp/foo/bar"), false));
		assert!(filter.should_include(Path::new("src/main.rs"), false));
	}

	#[test]
	fn test_rsync_basename_matching() {
		let mut filter = FilterEngine::new();
		// Pattern without '/' should match basename only (rsync behavior)
		filter.add_include("important.rs").unwrap();
		filter.add_exclude("*.rs").unwrap();

		// Should include important.rs even in subdirectory
		assert!(filter.should_include(Path::new("dir/important.rs"), false));
		assert!(filter.should_include(Path::new("deep/nested/important.rs"), false));

		// Should exclude other .rs files
		assert!(!filter.should_include(Path::new("code.rs"), false));
		assert!(!filter.should_include(Path::new("dir/code.rs"), false));
	}

	#[test]
	fn test_rsync_path_matching() {
		let mut filter = FilterEngine::new();
		// Pattern with '/' should match full path
		filter.add_exclude("dir1/*.rs").unwrap();
		filter.add_exclude("**/temp/*.log").unwrap();

		// Should exclude .rs files in dir1/ only
		assert!(!filter.should_include(Path::new("dir1/code.rs"), false));
		assert!(filter.should_include(Path::new("dir2/code.rs"), false)); // different dir

		// Should exclude .log files in any temp/ directory
		assert!(!filter.should_include(Path::new("temp/test.log"), false));
		assert!(!filter.should_include(Path::new("foo/temp/test.log"), false));
		assert!(filter.should_include(Path::new("temp/test.txt"), false)); // not .log
	}

	#[test]
	fn test_directory_only_patterns() {
		let mut filter = FilterEngine::new();
		// Pattern ending with '/' matches directory and all contents
		filter.add_exclude("dir1/").unwrap();
		filter.add_include("*.txt").unwrap();
		filter.add_exclude("*").unwrap();

		// dir1/ should be excluded along with all its contents
		assert!(!filter.should_include(Path::new("dir1/keep.txt"), false));
		assert!(!filter.should_include(Path::new("dir1/subdir/file.rs"), false));

		// *.txt in root or other directories should be included
		assert!(filter.should_include(Path::new("keep.txt"), false));
		assert!(filter.should_include(Path::new("dir2/keep.txt"), false));

		// Non-.txt files should be excluded by the wildcard
		assert!(!filter.should_include(Path::new("exclude.log"), false));
	}

	#[test]
	fn test_rsync_exact_scenario() {
		// Test the exact rsync scenario: --exclude="dir1/" --include="*.txt" --exclude="*"
		let mut filter = FilterEngine::new();
		filter.add_exclude("dir1/").unwrap();
		filter.add_include("*.txt").unwrap();
		filter.add_exclude("*").unwrap();

		// These should match rsync behavior
		assert!(filter.should_include(Path::new("keep.txt"), false), "keep.txt should be included");
		assert!(!filter.should_include(Path::new("dir1"), true), "dir1 directory should be excluded");
		assert!(!filter.should_include(Path::new("dir1/keep.txt"), false), "dir1/keep.txt should be excluded");
		assert!(!filter.should_include(Path::new("dir1/subdir/file.txt"), false), "dir1/subdir/file.txt should be excluded");
		assert!(filter.should_include(Path::new("dir2/keep.txt"), false), "dir2/keep.txt should be included");
		assert!(!filter.should_include(Path::new("exclude.log"), false), "exclude.log should be excluded");
	}

	#[test]
	fn test_directory_pattern_vs_file_pattern() {
		let mut filter = FilterEngine::new();
		// Pattern WITH trailing slash - directory only
		filter.add_exclude("build/").unwrap();

		// Should exclude the directory and everything in it
		assert!(!filter.should_include(Path::new("build/output.txt"), false));
		assert!(!filter.should_include(Path::new("build/nested/file.rs"), false));

		// Pattern WITHOUT trailing slash - matches file or directory basename
		let mut filter2 = FilterEngine::new();
		filter2.add_exclude("build").unwrap();

		// Should exclude anything named "build" (file or dir) - basename matching
		assert!(!filter2.should_include(Path::new("build"), false)); // file named "build"
		assert!(!filter2.should_include(Path::new("build"), true)); // directory named "build"
		assert!(!filter2.should_include(Path::new("other/build"), false)); // basename is "build"
		assert!(filter2.should_include(Path::new("build/output.txt"), false)); // basename is "output.txt", not "build"
		assert!(filter2.should_include(Path::new("building"), false)); // basename is "building", not "build"
	}
}