Initial commit for GitHub migration based on spice2x-25-03-03

src/spice2x/api/modules/capture.cpp

@@ -0,0 +1,82 @@
+#include "capture.h"
+#include <functional>
+#include "external/rapidjson/document.h"
+#include "hooks/graphics/graphics.h"
+#include "util/crypt.h"
+
+using namespace std::placeholders;
+using namespace rapidjson;
+
+namespace api::modules {
+
+    static thread_local std::vector<uint8_t> CAPTURE_BUFFER;
+
+    Capture::Capture() : Module("capture") {
+        functions["get_screens"] = std::bind(&Capture::get_screens, this, _1, _2);
+        functions["get_jpg"] = std::bind(&Capture::get_jpg, this, _1, _2);
+    }
+
+    /**
+     * get_screens()
+     */
+    void Capture::get_screens(Request &req, Response &res) {
+
+        // aquire screens
+        std::vector<int> screens;
+        graphics_screens_get(screens);
+
+        // add screens to response
+        for (auto &screen : screens) {
+            res.add_data(screen);
+        }
+    }
+
+    /**
+     * get_jpg([screen=0, quality=70, downscale=0, divide=1])
+     * screen: uint specifying the window
+     * quality: uint in range [0, 100]
+     * reduce: uint for dividing image size
+     */
+    void Capture::get_jpg(Request &req, Response &res) {
+        CAPTURE_BUFFER.reserve(1024 * 128);
+
+        // settings
+        int screen = 0;
+        int quality = 70;
+        int divide = 1;
+        if (req.params.Size() > 0 && req.params[0].IsUint())
+            screen = req.params[0].GetUint();
+        if (req.params.Size() > 1 && req.params[1].IsUint())
+            quality = req.params[1].GetUint();
+        if (req.params.Size() > 2 && req.params[2].IsUint())
+            divide = req.params[2].GetUint();
+
+        // receive JPEG data
+        uint64_t timestamp = 0;
+        int width = 0;
+        int height = 0;
+        graphics_capture_trigger(screen);
+        bool success = graphics_capture_receive_jpeg(screen, [] (uint8_t byte) {
+            CAPTURE_BUFFER.push_back(byte);
+        }, true, quality, true, divide, &timestamp, &width, &height);
+        if (!success) {
+            return;
+        }
+
+        // encode to base64
+        auto encoded = crypt::base64_encode(
+                CAPTURE_BUFFER.data(),
+                CAPTURE_BUFFER.size());
+
+        // clear buffer
+        CAPTURE_BUFFER.clear();
+
+        // add data to response
+        Value data;
+        data.SetString(encoded.c_str(), encoded.length(), res.doc()->GetAllocator());
+        res.add_data(timestamp);
+        res.add_data(width);
+        res.add_data(height);
+        res.add_data(data);
+    }
+}