aves-tyto 0.5.0

use std::fmt::Display;
use std::future::Future;
use std::sync::Arc;
use std::time::Duration;

use slack_morphism::prelude::{
    SlackApiToken, SlackApiTokenType, SlackApiTokenValue, SlackClient, SlackClientHyperConnector,
    SlackClientHyperHttpsConnector,
};
use tokio::time::timeout;

use crate::messages::ListenerType;
use crate::types::{MessagingConfig, MessagingCredentials, MessagingError, MessagingSubscriptionError, SlackConfig};

/// Shared Slack API client and its app and bot credentials.
pub(super) struct SlackConnection {
    pub(super) client: Arc<SlackClient<SlackClientHyperHttpsConnector>>,
    pub(super) token: SlackApiToken,
    pub(super) bot_token: SlackApiToken,
}

pub(super) fn configure(
    settings: &MessagingConfig,
    credentials: &MessagingCredentials,
) -> Result<Arc<SlackConnection>, MessagingError> {
    let MessagingCredentials::Slack { app_token, bot_token } = credentials else {
        return Err(MessagingError::ChannelConnectionFailed {
            listener_type: ListenerType::Slack,
            message: "credential protocol does not match the listener".to_owned(),
        });
    };

    let mut config = settings
        .slack
        .clone()
        .ok_or_else(|| MessagingError::ChannelConnectionFailed {
            listener_type: ListenerType::Slack,
            message: "Slack config block is missing".to_owned(),
        })?;

    config.app_token = Some(app_token.clone());
    config.bot_token = Some(bot_token.clone());

    prepare(&config)
}

/// Apply the Socket Mode listen deadline and add Slack connection context.
pub(super) async fn connect<F, T, E>(future: F) -> Result<T, MessagingSubscriptionError>
where
    F: Future<Output = Result<T, E>>,
    E: Display,
{
    let deadline = Duration::from_secs(60);
    timeout(deadline, future)
        .await
        .map_err(|_| MessagingSubscriptionError::Timeout {
            listener_type: ListenerType::Slack,
            operation: "listen_for".to_owned(),
            timeout: format!("{deadline:?}"),
        })?
        .map_err(|error| slack_connection(format!("Slack Socket Mode listen_for failed: {error}")))
}

fn prepare(config: &SlackConfig) -> Result<Arc<SlackConnection>, MessagingError> {
    let app_token = config
        .app_token
        .as_ref()
        .ok_or_else(|| MessagingError::ChannelConnectionFailed {
            listener_type: ListenerType::Slack,
            message: "Slack app token is missing".to_owned(),
        })?;

    let bot_token = config
        .bot_token
        .as_ref()
        .ok_or_else(|| MessagingError::ChannelConnectionFailed {
            listener_type: ListenerType::Slack,
            message: "Slack bot token is missing".to_owned(),
        })?;

    let client = Arc::new(SlackClient::new(SlackClientHyperConnector::new().map_err(|error| {
        MessagingError::ChannelConnectionFailed {
            listener_type: ListenerType::Slack,
            message: slack_connection(format!("Slack hyper connector init failed: {error}")).to_string(),
        }
    })?));
    let token =
        SlackApiToken::new(SlackApiTokenValue::new(app_token.to_string())).with_token_type(SlackApiTokenType::App);
    let bot_token =
        SlackApiToken::new(SlackApiTokenValue::new(bot_token.to_string())).with_token_type(SlackApiTokenType::Bot);

    Ok(Arc::new(SlackConnection {
        client,
        token,
        bot_token,
    }))
}

fn slack_connection(message: String) -> MessagingSubscriptionError {
    MessagingSubscriptionError::Connection {
        listener_type: ListenerType::Slack,
        message,
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn preparation_rejects_missing_app_token() {
        let config = SlackConfig {
            app_token: None,
            bot_token: Some("xoxb-test".to_owned().into()),
            ..Default::default()
        };
        assert!(
            matches!(prepare(&config), Err(MessagingError::ChannelConnectionFailed { listener_type: ListenerType::Slack, message }) if message.contains("app token is missing"))
        );
    }

    #[test]
    fn preparation_rejects_missing_bot_token() {
        let config = SlackConfig {
            app_token: Some("xapp-test".to_owned().into()),
            bot_token: None,
            ..Default::default()
        };
        assert!(
            matches!(prepare(&config), Err(MessagingError::ChannelConnectionFailed { listener_type: ListenerType::Slack, message }) if message.contains("bot token is missing"))
        );
    }

    #[test]
    fn preparation_checks_app_token_first_when_both_are_missing() {
        assert!(
            matches!(prepare(&SlackConfig::default()), Err(MessagingError::ChannelConnectionFailed { listener_type: ListenerType::Slack, message }) if message.contains("app token is missing"))
        );
    }
}