# Basic packages for your application
import boto3
import json
import random
import time
import os
import logging

# Add imports for OTel components into the application
from opentelemetry import trace
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
from opentelemetry.trace import SpanKind
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import BatchSpanProcessor
# Import the AWS X-Ray for OTel Python IDs Generator into the application.
from opentelemetry.sdk.extension.aws.trace import AwsXRayIdGenerator
from opentelemetry.instrumentation.botocore import BotocoreInstrumentor
from opentelemetry.sdk.extension.aws.trace.propagation.aws_xray_format import AwsXRayFormat

OTEL_EXPORTER_OTLP_ENDPOINT = os.environ.get('OTEL_EXPORTER_OTLP_ENDPOINT')

OTEL_METRICS_EXPORTER_OTLP_ENDPOINT = os.environ.get('OTEL_METRICS_EXPORTER_OTLP_ENDPOINT')

logging.basicConfig(level=logging.INFO)

# Sends generated traces in the OTLP format to an ADOT Collector running on port 55678
otlp_exporter = OTLPSpanExporter(endpoint=OTEL_EXPORTER_OTLP_ENDPOINT, insecure=True)
# Processes traces in batches as opposed to immediately one after the other
span_processor = BatchSpanProcessor(otlp_exporter)
# Configures the Global Tracer Provider
trace.set_tracer_provider(TracerProvider(active_span_processor=span_processor, id_generator=AwsXRayIdGenerator()))

BotocoreInstrumentor().instrument()

tracer = trace.get_tracer(__name__)

random.seed(time.time())
if __name__ == "__main__":
  sqs_client = boto3.client("sqs")
  sqs_queue = os.environ.get('QUEUE_NAME')
  queue_url = sqs_client.get_queue_url(
        QueueName=sqs_queue
    )

  queue_url = queue_url['QueueUrl']

  while True:
    with tracer.start_as_current_span("get message from queue"):
      messages = sqs_client.receive_message(
        QueueUrl=queue_url,
        MaxNumberOfMessages= 1
      )
      for message in messages['Messages']:
        logging.info("recieved message %s", message['Body'])
        sqs_client.delete_message(
            QueueUrl=queue_url,
            ReceiptHandle=message['ReceiptHandle']
        )

    time.sleep(5)