/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */

/*  CTraces
 *  =======
 *  Copyright 2022 The CTraces Authors
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

#include <ctraces/ctraces.h>
#include <ctraces/ctr_resource.h>

struct ctrace_resource *ctr_resource_create()
{
    struct ctrace_resource *res;
    struct ctrace_attributes *attr;

    res = calloc(1, sizeof(struct ctrace_resource));
    if (!res) {
        ctr_errno();
        return NULL;
    }

    attr = ctr_attributes_create();
    if (!attr) {
        ctr_resource_destroy(res);
        return NULL;
    }
    res->attr = attr;

    return res;
}

struct ctrace_resource *ctr_resource_create_default()
{
    struct ctrace_resource *res;

    res = ctr_resource_create();
    if (!res) {
        return NULL;
    }

    /* some default attributes */
    ctr_attributes_set_string(res->attr, "service.name", "Fluent Bit");
    ctr_attributes_set_int64(res->attr, "release_year", 2014);

    return res;
}

int ctr_resource_set_attributes(struct ctrace_resource *res, struct ctrace_attributes *attr)
{
    if (!attr) {
        return -1;
    }

    if (res->attr) {
        ctr_attributes_destroy(res->attr);
    }

    res->attr = attr;
    return 0;
}

void ctr_resource_set_dropped_attr_count(struct ctrace_resource *res, uint32_t count)
{
    res->dropped_attr_count = count;
}

void ctr_resource_destroy(struct ctrace_resource *res)
{
    if (res->attr) {
        ctr_attributes_destroy(res->attr);
    }
    free(res);
}

/*
 * resource_span API
 * -----------------
 */

/* creates a resource_span context */
struct ctrace_resource_span *ctr_resource_span_create(struct ctrace *ctx)
{
    struct ctrace_resource_span *resource_span;

    resource_span = calloc(1, sizeof(struct ctrace_resource_span));
    if (!resource_span) {
        ctr_errno();
        return NULL;
    }
    cfl_list_init(&resource_span->scope_spans);

    /* link to ctraces context */
    cfl_list_add(&resource_span->_head, &ctx->resource_spans);

    /* create an empty resource */
    resource_span->resource = ctr_resource_create();
    if (!resource_span->resource) {
        free(resource_span);
        return NULL;
    }

    return resource_span;
}

struct ctrace_resource *ctr_resource_span_get_resource(struct ctrace_resource_span *resource_span)
{
    return resource_span->resource;
}

/* Set the schema_url for a resource_span */
int ctr_resource_span_set_schema_url(struct ctrace_resource_span *resource_span, char *url)
{
    if (resource_span->schema_url) {
        cfl_sds_destroy(resource_span->schema_url);
    }

    resource_span->schema_url = cfl_sds_create(url);
    if (!resource_span->schema_url) {
        return -1;
    }

    return 0;
}

void ctr_resource_span_destroy(struct ctrace_resource_span *resource_span)
{
    struct cfl_list *tmp;
    struct cfl_list *head;
    struct ctrace_scope_span *scope_span;

    /* release resource if set */
    if (resource_span->resource) {
        ctr_resource_destroy(resource_span->resource);
    }

    if (resource_span->schema_url) {
        cfl_sds_destroy(resource_span->schema_url);
    }

    /* remove scope spans */
    cfl_list_foreach_safe(head, tmp, &resource_span->scope_spans) {
        scope_span = cfl_list_entry(head, struct ctrace_scope_span, _head);
        ctr_scope_span_destroy(scope_span);
    }

    free(resource_span);
}