import json
import os

import boto3
import requests
from opentelemetry import metrics, trace

# Get tracer and meter
tracer = trace.get_tracer(__name__)
meter = metrics.get_meter(__name__)

# Create a counter for custom metrics
request_counter = meter.create_counter(
    "test.requests",
    unit="1",
    description="Test request counter"
)

# Initialize S3 client (outside handler for connection reuse)
s3_client = boto3.client("s3")


def lambda_handler(event, context):
    # 1. Custom span with HTTP call - generates Application Signals service metrics
    with tracer.start_as_current_span("my-custom-span") as span:
        span.set_attribute("key1", "value1")
        span.add_event("custom-event", {"detail": "something happened"})
        requests.get("https://aws.amazon.com/")

    # 2. AWS SDK call - generates Application Signals dependency metrics
    s3_client.list_buckets()

    # 3. Custom metric emission - will be converted to EMF
    request_counter.add(1, {"operation": "test"})

    return {
        "statusCode": 200,
        "body": json.dumps({
            "message": "Metrics and traces emitted",
            "function_name": os.environ.get("AWS_LAMBDA_FUNCTION_NAME"),
            "trace_id": os.environ.get("_X_AMZN_TRACE_ID")
        })
    }