embystream 0.0.36

use std::collections::HashMap;

use crate::{
    api::emby::Operation,
    network::{HttpMethod, NetworkTarget, NetworkTask},
    system::SystemInfo,
    util::StringUtil,
};

const EMBY_PATH_PREFIX: &str = "emby";
const EMBY_USERS_SEGMENT: &str = "Users";
const EMBY_ITEMS_SEGMENT: &str = "Items";
const EMBY_PLAYBACK_INFO_SEGMENT: &str = "PlaybackInfo";
const PLAYBACK_INFO_MEDIA_SOURCE_ID_QUERY_KEY: &str = "MediaSourceId";
const ACCEPT_HEADER_VALUE: &str = "application/json";
const CONTENT_TYPE_HEADER_KEY: &str = "content-type";

/// Represents Emby API endpoints with their respective parameters.
#[derive(Debug, Clone)]
pub struct API {
    /// The base URL for the Emby API.
    base_url: String,
    /// The API key for authenticating with the Emby API.
    api_key: String,
    /// The specific API operation (GetUser or PlaybackInfo).
    operation: Operation,
}

impl API {
    /// Constructs an EmbyAPI instance for the GetUser endpoint.
    pub fn get_user(
        base_url: impl Into<String>,
        api_key: impl Into<String>,
        user_id: impl Into<String>,
    ) -> Self {
        API {
            base_url: base_url.into(),
            api_key: api_key.into(),
            operation: Operation::GetUser {
                user_id: user_id.into(),
            },
        }
    }

    /// Constructs an EmbyAPI instance for the PlaybackInfo endpoint.
    pub fn playback_info(
        base_url: impl Into<String>,
        api_key: impl Into<String>,
        item_id: impl Into<String>,
        media_source_id: impl Into<String>,
        method: HttpMethod,
        body: Option<Vec<u8>>,
        content_type: Option<String>,
    ) -> Self {
        API {
            base_url: base_url.into(),
            api_key: api_key.into(),
            operation: Operation::PlaybackInfo {
                item_id: item_id.into(),
                media_source_id: media_source_id.into(),
                method,
                body,
                content_type,
            },
        }
    }
}

impl NetworkTarget for API {
    fn base_url(&self) -> String {
        self.base_url.clone()
    }

    fn path(&self) -> String {
        match &self.operation {
            Operation::GetUser { user_id } => {
                format!("{EMBY_PATH_PREFIX}/{EMBY_USERS_SEGMENT}/{user_id}")
            }
            Operation::PlaybackInfo { item_id, .. } => {
                format!(
                    "{EMBY_PATH_PREFIX}/{EMBY_ITEMS_SEGMENT}/{item_id}/{EMBY_PLAYBACK_INFO_SEGMENT}"
                )
            }
        }
    }

    fn method(&self) -> HttpMethod {
        match &self.operation {
            Operation::GetUser { .. } => HttpMethod::Get,
            Operation::PlaybackInfo { method, .. } => *method,
        }
    }

    fn task(&self) -> NetworkTask {
        let mut params = HashMap::new();
        params.insert("api_key".to_string(), self.api_key.clone());
        match &self.operation {
            Operation::GetUser { .. } => NetworkTask::RequestParameters(params),
            Operation::PlaybackInfo {
                media_source_id,
                method,
                body,
                ..
            } => {
                params.insert(
                    PLAYBACK_INFO_MEDIA_SOURCE_ID_QUERY_KEY.to_string(),
                    media_source_id.clone(),
                );
                match method {
                    HttpMethod::Get => NetworkTask::RequestParameters(params),
                    HttpMethod::Post => {
                        NetworkTask::RequestBytesWithParameters(
                            body.clone().unwrap_or_default(),
                            params,
                        )
                    }
                    _ => NetworkTask::RequestParameters(params),
                }
            }
        }
    }

    fn headers(&self) -> Vec<(String, String)> {
        let sys_info = SystemInfo::new();
        let base_url =
            StringUtil::trim_trailing_slashes(&self.base_url).to_string();
        let mut headers = vec![
            ("accept".into(), ACCEPT_HEADER_VALUE.into()),
            ("origin".into(), base_url.clone()),
            ("referer".into(), format!("{base_url}/")),
            ("user-agent".into(), sys_info.get_user_agent()),
        ];

        if let Operation::PlaybackInfo {
            content_type: Some(content_type),
            ..
        } = &self.operation
        {
            let trimmed = content_type.trim();
            if !trimmed.is_empty() {
                headers.push((
                    CONTENT_TYPE_HEADER_KEY.into(),
                    trimmed.to_string(),
                ));
            }
        }

        headers
    }
}