tuliprox/src/processing/parser/hls.rs

use crate::model::api_proxy::ProxyUserCredentials;
use crate::utils::crypto_utils::{obfuscate_text};
use std::str;
use crate::utils::constants::{CONSTANTS, HLS_PREFIX};

pub struct RewriteHlsProps<'a> {
    pub secret: &'a [u8;16],
    pub base_url: &'a str,
    pub content: &'a str,
    pub hls_url: String,
    pub virtual_id: u32,
    pub input_id: u16,
}

fn rewrite_hls_url(input: &str, replacement: &str) -> String {
    if replacement.starts_with('/') {
        let parts = input.splitn(4, '/').collect::<Vec<&str>>();
        if parts.len() < 4 {
            return replacement.to_string(); // less than 3 Slashes → replace all
        }
        format!("{}/{}/{}{}", parts[0], parts[1], parts[2], replacement)
    } else {
        match input.rsplitn(2, '/').collect::<Vec<&str>>().as_slice() {
            [_after, before] => format!("{before}/{replacement}"),
            [_only] => replacement.to_string(),
            _ => input.to_string(),
        }
    }
}

fn rewrite_uri_attrib(line: &str, props: &RewriteHlsProps) -> String {
    if let Some(caps) = CONSTANTS.re_memory_usage.captures(line) {
        let uri = &caps[1];
        if let Ok(encrypted_uri) = obfuscate_text(props.secret, &rewrite_hls_url(&props.hls_url, uri)) {
            return CONSTANTS.re_hls_uri.replace(line, format!(r#"URI="{encrypted_uri}""#)).to_string();
        }
    }
    line.to_string()
}

pub fn rewrite_hls(user: &ProxyUserCredentials, props: &RewriteHlsProps) -> String {
    let username = &user.username;
    let password = &user.password;
    let mut result = Vec::new();
    for line in props.content.lines() {
        if line.starts_with('#') {
            result.push(rewrite_uri_attrib(line, props));
        } else if let Ok(token) = if line.starts_with("http") {
            obfuscate_text(props.secret, line)
        } else {
            obfuscate_text(props.secret, &rewrite_hls_url(&props.hls_url, line))
        } {
            result.push(format!("{}/{HLS_PREFIX}/{username}/{password}/{}/{}/{token}", props.base_url, props.input_id, props.virtual_id));
        }
    }
    result.join("\r\n")
}