use std::sync::Arc;

use axum::body::Body;
use axum::extract::Request;
use axum::http::header;
use axum::middleware::Next;
use axum::response::Response;

use crate::state::AppState;

const OG_PLACEHOLDER: &str = r#"<meta name="x-og-placeholder" content="__NARROWIT_OG_TAGS__"/>"#;

pub async fn og_middleware(request: Request, next: Next) -> Response {
    // Capture the query string before passing the request through
    let query_string = request.uri().query().unwrap_or("").to_string();

    // Get state from extensions
    let state = request.extensions().get::<Arc<AppState>>().cloned();

    let response = next.run(request).await;

    let content_type = response
        .headers()
        .get(header::CONTENT_TYPE)
        .and_then(|val| val.to_str().ok())
        .unwrap_or("");

    if !content_type.contains("text/html") {
        return response;
    }

    let state = match state {
        Some(st) => st,
        None => return response,
    };

    let index_html = match &state.index_html {
        Some(html) => html,
        None => return response,
    };

    // Build OG-injected HTML
    let og_image_url = if query_string.is_empty() {
        format!("{}/api/og-image", state.public_url)
    } else {
        format!("{}/api/og-image?{}", state.public_url, query_string)
    };

    let og_tags = format!(
        r#"<meta property="og:title" content="Narrowit — Every neighbourhood in England & Wales" />
    <meta property="og:description" content="Explore property prices, energy ratings, crime stats, school ratings, and more across England & Wales on one interactive map." />
    <meta property="og:type" content="website" />
    <meta property="og:image" content="{og_image_url}" />
    <meta property="og:image:width" content="1200" />
    <meta property="og:image:height" content="630" />
    <meta name="twitter:card" content="summary_large_image" />
    <meta name="twitter:title" content="Narrowit — Every neighbourhood in England & Wales" />
    <meta name="twitter:description" content="Explore property prices, energy ratings, crime stats, school ratings, and more across England & Wales on one interactive map." />"#
    );

    let html = index_html.replace(OG_PLACEHOLDER, &og_tags);

    let (parts, _body) = response.into_parts();
    Response::from_parts(parts, Body::from(html))
}