learnerd 0.2.1

The way to interact and learn stuff
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

//! Module for abstracting the "init" functionality to the [`learner`] database.

use super::*;

/// Function for the [`Commands::Init`] in the CLI.
pub async fn init(cli: Cli) -> Result<()> {
  let db_path = cli.path.unwrap_or_else(|| {
    let default_path = Database::default_path();
    println!(
      "{} Using default database path: {}",
      style(INFO_PREFIX).cyan(),
      style(default_path.display()).yellow()
    );
    default_path
  });

  if db_path.exists() {
    println!(
      "{} Database already exists at: {}",
      style(ERROR_PREFIX).yellow(),
      style(db_path.display()).yellow()
    );

    // Handle reinitialize confirmation
    let should_reinit = if cli.accept_defaults {
      false // Default to not reinitializing in automated mode
    } else {
      dialoguer::Confirm::new()
        .with_prompt("Do you want to reinitialize this database? This will erase all existing data")
        .default(false)
        .interact()?
    };

    if !should_reinit {
      println!("{} Keeping existing database", style("ℹ").blue());
      return Ok(());
    }

    // Handle INIT confirmation
    let should_proceed = if cli.accept_defaults {
      false // Default to not proceeding in automated mode
    } else {
      let input = dialoguer::Input::<String>::new()
        .with_prompt(format!(
          "{} Type {} to confirm reinitialization",
          style(WARNING_PREFIX).red(),
          style("INIT").red().bold()
        ))
        .interact_text()?;
      input == "INIT"
    };

    if !should_proceed {
      println!("{} Operation cancelled, keeping existing database", style("ℹ").blue());
      return Ok(());
    }

    // Remove existing database
    println!("{} Removing existing database", style(WARNING_PREFIX).yellow());
    std::fs::remove_file(&db_path)?;

    // Also remove any FTS auxiliary files
    let fts_files = glob::glob(&format!("{}*", db_path.display()))?;
    for file in fts_files.flatten() {
      std::fs::remove_file(file)?;
    }
  }

  // Create parent directories if they don't exist
  if let Some(parent) = db_path.parent() {
    trace!("Creating parent directories: {}", parent.display());
    std::fs::create_dir_all(parent)?;
  }

  println!(
    "{} Initializing database at: {}",
    style(INFO_PREFIX).cyan(),
    style(db_path.display()).yellow()
  );

  let db = Database::open(&db_path).await?;

  // Set up PDF directory
  let pdf_dir = Database::default_storage_path();
  println!(
    "\n{} PDF files will be stored in: {}",
    style(INFO_PREFIX).cyan(),
    style(pdf_dir.display()).yellow()
  );

  // TODO (autoparallel): I think we need this `allow` because though the returns are the same,
  // the initial `if` bypasses interaction
  #[allow(clippy::if_same_then_else)]
  let pdf_dir = if cli.accept_defaults {
    pdf_dir // Use default in automated mode
  } else if dialoguer::Confirm::new()
    .with_prompt("Use this location for PDF storage?")
    .default(true)
    .interact()?
  {
    pdf_dir
  } else {
    let input: String =
      dialoguer::Input::new().with_prompt("Enter path for PDF storage").interact_text()?;
    PathBuf::from_str(&input).unwrap() // TODO (autoparallel): fix this unwrap
  };

  std::fs::create_dir_all(&pdf_dir)?;
  db.set_storage_path(&pdf_dir.to_string_lossy().to_string()).await?;

  println!("{} Database initialized successfully!", style(SUCCESS_PREFIX).green());
  Ok(())
}