nicer formatting, moved Findings into its own file

This commit is contained in:
Lynne Megido 2021-02-15 03:12:27 +10:00
parent 3642f0112a
commit d7eb0de299
Signed by: lynnesbian
GPG Key ID: F0A184B5213D9F90
8 changed files with 48 additions and 39 deletions

1
Cargo.lock generated
View File

@ -176,6 +176,7 @@ dependencies = [
"env_logger",
"log",
"mime_guess",
"once_cell",
"rayon",
"smartstring",
"snailquote",

View File

@ -23,6 +23,7 @@ xdg-mime = {git = "https://github.com/ebassi/xdg-mime-rs", version = "0.3", rev
mime_guess = "2.0.3"
rayon = "1.5.0"
snailquote = "0.3.0"
once_cell = "1.5.2"
[dependencies.clap]
version = "3.0.0-beta.2"

17
src/findings.rs Normal file
View File

@ -0,0 +1,17 @@
use std::path::PathBuf;
use crate::inspectors::mime_extension_lookup;
use mime_guess::Mime;
use smartstring::alias::String;
pub struct Findings {
pub file: PathBuf, // TODO: replace with Path???? <'a> and all that
pub valid: bool,
pub mime: Mime,
}
impl Findings {
pub fn recommended_extension(&self) -> Option<String> {
mime_extension_lookup(self.mime.clone())
.map(|extensions| extensions[0].to_owned())
}
}

View File

@ -51,9 +51,8 @@ pub trait Format {
self.rename(
f,
&finding.file,
&finding.file.with_extension(ext.as_str())
&finding.file.with_extension(ext.as_str()),
)?
} else {
self.no_known_extension(f, &finding.file)?
}
@ -76,6 +75,7 @@ pub trait Format {
}
pub struct Script {}
impl Format for Script {
fn new() -> Self { Self {} }

View File

@ -20,7 +20,6 @@ use std::path::{Path, PathBuf};
use clap::Clap;
use log::{debug, info, trace, warn};
use mime_guess::Mime;
use rayon::prelude::*;
use smartstring::alias::String;
use walkdir::{DirEntry, WalkDir};
@ -29,35 +28,13 @@ use xdg_mime::SharedMimeInfo;
use crate::formats::{Format, Script};
use crate::parameters::OutputFormat;
use crate::scanerror::ScanError;
use crate::findings::Findings;
mod parameters;
mod inspectors;
mod formats;
mod scanerror;
pub struct Findings {
file: PathBuf, // TODO: replace with Path???? <'a> and all that
valid: bool,
mime: Mime,
}
impl Findings {
fn recommended_extension(&self) -> Option<String> {
inspectors::mime_extension_lookup(self.mime.clone())
.map(|extensions| extensions[0].to_owned())
}
}
impl Display for ScanError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}",
match self {
Self::File => "Couldn't read file",
Self::Mime => "Couldn't determine mime type"
}
)
}
}
mod findings;
// TODO: test if this actually works on a windows machine - not there's much of a point right now, considering
// xdg-mime-rs doesn't support windows
@ -67,7 +44,7 @@ fn is_hidden(entry: &DirEntry) -> bool {
std::fs::metadata(entry) // try to get metadata for file
.map_or(
false, // if getting metadata/attributes fails, assume it's not hidden
|f| f.file_attributes() & 0x2 > 0 // flag for hidden - https://docs.microsoft.com/windows/win32/fileio/file-attribute-constants
|f| f.file_attributes() & 0x2 > 0, // flag for hidden - https://docs.microsoft.com/windows/win32/fileio/file-attribute-constants
)
}
@ -89,11 +66,11 @@ fn wanted_file(args: &parameters::Parameters, entry: &DirEntry) -> bool {
let ext = extension_from_path(entry.path());
if ext.is_none() { return false } // don't scan files without extensions. TODO - this should be configurable
if ext.is_none() { return false; } // don't scan files without extensions. TODO - this should be configurable
if let Some(extensions) = &args.extensions {
// if the user has specified a list of extensions to check against, make sure this file ends in one of them.
return extensions.contains(&ext.unwrap().to_lowercase().into())
return extensions.contains(&ext.unwrap().to_lowercase().into());
}
true
}
@ -207,7 +184,7 @@ fn main() {
OutputFormat::Script => {
let s = Script::new();
s.write_all(&results, &mut BufWriter::new(stdout().lock())).expect("failed to output");
},
}
OutputFormat::Text => debug!("eewr")
}

View File

@ -6,7 +6,7 @@ use smartstring::{LazyCompact, SmartString};
#[derive(Clap, PartialEq, Debug)]
pub enum OutputFormat {
Script,
Text
Text,
}
#[derive(Clap, Debug)]

View File

@ -1,4 +1,17 @@
use std::fmt::{Display, Formatter, Result};
pub enum ScanError {
File,
Mime
}
impl Display for ScanError {
fn fmt(&self, f: &mut Formatter<'_>) -> Result {
write!(f, "{}",
match self {
Self::File => "Couldn't read file",
Self::Mime => "Couldn't determine mime type"
}
)
}
}