ogk 2.2.1

cli tool for open.go.kr
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
74
75
76
77
78

use crate::client;
use crate::utils::date;
use chrono::prelude::*;
use clap::Subcommand;
use std::error::Error;

#[derive(Subcommand)]
#[clap(about = "Fetch query to open.go.kr", author, long_about = None, version)]
pub enum Commands {
    #[clap(about = "Fetch bills with date range")]
    Bills {
        #[clap(long = "from", required = false)]
        from: Option<String>,
        #[clap(long = "page", required = false)]
        page: Option<i32>,
        #[clap(long = "to", required = false)]
        to: Option<String>,
        #[clap(long = "page-size", required = false)]
        page_size: Option<i32>,
        #[clap(long = "org", required = false)]
        org: Option<String>,
    },
}

async fn fetch_bills(
    client: &client::Client,
    page: &i32,
    from_date: &str,
    to_date: &str,
    page_size: &i32,
) -> Result<(), Box<dyn Error>> {
    let response = client
        .fetch_bills(page, from_date, to_date, page_size)
        .await?;

    let pretty_response = serde_json::to_string_pretty(&response)?;
    println!("{}", pretty_response);

    Ok(())
}

pub async fn run(args: &Commands) -> Result<(), Box<dyn Error>> {
    match args {
        Commands::Bills {
            from,
            page,
            page_size,
            to,
            org,
        } => {
            let from_date = match from {
                Some(date) => date.to_owned(),
                None => date::KstDateTime::from(Utc::now()).format(Some("%Y-%m-%d")),
            };
            let to_date = match to {
                Some(td) => td.to_owned(),
                None => date::KstDateTime::from(Utc::now()).format(Some("%Y-%m-%d")),
            };

            let _page = match page {
                Some(p) => p.to_owned(),
                None => 1 as i32,
            };

            let _page_size = match page_size {
                Some(ps) => ps.to_owned(),
                None => 10 as i32,
            };

            let mut client = client::Client::new().await?;
            client.auth_from_storage(org.as_deref()).await?;

            let _result = fetch_bills(&client, &_page, &from_date, &to_date, &_page_size).await;
        }
    }

    Ok(())
}