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

use crate::error::Result;
use crate::language_code::LanguageCode;
use url::Url;

static DUMP_INDEX_URL: &str = "https://dumps.wikimedia.org/backup-index.html";
static DUMP_BASE_URL: &str = "https://ftp.acc.umu.se/mirror/wikimedia.org/dumps";

pub fn dump_status_file(
    base_url: &DumpBaseUrl,
    language_code: &LanguageCode,
    date: &str,
) -> Result<Url> {
    let base_url = base_url.as_str();
    let language_abbreviation = language_code.to_wiktionary_abbreviation();
    Ok(Url::parse(&format!(
        "{base_url}/{language_abbreviation}wiktionary/{date}/dumpstatus.json"
    ))?)
}

pub fn available_dates(base_url: &DumpBaseUrl, language_code: &LanguageCode) -> Result<Url> {
    let base_url = base_url.as_str();
    let language_abbreviation = language_code.to_wiktionary_abbreviation();
    Ok(Url::parse(&format!(
        "{base_url}/{language_abbreviation}wiktionary/"
    ))?)
}

pub fn dump_url(base_url: &DumpBaseUrl, suffix: &str) -> Result<Url> {
    let base_url = base_url.as_str();
    Ok(Url::parse(&format!("{base_url}{suffix}"))?)
}

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum DumpIndexUrl {
    Default,
    Custom(String),
}

impl DumpIndexUrl {
    pub fn as_str(&self) -> &str {
        match self {
            DumpIndexUrl::Default => DUMP_INDEX_URL,
            DumpIndexUrl::Custom(custom) => custom,
        }
    }
}

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum DumpBaseUrl {
    Default,
    Custom(String),
}

impl DumpBaseUrl {
    pub fn as_str(&self) -> &str {
        match self {
            DumpBaseUrl::Default => DUMP_BASE_URL,
            DumpBaseUrl::Custom(custom) => custom,
        }
    }
}