path: root/src/YtdlpWrapper.hpp

#ifndef YTDLPWRAPPER_H
#define YTDLPWRAPPER_H
#include <string>
#include <pybind11/embed.h>
#include <nlohmann/json.hpp>

namespace Video {

    struct format {
    
        std::string format;
        std::string url;
        std::string vcodec;
        std::string acodec;
        std::string ext;
        std::optional<float> quality;
        std::optional<std::string> resolution;
        std::optional<int> height;
        std::optional<int> width;
    };

    struct thumbnail {

        std::string url;
        int preference;
        std::string id;
        std::optional<std::string> resolution;
        std::optional<int> height;
        std::optional<int> width;
    };

    struct video {

        std::string id;
        std::string title;
        std::string url;
        std::string channelId;
        std::string channelUrl;
        std::vector<format> formats;
        std::vector<thumbnail> thumbnails;
        std::optional<std::string> uploader;
        std::optional<std::string> uploaderId;
        std::optional<std::string> uploaderUrl;
        std::optional<int> duration;
        std::optional<int> viewcount;
        std::optional<std::string> description;
    };

    void to_json(nlohmann::json& j, const format& f);
    void to_json(nlohmann::json& j, const thumbnail& t);
    void to_json(nlohmann::json& j, const video& v);
    void from_json(const nlohmann::json& j, format& f);
    void from_json(const nlohmann::json& j, thumbnail& t);
    void from_json(const nlohmann::json& j, video& v);
};

class YtdlpWrapper {

    public:
        nlohmann::json getJsonSearch(const std::string& searchTerm, int limit = 1);
        Video::video getVideoByUrl(const std::string& url);
        std::vector<Video::video> searchVideos(const std::string& searchterm, int limit = 1);
    private:
        pybind11::object get_ytdl();
        pybind11::object ytdl = pybind11::none();
        pybind11::scoped_interpreter guard{};
};


#endif