feat(readarr): add Readarr tags and expose the command on the CLI

This commit is contained in:
Dark-Alex-17 committed 2026-08-20 22:47:41 -06:00
1 parent 29186ea03c
commit b9fa3b36c0
5 files changed
+268 -1

No files matched your search

+25 -1
View File
@@ -1,5 +1,6 @@
use anyhow::Result;
use log::info;
use serde_json::{Value, json};
use super::{NetworkEvent, NetworkResource};
use crate::models::readarr_models::{ReadarrSerdeable, ReadarrTaskName};
@@ -15,6 +16,7 @@ mod readarr_network_tests;
#[derive(Debug, Eq, PartialEq, Clone)]
pub enum ReadarrEvent {
AddTag(String),
GetDiskSpace,
GetHostConfig,
GetLogs(u64),
@@ -44,7 +46,7 @@ impl NetworkResource for ReadarrEvent {
ReadarrEvent::GetRootFolders => "/rootfolder",
ReadarrEvent::GetStatus => "/system/status",
ReadarrEvent::GetTasks => "/system/task",
ReadarrEvent::GetTags => "/tag",
ReadarrEvent::AddTag(_) | ReadarrEvent::GetTags => "/tag",
ReadarrEvent::GetUpdates => "/update",
}
}
@@ -62,6 +64,7 @@ impl Network<'_, '_> {
readarr_event: ReadarrEvent,
) -> Result<ReadarrSerdeable> {
match readarr_event {
ReadarrEvent::AddTag(tag) => self.add_readarr_tag(tag).await.map(ReadarrSerdeable::from),
ReadarrEvent::GetDiskSpace => self
.get_readarr_diskspace()
.await
@@ -177,4 +180,25 @@ impl Network<'_, '_> {
})
.await
}
async fn add_readarr_tag(&mut self, tag: String) -> Result<Tag> {
info!("Adding a new Readarr tag");
let event = ReadarrEvent::AddTag(String::new());
let request_props = self
.request_props_from(
event,
RequestMethod::Post,
Some(json!({ "label": tag })),
None,
None,
)
.await;
self
.handle_request::<Value, Tag>(request_props, |tag, mut app| {
app.data.readarr_data.tags_map.insert(tag.id, tag.label);
})
.await
}
}
@@ -22,6 +22,7 @@ mod tests {
#[case(ReadarrEvent::GetRootFolders, "/rootfolder")]
#[case(ReadarrEvent::GetStatus, "/system/status")]
#[case(ReadarrEvent::GetTasks, "/system/task")]
#[case(ReadarrEvent::AddTag(String::new()), "/tag")]
#[case(ReadarrEvent::GetTags, "/tag")]
#[case(ReadarrEvent::GetUpdates, "/update")]
fn test_resource(#[case] event: ReadarrEvent, #[case] expected_uri: &str) {
@@ -345,4 +346,68 @@ mod tests {
assert_err!(result);
assert_eq!(app.lock().await.data.readarr_data.tags_map, seeded_map);
}
#[tokio::test]
async fn test_handle_add_readarr_tag_event() {
let tag_json = json!({
"id": 2,
"label": "managarr-verify"
});
let response: Tag = serde_json::from_value(tag_json.clone()).unwrap();
let (mock, app, _server) = MockServarrApi::post()
.with_request_body(json!({ "label": "managarr-verify" }))
.returns(tag_json)
.build_for(ReadarrEvent::AddTag("managarr-verify".to_owned()))
.await;
{
let mut app = app.lock().await;
app.server_tabs.set_index(3);
app.data.readarr_data.tags_map = BiMap::from_iter([(1i64, "huntarr-missing".to_owned())]);
}
let mut network = test_network(&app);
let result = network
.handle_readarr_event(ReadarrEvent::AddTag("managarr-verify".to_owned()))
.await;
mock.assert_async().await;
let ReadarrSerdeable::Tag(tag) = result.unwrap() else {
panic!("Expected Tag");
};
assert_eq!(tag, response);
assert_eq!(
app.lock().await.data.readarr_data.tags_map,
BiMap::from_iter([
(1i64, "huntarr-missing".to_owned()),
(2i64, "managarr-verify".to_owned())
])
);
}
#[tokio::test]
async fn test_handle_add_readarr_tag_event_failure() {
let seeded_map = BiMap::from_iter([(1i64, "huntarr-missing".to_owned())]);
let (mock, app, _server) = MockServarrApi::post()
.with_request_body(json!({ "label": "managarr-verify" }))
.returns(json!({}))
.status(500)
.build_for(ReadarrEvent::AddTag("managarr-verify".to_owned()))
.await;
{
let mut app = app.lock().await;
app.server_tabs.set_index(3);
app.data.readarr_data.tags_map = seeded_map.clone();
}
let mut network = test_network(&app);
let result = network
.handle_readarr_event(ReadarrEvent::AddTag("managarr-verify".to_owned()))
.await;
mock.assert_async().await;
assert_err!(result);
assert_eq!(app.lock().await.data.readarr_data.tags_map, seeded_map);
}
}