From 833bdc6415cecbd35b0f7c332a9268194a61aa03 Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@verizon.net>
Date: Wed, 25 Jul 2012 01:15:42 -0400
Subject: [PATCH] Support other backgrounds.

---
 build.py                  |  6 +++-
 pages/copyvios.mako       |  6 ++--
 pages/debug.mako          |  4 +--
 pages/earwigbot.mako      |  4 +--
 pages/index.mako          |  4 +--
 pages/settings.mako       |  6 ++--
 pages/support/footer.mako |  8 +++--
 pages/support/header.mako | 14 ++++++--
 pages/surveyor.mako       |  4 +--
 pages/swmt.mako           |  4 +--
 static/js/potd.js         | 83 ++++++++++++++++++++++++++++++++++++-----------
 toolserver/cookies.py     |  2 +-
 toolserver/settings.py    | 13 ++++----
 13 files changed, 108 insertions(+), 50 deletions(-)

diff --git a/build.py b/build.py
index 6c1d96f..a3b293f 100755
--- a/build.py
+++ b/build.py
@@ -17,13 +17,17 @@ sys.path.insert(0, ".")
 from mako.template import Template
 from mako.lookup import TemplateLookup
 
+from toolserver.cookies import parse_cookies
+
 def main(environ, start_response):
     lookup = TemplateLookup(directories=["{{pages_dir}}"],
                             input_encoding="utf8")
     template = Template(filename="{{src}}", module_directory="{{temp_dir}}",
                         lookup=lookup, format_exceptions=True)
     headers = [("Content-Type", "text/html")]
-    page = template.render(environ=environ, headers=headers).encode("utf8")
+    cookies = parse_cookies(environ)
+    page = template.render(environ=environ, headers=headers,
+                           cookies=cookies).encode("utf8")
     start_response("200 OK", headers)
     return [page]
 
diff --git a/pages/copyvios.mako b/pages/copyvios.mako
index 3c2a058..deba046 100644
--- a/pages/copyvios.mako
+++ b/pages/copyvios.mako
@@ -1,9 +1,7 @@
-<%include file="/support/header.mako" args="environ=environ, title='Copyvio Detector', add_css=('copyvios.css',), add_js=('copyvios.js',)"/>\
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='Copyvio Detector', add_css=('copyvios.css',), add_js=('copyvios.js',)"/>\
 <%namespace module="toolserver.copyvios" import="main, highlight_delta"/>\
-<%namespace module="toolserver.cookies" import="parse_cookies"/>\
 <%namespace module="toolserver.misc" import="urlstrip"/>\
 <% query, bot, all_langs, all_projects, page, result = main(environ) %>
-<% cookies = parse_cookies(environ) %>
             <h1>Copyvio Detector</h1>
             <p>This tool attempts to detect <a href="//en.wikipedia.org/wiki/WP:COPYVIO">copyright violations</a> in articles. Simply give the title of the page you want to check and hit Submit. The tool will then search for its content elsewhere on the web and display a report if a similar webpage is found. If you also provide a URL, it will not query any search engines and instead display a report comparing the article to that particular webpage, like the <a href="//toolserver.org/~dcoetzee/duplicationdetector/">Duplication Detector</a>. Check out the <a href="//en.wikipedia.org/wiki/User:EarwigBot/Copyvios/FAQ">FAQ</a> for more information and technical details.</p>
             <form action="${environ['PATH_INFO']}" method="get">
@@ -129,4 +127,4 @@
                     </div>
                 </div>
             % endif
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/pages/debug.mako b/pages/debug.mako
index f6a11a9..25dafca 100644
--- a/pages/debug.mako
+++ b/pages/debug.mako
@@ -1,4 +1,4 @@
-<%include file="/support/header.mako" args="environ=environ, title='Debug'"/>
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='Debug'"/>
         <ul>
         % for key, value in environ.items():
             % if key not in ["wsgi.input", "wsgi.errors", "PATH"]:
@@ -8,4 +8,4 @@
             % endif
         % endfor
         </ul>
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/pages/earwigbot.mako b/pages/earwigbot.mako
index ded421a..da36883 100644
--- a/pages/earwigbot.mako
+++ b/pages/earwigbot.mako
@@ -48,8 +48,8 @@
         since ${since} (${uptime} uptime) on <tt>${host}</tt>
     % endif
 </%def>\
