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 abstracting the "add" functionality to the [`learner`] database.

use super::*;

// TODO (autoparallel): This could probably be made even more streamlined if we use the result/error
// type from `learner` more cleverly

/// Arguments that can be used for the [`Commands::Add`]
#[derive(Args, Clone)]
pub struct AddArgs {
  /// Paper identifier (arXiv ID, DOI, IACR ID)
  pub identifier: String,

  /// Force PDF download
  #[arg(long, group = "pdf_behavior")]
  pub pdf: bool,

  /// Skip PDF download
  #[arg(long, group = "pdf_behavior")]
  pub no_pdf: bool,
}

/// Function for the [`Commands::Add`] in the CLI.
pub async fn add<I: UserInteraction>(interaction: &mut I, add_args: AddArgs) -> Result<Paper> {
  let AddArgs { identifier, pdf, no_pdf } = add_args;

  if interaction.learner().retriever.is_empty() {
    return Err(LearnerdError::Learner(LearnerError::Config(
      "No retriever configured.".to_string(),
    )));
  }

  let (source, sanitized_identifier) =
    interaction.learner().retriever.sanitize_identifier(&identifier)?;
  let papers = Query::by_source(&source, &sanitized_identifier)
    .execute(&mut interaction.learner().database)
    .await?;

  if papers.is_empty() {
    interaction.reply(ResponseContent::Info(&format!("Fetching paper: {}", identifier)))?;
    let paper = interaction.learner().retriever.get_paper(&identifier).await?;
    interaction.reply(ResponseContent::Paper(&paper))?;

    let with_pdf = paper.pdf_url.is_some()
      && if pdf {
        true
      } else if no_pdf {
        false
      } else {
        interaction.confirm("Download PDF?")?
      };

    match if with_pdf {
      Add::complete(&paper).execute(&mut interaction.learner().database).await
    } else {
      Add::paper(&paper).execute(&mut interaction.learner().database).await
    } {
      Ok(_) => interaction.reply(ResponseContent::Success("Paper added successfully"))?,
      Err(e) => interaction.reply(ResponseContent::Error(LearnerdError::from(e)))?,
    }
    Ok(paper)
  } else {
    let paper = papers[0].clone();
    interaction.reply(ResponseContent::Info("Paper already exists in database"))?;

    let pdf_dir = interaction.learner().database.get_storage_path().await?;
    let pdf_path = pdf_dir.join(paper.filename());

    if pdf_path.exists() {
      interaction.reply(ResponseContent::Info(&format!("PDF exists at: {}", pdf_path.display())))?
    } else if paper.pdf_url.is_some() {
      let should_download = if pdf {
        true
      } else if no_pdf {
        false
      } else {
        interaction.confirm("PDF not found. Download it now?")?
      };

      if should_download {
        match Add::complete(&paper).execute(&mut interaction.learner().database).await {
          Ok(_) => interaction.reply(ResponseContent::Success("PDF downloaded successfully"))?,
          Err(e) => interaction.reply(ResponseContent::Error(LearnerdError::from(e)))?,
        }
      }
    }
    Ok(paper)
  }
}