embystream 0.0.22

Another Emby streaming application (frontend/backend separation) written in Rust.
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

use regex::{Error as RegexError, Regex};
use tokio::sync::OnceCell;

use crate::{PATH_REWRITER_LOGGER_DOMAIN, error_log};

#[derive(Clone, Debug)]
pub struct PathRewriter {
    pub enable: bool,
    pattern: String,
    replacement: String,
    regex: OnceCell<Regex>,
}

impl PathRewriter {
    pub fn new(enable: bool, pattern: &str, replacement: &str) -> Self {
        Self {
            enable,
            pattern: pattern.to_string(),
            replacement: replacement.to_string(),
            regex: OnceCell::new(),
        }
    }

    async fn compile(&self) -> Result<&Regex, RegexError> {
        self.regex
            .get_or_try_init(|| async { Regex::new(&self.pattern) })
            .await
    }

    pub async fn rewrite(&self, path: &str) -> String {
        match self.compile().await {
            Ok(re) => re.replace(path, &self.replacement).into_owned(),
            Err(e) => {
                error_log!(
                    PATH_REWRITER_LOGGER_DOMAIN,
                    "Error occurred during path rewrite: {:?}. ",
                    e
                );
                error_log!(
                    PATH_REWRITER_LOGGER_DOMAIN,
                    "Target path: {:?}, Pattern: {:?}, Regex: {:?}.",
                    path,
                    self.pattern,
                    self.replacement
                );
                path.to_string()
            }
        }
    }
}