text: /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
  
  /*  Fluent Bit
   *  ==========
   *  Copyright (C) 2015-2022 The Fluent Bit 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 <fluent-bit/flb_output_plugin.h>
  #include <fluent-bit/flb_utils.h>
  #include <fluent-bit/flb_pack.h>
  #include <fluent-bit/flb_sds.h>
  #include <fluent-bit/flb_kv.h>
  
  #include "opentelemetry.h"
  #include "opentelemetry_conf.h"
  
  static int config_add_labels(struct flb_output_instance *ins,
                               struct opentelemetry_context *ctx)
  {
      struct mk_list *head;
      struct flb_config_map_val *mv;
      struct flb_slist_entry *k = NULL;
      struct flb_slist_entry *v = NULL;
      struct flb_kv *kv;
  
      if (!ctx->add_labels || mk_list_size(ctx->add_labels) == 0) {
          return 0;
      }
  
      /* iterate all 'add_label' definitions */
      flb_config_map_foreach(head, mv, ctx->add_labels) {
          if (mk_list_size(mv->val.list) != 2) {
              flb_plg_error(ins, "'add_label' expects a key and a value, "
                            "e.g: 'add_label version 1.8.0'");
              return -1;
          }
  
          k = mk_list_entry_first(mv->val.list, struct flb_slist_entry, _head);
          v = mk_list_entry_last(mv->val.list, struct flb_slist_entry, _head);
  
          kv = flb_kv_item_create(&ctx->kv_labels, k->str, v->str);
          if (!kv) {
              flb_plg_error(ins, "could not append label %s=%s\n", k->str, v->str);
              return -1;
          }
      }
  
      return 0;
  }
  
  struct opentelemetry_context *flb_opentelemetry_context_create(
      struct flb_output_instance *ins, struct flb_config *config)
  {
      int ret;
      int ulen;
      int io_flags = 0;
      char *protocol = NULL;
      char *host = NULL;
      char *port = NULL;
      char *uri = NULL;
      char *tmp_uri = NULL;
      const char *tmp;
      struct flb_upstream *upstream;
      struct opentelemetry_context *ctx = NULL;
  
      /* Allocate plugin context */
      ctx = flb_calloc(1, sizeof(struct opentelemetry_context));
      if (!ctx) {
          flb_errno();
          return NULL;
      }
      ctx->ins = ins;
      mk_list_init(&ctx->kv_labels);
  
      ret = flb_output_config_map_set(ins, (void *) ctx);
      if (ret == -1) {
          flb_free(ctx);
          return NULL;
      }
  
      /* Parse 'add_label' */
      ret = config_add_labels(ins, ctx);
      if (ret == -1) {
          return NULL;
      }
  
      /*
       * Check if a Proxy have been set, if so the Upstream manager will use
       * the Proxy end-point and then we let the HTTP client know about it, so
       * it can adjust the HTTP requests.
       */
      tmp = flb_output_get_property("proxy", ins);
      if (tmp) {
          ret = flb_utils_url_split(tmp, &protocol, &host, &port, &uri);
          if (ret == -1) {
              flb_plg_error(ctx->ins, "could not parse proxy parameter: '%s'", tmp);
              flb_free(ctx);
              return NULL;
          }
  
          ctx->proxy_host = host;
          ctx->proxy_port = atoi(port);
          ctx->proxy = tmp;
          flb_free(protocol);
          flb_free(port);
          flb_free(uri);
          uri = NULL;
      }
      else {
          flb_output_net_default("127.0.0.1", 80, ins);
      }
  
      /* Check if SSL/TLS is enabled */
  #ifdef FLB_HAVE_TLS
      if (ins->use_tls == FLB_TRUE) {
          io_flags = FLB_IO_TLS;
      }
      else {
          io_flags = FLB_IO_TCP;
      }
  #else
      io_flags = FLB_IO_TCP;
  #endif
  
      if (ins->host.ipv6 == FLB_TRUE) {
          io_flags |= FLB_IO_IPV6;
      }
  
      if (ctx->proxy) {
          flb_plg_trace(ctx->ins, "Upstream Proxy=%s:%i",
                        ctx->proxy_host, ctx->proxy_port);
          upstream = flb_upstream_create(config,
                                         ctx->proxy_host,
                                         ctx->proxy_port,
                                         io_flags, ins->tls);
      }
      else {
          upstream = flb_upstream_create(config,
                                         ins->host.name,
                                         ins->host.port,
                                         io_flags, ins->tls);
      }
  
      if (!upstream) {
          flb_free(ctx);
          return NULL;
      }
  
      if (ins->host.uri) {
          uri = flb_strdup(ins->host.uri->full);
      }
      else {
          tmp = flb_output_get_property("uri", ins);
          if (tmp) {
              uri = flb_strdup(tmp);
          }
      }
  
      if (!uri) {
          uri = flb_strdup("/");
      }
      else if (uri[0] != '/') {
          ulen = strlen(uri);
          tmp_uri = flb_malloc(ulen + 2);
          tmp_uri[0] = '/';
          memcpy(tmp_uri + 1, uri, ulen);
          tmp_uri[ulen + 1] = '\0';
          flb_free(uri);
          uri = tmp_uri;
      }
  
      ctx->u = upstream;
      ctx->uri = uri;
      ctx->host = ins->host.name;
      ctx->port = ins->host.port;
  
      /* Set instance flags into upstream */
      flb_output_upstream_set(ctx->u, ins);
  
      return ctx;
  }
  
  void flb_opentelemetry_context_destroy(
      struct opentelemetry_context *ctx)
  {
      if (!ctx) {
          return;
      }
  
      flb_kv_release(&ctx->kv_labels);
  
      if (ctx->u) {
          flb_upstream_destroy(ctx->u);
      }
  
      flb_free(ctx->proxy_host);
      flb_free(ctx->uri);
      flb_free(ctx);
  }
  