From e470c5f65fa5605a52cc5d2ddc9cc49c125bb11b Mon Sep 17 00:00:00 2001
From: Suhas Ds <156927454+suhas-cyber@users.noreply.github.com>
Date: Wed, 6 Aug 2025 00:02:57 +0530
Subject: [PATCH] added page capture

---
 frontend/apps/reader/readerui.lua | 56 +++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/frontend/apps/reader/readerui.lua b/frontend/apps/reader/readerui.lua
index df8ed936c..6633dd2fb 100644
--- a/frontend/apps/reader/readerui.lua
+++ b/frontend/apps/reader/readerui.lua
@@ -531,6 +531,62 @@ function ReaderUI:registerKeyEvents()
             end
         end
     end
+    .    -- =================== CORRECT PAGE CAPTURE HOOK ===================
+    -- Add the required modules here, inside the init function
+    local UIManager = require("ui/uimanager")
+    local crengine_hook = require("ffi/crengine-hooks")
+    local logger = require("logger")
+
+    -- Use a small delay to ensure the paging module is fully initialized
+    UIManager:scheduleIn(0.1, function()
+        -- Get a reference to the paging module, which is self.paging
+        local ReaderPaging = self.paging
+        if ReaderPaging and not ReaderPaging.onGotoPageRel_original then
+            -- Store the original function so we can still call it
+            ReaderPaging.onGotoPageRel_original = ReaderPaging.onGotoPageRel
+
+            -- Create our new, wrapped function
+            ReaderPaging.onGotoPageRel = function(self_paging,...)
+                -- First, call the original function to perform the page turn
+                ReaderPaging.onGotoPageRel_original(self_paging,...)
+
+                -- Now, schedule our capture logic to run after the render is complete
+                UIManager:scheduleIn(0.1, function()
+                    -- Enable capture in the C++ backend
+                    crengine_hook.enable_capture(true)
+                    -- Force a repaint to trigger the Draw() method with the capture flag set
+                    self.view:paintTo(UIManager.screen)
+                    UIManager:forceRePaint()
+
+                    -- Schedule the buffer retrieval to run after the repaint has finished
+                    UIManager:scheduleIn(0.1, function()
+                        local current_page_data = crengine_hook.get_current_page_buffer()
+                        if current_page_data then
+                            logger.info("Captured current page buffer, size:", #current_page_data)
+                            pcall(function()
+                                local file = io.open("/tmp/current.raw", "wb")
+                                file:write(current_page_data)
+                                file:close()
+                            end)
+
+                            local next_page_data = crengine_hook.get_next_page_buffer()
+                            if next_page_data then
+                                pcall(function()
+                                    local file_next = io.open("/tmp/next.raw", "wb")
+                                    file_next:write(next_page_data)
+                                    file_next:close()
+                                end)
+                            end
+
+                            crengine_hook.clear_buffers()
+                            logger.info("Cleared C++ page buffers.")
+                        end
+                    end)
+                end)
+            end
+        end
+    end)
+    -- =================================================================
 end
 
 ReaderUI.onPhysicalKeyboardConnected = ReaderUI.registerKeyEvents