findfile 0.2.8

An ergonomic way to search for files
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use crate::ast::Expression;
use crate::play::Env;
use crate::play::PathInfo;
use crate::play::PlayError;
use crate::play::RunContext;
use crate::play::{PlayContext, PlayResult};
use crate::Value;
use os_str_bytes::OsStrBytes;
use std::collections::HashMap;
use std::ffi::OsString;
use std::io::{ErrorKind as IoErrorKind, Write};
use std::path::Path;
use std::path::PathBuf;

mod config;
pub use config::Config;

#[derive(Default, Debug)]
pub struct Program {
	config: Config,
	env: Env,
	vars: HashMap<String, Value>,
}

impl Program {
	pub fn new(config: Config, env: Env) -> Self {
		Self { config, env, vars: Default::default() }
	}

	pub fn assign_var(&mut self, name: &str, value: Value) {
		self.vars.insert(name.into(), value);
	}

	pub fn get_var(&mut self, name: &str) -> Option<Value> {
		self.vars.get(name).cloned()
	}

	pub fn config(&self) -> &Config {
		&self.config
	}

	pub fn env(&self) -> &Env {
		&self.env
	}

	pub fn run_file(&mut self, path: &Path) -> PlayResult<()> {
		self.play_expr(&std::fs::read_to_string(path)?)
	}

	pub fn play_expr(&mut self, source: &str) -> PlayResult<()> {
		let mut lctx = crate::parse::LexContext::new(source, self);
		let expr = Expression::parse_toplevel(&mut lctx)?;
		self.play(&expr)
	}

	fn _play<T: AsRef<Path> + ?Sized>(&mut self, expr: &Expression, start: &T) -> PlayResult<usize> {
		let mut num_matches = 0;

		for entry in std::fs::read_dir(start.as_ref())? {
			num_matches += self.handle(entry?.path(), expr)?;
		}

		Ok(num_matches)
	}

	fn handle(&mut self, name: PathBuf, expr: &Expression) -> PlayResult<usize> {
		let mut ctx = PlayContext::new(self, name)?;
		let matched = expr.run(&mut ctx, RunContext::Logical).map_or(false, |x| x.is_truthy());
		let pathinfo = ctx.into_pathinfo();

		// Invert `matched` if given the `!` flag.
		let matched = if self.config.is_inverted() { !matched } else { matched };

		let mut num_matches = 0;
		if matched {
			num_matches += 1;
		}

		if matched && self.config.should_print() {
			let mut stdout = std::io::stdout().lock();
			stdout.write_all(&pathinfo.path().as_os_str().to_raw_bytes())?;
			self.config.write_line_ending(stdout)?;
		}

		if pathinfo.is_dir() {
			// ensure we take it so the rest of the `pathinfo` struct can be dropped
			let path = pathinfo.path().clone();
			drop(pathinfo);
			match self._play(expr, &path) {
				Ok(match_count) => num_matches += match_count,
				Err(err) => self.config.handle_error(err)?,
			}
		}

		Ok(num_matches)
	}

	pub fn play(&mut self, expr: &Expression) -> PlayResult<()> {
		let start_positions = match expr.begin_position() {
			x if x.is_empty() => vec![".".into()],
			x => x,
		};

		let mut num_matches = 0;
		for start in start_positions {
			num_matches += self.handle(start.clone(), expr)?;
			num_matches += self._play(expr, &start)?;
		}

		if self.config.is_counting() {
			println!("{num_matches}");
		}

		Ok(())
	}
}