selene-core 0.9.0-alpha.2

selene-core is the backend for Selene, a local-first music player
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use std::str::FromStr;

use chrono::{DateTime, Utc};
use lunar_lib::{
    database::{Entry, Id},
    error,
    iterator_ext::IteratorExtensions,
};
use regex::Regex;
use serde::{Deserialize, Serialize};

use crate::library::{
    album::Album,
    artist::Artist,
    collection::rules::{EqOp, OrdOp, Rule, eq_many, eq_single, ord_single},
};

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub enum AlbumRule {
    /// Succeeds if the track id matches the [`EqOp`] of the input ids
    Id { ids: Vec<Id<Album>>, op: EqOp },

    /// Succeeds if the album name DOES/DOES NOT match ALL/ANY of the input regexes
    Name { regex: Vec<String>, op: EqOp },

    /// Succeeds if the album artists match the input artists using the [`EqOp`]
    Artists { artists: Vec<Id<Artist>>, op: EqOp },

    /// Succeeds if the disc total matches the [`OrdOp`]
    Discs { count: u32, op: OrdOp },

    /// Succeeds if the track total matches the [`OrdOp`]
    Tracks { count: u32, op: OrdOp },

    /// Succeeds if the album genres match the input genres using the [`EqOp`]
    Genre { genres: Vec<String>, op: EqOp },

    /// Succeeds if the album date matches the [`OrdOp`]
    Date { date: DateTime<Utc>, op: OrdOp },
}

impl Rule for AlbumRule {
    type Item = Entry<Album>;

    fn matches(&self, item: &Self::Item) -> bool {
        match self {
            AlbumRule::Id { ids: id, op } => eq_single(&item.id(), id, *op),
            AlbumRule::Name { regex, op } => {
                let regex: Vec<_> = match regex.iter().try_map(|r| Regex::from_str(r)) {
                    Ok(v) => v.collect(),
                    Err(err) => {
                        error!("Failed to create regex: {err}");
                        return false;
                    }
                };

                match op {
                    EqOp::EqAny => regex.iter().any(|r| r.is_match(&item.title)),
                    EqOp::EqAll => regex.iter().all(|r| r.is_match(&item.title)),
                    EqOp::NeqAny => !regex.iter().any(|r| r.is_match(&item.title)),
                    EqOp::NeqAll => !regex.iter().all(|r| r.is_match(&item.title)),
                }
            }
            AlbumRule::Artists { artists, op } => eq_many(item.artists(), artists, *op),
            AlbumRule::Discs { count, op } => {
                item.disc_total.is_some_and(|v| ord_single(v, *count, *op))
            }
            AlbumRule::Tracks { count, op } => {
                item.track_total.is_some_and(|v| ord_single(v, *count, *op))
            }
            AlbumRule::Genre { genres, op } => eq_many(&item.genre, genres, *op),
            AlbumRule::Date { date, op } => item.date.is_some_and(|v| ord_single(v, *date, *op)),
        }
    }
}