From 69f79f163def3377118de6946a328a13b3533254 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Tue, 21 Oct 2008 18:47:35 -0700
Subject: [PATCH] Add support for user stylesheets to ebook-viewer

---
 src/calibre/ebooks/html.py              |  2 +-
 src/calibre/gui2/viewer/config.ui       | 14 +++++++++++++-
 src/calibre/gui2/viewer/documentview.py | 20 ++++++++++++++++++--
 src/calibre/gui2/viewer/main.py         |  3 +--
 4 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/src/calibre/ebooks/html.py b/src/calibre/ebooks/html.py
index 02e6a1b580..dc5d33d93a 100644
--- a/src/calibre/ebooks/html.py
+++ b/src/calibre/ebooks/html.py
@@ -804,7 +804,7 @@ class Processor(Parser):
         # Workaround for anchor rendering bug in ADE
         css += '\n\na { color: inherit; text-decoration: inherit; cursor: default; }\na[href] { color: blue; text-decoration: underline; cursor:pointer; }'
         if self.opts.remove_paragraph_spacing:
-            css += '\n\np {text-indent: 2.1em; margin-top:1pt; margin-bottom:1pt; padding:0pt; border:0pt;}'
+            css += '\n\np {text-indent: 2em; margin-top:1pt; margin-bottom:1pt; padding:0pt; border:0pt;}'
         self.override_css = self.css_parser.parseString(self.preprocess_css(css))
         
         
diff --git a/src/calibre/gui2/viewer/config.ui b/src/calibre/gui2/viewer/config.ui
index 0063900432..5aa85f4834 100644
--- a/src/calibre/gui2/viewer/config.ui
+++ b/src/calibre/gui2/viewer/config.ui
@@ -6,7 +6,7 @@
     <x>0</x>
     <y>0</y>
     <width>479</width>
-    <height>436</height>
+    <height>574</height>
    </rect>
   </property>
   <property name="windowTitle" >
@@ -145,6 +145,18 @@
         </item>
        </layout>
       </item>
+      <item row="2" column="0" >
+       <widget class="QGroupBox" name="groupBox_2" >
+        <property name="title" >
+         <string>&amp;User stylesheet</string>
+        </property>
+        <layout class="QGridLayout" name="gridLayout_5" >
+         <item row="0" column="0" >
+          <widget class="QPlainTextEdit" name="css" />
+         </item>
+        </layout>
+       </widget>
+      </item>
      </layout>
     </widget>
    </item>
diff --git a/src/calibre/gui2/viewer/documentview.py b/src/calibre/gui2/viewer/documentview.py
index b0e85ba181..da8120c910 100644
--- a/src/calibre/gui2/viewer/documentview.py
+++ b/src/calibre/gui2/viewer/documentview.py
@@ -14,6 +14,7 @@ from PyQt4.QtWebKit import QWebPage, QWebView, QWebSettings
 from calibre.utils.config import Config, StringConfig
 from calibre.gui2.viewer.config_ui import Ui_Dialog
 from calibre.gui2.viewer.js import bookmarks, referencing
+from calibre.ptempfile import PersistentTemporaryFile
 
 def load_builtin_fonts():
     from calibre.ebooks.lrf.fonts.liberation import LiberationMono_BoldItalic
@@ -50,7 +51,10 @@ def config(defaults=None):
         c = Config('viewer', desc)
     else:
         c = StringConfig(defaults, desc)
-        
+    
+    c.add_opt('user_css', default='',
+              help=_('Set the user CSS stylesheet. This can be used to customize the look of all books.'))
+    
     fonts = c.add_group('FONTS', _('Font options'))
     fonts('serif_family', default='Liberation Serif', help=_('The serif font family'))
     fonts('sans_family', default='Liberation Sans', help=_('The sans-serif font family'))
@@ -74,6 +78,8 @@ class ConfigDialog(QDialog, Ui_Dialog):
         self.default_font_size.setValue(opts.default_font_size)
         self.mono_font_size.setValue(opts.mono_font_size)
         self.standard_font.setCurrentIndex({'serif':0, 'sans':1, 'mono':2}[opts.standard_font])
+        self.css.setPlainText(opts.user_css)
+        self.css.setToolTip(_('Set the user CSS stylesheet. This can be used to customize the look of all books.'))
         
     def accept(self, *args):
         c = config()
@@ -83,6 +89,7 @@ class ConfigDialog(QDialog, Ui_Dialog):
         c.set('default_font_size', self.default_font_size.value())
         c.set('mono_font_size', self.mono_font_size.value())
         c.set('standard_font', {0:'serif', 1:'sans', 2:'mono'}[self.standard_font.currentIndex()])
+        c.set('user_css', unicode(self.css.toPlainText()))
         return QDialog.accept(self, *args)
         
 
@@ -104,6 +111,7 @@ class Document(QWebPage):
         d = ConfigDialog(parent)
         if d.exec_() == QDialog.Accepted:
             self.set_font_settings()
+            self.set_user_stylesheet()
             self.triggerAction(QWebPage.Reload)
     
     def __init__(self, *args):
@@ -128,11 +136,19 @@ class Document(QWebPage):
         
         # Miscellaneous
         settings.setAttribute(QWebSettings.LinksIncludedInFocusChain, True)
+        self.set_user_stylesheet()
         
         # Load jQuery
         self.connect(self.mainFrame(), SIGNAL('javaScriptWindowObjectCleared()'), 
                      self.load_javascript_libraries)
-
+    
+    def set_user_stylesheet(self):
+        raw = config().parse().user_css
+        pt = PersistentTemporaryFile('_user_stylesheet.css')
+        pt.write(raw.encode('utf-8'))
+        pt.close()
+        self.settings().setUserStyleSheetUrl(QUrl.fromLocalFile(pt.name))
+    
     def load_javascript_libraries(self):
         from calibre.resources import jquery, jquery_scrollTo
         self.javascript(jquery)
diff --git a/src/calibre/gui2/viewer/main.py b/src/calibre/gui2/viewer/main.py
index 56ef579465..7204cdd142 100644
--- a/src/calibre/gui2/viewer/main.py
+++ b/src/calibre/gui2/viewer/main.py
@@ -419,8 +419,7 @@ class EbookViewer(MainWindow, Ui_EbookViewer):
         path = self.view.path()
         try:
             index = self.iterator.spine.index(path)
-        except ValueError:
-            print path
+        except (ValueError, AttributeError):
             return -1
         self.current_page = self.iterator.spine[index]
         self.current_index = index