Issue 3312 - Serve release update manifests for Adblock Browser

sitescripts/extensions/web/adblockbrowserUpdates.py

Index: sitescripts/extensions/web/adblockbrowserUpdates.py
===================================================================
--- a/sitescripts/extensions/web/adblockbrowserUpdates.py
+++ b/sitescripts/extensions/web/adblockbrowserUpdates.py
@@ -15,7 +15,9 @@
 # You should have received a copy of the GNU General Public License
 # along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
 
+import glob
 import hashlib
+import json
 import os
 import re
 from urlparse import parse_qs
@@ -42,24 +44,22 @@
 
 def _get_latest_build(builds_dir):
   latest_build = {"id": 0}
-  for file in os.listdir(builds_dir):
-    match = re.search(r"^adblockbrowser-.*?(\d+)-\w+\.apk$", file)
-    if match:
-      build_id = int(match.group(1))
-      if build_id > latest_build["id"]:
-        latest_build["id"] = build_id
-        latest_build["path"] = os.path.join(builds_dir, file)
+  for json_path in glob.glob(os.path.join(builds_dir, "adblockbrowser-*.json")):
+    with open(json_path) as json_file:
+      build_id = int(json.loads(json_file.read())["buildid"])
+    if build_id > latest_build["id"]:
+      latest_build["id"] = build_id
+      apk_path = os.path.splitext(json_path)[0] + ".apk"
+      latest_build["path"] = os.path.join(builds_dir, apk_path)
   if latest_build["id"] == 0:
     return {}
   return latest_build
 
-def _render_manifest(build=None):
+def _render_manifest(build=None, builds_url=None):
   if not build:
     return _MANIFEST_TEMPLATE.render()
 
-  nightlies_url = get_config().get("extensions", "nightliesURL")
-  build_url = "%s/adblockbrowser/%s?update" % (nightlies_url.rstrip("/"),
-                                               os.path.basename(build["path"]))
+  build_url = "%s/%s?update" % (builds_url, os.path.basename(build["path"]))
   with open(build["path"], "rb") as build_file:
     build_content = build_file.read()
   return _MANIFEST_TEMPLATE.render({
@@ -72,19 +72,16 @@
     }
   })
 
-def _get_update_manifest(current_build_id):
-  nightlies_dir = get_config().get("extensions", "nightliesDirectory")
-  builds_dir = os.path.join(nightlies_dir, "adblockbrowser")
+def _get_update_manifest(current_build_id, builds_dir, builds_url):
   if not os.path.isdir(builds_dir):
     return _render_manifest()
 
   latest_build = _get_latest_build(builds_dir)
   if not latest_build or current_build_id >= latest_build["id"]:
     return _render_manifest()
-  return _render_manifest(latest_build)
+  return _render_manifest(latest_build, builds_url)
 
-@url_handler("/devbuilds/adblockbrowser/updates.xml")
-def adblockbrowser_updates(environ, start_response):
+def _handle_request(environ, start_response, builds_dir, builds_url):
   params = parse_qs(environ.get("QUERY_STRING", ""))
   try:
     version = params.get("addonVersion", [""])[0]
@@ -92,7 +89,28 @@
   except:
     start_response("400 Processing Error", [("Content-Type", "text/plain")])
     return ["Failed to parse addonVersion."]
-  manifest = _get_update_manifest(build_id)
+  manifest = _get_update_manifest(build_id, builds_dir, builds_url)
   response = manifest.encode("utf-8")
   start_response("200 OK", [("Content-Type", "application/xml; charset=utf-8")])
   return [response]
+
+@url_handler("/adblockbrowser/updates.xml")
+def adblockbrowser_updates(environ, start_response):
+  config = get_config()
+
+  builds_dir = config.get("extensions", "downloadsDirectory")
+  builds_url = config.get("extensions", "downloadsURL").rstrip("/")
+
+  return _handle_request(environ, start_response, builds_dir, builds_url)
+
+@url_handler("/devbuilds/adblockbrowser/updates.xml")
+def adblockbrowser_devbuild_updates(environ, start_response):
+  config = get_config()
+
+  nightlies_dir = config.get("extensions", "nightliesDirectory")
+  builds_dir = os.path.join(nightlies_dir, "adblockbrowser")
+
+  nightlies_url = config.get("extensions", "nightliesURL").rstrip("/")
+  builds_url = "%s/adblockbrowser" % nightlies_url
+
+  return _handle_request(environ, start_response, builds_dir, builds_url)