r_dash2ts.cpp

#include <unistd.h>
#include <iomanip>
#include "streamplayer.h"
#include "addonhandler.h"
#include "logger.h"

extern bool saveonly;
extern bool use_TCP;

extern std::string headers;

std::string &Trim(std::string &, const char *const);

std::string urlEncode(const std::string& str) {
    std::ostringstream encodedStream;
    encodedStream << std::hex << std::uppercase << std::setfill('0');

    for (char c : str) {
        if (std::isalnum(c) || c == '-' || c == '_' || c == '.' || c == '~') {
            encodedStream << c;
        } else {
            encodedStream << '%' << std::setw(2) << static_cast<unsigned int>(static_cast<unsigned char>(c));
        }
    }

    return encodedStream.str();
}

void usage() {
    printf("Usage: r_dash2ts -u url_to_manifest.mpd\n");
    printf("                 -a user_agent\n");
    printf("                 -c cookies\n");
    printf("                 -r referer\n");
    printf("                 -k path_to_kodi\n");
    printf("                 -p probeFfmpeg only\n");
    printf("                 -s start position in seconds\n");
    printf("                 [-l <level>] \n");
}

int main(int argc, char *argv[]) {
    std::string url;
    std::string path_to_kodi;
    std::string user_agent;
    std::string cookies;
    std::string referer;
    int api_version = 0;
    int loglevel = 1;
    bool probeOnly = false;
    int startPosition;

    use_TCP = false;

    int c;
    while ((c = getopt(argc, argv, "u:a:c:r:k:l:ps:")) != -1) {
        switch (c) {
            case 'u': // URL to Manifest
                url = std::string(optarg);
                continue;

            case 'a': // user agent
                user_agent = std::string(optarg);
                continue;

            case 'c': // cookies
                cookies = std::string(optarg);
                continue;

            case 'r': // referer
                referer = std::string(optarg);
                continue;

            case 'k': // path to kodi
                path_to_kodi = std::string(optarg);
                continue;

            case 'l': // loglevel
                loglevel = atoi(optarg);
                continue;

            case 'p': // probe only
                probeOnly = true;
                continue;

            case 's': // start position
                startPosition = atoi(optarg);
                continue;

            default:
                usage();
                exit(EXIT_FAILURE);
        }
        break;
    }

    if (path_to_kodi.empty()) {
        // set default for VDR*ELEC
        path_to_kodi = "/storage/.kodi";
    }

    if (url.empty() || user_agent.empty() || cookies.empty() || referer.empty()) {
        usage();
        exit(EXIT_FAILURE);
    }

    switch (loglevel) {
        case 0:  logger.set_level(spdlog::level::critical); break;
        case 1:  logger.set_level(spdlog::level::err); break;
        case 2:  logger.set_level(spdlog::level::info); break;
        case 3:  logger.set_level(spdlog::level::debug); break;
        case 4:  logger.set_level(spdlog::level::trace); break;
        default: logger.set_level(spdlog::level::err); break;
    }

    auto handler = new AddonHandler(path_to_kodi); // Init AddonHandler

    // Load the inputstream.adaptive Library and get the API Version
    api_version = handler->LoadAddon();

    INFO("Found inputstream.adaptive Library, version {}", api_version);

    std::string httpHeader = "Referer=" + referer + "&User-Agent=" + user_agent + "&cookie=" + urlEncode(cookies);

    // kodi 22+
    handler->AddProp("inputstream.adaptive.common_headers", httpHeader.c_str());

    // kodi 20+
    // handler->AddProp("inputstream.adaptive.stream_params", "paramname=value&paramname2=value2");
    handler->AddProp("inputstream.adaptive.manifest_headers", httpHeader.c_str());

    // kodi 21+
    handler->AddProp("inputstream.adaptive.stream_headers", httpHeader.c_str());

    // all
    handler->AddProp("inputstream.adaptive.config", "{\"internal_cookies\":true}");

    handler->SetResolution(1920, 1080);

    bool ret = handler->OpenURL(const_cast<char *>(url.c_str()));

    if (ret) {
        INFO("Open URL {}");
        if (probeOnly) {
            // print result and exit
            printf("dash2ts result:%d:%d:%d:%ld:%ld:%d,\n", handler->GetTotalTime(0, true), handler->GetTime(), handler->GetCapabilities(), handler->PositionStream(), handler->LengthStream(), handler->IsRealTimeStream() ? 1 : 0);
            exit(EXIT_SUCCESS);
        }
    } else {
        ERROR("Unable to open URL {}", url);
        exit(EXIT_FAILURE);
    }

    // start streaming
    auto player = new StreamPlayer(startPosition);
    player->StreamPlay(handler);

    delete handler;
    delete player;

    exit(EXIT_SUCCESS);
}