-<%include file="/support/header.mako" args="environ=environ, title='EarwigBot Status'"/>
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='EarwigBot Status'"/>
             <h1>EarwigBot Status</h1>
             <p>EarwigBot ${get_status()}.</p>
             <p>Additional information: <a href="http://status.toolserver.org/">status.toolserver.org</a></p>
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/pages/index.mako b/pages/index.mako
index e401d4a..d518710 100644
--- a/pages/index.mako
+++ b/pages/index.mako
@@ -16,7 +16,7 @@
     pretty = path.split(root)[0]
 %>\
 <%def name="get_tools()"><% return tools %></%def>\
-<%include file="/support/header.mako" args="environ=environ, title='Home', slug='home'"/>
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='Home', slug='home'"/>
             <h1>My Tools</h1>
             % for tool in tools:
                 % if tool != "DIVIDER":
@@ -29,4 +29,4 @@
                     % endif
                 % endif
             % endfor
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/pages/settings.mako b/pages/settings.mako
index ad718a3..23e33c5 100644
--- a/pages/settings.mako
+++ b/pages/settings.mako
@@ -1,7 +1,7 @@
-<%include file="/support/header.mako" args="environ=environ, title='Settings'"/>\
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='Settings'"/>\
 <%namespace module="toolserver.settings" import="main"/>\
 <%! from json import dumps, loads %>
-<% bot, cookies, status, langs, projects = main(environ, headers) %>
+<% bot, status, langs, projects = main(environ, headers, cookies) %>
             % if status:
                 <div class="green-box">
                     <p>${status}</p>
@@ -86,4 +86,4 @@
             % else:
                 <p>No cookies!</p>
             % endif
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/pages/support/footer.mako b/pages/support/footer.mako
index e93f37f..5add092 100644
--- a/pages/support/footer.mako
+++ b/pages/support/footer.mako
@@ -1,4 +1,4 @@
-<%page args="environ"/>\
+<%page args="environ, cookies"/>\
 <%!
     from os import path
 %>\
@@ -18,7 +18,9 @@
                         <p>Copyright &copy; 2009&ndash;2012 <a href="//en.wikipedia.org/wiki/User:The_Earwig">Ben Kurtovic</a> &bull; \
                             <a href="mailto:earwig@toolserver.org">Contact</a> &bull; \
                             <a href="https://github.com/earwig/toolserver">View Source</a> &bull; \
-                            <a id="bg_image_link" href="">Background</a> &bull; \
+                            % if ("EarwigBackground" in cookies and cookies["EarwigBackground"].value in ["potd", "list"]) or "EarwigBackground" not in cookies:
+                                <a id="bg_image_link" href="">Background</a> &bull; \
+                            % endif
                             <a href="http://validator.w3.org/check?uri=referer">Valid XHTML 1.0 Strict</a>
                         </p>
                     </td>
@@ -29,4 +31,4 @@
             </table>
         </div>
     </body>
-</html>
\ No newline at end of file
+</html>
diff --git a/pages/support/header.mako b/pages/support/header.mako
index 263dc2d..cb94f53 100644
--- a/pages/support/header.mako
+++ b/pages/support/header.mako
@@ -1,4 +1,4 @@
-<%page args="environ, title, slug=None, add_css=(), add_js=()"/>\
+<%page args="environ, cookies, title, slug=None, add_css=(), add_js=()"/>\
 <%namespace name="index" file="/index.mako" import="get_tools"/>\
 <%!
     from os import path
@@ -28,7 +28,17 @@
             <script src="${root}/static/js/${filename}" type="text/javascript"></script>
         % endfor
     </head>
-    <body onload="potd_set_background()">
+    % if "EarwigBackground" in cookies:
+        % if cookies["EarwigBackground"].value == "list":
+            <body onload="set_background_list()">
+        % elif cookies["EarwigBackground"].value == "plain":
+            <body style="background-image: url('${root}/static/images/background.png');">
+        % else
+            <body onload="set_background_potd()">
+        % endif
+    % else
+        <body onload="set_background_potd()">
+    % endif
         <div id="header">
             <p id="heading"><a class="dark" href="${pretty}">earwig</a><span class="light">@</span><a class="mid" href="https://wiki.toolserver.org/">toolserver</a><span class="light">:</span><a class="dark" href="${this}">${slug}</a></p>
             <p id="links"><span class="light">&gt;</span>
