// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.

#include "glass/networktables/NTStringChooser.h"

using namespace glass;

NTStringChooserModel::NTStringChooserModel(wpi::StringRef path)
    : NTStringChooserModel{nt::GetDefaultInstance(), path} {}

NTStringChooserModel::NTStringChooserModel(NT_Inst inst, wpi::StringRef path)
    : m_nt{inst},
      m_default{m_nt.GetEntry(path + "/default")},
      m_selected{m_nt.GetEntry(path + "/selected")},
      m_active{m_nt.GetEntry(path + "/active")},
      m_options{m_nt.GetEntry(path + "/options")} {
  m_nt.AddListener(m_default);
  m_nt.AddListener(m_selected);
  m_nt.AddListener(m_active);
  m_nt.AddListener(m_options);
}

void NTStringChooserModel::SetDefault(wpi::StringRef val) {
  nt::SetEntryValue(m_default, nt::Value::MakeString(val));
}

void NTStringChooserModel::SetSelected(wpi::StringRef val) {
  nt::SetEntryValue(m_selected, nt::Value::MakeString(val));
}

void NTStringChooserModel::SetActive(wpi::StringRef val) {
  nt::SetEntryValue(m_active, nt::Value::MakeString(val));
}

void NTStringChooserModel::SetOptions(wpi::ArrayRef<std::string> val) {
  nt::SetEntryValue(m_options, nt::Value::MakeStringArray(val));
}

void NTStringChooserModel::Update() {
  for (auto&& event : m_nt.PollListener()) {
    if (event.entry == m_default) {
      if ((event.flags & NT_NOTIFY_DELETE) != 0) {
        m_defaultValue.clear();
      } else if (event.value && event.value->IsString()) {
        m_defaultValue = event.value->GetString();
      }
    } else if (event.entry == m_selected) {
      if ((event.flags & NT_NOTIFY_DELETE) != 0) {
        m_selectedValue.clear();
      } else if (event.value && event.value->IsString()) {
        m_selectedValue = event.value->GetString();
      }
    } else if (event.entry == m_active) {
      if ((event.flags & NT_NOTIFY_DELETE) != 0) {
        m_activeValue.clear();
      } else if (event.value && event.value->IsString()) {
        m_activeValue = event.value->GetString();
      }
    } else if (event.entry == m_options) {
      if ((event.flags & NT_NOTIFY_DELETE) != 0) {
        m_optionsValue.clear();
      } else if (event.value && event.value->IsStringArray()) {
        m_optionsValue = event.value->GetStringArray();
      }
    }
  }
}

bool NTStringChooserModel::Exists() {
  return m_nt.IsConnected() && nt::GetEntryType(m_options) != NT_UNASSIGNED;
}