ik-mini-epub-cli 0.2.0-alpha.2

A CLI tool to download IK Storeis into epubs.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

mod cli;

use crate::cli::{Commands, DoArgs, WPEpubCli};
use anyhow::Context;
use anyhow::Result;
use clap::Parser;
use ik_mini::InkittClient;
use ik_mini_epub::{download_story_to_folder, login};
use reqwest::Client;
use tracing::info;
use tracing_subscriber::EnvFilter;

// --- Application Entry Point ---
#[tokio::main]
async fn main() -> Result<()> {
    // 1. Initialize the tracing subscriber with EnvFilter.
    // This will default to showing INFO logs for your crate (`ik_epub_cli`)
    // and WARN for all other crates.
    let filter = EnvFilter::try_from_default_env()
        .unwrap_or_else(|_| EnvFilter::new("ik_epub_cli=info,warn"));

    tracing_subscriber::fmt().with_env_filter(filter).init();

    // 2. Create the reqwest client ONCE and pass it down. No more globals!
    let http_client = Client::builder()
        .cookie_store(true)
        .user_agent("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/140.0.0.0 Safari/537.36")
        .build()?;

    // 3. Parse CLI arguments
    let cli = WPEpubCli::parse();
    info!("CLI arguments parsed successfully");

    // 4. Match the subcommand and delegate to handler functions
    match cli.command {
        Commands::Do(args) => handle_do_command(&http_client, args)
            .await
            .context("Failed to process story")?,
    }

    Ok(())
}

async fn handle_do_command(client: &Client, args: DoArgs) -> Result<()> {
    info!(id = args.id, "Handling 'do' command");

    info!(user = %args.email, "Authenticating & Creating InkittClient...");
    let ik_client = InkittClient::new();
    login(&ik_client, &args.email, &args.password)
        .await
        .context("Login failed. Check your credentials.")?;
    info!("Login successful. Starting download...");

    let output_dir = args
        .output_path
        .unwrap_or_else(|| std::env::current_dir().expect("Failed to get current directory"));

    let story_response = download_story_to_folder(
        &ik_client,
        client,
        args.id,
        args.include_images,
        args.semaphore as usize,
        &output_dir,
    )
    .await?;

    let story_title = story_response.metadata.title;

    info!(title = %story_title, "Story processing completed successfully");

    Ok(())
}