kis_credentials.hpp

Go to the documentation of this file.

#pragma once

 
#include <algotradelib/kis>
 
#include <cstdlib>
#include <filesystem>
#include <optional>
#include <ostream>
#include <stdexcept>
#include <string>
#include <string_view>
 
namespace kis_credentials {

inline std::optional<std::filesystem::path> environment_directory() {
  const char* variable = std::getenv("ATL_CREDENTIALS_DIR");
  if (variable != nullptr && *variable == '\0') {
    throw std::runtime_error("ATL_CREDENTIALS_DIR is set but empty");
  }
  return variable != nullptr ? std::optional<std::filesystem::path>(variable) : std::nullopt;
}

inline std::optional<std::filesystem::path> take_directory(int& argc, char** argv) {
  std::optional<std::filesystem::path> directory;
  int kept = 1;
  for (int i = 1; i < argc; ++i) {
    if (std::string_view(argv[i]) == "--credentials") {
      if (i + 1 >= argc || std::string_view(argv[i + 1]).empty() ||
          std::string_view(argv[i + 1]).rfind("--", 0) == 0) {
        throw std::runtime_error("--credentials needs a directory");
      }
      directory = argv[++i];
    } else {
      argv[kept++] = argv[i];
    }
  }
  argc = kept;
  return directory ? directory : environment_directory();
}

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

inline std::filesystem::path required_file(const std::filesystem::path& directory, const char* name) {
  const auto path = directory / "kis" / name;
  if (!std::filesystem::exists(path)) {
    throw std::runtime_error(
      "no " + path.string() + "; the credentials directory is given, so the environment is not read"
    );
  }
  return path;
}

inline void load_config(algotradelib::kis::KisConfig& config, const std::filesystem::path& path, bool demo) {
  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"
    );
  }
}

inline void load(
  const std::optional<std::filesystem::path>& directory,
  algotradelib::kis::KisConfig& config,
  algotradelib::kis::KisAccount* account,
  std::ostream& out
) {
  if (!directory) {
    config.from_env();
    if (account != nullptr) {
      account->from_env();
    }
    out << "credentials: environment variables\n";
    return;
  }
  const bool demo = algotradelib::kis::KisConfig::demo_from_env();
  const auto path = required_file(*directory, demo ? "demo.json" : "real.json");
  load_config(config, path, demo);
  if (account != nullptr) {
    account->from_json(path.string());
  }
  out << "credentials: " << path.string() << '\n';
}

inline std::optional<algotradelib::kis::KisConfig> market_data_config(
  const std::optional<std::filesystem::path>& directory
) {
  if (!directory || !std::filesystem::exists(*directory / "kis" / "real.json")) {
    return std::nullopt;
  }
  algotradelib::kis::KisConfig config;
  load_config(config, *directory / "kis" / "real.json", false);
  return config;
}
 
} // namespace kis_credentials