learnerd 0.3.5

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

//! Module for setting up a [`learner`] environment

use super::*;

/// Arguments that can be used for the [`Commands::Add`]
#[derive(Args, Clone)]
pub struct InitArgs {
  /// Path to use to store [`Database`].
  /// Defaults to [`Database::default_path`].
  #[arg(long)]
  pub db_path: Option<PathBuf>,

  /// Path to use to store documents.
  /// Defaults to [`Database::default_storage_path`].
  #[arg(long)]
  pub storage_path: Option<PathBuf>,

  /// Whether to use the default set of retrievier configurations.
  /// Defaults to `false`.
  #[arg(long)]
  pub no_default_retrievers: bool,
}

/// Function for the [`Commands::Init`] in the CLI.
pub async fn init<I: UserInteraction>(interaction: &mut I, init_args: InitArgs) -> Result<()> {
  let InitArgs { db_path, storage_path, no_default_retrievers } = init_args;
  // Throughout, assume we are using default config path (`~/.learner`)

  // Set database storage location
  let config = if let Some(db_path) = db_path {
    Config::default().with_database_path(&db_path)
  } else if !interaction.confirm(&format!(
    "Would you like to use the default path {:?} for storing the Learner database?",
    Database::default_path(),
  ))? {
    interaction.reply(ResponseContent::Info(
      "Please pass in your intended database storage path using --db-path",
    ))?;
    return Ok(());
  } else {
    Config::default()
  };

  if config.database_path.exists()
    && !interaction.confirm(
      "Database already exists at this location, do you want to overwrite this database?",
    )?
  {
    interaction.reply(ResponseContent::Info(
      "Please choose a different location for this new Learner database using --db-path",
    ))?;
    return Ok(());
  }

  // Set document storage location
  let config = if let Some(storage_path) = storage_path {
    config.with_storage_path(&storage_path)
  } else if !interaction.confirm(&format!(
    "Would you like to use the default path {:?} for storing documents?",
    Database::default_storage_path(),
  ))? {
    interaction.reply(ResponseContent::Info(
      "Please pass in your intended database storage path using --storage-path",
    ))?;
    return Ok(());
  } else {
    config
  };

  // Create learner with this configuration and with the default retrievers (arXiv, DOI, IACR)
  if !no_default_retrievers {
    interaction
      .reply(ResponseContent::Info("Using the default set of retrievers (arXiv, DOI, and DOI)."))?;
    std::fs::create_dir_all(Config::default_path()?.join("retrievers"))?;
    std::fs::write(config.retrievers_path.join("arxiv.toml"), learner::ARXIV_CONFIG)?;
    std::fs::write(config.retrievers_path.join("doi.toml"), learner::DOI_CONFIG)?;
    std::fs::write(config.retrievers_path.join("iacr.toml"), learner::IACR_CONFIG)?;
  }
  Learner::builder().with_config(config.clone()).build().await?;
  std::fs::write(Config::default_path()?.join("config.toml"), toml::to_string(&config)?)?;
  interaction.reply(ResponseContent::Success(&format!(
    "Created Learner configuration with\nConfig path: {:?}\nDatabase path: {:?}\nDocument storage \
     path: {:?}",
    Config::default_path()?,
    config.database_path,
    config.storage_path,
  )))?;
  Ok(())
}