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

use crate::{Sauce, SauceResult, SauceError};
use async_trait::async_trait;
use reqwest::Client;
use select::document::Document;
use select::predicate::*;
use serde::{Deserialize, Serialize};

const BASE_ADDRESS: &str = "https://yandex.com/images/search";

/// Gets sauces from yandex.com
#[derive(Debug)]
pub struct Yandex;

#[async_trait]
impl Sauce for Yandex {
    async fn build_url(&self, url: &str) -> Result<String, SauceError> {
        let blocks = r#"{"blocks":[{"block":"b-page_type_search-by-image__link"}]}"#;
        let get_url = format!(
            r#"{}?url={}&rpt=imageview&format=json&request={}"#,
            BASE_ADDRESS, url, blocks
        );

        let cli = Client::new();

        let resp = cli
            .get(&get_url)
            .send()
            .await?;

        let json = resp
            .json::<YandexBuildUrl>()
            .await?;

        Ok(format!("{}?{}", BASE_ADDRESS, json.blocks[0].params.url))
    }

    async fn check_sauce(&self, url: String) -> Result<SauceResult, SauceError> {
        let url = self.build_url(&url).await?;

        let cli = Client::new();

        let resp = cli
            .get(&url)
            .send()
            .await?;

        let html = Document::from(
            resp.text()
                .await?
                .as_str(),
        );

        let _similar = html.find(And(Class("CbirItem"), Class("CbirOtherSizes")));

        todo!("Currently unimplemented")
    }
}

#[derive(Debug, Default, Serialize, Deserialize)]
struct YandexBuildUrl {
    pub blocks: Vec<YandexBlock>,
}

#[derive(Debug, Default, Serialize, Deserialize)]
struct YandexBlock {
    pub params: YandexBlockParams,
}

#[derive(Debug, Default, Serialize, Deserialize)]
struct YandexBlockParams {
    pub url: String,
}