satdump/src-core/common/tracking/tle.h

#pragma once

#include <deque>
#include <string>
#include <optional>
#include "nlohmann/json.hpp"
#include "dll_export.h"

namespace satdump
{
    // Simple struct to hold TLE definitions
    struct TLE
    {
        int norad = -1;
        std::string name;
        std::string line1;
        std::string line2;
    };

    inline void to_json(nlohmann::json &j, const TLE &v)
    {
        j["norad"] = v.norad;
        j["name"] = v.name;
        j["line1"] = v.line1;
        j["line2"] = v.line2;
    }

    inline void from_json(const nlohmann::json &j, TLE &v)
    {
        v.norad = j["norad"].get<int>();
        v.name = j["name"].get<std::string>();
        v.line1 = j["line1"].get<std::string>();
        v.line2 = j["line2"].get<std::string>();
    }

    struct TLEsUpdatedEvent
    {
    };

    // TLE Registry class, which simply extends a std::vector with some QOL functions
    class TLERegistry : public std::vector<TLE>
    {
    public:
        std::optional<TLE> get_from_norad(int norad);
        std::optional<TLE> get_from_norad_time(int norad, time_t timestamp);
    };

    SATDUMP_DLL extern TLERegistry general_tle_registry;

    int parseTLEStream(std::istream &inputStream, TLERegistry &new_registry); // Helper - Takes an input stream and parses out the valid TLEs
    void updateTLEFile(std::string path);                                     // Updates the TLE file now based on the URLs in the config
    void autoUpdateTLE(std::string path);                                     // Updates the TLE file if it's old enough, per user setting
    void loadTLEFileIntoRegistry(std::string path);                           // Loads the TLE file into the general registry

    void fetchTLENow(int norad); // Utils, in case you want to fetch & load a TLE into the regristry right now. Should NOT be used in most cases
}