phonerr 0.1.2

A CLI to query mobile phone region.
Documentation 
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
79
80
81
82
83
84

use std::{process::exit, time};

use clap::{CommandFactory, Parser};
use colored::Colorize;
use phonerr::{util::wait_blink, PhoneData};

#[derive(Parser)]
#[command(version, about, long_about = None)]
#[command(propagate_version = true)]
struct Cli {
    /// 更新号码库
    #[clap(long, conflicts_with_all = vec!["info"])]
    update: bool,

    /// 更新号码库链接，默认 https://raw.githubusercontent.com/ls0f/phone/master/phone/phone.dat
    #[clap(long)]
    update_url: Option<String>,

    /// 查看号码库信息
    #[clap(long, conflicts_with_all = vec!["update", "update_url"])]
    info: bool,

    /// 不显示elapsed time
    #[clap(long)]
    no_elapsed_time: bool,

    /// 手机号码
    phone: Option<String>,
}

#[tokio::main]
async fn main() -> Result<(), anyhow::Error> {
    let start = time::Instant::now();
    let cli = Cli::parse();
    if cli.phone.is_some() && cli.phone.clone().unwrap().replace(" ", "").len() != 11 {
        eprintln!("{}", "手机号码格式错误".red());
        exit(1);
    }
    let mut client = PhoneData::new(None);
    if cli.update {
        #[cfg(not(feature = "download-progress"))]
        let wait = wait_blink("下载中，请稍候⏬...", 3);
        client.download_file(cli.update_url, false).await?;
        #[cfg(not(feature = "download-progress"))]
        {
            wait.sender.send(true).unwrap();
            wait.handle.await?;
            println!(
                "{} {}",
                "下载完成 ✅".green(),
                format!("{}ms elapsed.", start.elapsed().as_millis()).bright_black()
            );
        }
        return Ok(());
    }
    if cli.info {
        client.print_db_info().await?;
        return Ok(());
    }
    if cli.phone.is_none() {
        Cli::command().print_help().unwrap();
        exit(0);
    }
    let wait = wait_blink("查询中，请稍候🔎...", 3);
    let record = client
        .query(&cli.phone.unwrap().replace(" ", ""), true)
        .await;
    wait.sender.send(true).unwrap();
    wait.handle.await?;
    match record {
        Err(e) => {
            eprintln!("{}", e.to_string().red());
            exit(1)
        }
        Ok(data) => data.display(),
    }
    if !cli.no_elapsed_time {
        println!(
            "{}",
            format!("{}ms elapsed.", start.elapsed().as_millis()).bright_black()
        );
    }
    Ok(())
}