From 4c9231a8f85ce93dc1f047f5b052ab5cfe4f8891 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A4=8F=E9=B2=81=E8=B1=AB?= <fiercex@outlook.com>
Date: Thu, 18 May 2023 19:10:09 +0800
Subject: [PATCH] [plugin] Exporter: add memos export (#10411)

Add an export method for memos, which originates from [flomo](https://flomoapp.com/) and now has an open-source self-deployment solution:[memos](https://github.com/usememos/memos)
---
 plugins/exporter.koplugin/main.lua         |   3 +-
 plugins/exporter.koplugin/target/memos.lua | 129 +++++++++++++++++++++
 2 files changed, 131 insertions(+), 1 deletion(-)
 create mode 100644 plugins/exporter.koplugin/target/memos.lua

diff --git a/plugins/exporter.koplugin/main.lua b/plugins/exporter.koplugin/main.lua
index aedcc4242..224b7d1f3 100644
--- a/plugins/exporter.koplugin/main.lua
+++ b/plugins/exporter.koplugin/main.lua
@@ -38,7 +38,7 @@ local _ = require("gettext")
 
 -- migrate settings from old "evernote.koplugin" or from previous (monolithic) "exporter.koplugin"
 local function migrateSettings()
-    local formats = { "html", "joplin", "json", "readwise", "text", "my_clippings" }
+    local formats = { "html", "joplin", "json", "memos", "my_clippings", "readwise", "text" }
 
     local settings = G_reader_settings:readSetting("exporter")
     if not settings then
@@ -104,6 +104,7 @@ local Exporter = WidgetContainer:extend{
         markdown = require("target/markdown"),
         readwise = require("target/readwise"),
         text = require("target/text"),
+        memos = require("target/memos"),
         my_clippings = require("target/my_clippings"),
     },
 }
diff --git a/plugins/exporter.koplugin/target/memos.lua b/plugins/exporter.koplugin/target/memos.lua
new file mode 100644
index 000000000..d3c65ab93
--- /dev/null
+++ b/plugins/exporter.koplugin/target/memos.lua
@@ -0,0 +1,129 @@
+local InputDialog = require("ui/widget/inputdialog")
+local UIManager = require("ui/uimanager")
+local http = require("socket.http")
+local json = require("json")
+local logger = require("logger")
+local ltn12 = require("ltn12")
+local socket = require("socket")
+local socketutil = require("socketutil")
+local _ = require("gettext")
+
+-- readwise exporter
+local MemosExporter = require("base"):new {
+    name = "memos",
+    is_remote = true,
+}
+
+local function makeRequest(method, request_body, api)
+    local sink = {}
+    local request_body_json = json.encode(request_body)
+    local source = ltn12.source.string(request_body_json)
+    socketutil:set_timeout(socketutil.LARGE_BLOCK_TIMEOUT, socketutil.LARGE_TOTAL_TIMEOUT)
+    local request = {
+        url     = api,
+        method  = method,
+        sink    = ltn12.sink.table(sink),
+        source  = source,
+        headers = {
+            ["Content-Type"] = "application/json"
+        },
+    }
+    local code, headers, status = socket.skip(1, http.request(request))
+    socketutil:reset_timeout()
+
+    if code ~= 200 then
+        logger.warn("Memos: HTTP response code <> 200. Response status:", status or code or "network unreachable")
+        logger.dbg("Response headers:", headers)
+        return nil, status
+    end
+
+    local response = json.decode(sink[1])
+    return response
+end
+
+function MemosExporter:isReadyToExport()
+    if self.settings.api then return true end
+    return false
+end
+
+function MemosExporter:getMenuTable()
+    return {
+        text = _("Memos"),
+        checked_func = function() return self:isEnabled() end,
+        sub_item_table = {
+            {
+                text = _("Set Memos API URL"),
+                keep_menu_open = true,
+                callback = function()
+                    local auth_dialog
+                    auth_dialog = InputDialog:new {
+                        title = _("Set API URL for Memos"),
+                        input = self.settings.api,
+                        buttons = {
+                            {
+                                {
+                                    text = _("Cancel"),
+                                    id = "close",
+                                    callback = function()
+                                        UIManager:close(auth_dialog)
+                                    end
+                                },
+                                {
+                                    text = _("Set API URL"),
+                                    callback = function()
+                                        self.settings.api = auth_dialog:getInputText()
+                                        self:saveSettings()
+                                        UIManager:close(auth_dialog)
+                                    end
+                                }
+                            }
+                        }
+                    }
+                    UIManager:show(auth_dialog)
+                    auth_dialog:onShowKeyboard()
+                end
+            },
+            {
+                text = _("Export to Memos"),
+                checked_func = function() return self:isEnabled() end,
+                callback = function() self:toggleEnabled() end,
+            },
+
+        }
+    }
+end
+
+function MemosExporter:createHighlights(booknotes)
+    local number = 0
+    local error_number = 0
+    for _, chapter in ipairs(booknotes) do
+        for _, clipping in ipairs(chapter) do
+            local highlight = clipping.text .. "\n\n"
+            if clipping.note then
+                highlight = highlight .. clipping.note .. "\n\n"
+            end
+            highlight =  highlight .. booknotes.title .. " (page: " .. clipping.page .. "）\n\n #" .. booknotes.title .. " #koreader"
+            local result, err = makeRequest("POST", { content = highlight }, self.settings.api)
+            if not result then
+                logger.warn("error creating highlights", err)
+                error_number = error_number + 1
+            end
+            number = number + 1
+        end
+    end
+    local success_number = number - error_number
+    logger.dbg("createHighlights success number: " .. success_number .. " createHighlights error number: " .. error_number)
+    return true
+end
+
+function MemosExporter:export(t)
+    if not self:isReadyToExport() then return false end
+
+    for _, booknotes in ipairs(t) do
+        local ok = self:createHighlights(booknotes)
+        if not ok then return false end
+    end
+    return true
+end
+
+return MemosExporter