koreader/frontend/ui/widget/screenshoter.lua

local BD = require("ui/bidi")
local ButtonDialog = require("ui/widget/buttondialog")
local DataStorage = require("datastorage")
local GestureRange = require("ui/gesturerange")
local InputContainer = require("ui/widget/container/inputcontainer")
local UIManager = require("ui/uimanager")
local filemanagerutil = require("apps/filemanager/filemanagerutil")
local Screen = require("device").screen
local _ = require("gettext")

local Screenshoter = InputContainer:extend{
    prefix = "Screenshot",
    default_dir = DataStorage:getFullDataDir() .. "/screenshots",
}

function Screenshoter:init()
    local diagonal = math.sqrt(Screen:getWidth()^2 + Screen:getHeight()^2)
    self.ges_events = {
        TapDiagonal = {
            GestureRange:new{
                ges = "two_finger_tap",
                scale = {diagonal - Screen:scaleBySize(200), diagonal},
                rate = 1.0,
            }
        },
        SwipeDiagonal = {
            GestureRange:new{
                ges = "swipe",
                scale = {diagonal - Screen:scaleBySize(200), diagonal},
                rate = 1.0,
            }
        },
    }
end

function Screenshoter:getScreenshotDir()
    local screenshot_dir = G_reader_settings:readSetting("screenshot_dir")
    return screenshot_dir and screenshot_dir:gsub("/$", "") or self.default_dir
end

function Screenshoter:onScreenshot(screenshot_name, caller_callback)
    if not screenshot_name then
        screenshot_name = os.date(self:getScreenshotDir() .. "/" .. self.prefix .. "_%Y-%m-%d_%H%M%S.png")
    end
    Screen:shot(screenshot_name)
    local file = self.ui and self.ui.document and self.ui.document.file -- currently opened book
    local dialog
    local buttons = {
        {
            {
                text = _("Delete"),
                callback = function()
                    os.remove(screenshot_name)
                    dialog:onClose()
                end,
            },
            {
                text = _("Set as book cover"),
                enabled = file and true or false,
                callback = function()
                    self.ui.bookinfo:setCustomCoverFromImage(file, screenshot_name)
                    os.remove(screenshot_name)
                    dialog:onClose()
                end,
            },
        },
        {
            {
                text = _("View"),
                callback = function()
                    local ImageViewer = require("ui/widget/imageviewer")
                    local image_viewer = ImageViewer:new{
                        file = screenshot_name,
                        modal = true,
                        with_title_bar = false,
                        buttons_visible = true,
                    }
                    UIManager:show(image_viewer)
                end,
            },
            {
                text = _("Set as screensaver"),
                callback = function()
                    G_reader_settings:saveSetting("screensaver_type", "image_file")
                    G_reader_settings:saveSetting("screensaver_image", screenshot_name)
                    dialog:onClose()
                end,
            },
        },
    }
    dialog = ButtonDialog:new{
        title = _("Screenshot saved to:") .. "\n\n" .. BD.filepath(screenshot_name) .. "\n",
        buttons = buttons,
        tap_close_callback = function()
            if caller_callback then
                caller_callback()
            end
            local current_path = self.ui and self.ui.file_chooser and self.ui.file_chooser.path
            if current_path and current_path .. "/" == screenshot_name:match(".*/") then
                self.ui.file_chooser:refreshPath()
            end
        end,
    }
    UIManager:show(dialog)
    -- trigger full refresh
    UIManager:setDirty(nil, "full")
    return true
end

function Screenshoter:chooseFolder()
    local title_header = _("Current screenshot folder:")
    local current_path = G_reader_settings:readSetting("screenshot_dir")
    local default_path = self.default_dir
    local caller_callback = function(path)
        G_reader_settings:saveSetting("screenshot_dir", path)
    end
    filemanagerutil.showChooseDialog(title_header, caller_callback, current_path, default_path)
end

function Screenshoter:onTapDiagonal()
    return self:onScreenshot()
end

function Screenshoter:onSwipeDiagonal()
    return self:onScreenshot()
end

return Screenshoter