From 88d9cdf533c98580f265bab2c7eb4578c8e24937 Mon Sep 17 00:00:00 2001
From: chrox <chrox.huang@gmail.com>
Date: Wed, 10 Sep 2014 13:26:50 +0800
Subject: [PATCH] add shortcut to add local calibre OPDS catalog

---
 frontend/ui/widget/opdsbrowser.lua | 83 +++++++++++++++++++++++++++++-
 1 file changed, 81 insertions(+), 2 deletions(-)

diff --git a/frontend/ui/widget/opdsbrowser.lua b/frontend/ui/widget/opdsbrowser.lua
index 378be08e6..a4953294f 100644
--- a/frontend/ui/widget/opdsbrowser.lua
+++ b/frontend/ui/widget/opdsbrowser.lua
@@ -36,6 +36,8 @@ local CatalogCache = Cache:new{
 
 local OPDSBrowser = Menu:extend{
     opds_servers = {},
+    calibre_name = _("Local calibre catalog"),
+
     catalog_type = "application/atom%+xml",
     search_type = "application/opensearchdescription%+xml",
     acquisition_rel = "http://opds-spec.org/acquisition",
@@ -71,6 +73,19 @@ function OPDSBrowser:addServerFromInput(fields)
     self:init()
 end
 
+function OPDSBrowser:editCalibreFromInput(fields)
+    DEBUG("input calibre server", fields)
+    local calibre = G_reader_settings:readSetting("calibre_opds") or {}
+    if fields[1] then
+        calibre.host = fields[1]
+    end
+    if tonumber(fields[2]) then
+        calibre.port = fields[2]
+    end
+    G_reader_settings:saveSetting("calibre_opds", calibre)
+    self:init()
+end
+
 function OPDSBrowser:addNewCatalog()
     self.add_server_dialog = MultiInputDialog:new{
         title = _("Add OPDS catalog"),
@@ -110,6 +125,47 @@ function OPDSBrowser:addNewCatalog()
     UIManager:show(self.add_server_dialog)
 end
 
+function OPDSBrowser:editCalibreServer()
+    local calibre = G_reader_settings:readSetting("calibre_opds") or {}
+    self.add_server_dialog = MultiInputDialog:new{
+        title = _("Edit local calibre host and port"),
+        fields = {
+            {
+                -- TODO: get IP address of current device
+                text = calibre.host or "192.168.1.1",
+                hint = _("Calibre host"),
+            },
+            {
+                text = calibre.port and tostring(calibre.port) or "8080",
+                hint = _("Calibre port"),
+            },
+        },
+        buttons = {
+            {
+                {
+                    text = _("Cancel"),
+                    callback = function()
+                        self.add_server_dialog:onClose()
+                        UIManager:close(self.add_server_dialog)
+                    end
+                },
+                {
+                    text = _("Apply"),
+                    callback = function()
+                        self.add_server_dialog:onClose()
+                        UIManager:close(self.add_server_dialog)
+                        self:editCalibreFromInput(MultiInputDialog:getFields())
+                    end
+                },
+            },
+        },
+        width = Screen:getWidth() * 0.95,
+        height = Screen:getHeight() * 0.2,
+    }
+    self.add_server_dialog:onShowKeyboard()
+    UIManager:show(self.add_server_dialog)
+end
+
 function OPDSBrowser:genItemTableFromRoot()
     local item_table = {}
     for i, server in ipairs(self.opds_servers) do
@@ -131,6 +187,25 @@ function OPDSBrowser:genItemTableFromRoot()
             editable = true,
         })
     end
+    local calibre_opds = G_reader_settings:readSetting("calibre_opds") or {}
+    local calibre_callback = nil
+    if not calibre_opds.host or not calibre_opds.port then
+        table.insert(item_table, {
+            text = self.calibre_name,
+            callback = function()
+                self:editCalibreServer()
+            end,
+            deletable = false,
+        })
+    else
+        table.insert(item_table, {
+            text = self.calibre_name,
+            url = string.format("http://%s:%d/opds",
+                calibre_opds.host, calibre_opds.port),
+            editable = true,
+            deletable = false,
+        })
+    end
     table.insert(item_table, {
         text = _("Add new OPDS catalog"),
         callback = function()
@@ -458,7 +533,7 @@ function OPDSBrowser:editOPDSServer(item)
                     end
                 },
                 {
-                    text = _("Edit"),
+                    text = _("Apply"),
                     callback = function()
                         self.edit_server_dialog:onClose()
                         UIManager:close(self.edit_server_dialog)
@@ -496,7 +571,11 @@ function OPDSBrowser:onMenuHold(item)
                         enabled = item.editable,
                         callback = function()
                             UIManager:close(self.opds_server_dialog)
-                            self:editOPDSServer(item)
+                            if item.text ~= self.calibre_name then
+                                self:editOPDSServer(item)
+                            else
+                                self:editCalibreServer(item)
+                            end
                         end
                     },
                     {