watchexec/crates/cli/src/filterer/proglib/file.rs

use std::{
	fs::{metadata, File, FileType, Metadata},
	io::{BufReader, Read},
	iter::once,
	time::{SystemTime, UNIX_EPOCH},
};

use jaq_interpret::{Error, Native, ParseCtx, Val};
use serde_json::{json, Value};
use tracing::{debug, error, trace};

use super::macros::*;

pub fn load(jaq: &mut ParseCtx) {
	trace!("jaq: add file_read filter");
	jaq.insert_native(
		"file_read".into(),
		1,
		Native::new({
			move |args, (ctx, val)| {
				let path = match &val {
					Val::Str(v) => v.to_string(),
					_ => return_err!(Err(Error::str("expected string (path) but got {val:?}"))),
				};

				let bytes = match int_arg!(args, 0, ctx, &val) {
					Ok(v) => v,
					Err(e) => return_err!(Err(e)),
				};

				Box::new(once(Ok(match File::open(&path) {
					Ok(file) => {
						let buf_reader = BufReader::new(file);
						let mut limited = buf_reader.take(bytes);
						let mut buffer = String::with_capacity(bytes as _);
						match limited.read_to_string(&mut buffer) {
							Ok(read) => {
								debug!("jaq: read {read} bytes from {path:?}");
								Val::Str(buffer.into())
							}
							Err(err) => {
								error!("jaq: failed to read from {path:?}: {err:?}");
								Val::Null
							}
						}
					}
					Err(err) => {
						error!("jaq: failed to open file {path:?}: {err:?}");
						Val::Null
					}
				})))
			}
		}),
	);

	trace!("jaq: add file_meta filter");
	jaq.insert_native(
		"file_meta".into(),
		0,
		Native::new({
			move |_, (_, val)| {
				let path = match &val {
					Val::Str(v) => v.to_string(),
					_ => return_err!(Err(Error::str("expected string (path) but got {val:?}"))),
				};

				Box::new(once(Ok(match metadata(&path) {
					Ok(meta) => Val::from(json_meta(meta)),
					Err(err) => {
						error!("jaq: failed to open {path:?}: {err:?}");
						Val::Null
					}
				})))
			}
		}),
	);

	trace!("jaq: add file_size filter");
	jaq.insert_native(
		"file_size".into(),
		0,
		Native::new({
			move |_, (_, val)| {
				let path = match &val {
					Val::Str(v) => v.to_string(),
					_ => return_err!(Err(Error::str("expected string (path) but got {val:?}"))),
				};

				Box::new(once(Ok(match metadata(&path) {
					Ok(meta) => Val::Int(meta.len() as _),
					Err(err) => {
						error!("jaq: failed to open {path:?}: {err:?}");
						Val::Null
					}
				})))
			}
		}),
	);
}

fn json_meta(meta: Metadata) -> Value {
	let perms = meta.permissions();
	let mut val = json!({
		"type": filetype_str(meta.file_type()),
		"size": meta.len(),
		"modified": fs_time(meta.modified()),
		"accessed": fs_time(meta.accessed()),
		"created": fs_time(meta.created()),
		"dir": meta.is_dir(),
		"file": meta.is_file(),
		"symlink": meta.is_symlink(),
		"readonly": perms.readonly(),
	});

	#[cfg(unix)]
	{
		use std::os::unix::fs::PermissionsExt;
		let map = val.as_object_mut().unwrap();
		map.insert(
			"mode".to_string(),
			Value::String(format!("{:o}", perms.mode())),
		);
		map.insert("mode_byte".to_string(), Value::from(perms.mode()));
		map.insert(
			"executable".to_string(),
			Value::Bool(perms.mode() & 0o111 != 0),
		);
	}

	val
}

fn filetype_str(filetype: FileType) -> &'static str {
	#[cfg(unix)]
	{
		use std::os::unix::fs::FileTypeExt;
		if filetype.is_char_device() {
			return "char";
		} else if filetype.is_block_device() {
			return "block";
		} else if filetype.is_fifo() {
			return "fifo";
		} else if filetype.is_socket() {
			return "socket";
		}
	}

	#[cfg(windows)]
	{
		use std::os::windows::fs::FileTypeExt;
		if filetype.is_symlink_dir() {
			return "symdir";
		} else if filetype.is_symlink_file() {
			return "symfile";
		}
	}

	if filetype.is_dir() {
		"dir"
	} else if filetype.is_file() {
		"file"
	} else if filetype.is_symlink() {
		"symlink"
	} else {
		"unknown"
	}
}

fn fs_time(time: std::io::Result<SystemTime>) -> Option<u64> {
	time.ok()
		.and_then(|time| time.duration_since(UNIX_EPOCH).ok())
		.map(|dur| dur.as_secs())
}