Merge pull request #51 from oscar-project/dev-lib

feat(lib): make oscar-tools available as lib

Cargo.toml

@@ -1,12 +1,23 @@
 [package]
-authors = ["Pedro J. Ortiz <[email protected]>", "Julien Abadji <[email protected]>"]
+authors = [
+    "Pedro J. Ortiz <[email protected]>",
+    "Julien Abadji <[email protected]>",
+]
 edition = "2021"
 name = "oscar-tools"
 version = "0.3.0"
 repository = "https://github.com/oscar-project/oscar-tools"
 description = "Tools for processing OSCAR Corpora"
 license = "Apache-2.0"
 
+[[bin]]
+name = "oscar-tools"
+path = "src/main.rs"
+
+[lib]
+name = "oscar_tools"
+path = "src/lib.rs"
+
 [features]
 zstd = ["dep:zstd"]
 
@@ -21,7 +32,7 @@ rayon = "1.5.1"
 runiq-lib = "1.2.2"
 serde_json = "1.0.78"
 sha2 = "0.10.1"
-zstd = {version="0.11.2", optional=true}
+zstd = { version = "0.11.2", optional = true }
 walkdir = "2.3.3"
 
 [dependencies.clap]
@@ -49,5 +60,9 @@ ci = ["github"]
 # The installers to generate for each app
 installers = []
 # Target platforms to build apps for (Rust target-triple syntax)
-targets = ["x86_64-unknown-linux-gnu", "x86_64-apple-darwin", "x86_64-pc-windows-msvc", "aarch64-apple-darwin"]
-
+targets = [
+    "x86_64-unknown-linux-gnu",
+    "x86_64-apple-darwin",
+    "x86_64-pc-windows-msvc",
+    "aarch64-apple-darwin",
+]

src/lib.rs

@@ -0,0 +1,4 @@
+mod error;
+mod ops;
+
+pub use ops::Checksum;

src/ops/checksum.rs

@@ -6,6 +6,7 @@ use std::{
     path::{Path, PathBuf},
 };
 
+use log::{debug, error, info, warn};
 use rayon::{iter::ParallelIterator, prelude::ParallelBridge};
 use sha2::{Digest, Sha384};
 

src/ops/compress.rs

@@ -6,6 +6,7 @@ use std::{
 };
 
 use flate2::{write::GzEncoder, Compression};
+use log::{debug, error, info, warn};
 use rayon::iter::{IntoParallelIterator, ParallelIterator};
 use walkdir::WalkDir;
 
@@ -79,7 +80,7 @@ pub trait Compress {
 
         Ok(())
     }
-    
+
     /// Recursively compresses files in provided folder.
     /// If `del_src` is set to `true`, removes the compressed files at `src` upon compression completion.
     /// `src` has to exist and be a folder

src/ops/mod.rs

@@ -9,7 +9,7 @@ mod filter_tags;
 mod sampling;
 mod split;
 
-pub(crate) use checksum::Checksum;
+pub use checksum::Checksum;
 pub(crate) use compress::Compress;
 pub(crate) use dedup::Dedup;
 pub(crate) use extract_text::ExtractText;

src/ops/split.rs

@@ -2,6 +2,7 @@
 
  These operations split the corpus into smaller files of a defined max size.
 !*/
+use log::{debug, error, info, warn};
 use std::{
     borrow::Cow,
     fs::File,