69 lines
1.8 KiB
Rust
69 lines
1.8 KiB
Rust
use std::env;
|
|
use std::ffi::OsStr;
|
|
use std::fs::{DirEntry, File};
|
|
use std::io::{self, Read};
|
|
use std::path::Path;
|
|
use std::process::ExitCode;
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
|
|
|
use clap::{Parser, Subcommand};
|
|
|
|
#[derive(Parser)]
|
|
pub struct Args {
|
|
#[arg(short, long)]
|
|
verbose: bool,
|
|
#[command(subcommand)]
|
|
command: Commands,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
pub enum Commands {
|
|
CheckApiRefs,
|
|
}
|
|
|
|
pub static EXIT_OK: AtomicBool = AtomicBool::new(true);
|
|
|
|
fn main() -> io::Result<ExitCode> {
|
|
let args = Args::parse();
|
|
match args.command {
|
|
Commands::CheckApiRefs => walk_wsp(&mut |_| Ok(true), &mut |file| {
|
|
if file.path().extension() == Some(OsStr::new("rs")) && file.file_name() != "lib.rs" {
|
|
let mut contents = String::new();
|
|
File::open(file.path())?.read_to_string(&mut contents)?;
|
|
for (l, line) in contents.lines().enumerate() {
|
|
if line.trim().starts_with("use") {
|
|
if let Some(c) = line.find("orchid_api") {
|
|
if Some(c) != line.find("orchid_api_") {
|
|
let dname = file.path().to_string_lossy().to_string();
|
|
eprintln!("orchid_api imported in {dname} at {};{}", l + 1, c + 1)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Ok(())
|
|
})?,
|
|
}
|
|
Ok(if EXIT_OK.load(Ordering::Relaxed) { ExitCode::SUCCESS } else { ExitCode::FAILURE })
|
|
}
|
|
|
|
fn walk_wsp(
|
|
dir_filter: &mut impl FnMut(&DirEntry) -> io::Result<bool>,
|
|
file_handler: &mut impl FnMut(DirEntry) -> io::Result<()>,
|
|
) -> io::Result<()> {
|
|
return recurse(&env::current_dir()?, dir_filter, file_handler);
|
|
fn recurse(
|
|
dir: &Path,
|
|
dir_filter: &mut impl FnMut(&DirEntry) -> io::Result<bool>,
|
|
file_handler: &mut impl FnMut(DirEntry) -> io::Result<()>,
|
|
) -> io::Result<()> {
|
|
for file in dir.read_dir()?.collect::<Result<Vec<_>, _>>()? {
|
|
if file.metadata()?.is_dir() && dir_filter(&file)? {
|
|
recurse(&file.path(), dir_filter, file_handler)?;
|
|
}
|
|
file_handler(file)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|