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

//! 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

/// Function for the [`Commands::Add`] in the CLI.
pub async fn add<I: UserInteraction>(
  interaction: &I,
  mut learner: Learner,
  identifier: &str,
  pdf: bool,
  no_pdf: bool,
) -> Result<()> {
  let (source, sanitized_identifier) = learner.retriever.sanitize_identifier(identifier)?;
  let papers =
    Query::by_source(&source, &sanitized_identifier).execute(&mut learner.database).await?;

  if papers.is_empty() {
    interaction.reply(ResponseContent::Info(&format!("Fetching paper: {}", identifier)))?;
    let paper = 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 learner.database).await
    } else {
      Add::paper(&paper).execute(&mut learner.database).await
    } {
      Ok(_) => interaction.reply(ResponseContent::Success("Paper added successfully")),
      Err(e) => interaction.reply(ResponseContent::Error(LearnerdError::from(e))),
    }
  } else {
    let paper = &papers[0];
    interaction.reply(ResponseContent::Info("Paper already exists in database"))?;

    let pdf_dir = 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 learner.database).await {
          Ok(_) => interaction.reply(ResponseContent::Success("PDF downloaded successfully")),
          Err(e) => interaction.reply(ResponseContent::Error(LearnerdError::from(e))),
        }
      } else {
        Ok(())
      }
    } else {
      Ok(())
    }
  }
}