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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*
This file is part of jellything (https://codeberg.org/metamuffin/jellything)
which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
Copyright (C) 2023 metamuffin <metamuffin.org>
*/
use anyhow::Context;
use jellycommon::chrono::{format::Parsed, DateTime, Utc};
use log::info;
use serde::Deserialize;
use std::io::Write;
#[derive(Debug, Clone, Deserialize)]
pub struct TmdbQuery {
pub page: usize,
pub results: Vec<TmdbQueryResult>,
pub total_pages: usize,
pub total_results: usize,
}
#[derive(Debug, Clone, Deserialize)]
pub struct TmdbQueryResult {
pub adult: bool,
pub backdrop_path: Option<String>,
pub genre_ids: Vec<u64>,
pub id: u64,
pub original_language: Option<String>,
pub original_title: Option<String>,
pub overview: String,
pub popularity: f64,
pub poster_path: Option<String>,
pub release_date: Option<String>,
pub title: Option<String>,
pub name: Option<String>,
pub vote_average: f64,
pub vote_count: usize,
}
#[derive(Debug, Clone, Deserialize)]
pub struct TmdbDetails {
pub adult: bool,
pub backdrop_path: Option<String>,
pub genres: Vec<TmdbGenre>,
pub id: u64,
pub original_language: Option<String>,
pub original_title: Option<String>,
pub overview: String,
pub popularity: f64,
pub poster_path: Option<String>,
pub release_date: Option<String>,
pub title: Option<String>,
pub name: Option<String>,
pub vote_average: f64,
pub vote_count: usize,
pub budget: Option<usize>,
pub homepage: Option<String>,
pub imdb_id: Option<String>,
pub production_companies: Vec<TmdbProductionCompany>,
pub revenue: Option<usize>,
pub tagline: Option<String>,
}
#[derive(Debug, Clone, Deserialize)]
pub struct TmdbGenre {
pub id: u64,
pub name: String,
}
#[derive(Debug, Clone, Deserialize)]
pub struct TmdbProductionCompany {
pub id: u64,
pub name: String,
pub logo_path: Option<String>,
}
pub fn tmdb_search(kind: &str, query: &str, key: &str) -> anyhow::Result<TmdbQuery> {
info!("searching tmdb: {query:?}");
Ok(reqwest::blocking::get(&format!(
"https://api.themoviedb.org/3/search/{kind}?query={}&api_key={key}",
query.replace(" ", "+")
))?
.json::<TmdbQuery>()?)
}
pub fn tmdb_details(kind: &str, id: u64, key: &str) -> anyhow::Result<TmdbDetails> {
info!("fetching details: {id:?}");
Ok(reqwest::blocking::get(&format!(
"https://api.themoviedb.org/3/{kind}/{id}?api_key={key}"
))?
.json()?)
}
pub fn tmdb_image(path: &str, out: &mut impl Write) -> anyhow::Result<()> {
info!("downloading image {path:?}");
let mut res = reqwest::blocking::get(&format!("https://image.tmdb.org/t/p/original{path}"))?;
res.copy_to(out)?;
Ok(())
}
pub fn parse_release_date(d: &str) -> anyhow::Result<DateTime<Utc>> {
let (year, month, day) = (&d[0..4], &d[5..7], &d[8..10]);
let (year, month, day) = (
year.parse().context("parsing year")?,
month.parse().context("parsing month")?,
day.parse().context("parsing day")?,
);
let mut p = Parsed::new();
p.year = Some(year);
p.month = Some(month);
p.day = Some(day);
p.hour_div_12 = Some(0);
p.hour_mod_12 = Some(0);
p.minute = Some(0);
p.second = Some(0);
Ok(p.to_datetime_with_timezone(&Utc)?)
}
|
