healthkit-to-sqlite-cli 0.0.4

Command-line tool to convert Apple HealthKit data to a SQLite database. 
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
85
86
87
88
89
90
91

use clap::Parser;
use console::Term;
use dialoguer::theme::ColorfulTheme;
use dialoguer::Confirm;
use indicatif::{ProgressBar, ProgressDrawTarget, ProgressStyle};
use sqlx::migrate::MigrateDatabase;
use std::path::PathBuf;
use std::time::Duration;

mod core;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Cli {
    #[arg(help = "Path to the HealthKit export.zip data")]
    export_zip: PathBuf,
    #[arg(help = "URL to the SQLite database", env = "DATABASE_URL")]
    db_url: String,
    #[arg(
        help = "Prompts the user to drop the database if it already exists",
        short,
        long
    )]
    drop: bool,
    #[arg(help = "Ignore prompts", short, long)]
    force: bool,
    #[arg(help = "Disable CLI output to stdout", short, long)]
    quiet: bool,
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    dotenv::dotenv().ok();
    env_logger::init();

    let cli = Cli::parse();
    let term = Term::stdout();

    // Abort the program if the database already exists and the user didn't specify the --force flag
    let database_uri = &cli.db_url;
    if sqlx::Sqlite::database_exists(database_uri).await? {
        let drop_prompt = format!("The database at \"{}\" already exists. Do you want to drop it? This will delete all data in the database.", database_uri);
        if cli.drop
            && (cli.force
                || Confirm::with_theme(&ColorfulTheme::default())
                    .with_prompt(drop_prompt)
                    .default(false)
                    .interact()
                    .unwrap())
        {
            if !cli.quiet {
                term.write_line(&format!("Dropping database at \"{}\"...", database_uri))?;
            }
            sqlx::Sqlite::drop_database(database_uri).await?;
        } else {
            term.write_line(&format!(
                "The database at \"{}\" already exists. Please delete it or specify a different database URL.",
                database_uri)
            )?;
            return Ok(());
        }
    }

    let pb = ProgressBar::new_spinner();
    if cli.quiet {
        pb.set_draw_target(ProgressDrawTarget::hidden());
    }
    pb.enable_steady_tick(Duration::from_millis(120));
    pb.set_style(
        ProgressStyle::with_template("[{elapsed_precise}] {spinner:.blue} {msg}")
            .unwrap()
            .tick_strings(&[
                "▹▹▹▹▹",
                "▸▹▹▹▹",
                "▹▸▹▹▹",
                "▹▹▸▹▹",
                "▹▹▹▸▹",
                "▹▹▹▹▸",
                "▪▪▪▪▪",
            ]),
    );
    pb.set_message(format!(
        "Creating SQLite database \"{}\" from \"{}\"...",
        cli.db_url,
        cli.export_zip.display(),
    ));

    core::healthkit_to_sqlite(database_uri, &cli.export_zip).await?;
    pb.finish_with_message(format!("Created SQLite database {}", cli.db_url));
    Ok(())
}