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

#pragma once

#include <bpfcore/vmlinux.h>
#include <bpfcore/bpf_helpers.h>
#include <bpfcore/bpf_builtins.h>

#include <common/globals.h>
#include <common/tp_info.h>

typedef struct obi_ctx_info {
    unsigned char trace_id[TRACE_ID_SIZE_BYTES];
    unsigned char span_id[SPAN_ID_SIZE_BYTES];
} obi_ctx_info_t;

// NOTE: this map spec is part of an OTEP (https://github.com/open-telemetry/opentelemetry-specification/pull/4855).
// Changing its spec may break other components relying on it.
struct {
    __uint(type, BPF_MAP_TYPE_LRU_HASH);
    __type(key, u64);
    __type(value, obi_ctx_info_t);
    __uint(max_entries, 1 << 14);
    __uint(pinning, LIBBPF_PIN_BY_NAME);
} traces_ctx_v1 SEC(".maps");

static __always_inline obi_ctx_info_t *obi_ctx__get(const u64 pid_tgid) {
    return bpf_map_lookup_elem(&traces_ctx_v1, &pid_tgid);
}

static __always_inline long obi_ctx__set(const u64 pid_tgid, const tp_info_t *info) {
    if (!g_traces_ctx_v1_enabled) {
        return 0;
    }
    obi_ctx_info_t obi_info = {};
    bpf_memcpy(obi_info.trace_id, info->trace_id, TRACE_ID_SIZE_BYTES);
    bpf_memcpy(obi_info.span_id, info->span_id, SPAN_ID_SIZE_BYTES);
    return bpf_map_update_elem(&traces_ctx_v1, &pid_tgid, &obi_info, BPF_ANY);
}

static __always_inline long obi_ctx__del(const u64 pid_tgid) {
    if (!g_traces_ctx_v1_enabled) {
        return 0;
    }
    return bpf_map_delete_elem(&traces_ctx_v1, &pid_tgid);
}