diff --git a/pages/surveyor.mako b/pages/surveyor.mako
index 6d3fccc..a5036dc 100644
--- a/pages/surveyor.mako
+++ b/pages/surveyor.mako
@@ -1,4 +1,4 @@
-<%include file="/support/header.mako" args="environ=environ, title='Contribution Surveyor'"/>
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='Contribution Surveyor'"/>
             <h1>Contribution Surveyor</h1>
             <p>This tool does not exist yet. Watch this space for updates.</p>
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/pages/swmt.mako b/pages/swmt.mako
index 0d1145c..e8bb86e 100644
--- a/pages/swmt.mako
+++ b/pages/swmt.mako
@@ -1,4 +1,4 @@
-<%include file="/support/header.mako" args="environ=environ, title='SWMT Helper'"/>
+<%include file="/support/header.mako" args="environ=environ, cookies=cookies, title='SWMT Helper'"/>
             <h1>SWMT Helper</h1>
             <p>This tool does not exist yet. Watch this space for updates.</p>
-<%include file="/support/footer.mako" args="environ=environ"/>
+<%include file="/support/footer.mako" args="environ=environ, cookies=cookies"/>
diff --git a/static/js/potd.js b/static/js/potd.js
index 055e917..8276179 100644
--- a/static/js/potd.js
+++ b/static/js/potd.js
@@ -1,4 +1,41 @@
-function potd_set_background() {
+function set_background_potd() {
+    if (cache_cookie()) return;
+    var d = new Date();
+    var callback = "earwigpotd1";
+    var date = (d.getUTCFullYear()) + "-" + zero_pad(d.getUTCMonth() + 1, 2) + "-" + zero_pad(d.getUTCDate(), 2);
+    var base = "//commons.wikimedia.org/w/api.php?action=query&prop=revisions&rvprop=content&format=json&titles=Template:Potd/";
+    var url = base + date + "&callback=" + callback;
+
+    var script = document.createElement("script");
+    var head = document.getElementsByTagName("head")[0];
+
+    window[callback] = function(data) {
+        head.removeChild(script);
+        parse_potd_file_name(data);
+    };
+
+    script.src = url;
+    head.appendChild(script);
+}
+
+function set_background_list() {
+    if (cache_cookie()) return;
+    var base = "//commons.wikimedia.org/w/api.php?action=query&prop=revisions&rvprop=content&format=json&titles=User:The+Earwig/POTD";
+    var url = base + "&callback=" + callback;
+
+    var script = document.createElement("script");
+    var head = document.getElementsByTagName("head")[0];
+
+    window[callback] = function(data) {
+        head.removeChild(script);
+        parse_list_file_name(data);
+    };
+
+    script.src = url;
+    head.appendChild(script);
+}
+
+function cache_cookie() {
     var cookie = get_cookie("EarwigBackgroundCache");
     if (cookie) {
         try {
@@ -10,37 +47,50 @@ function potd_set_background() {
             var imgheight = data.imgheight;
             if (filename && url && descurl && imgwidth && imgheight) {
                 set_background(filename, url, descurl, imgwidth, imgheight);
-                return;
+                return true;
             }
         }
         catch (SyntaxError) {}
     }
+    return false;
+}
 
-    var d = new Date();
-    var callback = "earwigpotd1";
-    var date = (d.getUTCFullYear()) + "-" + zero_pad(d.getUTCMonth() + 1, 2) + "-" + zero_pad(d.getUTCDate(), 2);
-    var base = "//commons.wikimedia.org/w/api.php?action=query&prop=revisions&rvprop=content&format=json&titles=Template:Potd/";
-    var url = base + date + "&callback=" + callback;
+function parse_potd_file_name(data) {
+    var content = "";
+    var res = data["query"]["pages"];
+    for (pageid in res) {
+        content = res[pageid]["revisions"][0]["*"];
+    }
+    var filename = /\{\{Potd filename\|(1=)?(.*?)\|.*?\}\}/.exec(content)[2];
+
+    var callback = "earwigpotd2";
+    var base = "//commons.wikimedia.org/w/api.php?action=query&prop=imageinfo&iiprop=url|size&format=json&titles=File:";
+    var url = base + escape(filename) + "&callback=" + callback;
 
     var script = document.createElement("script");
     var head = document.getElementsByTagName("head")[0];
 
     window[callback] = function(data) {
         head.removeChild(script);
-        parse_file_name(data);
+        parse_file_url(data, escape(filename.replace(/ /g, "_")));
     };
 
     script.src = url;
     head.appendChild(script);
 }
 
-function parse_file_name(data) {
+function parse_list_file_name(data) {
     var content = "";
     var res = data["query"]["pages"];
     for (pageid in res) {
         content = res[pageid]["revisions"][0]["*"];
     }
-    var filename = /\{\{Potd filename\|(1=)?(.*?)\|.*?\}\}/.exec(content)[2];
+
+    var filenames = [];
+    while ((match = /\*\s*\[\[File:(.*?)\]\]/g.exec(content)) !== null) {
+        filenames.push(match[1]);
+    }
+    var filename = filenames[Math.floor(Math.random() * filenames.length)];
 
     var callback = "earwigpotd2";
     var base = "//commons.wikimedia.org/w/api.php?action=query&prop=imageinfo&iiprop=url|size&format=json&titles=File:";
@@ -116,19 +166,14 @@ function zero_pad(value, length) {
 
 function get_window_size() {
     // See http://www.javascripter.net/faq/browserw.htm
-    var width = 1024,
-        height = 768;
     if (document.body && document.body.offsetWidth && document.body.offsetHeight) {
-        width = document.body.offsetWidth;
-        height = document.body.offsetHeight;
+        return [document.body.offsetWidth, document.body.offsetHeight];
     }
     if (document.compatMode=="CSS1Compat" && document.documentElement && document.documentElement.offsetWidth && document.documentElement.offsetHeight) {
-        width = document.documentElement.offsetWidth;
-        height = document.documentElement.offsetHeight;
+        return [document.documentElement.offsetWidth, document.documentElement.offsetHeight];
     }
     if (window.innerWidth && window.innerHeight) {
-        width = window.innerWidth;
-        height = window.innerHeight;
+        return [window.innerWidth, window.innerHeight];
     }
-    return [width, height];
+    return [1024, 768];
 }
diff --git a/toolserver/cookies.py b/toolserver/cookies.py
index 23fb9cc..2a84c2a 100644
--- a/toolserver/cookies.py
+++ b/toolserver/cookies.py
@@ -38,7 +38,7 @@ class _CookieManager(SimpleCookie):
         return self._path
 
 
-def parse_cookies(context, environ):
+def parse_cookies(environ):
     return _CookieManager(environ)
 
 def set_cookie(headers, cookies, key, value, days=0):
diff --git a/toolserver/settings.py b/toolserver/settings.py
index 80f9f5c..22edf4e 100644
--- a/toolserver/settings.py
+++ b/toolserver/settings.py
@@ -2,18 +2,17 @@
 
 from markupsafe import escape
 
-from .cookies import parse_cookies, set_cookie, delete_cookie
+from .cookies import set_cookie, delete_cookie
 from .misc import get_bot, Query
 from .sites import get_sites
 
-def main(context, environ, headers):
+def main(context, environ, headers, cookies):
     query = Query(environ, method="POST")
-    cookies = parse_cookies(context, environ)
 
     if query.action == "set":
-        status = _do_set(query, cookies, headers)
+        status = _do_set(query, headers, cookies)
     elif query.action == "delete":
-        status = _do_delete(query, cookies, headers)
+        status = _do_delete(query, headers, cookies)
     else:
         status = None
 
@@ -21,7 +20,7 @@ def main(context, environ, headers):
     langs, projects = get_sites(bot)
     return bot, cookies, status, langs, projects
 
-def _do_set(query, cookies, headers):
+def _do_set(query, headers, cookies):
     changes = set()
     if query.lang:
         key = "EarwigDefaultLang"
@@ -38,7 +37,7 @@ def _do_set(query, cookies, headers):
         return "Updated {0}.".format(changes)
     return None
 
-def _do_delete(query, cookies, headers):
+def _do_delete(query, headers, cookies):
     if query.cookie in cookies:
         delete_cookie(headers, cookies, query.cookie.encode("utf8"))
         template = "Deleted cookie <b><tt>{0}</tt></b>."