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, token) 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", ["Authorization"] = "Bearer " .. token }, } 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 and self.settings.token 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 = _("Set Memos token"), keep_menu_open = true, callback = function() local token_dialog token_dialog = InputDialog:new { title = _("Memos token"), input = self.settings.token, buttons = { { { text = _("Cancel"), id = "close", callback = function() UIManager:close(token_dialog) end }, { text = _("Set token"), callback = function() self.settings.token = token_dialog:getInputText() self:saveSettings() UIManager:close(token_dialog) end } } } } UIManager:show(token_dialog) token_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, self.settings.token) 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