// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

#pragma once

#include <string>

#include "opentelemetry/common/key_value_iterable.h"
#include "opentelemetry/nostd/string_view.h"
#include "opentelemetry/sdk/trace/sampler.h"
#include "opentelemetry/trace/span_context.h"
#include "opentelemetry/trace/span_context_kv_iterable.h"
#include "opentelemetry/trace/span_metadata.h"
#include "opentelemetry/trace/trace_id.h"

class CustomSampler : public opentelemetry::sdk::trace::Sampler
{
public:
  CustomSampler(const std::string &comment) : comment_(comment) {}
  CustomSampler(CustomSampler &&)                      = delete;
  CustomSampler(const CustomSampler &)                 = delete;
  CustomSampler &operator=(CustomSampler &&)           = delete;
  CustomSampler &operator=(const CustomSampler &other) = delete;
  ~CustomSampler() override                            = default;

  opentelemetry::sdk::trace::SamplingResult ShouldSample(
      const opentelemetry::trace::SpanContext &parent_context,
      opentelemetry::trace::TraceId trace_id,
      opentelemetry::nostd::string_view name,
      opentelemetry::trace::SpanKind span_kind,
      const opentelemetry::common::KeyValueIterable &attributes,
      const opentelemetry::trace::SpanContextKeyValueIterable &links) noexcept override;

  opentelemetry::nostd::string_view GetDescription() const noexcept override;

private:
  std::string comment_;
};