...
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
use clap::{App, Arg, SubCommand};
|
||||
use doctree::{DocTree, RedisStorage, Result};
|
||||
use doctree::{DocTree, RedisStorage, Result, from_directory};
|
||||
use std::path::Path;
|
||||
|
||||
fn main() -> Result<()> {
|
||||
@@ -17,11 +17,41 @@ fn main() -> Result<()> {
|
||||
SubCommand::with_name("list")
|
||||
.about("List collections"),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("scan-collections")
|
||||
.about("Recursively scan directories for .collection files")
|
||||
.arg(Arg::with_name("path").required(true).help("Root path to scan for collections")),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("scan-and-info")
|
||||
.about("Scan collections and show detailed information")
|
||||
.arg(Arg::with_name("path").required(true).help("Root path to scan for collections"))
|
||||
.arg(Arg::with_name("collection").help("Name of the collection (optional)")),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("info")
|
||||
.about("Show detailed information about collections")
|
||||
.arg(Arg::with_name("collection").help("Name of the collection (optional)")),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("get")
|
||||
.about("Get page content")
|
||||
.arg(Arg::with_name("collection").required(true).help("Name of the collection"))
|
||||
.arg(Arg::with_name("page").required(true).help("Name of the page")),
|
||||
.arg(Arg::with_name("collection")
|
||||
.short("c".chars().next().unwrap())
|
||||
.long("collection")
|
||||
.takes_value(true)
|
||||
.help("Name of the collection (optional)"))
|
||||
.arg(Arg::with_name("page")
|
||||
.short("p".chars().next().unwrap())
|
||||
.long("page")
|
||||
.required(true)
|
||||
.takes_value(true)
|
||||
.help("Name of the page"))
|
||||
.arg(Arg::with_name("format")
|
||||
.short("f".chars().next().unwrap())
|
||||
.long("format")
|
||||
.takes_value(true)
|
||||
.help("Output format (html or markdown, default: markdown)")),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("html")
|
||||
@@ -29,6 +59,15 @@ fn main() -> Result<()> {
|
||||
.arg(Arg::with_name("collection").required(true).help("Name of the collection"))
|
||||
.arg(Arg::with_name("page").required(true).help("Name of the page")),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("delete-collection")
|
||||
.about("Delete a collection from Redis")
|
||||
.arg(Arg::with_name("collection").required(true).help("Name of the collection")),
|
||||
)
|
||||
.subcommand(
|
||||
SubCommand::with_name("reset")
|
||||
.about("Delete all collections from Redis"),
|
||||
)
|
||||
.get_matches();
|
||||
|
||||
// Create a Redis storage instance
|
||||
@@ -59,17 +98,257 @@ fn main() -> Result<()> {
|
||||
}
|
||||
}
|
||||
} else if let Some(matches) = matches.subcommand_matches("get") {
|
||||
let collection = matches.value_of("collection").unwrap();
|
||||
let collection = matches.value_of("collection");
|
||||
let page = matches.value_of("page").unwrap();
|
||||
let format = matches.value_of("format").unwrap_or("markdown");
|
||||
|
||||
let content = doctree.page_get(Some(collection), page)?;
|
||||
println!("{}", content);
|
||||
if format.to_lowercase() == "html" {
|
||||
let html = doctree.page_get_html(collection, page)?;
|
||||
println!("{}", html);
|
||||
} else {
|
||||
let content = doctree.page_get(collection, page)?;
|
||||
println!("{}", content);
|
||||
}
|
||||
} else if let Some(matches) = matches.subcommand_matches("html") {
|
||||
let collection = matches.value_of("collection").unwrap();
|
||||
let page = matches.value_of("page").unwrap();
|
||||
|
||||
let html = doctree.page_get_html(Some(collection), page)?;
|
||||
println!("{}", html);
|
||||
} else if let Some(matches) = matches.subcommand_matches("delete-collection") {
|
||||
let collection = matches.value_of("collection").unwrap();
|
||||
|
||||
println!("Deleting collection '{}' from Redis...", collection);
|
||||
doctree.delete_collection(collection)?;
|
||||
println!("Collection '{}' deleted successfully", collection);
|
||||
} else if let Some(_) = matches.subcommand_matches("reset") {
|
||||
println!("Deleting all collections from Redis...");
|
||||
doctree.delete_all_collections()?;
|
||||
println!("All collections deleted successfully");
|
||||
} else if let Some(matches) = matches.subcommand_matches("scan-collections") {
|
||||
let path = matches.value_of("path").unwrap();
|
||||
|
||||
println!("Recursively scanning for collections in: {}", path);
|
||||
|
||||
// Use the from_directory function to create a DocTree with all collections
|
||||
let doctree = from_directory(Path::new(path))?;
|
||||
|
||||
// Print the discovered collections
|
||||
let collections = doctree.list_collections();
|
||||
if collections.is_empty() {
|
||||
println!("No collections found");
|
||||
} else {
|
||||
println!("Discovered collections:");
|
||||
for collection in collections {
|
||||
println!("- {}", collection);
|
||||
}
|
||||
}
|
||||
} else if let Some(matches) = matches.subcommand_matches("scan-and-info") {
|
||||
let path = matches.value_of("path").unwrap();
|
||||
let collection_name = matches.value_of("collection");
|
||||
|
||||
println!("Recursively scanning for collections in: {}", path);
|
||||
|
||||
// Use the from_directory function to create a DocTree with all collections
|
||||
let doctree = from_directory(Path::new(path))?;
|
||||
|
||||
// Print the discovered collections
|
||||
let collections = doctree.list_collections();
|
||||
if collections.is_empty() {
|
||||
println!("No collections found");
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
println!("Discovered collections:");
|
||||
for collection in &collections {
|
||||
println!("- {}", collection);
|
||||
}
|
||||
|
||||
println!("\nDetailed Collection Information:");
|
||||
|
||||
if let Some(name) = collection_name {
|
||||
// Show info for a specific collection
|
||||
match doctree.get_collection(name) {
|
||||
Ok(collection) => {
|
||||
println!("Collection Information for '{}':", name);
|
||||
println!(" Path: {:?}", collection.path);
|
||||
println!(" Redis Key: collections:{}", collection.name);
|
||||
|
||||
// List documents
|
||||
match collection.page_list() {
|
||||
Ok(pages) => {
|
||||
println!(" Documents ({}):", pages.len());
|
||||
for page in pages {
|
||||
match collection.page_get_path(&page) {
|
||||
Ok(path) => {
|
||||
println!(" - {} => Redis: collections:{} / {}", path, collection.name, page);
|
||||
},
|
||||
Err(_) => {
|
||||
println!(" - {}", page);
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
Err(e) => println!(" Error listing documents: {}", e),
|
||||
}
|
||||
|
||||
// List files
|
||||
match collection.file_list() {
|
||||
Ok(files) => {
|
||||
// Filter images
|
||||
let images: Vec<String> = files.iter()
|
||||
.filter(|f|
|
||||
f.ends_with(".png") || f.ends_with(".jpg") ||
|
||||
f.ends_with(".jpeg") || f.ends_with(".gif") ||
|
||||
f.ends_with(".svg"))
|
||||
.cloned()
|
||||
.collect();
|
||||
|
||||
println!(" Images ({}):", images.len());
|
||||
for image in images {
|
||||
println!(" - {} => Redis: collections:{} / {}", image, collection.name, image);
|
||||
}
|
||||
|
||||
// Filter other files
|
||||
let other_files: Vec<String> = files.iter()
|
||||
.filter(|f|
|
||||
!f.ends_with(".png") && !f.ends_with(".jpg") &&
|
||||
!f.ends_with(".jpeg") && !f.ends_with(".gif") &&
|
||||
!f.ends_with(".svg"))
|
||||
.cloned()
|
||||
.collect();
|
||||
|
||||
println!(" Other Files ({}):", other_files.len());
|
||||
for file in other_files {
|
||||
println!(" - {} => Redis: collections:{} / {}", file, collection.name, file);
|
||||
}
|
||||
},
|
||||
Err(e) => println!(" Error listing files: {}", e),
|
||||
}
|
||||
},
|
||||
Err(e) => println!("Error: {}", e),
|
||||
}
|
||||
} else {
|
||||
// Show info for all collections
|
||||
for name in collections {
|
||||
if let Ok(collection) = doctree.get_collection(&name) {
|
||||
println!("- {} (Redis Key: collections:{})", name, collection.name);
|
||||
println!(" Path: {:?}", collection.path);
|
||||
|
||||
// Count documents and images
|
||||
if let Ok(pages) = collection.page_list() {
|
||||
println!(" Documents: {}", pages.len());
|
||||
}
|
||||
|
||||
if let Ok(files) = collection.file_list() {
|
||||
let image_count = files.iter()
|
||||
.filter(|f|
|
||||
f.ends_with(".png") || f.ends_with(".jpg") ||
|
||||
f.ends_with(".jpeg") || f.ends_with(".gif") ||
|
||||
f.ends_with(".svg"))
|
||||
.count();
|
||||
println!(" Images: {}", image_count);
|
||||
println!(" Other Files: {}", files.len() - image_count);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} else if let Some(matches) = matches.subcommand_matches("info") {
|
||||
let collection_name = matches.value_of("collection");
|
||||
|
||||
if let Some(name) = collection_name {
|
||||
// Show info for a specific collection
|
||||
match doctree.get_collection(name) {
|
||||
Ok(collection) => {
|
||||
println!("Collection Information for '{}':", name);
|
||||
println!(" Path: {:?}", collection.path);
|
||||
println!(" Redis Key: collections:{}", collection.name);
|
||||
|
||||
// List documents
|
||||
match collection.page_list() {
|
||||
Ok(pages) => {
|
||||
println!(" Documents ({}):", pages.len());
|
||||
for page in pages {
|
||||
match collection.page_get_path(&page) {
|
||||
Ok(path) => {
|
||||
println!(" - {} => Redis: collections:{} / {}", path, collection.name, page);
|
||||
},
|
||||
Err(_) => {
|
||||
println!(" - {}", page);
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
Err(e) => println!(" Error listing documents: {}", e),
|
||||
}
|
||||
|
||||
// List files
|
||||
match collection.file_list() {
|
||||
Ok(files) => {
|
||||
// Filter images
|
||||
let images: Vec<String> = files.iter()
|
||||
.filter(|f|
|
||||
f.ends_with(".png") || f.ends_with(".jpg") ||
|
||||
f.ends_with(".jpeg") || f.ends_with(".gif") ||
|
||||
f.ends_with(".svg"))
|
||||
.cloned()
|
||||
.collect();
|
||||
|
||||
println!(" Images ({}):", images.len());
|
||||
for image in images {
|
||||
println!(" - {} => Redis: collections:{} / {}", image, collection.name, image);
|
||||
}
|
||||
|
||||
// Filter other files
|
||||
let other_files: Vec<String> = files.iter()
|
||||
.filter(|f|
|
||||
!f.ends_with(".png") && !f.ends_with(".jpg") &&
|
||||
!f.ends_with(".jpeg") && !f.ends_with(".gif") &&
|
||||
!f.ends_with(".svg"))
|
||||
.cloned()
|
||||
.collect();
|
||||
|
||||
println!(" Other Files ({}):", other_files.len());
|
||||
for file in other_files {
|
||||
println!(" - {} => Redis: collections:{} / {}", file, collection.name, file);
|
||||
}
|
||||
},
|
||||
Err(e) => println!(" Error listing files: {}", e),
|
||||
}
|
||||
},
|
||||
Err(e) => println!("Error: {}", e),
|
||||
}
|
||||
} else {
|
||||
// Show info for all collections
|
||||
let collections = doctree.list_collections();
|
||||
if collections.is_empty() {
|
||||
println!("No collections found");
|
||||
} else {
|
||||
println!("Collections:");
|
||||
for name in collections {
|
||||
if let Ok(collection) = doctree.get_collection(&name) {
|
||||
println!("- {} (Redis Key: collections:{})", name, collection.name);
|
||||
println!(" Path: {:?}", collection.path);
|
||||
|
||||
// Count documents and images
|
||||
if let Ok(pages) = collection.page_list() {
|
||||
println!(" Documents: {}", pages.len());
|
||||
}
|
||||
|
||||
if let Ok(files) = collection.file_list() {
|
||||
let image_count = files.iter()
|
||||
.filter(|f|
|
||||
f.ends_with(".png") || f.ends_with(".jpg") ||
|
||||
f.ends_with(".jpeg") || f.ends_with(".gif") ||
|
||||
f.ends_with(".svg"))
|
||||
.count();
|
||||
println!(" Images: {}", image_count);
|
||||
println!(" Other Files: {}", files.len() - image_count);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
println!("No command specified. Use --help for usage information.");
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user