autopulse_service/settings/triggers/
mod.rs

1/// Autoscan - autoscan compatibility trigger
2///
3/// Provides a trigger with compatibility for applications that are built for autoscan
4///
5/// # Example
6///
7/// ```yml
8/// triggers:
9///   my_autoscan:
10///     type: autoscan
11/// ```
12///
13/// or
14///
15/// ```yml
16/// triggers:
17///   my_autoscan:
18///     type: autoscan
19///     rewrite:
20///       from: "/downloads/all"
21///       to: "/all"
22///     timer:
23///       wait: 30
24///     excludes: [ "ignored_target" ]
25/// ```
26///
27/// See [`Autoscan`] for all options
28pub mod autoscan;
29/// Lidarr - Lidarr trigger
30///
31/// This trigger is used to process a file from Lidarr
32///
33/// # Example
34///
35/// ```yml
36/// triggers:
37///   my_lidarr:
38///     type: lidarr
39/// ```
40///
41/// or
42///
43/// ```yml
44/// triggers:
45///   my_lidarr:
46///     type: lidarr
47///     rewrite:
48///       from: "/downloads/music"
49///       to: "/music"
50///     timer:
51///       wait: 30
52///     excludes: [ "ignored_target" ]
53/// ```
54///
55/// See [`Lidarr`] for all options
56pub mod lidarr;
57/// Manual - Manual trigger
58///
59/// This trigger is used to manually process a file. Often used when implementing a custom trigger
60///
61/// Note: A default route of `/triggers/manual` is provided
62///
63/// # Example
64///
65/// ```yml
66/// triggers:
67///   my_manual:
68///     type: manual
69/// ```
70///
71/// or
72///
73/// ```yml
74/// triggers:
75///   my_manual:
76///     type: manual
77///     rewrite:
78///       from: "/downloads/stuff"
79///       to: "/stuff"
80///     timer:
81///       wait: 30
82///     excludes: [ "ignored_target" ]
83/// ```
84///
85/// See [`Manual`] for all options
86/// and
87/// [`ManualQueryParams`](manual::ManualQueryParams) for query parameters
88pub mod manual;
89/// Notify - Notify trigger
90///
91/// Cross-platform monitoring for a directory to process based on file events
92///
93/// # Example
94///
95/// ```yml
96/// triggers:
97///   my_notify:
98///     type: notify
99///     paths:
100///       - "/path/to/monitor"
101/// ```
102///
103/// or
104///
105/// ```yml
106/// triggers:
107///   my_notify:
108///     type: notify
109///     paths:
110///       - "/downloads"
111///     recursive: false
112///     rewrite:
113///       from: "/downloads"
114///       to: "/media"
115///     timer:
116///       wait: 30
117///     excludes: [ "ignored_target" ]
118/// ```
119///
120/// See [`Notify`] for all options
121pub mod notify;
122/// Radarr - Radarr trigger
123///
124/// This trigger is used to process a file from Radarr
125///
126/// # Example
127///
128/// ```yml
129/// triggers:
130///   my_radarr:
131///     type: radarr
132/// ```
133///
134/// or
135///
136/// ```yml
137/// triggers:
138///   my_radarr:
139///     type: radarr
140///     rewrite:
141///       from: "/downloads/movies"
142///       to: "/movies"
143///     timer:
144///       wait: 30
145///     excludes: [ "ignored_target" ]
146/// ```
147///
148/// See [`Radarr`] for all options
149pub mod radarr;
150/// Readarr - Readarr trigger
151///
152/// This trigger is used to process a file from Readarr
153///
154/// # Example
155///
156/// ```yml
157/// triggers:
158///   my_readarr:
159///     type: readarr
160/// ```
161///
162/// or
163///
164/// ```yml
165/// triggers:
166///   my_readarr:
167///     type: readarr
168///     rewrite:
169///       from: "/downloads/books"
170///       to: "/books"
171///     timer:
172///       wait: 30
173///     excludes: [ "ignored_target" ]
174/// ```
175///
176/// See [`Readarr`] for all options
177pub mod readarr;
178/// Sonarr - Sonarr trigger
179///
180/// This trigger is used to process a file from Sonarr
181///
182/// # Example
183///
184/// ```yml
185/// triggers:
186///   my_sonarr:
187///     type: sonarr
188/// ```
189///
190/// or
191///
192/// ```yml
193/// triggers:
194///   my_sonarr:
195///     type: sonarr
196///     rewrite:
197///       from: "/downloads/shows"
198///       to: "/shows"
199///     timer:
200///       wait: 30
201///     excludes: [ "ignored_target" ]
202/// ```
203///
204/// See [`Sonarr`] for all options
205pub mod sonarr;
206
207use crate::settings::timer::Timer;
208use crate::settings::{rewrite::Rewrite, triggers::autoscan::Autoscan};
209use serde::{Deserialize, Serialize};
210use {
211    lidarr::{Lidarr, LidarrRequest},
212    manual::Manual,
213    notify::Notify,
214    radarr::{Radarr, RadarrRequest},
215    readarr::{Readarr, ReadarrRequest},
216    sonarr::{Sonarr, SonarrRequest},
217};
218
219pub trait TriggerRequest {
220    fn from_json(json: serde_json::Value) -> anyhow::Result<Self>
221    where
222        Self: Sized;
223
224    // where the bool represents whether to check found status
225    fn paths(&self) -> Vec<(String, bool)>;
226}
227
228#[derive(Serialize, Deserialize)]
229#[serde(rename_all = "lowercase")]
230pub enum TriggerType {
231    Manual,
232    Autoscan,
233    Radarr,
234    Bazarr,
235    Sonarr,
236    Lidarr,
237    Readarr,
238    Notify,
239}
240
241#[derive(Serialize, Deserialize, Clone)]
242#[serde(tag = "type", rename_all = "lowercase")]
243pub enum Trigger {
244    Manual(Manual),
245    Autoscan(Autoscan),
246    Bazarr(Manual),
247    Radarr(Radarr),
248    Sonarr(Sonarr),
249    Lidarr(Lidarr),
250    Readarr(Readarr),
251    Notify(Notify),
252}
253
254impl Trigger {
255    pub const fn get_rewrite(&self) -> Option<&Rewrite> {
256        match &self {
257            Self::Sonarr(trigger) => trigger.rewrite.as_ref(),
258            Self::Radarr(trigger) => trigger.rewrite.as_ref(),
259            Self::Lidarr(trigger) => trigger.rewrite.as_ref(),
260            Self::Bazarr(trigger) => trigger.rewrite.as_ref(),
261            Self::Readarr(trigger) => trigger.rewrite.as_ref(),
262            Self::Autoscan(trigger) => trigger.rewrite.as_ref(),
263            Self::Manual(trigger) => trigger.rewrite.as_ref(),
264            Self::Notify(trigger) => trigger.rewrite.as_ref(),
265        }
266    }
267
268    pub fn get_timer(&self, event_name: Option<String>) -> Timer {
269        let mut base_timer = match self.clone() {
270            Self::Sonarr(trigger) => trigger.timer,
271            Self::Radarr(trigger) => trigger.timer,
272            Self::Lidarr(trigger) => trigger.timer,
273            Self::Bazarr(trigger) => trigger.timer,
274            Self::Readarr(trigger) => trigger.timer,
275            Self::Manual(trigger) => trigger.timer,
276            Self::Notify(trigger) => trigger.timer,
277            Self::Autoscan(trigger) => trigger.timer,
278        }
279        .unwrap_or_default();
280
281        let event_specific_timer = match &self {
282            Self::Sonarr(trigger) => event_name
283                .as_ref()
284                .and_then(|event| trigger.event_timers.as_ref().and_then(|map| map.get(event))),
285            Self::Radarr(trigger) => event_name
286                .as_ref()
287                .and_then(|event| trigger.event_timers.as_ref().and_then(|map| map.get(event))),
288            Self::Lidarr(trigger) => event_name
289                .as_ref()
290                .and_then(|event| trigger.event_timers.as_ref().and_then(|map| map.get(event))),
291            Self::Readarr(trigger) => event_name
292                .as_ref()
293                .and_then(|event| trigger.event_timers.as_ref().and_then(|map| map.get(event))),
294            _ => None,
295        };
296
297        if let Some(event_timer) = event_specific_timer {
298            base_timer = base_timer.chain(event_timer);
299        }
300
301        base_timer
302    }
303
304    pub fn paths(&self, body: serde_json::Value) -> anyhow::Result<(String, Vec<(String, bool)>)> {
305        let event_name = body["eventType"].as_str().unwrap_or("unknown").to_string();
306
307        let paths = match &self {
308            Self::Sonarr(_) => Ok(SonarrRequest::from_json(body)?.paths()),
309            Self::Radarr(_) => Ok(RadarrRequest::from_json(body)?.paths()),
310            Self::Lidarr(_) => Ok(LidarrRequest::from_json(body)?.paths()),
311            Self::Readarr(_) => Ok(ReadarrRequest::from_json(body)?.paths()),
312            Self::Manual(_) | Self::Notify(_) | Self::Autoscan(_) | Self::Bazarr(_) => {
313                Err(anyhow::anyhow!("Manual trigger does not have paths"))
314            }
315        }?;
316
317        Ok((event_name, paths))
318    }
319
320    pub const fn excludes(&self) -> &Vec<String> {
321        match &self {
322            Self::Manual(trigger) => &trigger.excludes,
323            Self::Radarr(trigger) => &trigger.excludes,
324            Self::Sonarr(trigger) => &trigger.excludes,
325            Self::Lidarr(trigger) => &trigger.excludes,
326            Self::Bazarr(trigger) => &trigger.excludes,
327            Self::Readarr(trigger) => &trigger.excludes,
328            Self::Notify(trigger) => &trigger.excludes,
329            Self::Autoscan(trigger) => &trigger.excludes,
330        }
331    }
332}
autopulse_service/settings/triggers/mod.rs

autopulse_service/settings/triggers/
mod.rs
