// Copyright 2021 The OPA Authors.  All rights reserved.
// Use of this source code is governed by an Apache2
// license that can be found in the LICENSE file.

// Package tracing enables dependency-injection at runtime. When used
// together with an underscore-import of `github.com/open-policy-agent/opa/features/tracing`,
// the server and its runtime will emit OpenTelemetry spans to the
// configured sink.
package tracing

import (
	"net/http"

	v1 "github.com/open-policy-agent/opa/v1/tracing"
)

// Options are options for the HTTPTracingService, passed along as-is.
type Options = v1.Options

// NewOptions is a helper method for constructing `tracing.Options`
func NewOptions(opts ...any) Options {
	return v1.NewOptions(opts...)
}

// HTTPTracingService defines how distributed tracing comes in, server- and client-side
type HTTPTracingService = v1.HTTPTracingService

// RegisterHTTPTracing enables a HTTPTracingService for further use.
func RegisterHTTPTracing(ht HTTPTracingService) {
	v1.RegisterHTTPTracing(ht)
}

// NewTransport returns another http.RoundTripper, instrumented to emit tracing
// spans according to Options. Provided by the HTTPTracingService registered with
// this package via RegisterHTTPTracing.
func NewTransport(tr http.RoundTripper, opts Options) http.RoundTripper {
	return v1.NewTransport(tr, opts)
}

// NewHandler returns another http.Handler, instrumented to emit tracing spans
// according to Options. Provided by the HTTPTracingService registered with
// this package via RegisterHTTPTracing.
func NewHandler(f http.Handler, label string, opts Options) http.Handler {
	return v1.NewHandler(f, label, opts)
}
