termchan 0.2.0

ターミナルで動く軽量2chクライアント
Documentation 
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

use anyhow::Context;
use reqwest::header::{HeaderName, COOKIE, HOST};

use super::encoder;

pub struct Reciever {
    url: String,
    html: String,
}

impl Reciever {
    pub async fn get(url: &str) -> anyhow::Result<Reciever> {
        // URLからホスト名を取得
        let url = url.to_owned();
        let host = url::Url::parse(&url)
            .context("url parse error")?
            .host_str()
            .context("host parse error")?
            .to_string();

        let cookie = vec![("READJS", "off"), ("SUBBACK_STYLE", "1")];
        let cookie = encoder::cookie_from_vec(cookie);

        let mut headers: Vec<(HeaderName, String)> = encoder::base_headers();
        headers.append(&mut vec![(HOST, host.to_string()), (COOKIE, cookie)]);
        let headers = encoder::headers_from_vec(headers)?;

        let client = reqwest::Client::builder()
            .default_headers(headers)
            .build()
            .context("failed to create client")?
            .get(&url);

        // バイナリで取得したHTMLをUTF-8に変換
        let res = client.send().await.context("failed to get html")?;
        let bytes = res.bytes().await.context("failed to get bytes")?;
        let html = encoder::sjis_to_utf8(&bytes)?;
        Ok(Self {
            url: url.to_string(),
            html,
        })
    }

    pub fn url(&self) -> String {
        self.url.clone()
    }

    pub fn html(&self) -> String {
        self.html.clone()
    }
}