feature: custom ignore path

Author: JasonShin

sqlx-ts-common/src/cli.rs

@@ -64,4 +64,8 @@ pub struct Cli {
  /// Primary DB database name
  #[clap(long)]
  pub db_name: Option<String>,
+
+ /// Folder paths to ignore
+ #[clap(long, parse(from_os_str), multiple_values = true)]
+ pub ignore: Vec<std::path::PathBuf>,
 }

src/main.rs

@@ -10,6 +10,7 @@ use clap::{ArgEnum, Args, Parser, Subcommand};
 use dotenv::dotenv;
 use sqlx_ts_common::cli::Cli;
 use sqlx_ts_core::execute::execute;
+use std::path::Path;
 
 use crate::{parser::parse_source, scan_folder::scan_folder};
 
@@ -19,16 +20,20 @@ fn main() {
  let cli_args = Cli::parse();
  let source_folder = &cli_args.path;
  let ext = &cli_args.ext;
+ let ignore_paths = &cli_args.ignore;
+
  println!(
  "Scanning {:?} for sqls with extension {:?}",
  source_folder, ext
  );
 
- let files = scan_folder(&source_folder, ext);
+ let files = scan_folder(&source_folder, ext, ignore_paths);
+
+ println!("checking files {:?}", files);
 
  if files.is_empty() {
  println!("No targets detected, is it an empty folder?");
- return
+ return;
  }
 
  let explain_results: Vec<bool> = files

src/scan_folder.rs

@@ -3,18 +3,31 @@ use std::path::{Path, PathBuf};
 use sqlx_ts_common::cli::JsExtension;
 use walkdir::WalkDir;
 
-pub fn scan_folder<'a>(folder: &'a PathBuf, js_extension: &JsExtension) -> Vec<PathBuf> {
+pub fn scan_folder<'a>(
+ folder: &'a PathBuf,
+ js_extension: &'a JsExtension,
+ ignore_paths: &'a Vec<PathBuf>,
+) -> Vec<PathBuf> {
  let node_modules_path = folder.join(Path::new("node_modules"));
  let path = Path::new(folder);
  let result: Vec<_> = WalkDir::new(path)
  .follow_links(true)
  .into_iter()
  .filter_map(|e| e.ok())
  .filter(|entry| {
+ // 1. ignore node modules
  if entry.path().starts_with(node_modules_path.as_path()) {
  return false;
  }
 
+ // 2. any custom ignore paths set by user should be ignored
+ let should_ignore = ignore_paths
+ .iter()
+ .any(|ignore| entry.path().starts_with(ignore));
+ if should_ignore {
+ return false;
+ }
+
  let f_name = entry.file_name().to_string_lossy();
  if f_name.ends_with(js_extension.to_string().as_str()) {
  true