/*
 * Copyright (c) 2018 Cisco and/or its affiliates.
 * 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 "vom/stat_client.hpp"

namespace VOM {

stat_client::stat_data_t::stat_data_t(const stat_segment_data_t& stat_seg_data)
  : m_name(stat_seg_data.name)
  , m_type(stat_seg_data.type)
{
  switch (m_type) {
    case STAT_DIR_TYPE_SCALAR_INDEX:
      m_scalar_value = stat_seg_data.scalar_value;
      break;
    case STAT_DIR_TYPE_COUNTER_VECTOR_SIMPLE:
      m_simple_counter_vec = stat_seg_data.simple_counter_vec;
      break;
    case STAT_DIR_TYPE_COUNTER_VECTOR_COMBINED:
      m_combined_counter_vec = stat_seg_data.combined_counter_vec;
      break;
    case STAT_DIR_TYPE_ERROR_INDEX:
      m_error_value = stat_seg_data.error_value;
      break;
    case STAT_DIR_TYPE_ILLEGAL:
      break;
  }
}

const std::string&
stat_client::stat_data_t::name() const
{
  return m_name;
}

const stat_directory_type_t&
stat_client::stat_data_t::type() const
{
  return m_type;
}

double
stat_client::stat_data_t::get_stat_segment_scalar_data() const
{
  return m_scalar_value;
}

uint64_t
stat_client::stat_data_t::get_stat_segment_error_data() const
{
  return m_error_value;
}

uint64_t**
stat_client::stat_data_t::get_stat_segment_simple_counter_data() const
{
  return m_simple_counter_vec;
}

vlib_counter_t**
stat_client::stat_data_t::get_stat_segment_combined_counter_data() const
{
  return m_combined_counter_vec;
}

stat_client::stat_client(std::string& socket_name)
  : m_socket_name(socket_name)
  , m_patterns()
  , m_stat_connect(0)
  , m_counter_vec()
  , m_stat_seg_data(nullptr)
  , m_stat_data()
{
  m_patterns.push_back("/if");
}

stat_client::stat_client(std::vector<std::string>& pattern)
  : m_socket_name("/run/vpp/stats.sock")
  , m_patterns(pattern)
  , m_stat_connect(0)
  , m_counter_vec()
  , m_stat_seg_data(nullptr)
  , m_stat_data()
{
}

stat_client::stat_client(std::string socket_name,
                         std::vector<std::string> patterns)
  : m_socket_name(socket_name)
  , m_patterns(patterns)
  , m_stat_connect(0)
  , m_counter_vec()
  , m_stat_seg_data(nullptr)
  , m_stat_data()
{
}

stat_client::stat_client()
  : m_socket_name("/run/vpp/stats.sock")
  , m_patterns()
  , m_stat_connect(0)
  , m_counter_vec()
  , m_stat_seg_data(nullptr)
  , m_stat_data()
{
  m_patterns.push_back("/if");
}

stat_client::~stat_client()
{
  stat_segment_vec_free(m_counter_vec);
  data_free();
  if (m_stat_connect)
    stat_segment_disconnect();
}

stat_client::stat_client(const stat_client& o)
  : m_socket_name(o.m_socket_name)
  , m_patterns(o.m_patterns)
{
}

int
stat_client::connect()
{
  if (stat_segment_connect(m_socket_name.c_str()) == 0) {
    m_stat_connect = 1;
    ls();
  }
  return m_stat_connect;
}

void
stat_client::disconnect()
{
  if (m_stat_connect)
    stat_segment_disconnect();
}

int
stat_client::vec_len(void* vec)
{
  return stat_segment_vec_len(vec);
}

void
stat_client::vec_free(void* vec)
{
  stat_segment_vec_free(vec);
}

void
stat_client::ls()
{
  uint8_t** string_vec = { 0 };
  for (auto& pattern : m_patterns) {
    string_vec = stat_segment_string_vector(string_vec, pattern.c_str());
  }
  m_counter_vec = stat_segment_ls(string_vec);
  stat_segment_vec_free(string_vec);
}

const stat_client::stat_data_vec_t&
stat_client::dump()
{
  stat_segment_data_free(m_stat_seg_data);
  if (m_stat_data.size()) {
    m_stat_data.clear();
  }
  m_stat_seg_data = stat_segment_dump(m_counter_vec);
  if (!m_stat_seg_data) {
    ls();
    return m_stat_data;
  }
  for (int i = 0; i < stat_segment_vec_len(m_stat_seg_data); i++) {
    stat_data_t sd(m_stat_seg_data[i]);
    m_stat_data.push_back(sd);
  }
  return m_stat_data;
}

const stat_client::stat_data_vec_t&
stat_client::dump_entry(uint32_t index)
{
  stat_segment_data_free(m_stat_seg_data);
  if (m_stat_data.size()) {
    m_stat_data.clear();
  }
  m_stat_seg_data = stat_segment_dump_entry(index);
  if (!m_stat_seg_data) {
    ls();
    return m_stat_data;
  }
  for (int i = 0; i < stat_segment_vec_len(m_stat_seg_data); i++) {
    stat_data_t sd(m_stat_seg_data[i]);
    m_stat_data.push_back(sd);
  }
  return m_stat_data;
}

void
stat_client::data_free()
{
  stat_segment_data_free(m_stat_seg_data);
}

double
stat_client::heartbeat()
{
  return stat_segment_heartbeat();
}

std::string
stat_client::index_to_name(uint32_t index)
{
  return stat_segment_index_to_name(index);
}

} // namespace VOM

/*
 * fd.io coding-style-patch-verification: ON
 *
 * Local Variables:
 * eval: (c-set-style "mozilla")
 * End:
 */