1use std::borrow::Cow;
7use std::ffi::{OsStr, OsString};
8use std::path::{Path, PathBuf};
9
10use rustc_hash::{FxHashMap, FxHashSet};
11
12use fallow_config::{ExternalPluginDef, PackageJson, PluginDetection, UsedClassMemberRule};
13
14use crate::discover::SOURCE_EXTENSIONS;
15
16use super::super::{PathRule, Plugin, PluginResult, PluginUsedExportRule, UsedExportRule};
17use super::{AggregatedPluginResult, PluginRegexValidationError, PluginRegexValidationErrorInput};
18
19#[must_use]
28pub fn is_source_ext_root_pattern(pat: &str) -> bool {
29 if pat.is_empty() || pat.contains('/') {
30 return false;
31 }
32 for expanded in expand_brace_pattern(pat) {
33 if expanded.starts_with('.') {
34 return false;
35 }
36 let Some(ext) = std::path::Path::new(&expanded).extension() else {
37 return false;
38 };
39 let Some(ext_str) = ext.to_str() else {
40 return false;
41 };
42 if !SOURCE_EXTENSIONS.contains(&ext_str) {
43 return false;
44 }
45 }
46 true
47}
48
49#[must_use]
51pub fn prepare_config_pattern(pat: &str) -> Cow<'_, str> {
52 if is_source_ext_root_pattern(pat) {
53 Cow::Owned(format!("**/{pat}"))
54 } else {
55 Cow::Borrowed(pat)
56 }
57}
58
59pub fn process_static_patterns(
61 plugin: &dyn Plugin,
62 root: &Path,
63 result: &mut AggregatedPluginResult,
64) {
65 let pname = plugin.name().to_string();
66 result.active_plugins.push(pname.clone());
67 result
68 .entry_point_roles
69 .insert(pname.clone(), plugin.entry_point_role());
70
71 collect_static_plugin_rules(plugin, &pname, result);
72 collect_static_plugin_metadata(plugin, root, result);
73}
74
75fn collect_static_plugin_rules(
78 plugin: &dyn Plugin,
79 pname: &str,
80 result: &mut AggregatedPluginResult,
81) {
82 for rule in plugin.entry_pattern_rules() {
83 result.entry_patterns.push((rule, pname.to_string()));
84 }
85 for pat in plugin.config_patterns() {
86 result.config_patterns.push((*pat).to_string());
87 }
88 for pat in plugin.always_used() {
89 result
90 .always_used
91 .push(((*pat).to_string(), pname.to_string()));
92 }
93 for rule in plugin.used_export_rules() {
94 result
95 .used_exports
96 .push(PluginUsedExportRule::new(pname.to_string(), rule));
97 }
98 for member in plugin.used_class_members() {
99 result
100 .used_class_members
101 .push(UsedClassMemberRule::from(*member));
102 }
103 for rule in plugin.used_class_member_rules() {
104 result.used_class_members.push(rule);
105 }
106 for pat in plugin.fixture_glob_patterns() {
107 result
108 .fixture_patterns
109 .push(((*pat).to_string(), pname.to_string()));
110 }
111}
112
113fn collect_static_plugin_metadata(
116 plugin: &dyn Plugin,
117 root: &Path,
118 result: &mut AggregatedPluginResult,
119) {
120 for dep in plugin.tooling_dependencies() {
121 result.tooling_dependencies.push((*dep).to_string());
122 }
123 for prefix in plugin.virtual_module_prefixes() {
124 result.virtual_module_prefixes.push((*prefix).to_string());
125 }
126 for suffix in plugin.virtual_package_suffixes() {
127 result.virtual_package_suffixes.push((*suffix).to_string());
128 }
129 for pattern in plugin.generated_import_patterns() {
130 result
131 .generated_import_patterns
132 .push((*pattern).to_string());
133 }
134 for prefix in plugin.generated_type_import_prefixes() {
135 result
136 .generated_type_import_prefixes
137 .push((*prefix).to_string());
138 }
139 for (prefix, replacement) in plugin.path_aliases(root) {
140 result.path_aliases.push((prefix.to_string(), replacement));
141 }
142 result.auto_imports.extend(plugin.auto_imports(root));
143 result
144 .provided_dependencies
145 .extend(plugin.provided_dependencies());
146}
147
148pub fn process_package_json_metadata(
150 active: &[&dyn Plugin],
151 pkg: &PackageJson,
152 root: &Path,
153 result: &mut AggregatedPluginResult,
154 regex_errors: &mut Vec<PluginRegexValidationError>,
155) {
156 for plugin in active {
157 let package_referenced = plugin.package_json_referenced_dependencies(pkg, root);
158 if !package_referenced.is_empty() {
159 let pkg_path = root.join("package.json");
160 result.package_referenced_dependencies.extend(
161 package_referenced
162 .into_iter()
163 .map(|dep| (pkg_path.clone(), dep)),
164 );
165 }
166 let plugin_result = plugin.resolve_package_json(pkg, root);
167 if plugin_result.is_empty() {
168 continue;
169 }
170 tracing::debug!(
171 plugin = plugin.name(),
172 deps = plugin_result.referenced_dependencies.len(),
173 "resolved package.json metadata"
174 );
175 if let Err(mut errors) = process_config_result(plugin.name(), plugin_result, result, None) {
176 regex_errors.append(&mut errors);
177 }
178 }
179}
180
181pub fn is_external_plugin_active(
186 ext: &ExternalPluginDef,
187 all_deps: &[String],
188 root: &Path,
189 discovered_files: &[PathBuf],
190) -> bool {
191 if let Some(detection) = &ext.detection {
192 let all_dep_refs: Vec<&str> = all_deps.iter().map(String::as_str).collect();
193 check_plugin_detection(detection, &all_dep_refs, root, discovered_files)
194 } else if !ext.enablers.is_empty() {
195 ext.enablers.iter().any(|enabler| {
196 if enabler.ends_with('/') {
197 all_deps.iter().any(|d| d.starts_with(enabler))
198 } else {
199 all_deps.iter().any(|d| d == enabler)
200 }
201 })
202 } else {
203 false
204 }
205}
206
207pub fn process_external_plugins(
209 external_plugins: &[ExternalPluginDef],
210 all_deps: &[String],
211 root: &Path,
212 discovered_files: &[PathBuf],
213 result: &mut AggregatedPluginResult,
214) {
215 for ext in external_plugins {
216 let is_active = is_external_plugin_active(ext, all_deps, root, discovered_files);
217 if is_active {
218 result.active_plugins.push(ext.name.clone());
219 result
220 .entry_point_roles
221 .insert(ext.name.clone(), ext.entry_point_role);
222 result.entry_patterns.extend(
223 ext.entry_points
224 .iter()
225 .map(|p| (PathRule::new(p.clone()), ext.name.clone())),
226 );
227 if !ext.manifest_entries.is_empty() {
228 result.entry_patterns.extend(
229 crate::plugins::manifest_entries::evaluate_manifest_entries(ext, root)
230 .into_iter()
231 .map(|rule| (rule, ext.name.clone())),
232 );
233 }
234 result.config_patterns.extend(ext.config_patterns.clone());
235 result.always_used.extend(
236 ext.config_patterns
237 .iter()
238 .chain(ext.always_used.iter())
239 .map(|p| (p.clone(), ext.name.clone())),
240 );
241 result
242 .tooling_dependencies
243 .extend(ext.tooling_dependencies.clone());
244 for ue in &ext.used_exports {
245 result.used_exports.push(PluginUsedExportRule::new(
246 ext.name.clone(),
247 UsedExportRule::new(ue.pattern.clone(), ue.exports.clone()),
248 ));
249 }
250 result
251 .used_class_members
252 .extend(ext.used_class_members.iter().cloned());
253 }
254 }
255}
256
257pub struct ConfigCandidateIndex {
269 dirs: FxHashMap<PathBuf, FxHashSet<OsString>>,
270}
271
272impl ConfigCandidateIndex {
273 #[must_use]
277 pub fn build<'a>(paths: impl IntoIterator<Item = &'a Path>) -> Self {
278 let mut dirs: FxHashMap<PathBuf, FxHashSet<OsString>> = FxHashMap::default();
279 for path in paths {
280 if let (Some(parent), Some(name)) = (path.parent(), path.file_name()) {
281 dirs.entry(parent.to_path_buf())
282 .or_default()
283 .insert(name.to_os_string());
284 }
285 }
286 Self { dirs }
287 }
288
289 #[must_use]
293 pub fn dir_contains(&self, dir: &Path, name: &OsStr) -> bool {
294 self.dirs.get(dir).is_some_and(|names| names.contains(name))
295 }
296
297 #[must_use]
302 pub fn any_descendant_contains(&self, root: &Path, name: &OsStr) -> bool {
303 self.dirs
304 .iter()
305 .any(|(dir, names)| dir.starts_with(root) && names.contains(name))
306 }
307
308 fn glob_matches_in_dir(&self, dir: &Path, matcher: &globset::GlobMatcher) -> Vec<PathBuf> {
309 self.dirs.get(dir).map_or_else(Vec::new, |names| {
310 names
311 .iter()
312 .filter(|name| matcher.is_match(Path::new(name)))
313 .map(|name| dir.join(name))
314 .collect()
315 })
316 }
317}
318
319pub fn discover_config_files<'a>(
339 config_matchers: &[(&'a dyn Plugin, Vec<globset::GlobMatcher>)],
340 resolved_plugins: &FxHashSet<&str>,
341 roots: &[&Path],
342 production_mode: bool,
343 candidate_index: Option<&ConfigCandidateIndex>,
344) -> Vec<(PathBuf, &'a dyn Plugin)> {
345 use rayon::prelude::*;
346 let mut pending: Vec<(&'a dyn Plugin, &Path, String)> = Vec::new();
347 for (plugin, _) in config_matchers {
348 if resolved_plugins.contains(plugin.name()) {
349 continue;
350 }
351 for root in roots {
352 for pat in plugin.config_patterns() {
353 if !production_mode && is_source_ext_root_pattern(pat) {
354 continue;
355 }
356 pending.push((*plugin, *root, pat.to_string()));
357 }
358 }
359 }
360
361 let hits: Vec<(PathBuf, &'a dyn Plugin)> = pending
362 .par_iter()
363 .flat_map_iter(|(plugin, root, pat)| {
364 expand_brace_pattern(pat)
365 .into_iter()
366 .flat_map(|expanded| match candidate_index {
367 Some(index) if !pattern_needs_filesystem(&expanded) => {
373 match_pattern_in_index(root, &expanded, index)
374 }
375 _ => discover_pattern_matches(root, &expanded),
376 })
377 .map(move |path| (path, *plugin))
378 .collect::<Vec<_>>()
379 })
380 .collect();
381
382 let mut seen: FxHashSet<(PathBuf, &'a str)> = FxHashSet::default();
383 let mut config_files: Vec<(PathBuf, &'a dyn Plugin)> = Vec::with_capacity(hits.len());
384 for (path, plugin) in hits {
385 if seen.insert((path.clone(), plugin.name())) {
386 config_files.push((path, plugin));
387 }
388 }
389 config_files
390}
391
392fn pattern_has_glob(pattern: &str) -> bool {
393 pattern.contains('*') || pattern.contains('?') || pattern.contains('[')
394}
395
396fn pattern_needs_filesystem(pattern: &str) -> bool {
402 let mut components = pattern.split('/').peekable();
403 let mut needs_fs = false;
404 while let Some(component) = components.next() {
405 if components.peek().is_none() {
406 break; }
408 if component.starts_with('.')
409 && component != "."
410 && component != ".."
411 && !crate::discover::is_allowed_hidden_dir(OsStr::new(component))
412 {
413 needs_fs = true;
414 break;
415 }
416 }
417 needs_fs
418}
419
420fn match_pattern_in_index(
425 root: &Path,
426 pattern: &str,
427 index: &ConfigCandidateIndex,
428) -> Vec<PathBuf> {
429 if !pattern_has_glob(pattern) {
430 let path = root.join(pattern);
431 return match (path.parent(), path.file_name()) {
432 (Some(dir), Some(name)) if index.dir_contains(dir, name) => vec![path],
433 _ => Vec::new(),
434 };
435 }
436
437 if let Some(stripped) = pattern.strip_prefix("**/") {
438 return match_pattern_in_index(root, stripped, index);
439 }
440
441 let (dir, file_pattern) = match pattern.rsplit_once('/') {
442 Some((parent, file_pattern)) if !pattern_has_glob(parent) => {
443 (root.join(parent), file_pattern)
444 }
445 Some(_) => return Vec::new(),
446 None => (root.to_path_buf(), pattern),
447 };
448
449 let Ok(matcher) = globset::Glob::new(file_pattern).map(|g| g.compile_matcher()) else {
450 return Vec::new();
451 };
452 index.glob_matches_in_dir(&dir, &matcher)
453}
454
455fn discover_pattern_matches(root: &Path, pattern: &str) -> Vec<PathBuf> {
456 if !pattern_has_glob(pattern) {
457 let path = root.join(pattern);
458 return if path.is_file() {
459 vec![path]
460 } else {
461 Vec::new()
462 };
463 }
464
465 if let Some(stripped) = pattern.strip_prefix("**/") {
466 return discover_pattern_matches(root, stripped);
467 }
468
469 let (dir, file_pattern) = match pattern.rsplit_once('/') {
470 Some((parent, file_pattern)) if !pattern_has_glob(parent) => {
471 (root.join(parent), file_pattern)
472 }
473 Some(_) => return Vec::new(),
474 None => (root.to_path_buf(), pattern),
475 };
476
477 scan_dir_for_pattern(&dir, file_pattern)
478}
479
480fn scan_dir_for_pattern(dir: &Path, file_pattern: &str) -> Vec<PathBuf> {
481 let Ok(matcher) = globset::Glob::new(file_pattern).map(|g| g.compile_matcher()) else {
482 return Vec::new();
483 };
484 let Ok(entries) = std::fs::read_dir(dir) else {
485 return Vec::new();
486 };
487
488 entries
489 .filter_map(Result::ok)
490 .map(|entry| entry.path())
491 .filter(|path| path.is_file())
492 .filter(|path| {
493 path.file_name()
494 .is_some_and(|name| matcher.is_match(std::path::Path::new(name)))
495 })
496 .collect()
497}
498
499fn expand_brace_pattern(pattern: &str) -> Vec<String> {
500 let Some(open) = pattern.find('{') else {
501 return vec![pattern.to_string()];
502 };
503 let Some(close_rel) = pattern[open + 1..].find('}') else {
504 return vec![pattern.to_string()];
505 };
506 let close = open + 1 + close_rel;
507
508 let prefix = &pattern[..open];
509 let suffix = &pattern[close + 1..];
510 let inner = &pattern[open + 1..close];
511 let mut expanded = Vec::new();
512 for option in inner.split(',') {
513 for tail in expand_brace_pattern(suffix) {
514 expanded.push(format!("{prefix}{option}{tail}"));
515 }
516 }
517 expanded
518}
519
520fn collect_path_rule_regex_errors(
525 rule: &crate::plugins::PathRule,
526 plugin_name: &str,
527 config_path: Option<&Path>,
528 rule_kind: &'static str,
529 errors: &mut Vec<PluginRegexValidationError>,
530) {
531 for pattern in &rule.exclude_regexes {
532 if let Err(source) = regex::Regex::new(pattern) {
533 errors.push(PluginRegexValidationError::new(
534 PluginRegexValidationErrorInput {
535 plugin_name,
536 config_path,
537 rule_kind,
538 field: "exclude_regexes",
539 rule_pattern: &rule.pattern,
540 regex_pattern: pattern,
541 source: &source,
542 },
543 ));
544 }
545 }
546 for pattern in &rule.exclude_segment_regexes {
547 if let Err(source) = regex::Regex::new(pattern) {
548 errors.push(PluginRegexValidationError::new(
549 PluginRegexValidationErrorInput {
550 plugin_name,
551 config_path,
552 rule_kind,
553 field: "exclude_segment_regexes",
554 rule_pattern: &rule.pattern,
555 regex_pattern: pattern,
556 source: &source,
557 },
558 ));
559 }
560 }
561}
562
563pub fn process_config_result(
569 plugin_name: &str,
570 plugin_result: PluginResult,
571 result: &mut AggregatedPluginResult,
572 config_path: Option<&Path>,
573) -> Result<(), Vec<PluginRegexValidationError>> {
574 let mut regex_errors = Vec::new();
575
576 for rule in &plugin_result.entry_patterns {
577 collect_path_rule_regex_errors(
578 rule,
579 plugin_name,
580 config_path,
581 "entry_patterns[]",
582 &mut regex_errors,
583 );
584 }
585 for rule in &plugin_result.used_exports {
586 collect_path_rule_regex_errors(
587 &rule.path,
588 plugin_name,
589 config_path,
590 "used_exports[].path",
591 &mut regex_errors,
592 );
593 }
594 if !regex_errors.is_empty() {
595 return Err(regex_errors);
596 }
597 merge_plugin_result_fields(plugin_name, plugin_result, result);
598 Ok(())
599}
600
601fn merge_plugin_result_fields(
605 pname: &str,
606 plugin_result: PluginResult,
607 result: &mut AggregatedPluginResult,
608) {
609 if plugin_result.replace_entry_patterns && !plugin_result.entry_patterns.is_empty() {
610 result.entry_patterns.retain(|(_, name)| name != pname);
611 }
612 if plugin_result.replace_used_export_rules && !plugin_result.used_exports.is_empty() {
613 result.used_exports.retain(|rule| rule.plugin_name != pname);
614 }
615 result.entry_patterns.extend(
616 plugin_result
617 .entry_patterns
618 .into_iter()
619 .map(|rule| (rule, pname.to_string())),
620 );
621 result.used_exports.extend(
622 plugin_result
623 .used_exports
624 .into_iter()
625 .map(|rule| PluginUsedExportRule::new(pname.to_string(), rule)),
626 );
627 result
628 .used_class_members
629 .extend(plugin_result.used_class_members);
630 result
631 .referenced_dependencies
632 .extend(plugin_result.referenced_dependencies);
633 result.discovered_always_used.extend(
634 plugin_result
635 .always_used_files
636 .into_iter()
637 .map(|p| (p, pname.to_string())),
638 );
639 for (prefix, replacement) in plugin_result.path_aliases {
640 result
641 .path_aliases
642 .retain(|(existing_prefix, _)| existing_prefix != &prefix);
643 result.path_aliases.push((prefix, replacement));
644 }
645 result.setup_files.extend(
646 plugin_result
647 .setup_files
648 .into_iter()
649 .map(|p| (p, pname.to_string())),
650 );
651 result.fixture_patterns.extend(
652 plugin_result
653 .fixture_patterns
654 .into_iter()
655 .map(|p| (p, pname.to_string())),
656 );
657 result
658 .scss_include_paths
659 .extend(plugin_result.scss_include_paths);
660 result
661 .static_dir_mappings
662 .extend(plugin_result.static_dir_mappings);
663 result
664 .provided_dependencies
665 .extend(plugin_result.provided_dependencies);
666}
667
668pub fn check_has_config_file(
670 plugin: &dyn Plugin,
671 config_matchers: &[(&dyn Plugin, Vec<globset::GlobMatcher>)],
672 relative_files: &[(PathBuf, String)],
673) -> bool {
674 !plugin.config_patterns().is_empty()
675 && config_matchers.iter().any(|(p, matchers)| {
676 p.name() == plugin.name()
677 && relative_files
678 .iter()
679 .any(|(_, rel)| matchers.iter().any(|m| m.is_match(rel.as_str())))
680 })
681}
682
683pub fn check_plugin_detection(
685 detection: &PluginDetection,
686 all_deps: &[&str],
687 root: &Path,
688 discovered_files: &[PathBuf],
689) -> bool {
690 match detection {
691 PluginDetection::Dependency { package } => all_deps.iter().any(|d| *d == package),
692 PluginDetection::FileExists { pattern } => {
693 if let Ok(matcher) = globset::Glob::new(pattern).map(|g| g.compile_matcher()) {
694 for file in discovered_files {
695 let relative = file.strip_prefix(root).unwrap_or(file);
696 if matcher.is_match(relative) {
697 return true;
698 }
699 }
700 }
701 let full_pattern = root.join(pattern).to_string_lossy().to_string();
702 glob::glob(&full_pattern)
703 .ok()
704 .is_some_and(|mut g| g.next().is_some())
705 }
706 PluginDetection::All { conditions } => conditions
707 .iter()
708 .all(|c| check_plugin_detection(c, all_deps, root, discovered_files)),
709 PluginDetection::Any { conditions } => conditions
710 .iter()
711 .any(|c| check_plugin_detection(c, all_deps, root, discovered_files)),
712 }
713}
714
715#[cfg(test)]
716mod tests {
717 use super::*;
718
719 #[test]
720 fn pattern_needs_filesystem_only_for_non_allowlisted_hidden_dirs() {
721 assert!(pattern_needs_filesystem(".config/prisma.ts"));
724 assert!(!pattern_needs_filesystem("tsconfig.json"));
727 assert!(!pattern_needs_filesystem("prisma/schema.prisma"));
728 assert!(!pattern_needs_filesystem(".eslintrc.json"));
729 assert!(!pattern_needs_filesystem("**/project.json"));
730 assert!(!pattern_needs_filesystem(".storybook/main.ts"));
731 assert!(!pattern_needs_filesystem("a/b/c.json"));
732 }
733
734 #[test]
735 fn config_candidate_index_matches_plain_nested_and_glob_shapes() {
736 let root = Path::new("/project");
737 let index = ConfigCandidateIndex::build([
738 Path::new("/project/tsconfig.json"),
739 Path::new("/project/packages/a/tsconfig.json"),
740 Path::new("/project/prisma/schema.prisma"),
741 Path::new("/project/src/main.ts"),
742 ]);
743
744 assert_eq!(
746 match_pattern_in_index(root, "tsconfig.json", &index),
747 vec![PathBuf::from("/project/tsconfig.json")]
748 );
749 assert_eq!(
750 match_pattern_in_index(Path::new("/project/packages/a"), "tsconfig.json", &index),
751 vec![PathBuf::from("/project/packages/a/tsconfig.json")]
752 );
753 assert_eq!(
755 match_pattern_in_index(root, "prisma/schema.prisma", &index),
756 vec![PathBuf::from("/project/prisma/schema.prisma")]
757 );
758 assert_eq!(
759 match_pattern_in_index(root, "**/tsconfig.json", &index),
760 vec![PathBuf::from("/project/tsconfig.json")]
761 );
762 assert_eq!(
763 match_pattern_in_index(Path::new("/project/prisma"), "*.prisma", &index),
764 vec![PathBuf::from("/project/prisma/schema.prisma")]
765 );
766 assert!(match_pattern_in_index(root, "missing.json", &index).is_empty());
768 }
769}