use actix_web::http::header::HeaderMap;
use actix_web::HttpRequest;
use opentelemetry::global::BoxedTracer;
use opentelemetry::{global, Context, KeyValue};
use opentelemetry::propagation::TextMapPropagator;
use opentelemetry_appender_log::OpenTelemetryLogBridge;
use opentelemetry_otlp::WithExportConfig;
use opentelemetry_sdk::logs::{BatchLogProcessor, SdkLoggerProvider};
use opentelemetry_sdk::propagation::TraceContextPropagator;
use opentelemetry_sdk::Resource;

pub fn global_tracer() -> BoxedTracer {
    global::tracer("image-server")
}

pub fn init_tracing() {
    let resources = Resource::builder()
        .with_attributes([
            KeyValue::new("service.name", "image-server"),
            KeyValue::new("service.version", env!("CARGO_PKG_VERSION")),
        ])
        .build();

    let span_exporter = opentelemetry_otlp::SpanExporter::builder()
        .with_tonic()
        .with_endpoint(std::env::var("OTLP_OTLS_ENDPOINT").unwrap())
        .build()
        .unwrap();

    let tracer_provider = opentelemetry_sdk::trace::SdkTracerProvider::builder()
        .with_batch_exporter(span_exporter)
        .with_resource(resources)
        .build();

    global::set_tracer_provider(tracer_provider);
}

pub fn init_logs() {
    let otlp_exporter = opentelemetry_otlp::LogExporter::builder()
        .with_tonic()
        .with_endpoint(std::env::var("OTLP_OTLS_ENDPOINT").unwrap())
        .build()
        .unwrap();

    let exporter = opentelemetry_stdout::LogExporter::default();

    let resources = Resource::builder()
        .with_attributes([
            KeyValue::new("service.name", "image-server"),
            KeyValue::new("service.version", "1.0"),
        ])
        .build();

    let log_provider = SdkLoggerProvider::builder()
        .with_log_processor(BatchLogProcessor::builder(exporter).build())
        .with_log_processor(BatchLogProcessor::builder(otlp_exporter).build())
        .with_resource(resources)
        .build();

    let otel_log_appender = OpenTelemetryLogBridge::new(&log_provider);
    log::set_boxed_logger(Box::new(otel_log_appender)).expect("Unable to set boxed logger");
    //TODO: Still set this with the env? Ideally we still have a clean/simple local logger for local dev
    log::set_max_level(log::LevelFilter::Info);
}

struct HeaderExtractor<'a>(&'a HeaderMap);

impl<'a> opentelemetry::propagation::Extractor for HeaderExtractor<'a> {
    fn get(&self, key: &str) -> Option<&str> {
        self.0.get(key).and_then(|v| v.to_str().ok())
    }

    fn keys(&self) -> Vec<&str> {
        self.0.keys().map(|k| k.as_str()).collect()
    }
}
pub fn extract_context_from_request(req: &HttpRequest) -> Context {
    let propagator = TraceContextPropagator::new();
    propagator.extract(&HeaderExtractor(req.headers()))
}