kiwoom_credentials.hpp

Go to the documentation of this file.

#pragma once

 
#include "kis_credentials.hpp"
 
#include <algotradelib/kiwoom>
 
#include <filesystem>
#include <optional>
#include <ostream>
#include <stdexcept>
#include <string>
#include <string_view>
 
namespace kiwoom_credentials {

inline bool valid_label(std::string_view label) {
  if (label.empty() || label.front() == '-') {
    return false;
  }
  for (const char c : label) {
    if (!((c >= 'a' && c <= 'z') || (c >= '0' && c <= '9') || c == '-')) {
      return false;
    }
  }
  return true;
}

inline std::optional<std::string> take_label(int& argc, char** argv) {
  std::optional<std::string> label;
  int kept = 1;
  for (int i = 1; i < argc; ++i) {
    if (std::string_view(argv[i]) == "--kiwoom-account") {
      if (i + 1 >= argc || std::string_view(argv[i + 1]).rfind("--", 0) == 0) {
        throw std::runtime_error("--kiwoom-account needs a label, such as kr");
      }
      label = argv[++i];
      if (!valid_label(*label)) {
        throw std::runtime_error("--kiwoom-account takes lowercase letters, digits and '-', not \"" + *label + "\"");
      }
    } else {
      argv[kept++] = argv[i];
    }
  }
  argc = kept;
  return label;
}

inline void refuse_unknown_flags(int argc, char** argv) {
  for (int i = 1; i < argc; ++i) {
    if (std::string_view(argv[i]).rfind("--", 0) == 0) {
      throw std::runtime_error(std::string("unknown option ") + argv[i]);
    }
  }
}

inline algotradelib::network::HttpClientOptions paced_options() {
  algotradelib::network::HttpClientOptions options;
  options.rate_limit_ = algotradelib::network::RateLimitBudget{1, 1, 0};
  return options;
}

inline void load(
  const std::optional<std::filesystem::path>& directory,
  const std::optional<std::string>& label,
  algotradelib::kiwoom::KiwoomConfig& config,
  std::ostream& out
) {
  if (!directory) {
    if (label) {
      throw std::runtime_error(
        "--kiwoom-account selects a file in a credentials directory; give --credentials DIR or set "
        "ATL_CREDENTIALS_DIR"
      );
    }
    config.from_env();
    out << "credentials: environment variables\n";
    return;
  }
  const std::string name = label.value_or("kr");
  const bool demo = algotradelib::kiwoom::KiwoomConfig::demo_from_env();
  const auto path = *directory / "kiwoom" / ((demo ? "demo-" : "real-") + name + ".json");
  if (!std::filesystem::exists(path)) {
    throw std::runtime_error(
      "no " + path.string() + "; the credentials directory is given, so the environment is not read"
    );
  }
  config.from_json(path.string());
  if (config.is_demo_ != demo) {
    throw std::runtime_error(
      path.string() + " says \"demo\": " + (config.is_demo_ ? "true" : "false") +
      ", which contradicts its name"
    );
  }
  out << "credentials: " << path.string() << '\n';
  if (name != "kr") {
    out << "note: the Kiwoom adapter calls domestic-stock endpoints only\n";
  }
}
 
}  // namespace kiwoom_credentials