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
117
118
119
120
121
122
123
124
|
/*
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) 2025 metamuffin <metamuffin.org>
*/
use anyhow::Result;
use bincode::{Decode, Encode};
use jellybase::cache::async_cache_memory;
use reqwest::{
header::{HeaderMap, HeaderName, HeaderValue},
Client, ClientBuilder,
};
use serde::Deserialize;
use std::{path::Path, process::Stdio, sync::Arc, time::Duration};
use tokio::{
io::AsyncReadExt,
process::Command,
sync::Semaphore,
time::{sleep_until, Instant},
};
pub(crate) struct AcoustID {
client: Client,
key: String,
rate_limit: Arc<Semaphore>,
}
#[derive(Debug, Hash, Clone, Encode, Decode, Deserialize)]
pub(crate) struct Fingerprint {
duration: u32,
fingerprint: String,
}
#[derive(Deserialize, Encode, Decode)]
pub(crate) struct AcoustIDLookupResultRecording {
id: String,
}
#[derive(Deserialize, Encode, Decode)]
pub(crate) struct AcoustIDLookupResult {
id: String,
score: f32,
recordings: Vec<AcoustIDLookupResultRecording>,
}
#[derive(Deserialize, Encode, Decode)]
pub(crate) struct AcoustIDLookupResponse {
status: String,
results: Vec<AcoustIDLookupResult>,
}
impl AcoustID {
pub fn new(api_key: &str) -> Self {
let client = ClientBuilder::new()
.default_headers(HeaderMap::from_iter([(
HeaderName::from_static("accept"),
HeaderValue::from_static("application/json"),
)]))
.build()
.unwrap();
Self {
client,
// send at most 3 req/s according to acoustid docs, each lock is therefore held for 1s
// this implementation also never sends more than 3 requests in-flight.
rate_limit: Arc::new(Semaphore::new(3)),
key: api_key.to_owned(),
}
}
pub async fn get_atid_mbid(&self, fp: Fingerprint) -> Result<Option<(String, String)>> {
let res = self.lookup(fp).await?;
for r in &res.results {
if let Some(k) = r.recordings.get(0) {
return Ok(Some((r.id.clone(), k.id.clone())));
}
}
Ok(None)
}
pub async fn lookup(&self, fp: Fingerprint) -> Result<Arc<AcoustIDLookupResponse>> {
async_cache_memory("api-acoustid", fp.clone(), || async move {
let _permit = self.rate_limit.clone().acquire_owned().await?;
let permit_drop_ts = Instant::now() + Duration::SECOND;
let duration = fp.duration;
let fingerprint = &fp.fingerprint;
let client = &self.key;
let meta = "recordingids";
let body = format!("format=json&client={client}&duration={duration}&fingerprint={fingerprint}&meta={meta}");
let resp = self
.client
.post(format!("https://api.acoustid.org/v2/lookup"))
.body(body)
.send()
.await?.error_for_status()?.json::<AcoustIDLookupResponse>().await?;
tokio::task::spawn(async move {
sleep_until(permit_drop_ts).await;
drop(_permit);
});
Ok(resp)
})
.await
}
}
#[allow(unused)]
pub(crate) async fn acoustid_fingerprint(path: &Path) -> Result<Arc<Fingerprint>> {
async_cache_memory("fpcalc", path, || async move {
let child = Command::new("fpcalc")
.arg("-json")
.arg(path)
.stdout(Stdio::piped())
.spawn()?;
let mut buf = Vec::new();
child.stdout.unwrap().read_to_end(&mut buf).await?;
let out: Fingerprint = serde_json::from_slice(&buf)?;
Ok(out)
})
.await
}
|
