mirror of
https://github.com/YGGverse/pulsarss.git
synced 2026-03-31 17:15:29 +00:00
add custom index filename support
This commit is contained in:
parent
86b5d41aec
commit
dbd9fa3960
3 changed files with 11 additions and 11 deletions
|
|
@ -3,9 +3,9 @@ use clap::Parser;
|
|||
#[derive(Parser, Debug)]
|
||||
#[command(version, about, long_about = None)]
|
||||
pub struct Argument {
|
||||
/// Generate `index.gmi` files (`false` by default)
|
||||
#[arg(short, long, default_value_t = false)]
|
||||
pub index: bool,
|
||||
/// Generate index files (`index.gmi` by default, empty to disable)
|
||||
#[arg(short, long, default_value_t = String::from("index.gmi"))]
|
||||
pub index: String,
|
||||
|
||||
/// Limit channel items (unlimited by default)
|
||||
#[arg(short, long, default_value_t = 0)]
|
||||
|
|
|
|||
10
src/main.rs
10
src/main.rs
|
|
@ -94,14 +94,14 @@ fn crawl(argument: &Argument, output: &Output) -> Result<(), Box<dyn Error>> {
|
|||
|
||||
File::create(destination.item())?.write_all(data.join("\n\n").as_bytes())?;
|
||||
|
||||
if argument.index {
|
||||
index.insert(destination.path); // request `index.gmi` update
|
||||
if !argument.index.is_empty() {
|
||||
index.insert(destination.path); // request index file update
|
||||
}
|
||||
}
|
||||
|
||||
// renew pending `index.gmi` files on items crawl completed
|
||||
// renew pending index files on items crawl completed
|
||||
for path in index {
|
||||
let subject = format!("{path}index.gmi");
|
||||
let subject = format!("{path}{}", argument.index);
|
||||
|
||||
let mut index = File::create(&subject)?;
|
||||
let mut data = Vec::with_capacity(argument.limit);
|
||||
|
|
@ -110,7 +110,7 @@ fn crawl(argument: &Argument, output: &Output) -> Result<(), Box<dyn Error>> {
|
|||
for file in read_dir(&path)? {
|
||||
let name = file?.file_name().into_string().unwrap();
|
||||
|
||||
if name == "index.gmi" {
|
||||
if name == argument.index {
|
||||
continue;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue