OLD | NEW |
1 # This file is part of the Adblock Plus web scripts, | 1 # This file is part of the Adblock Plus web scripts, |
2 # Copyright (C) 2006-2016 Eyeo GmbH | 2 # Copyright (C) 2006-2016 Eyeo GmbH |
3 # | 3 # |
4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
7 # | 7 # |
8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
12 # | 12 # |
13 # You should have received a copy of the GNU General Public License | 13 # You should have received a copy of the GNU General Public License |
14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
15 | 15 |
16 import glob | 16 import glob |
17 import hashlib | 17 import hashlib |
18 import json | 18 import json |
19 import os | 19 import os |
20 import re | 20 import re |
21 from urlparse import parse_qs | 21 from urlparse import parse_qs |
22 | 22 |
23 from jinja2 import Template | 23 from jinja2 import Template |
24 | 24 |
25 from sitescripts.utils import get_config | 25 from sitescripts.utils import get_config |
26 from sitescripts.web import url_handler | 26 from sitescripts.web import url_handler |
27 | 27 |
28 _MANIFEST_TEMPLATE = Template("""<?xml version="1.0"?> | 28 _MANIFEST_TEMPLATE = Template('''<?xml version="1.0"?> |
29 <updates> | 29 <updates> |
30 {% if build %} | 30 {% if build %} |
31 <update buildID="{{ build.build_id }}"> | 31 <update buildID="{{ build.build_id }}"> |
32 <patch | 32 <patch |
33 URL="{{ build.url }}" | 33 URL="{{ build.url }}" |
34 hashFunction="{{ build.hash_function }}" | 34 hashFunction="{{ build.hash_function }}" |
35 hashValue="{{ build.hash_value }}" | 35 hashValue="{{ build.hash_value }}" |
36 size="{{ build.size }}"/> | 36 size="{{ build.size }}"/> |
37 </update> | 37 </update> |
38 {% endif %} | 38 {% endif %} |
39 </updates> | 39 </updates> |
40 | 40 |
41 """, autoescape=True) | 41 ''', autoescape=True) |
42 | 42 |
43 | 43 |
44 def _get_latest_build(builds_dir): | 44 def _get_latest_build(builds_dir): |
45 latest_build = {"id": 0} | 45 latest_build = {'id': 0} |
46 for json_path in glob.glob(os.path.join(builds_dir, "adblockbrowser-*.json")
): | 46 for json_path in glob.glob(os.path.join(builds_dir, 'adblockbrowser-*.json')
): |
47 with open(json_path) as json_file: | 47 with open(json_path) as json_file: |
48 build_id = int(json.loads(json_file.read())["buildid"]) | 48 build_id = int(json.loads(json_file.read())['buildid']) |
49 if build_id > latest_build["id"]: | 49 if build_id > latest_build['id']: |
50 latest_build["id"] = build_id | 50 latest_build['id'] = build_id |
51 apk_path = os.path.splitext(json_path)[0] + ".apk" | 51 apk_path = os.path.splitext(json_path)[0] + '.apk' |
52 latest_build["path"] = os.path.join(builds_dir, apk_path) | 52 latest_build['path'] = os.path.join(builds_dir, apk_path) |
53 if latest_build["id"] == 0: | 53 if latest_build['id'] == 0: |
54 return {} | 54 return {} |
55 return latest_build | 55 return latest_build |
56 | 56 |
57 | 57 |
58 def _render_manifest(build=None, builds_url=None): | 58 def _render_manifest(build=None, builds_url=None): |
59 if not build: | 59 if not build: |
60 return _MANIFEST_TEMPLATE.render() | 60 return _MANIFEST_TEMPLATE.render() |
61 | 61 |
62 build_url = "%s/%s?update" % (builds_url, os.path.basename(build["path"])) | 62 build_url = '%s/%s?update' % (builds_url, os.path.basename(build['path'])) |
63 with open(build["path"], "rb") as build_file: | 63 with open(build['path'], 'rb') as build_file: |
64 build_content = build_file.read() | 64 build_content = build_file.read() |
65 return _MANIFEST_TEMPLATE.render({ | 65 return _MANIFEST_TEMPLATE.render({ |
66 "build": { | 66 'build': { |
67 "build_id": build["id"], | 67 'build_id': build['id'], |
68 "url": build_url, | 68 'url': build_url, |
69 "hash_function": "SHA512", | 69 'hash_function': 'SHA512', |
70 "hash_value": hashlib.sha512(build_content).hexdigest(), | 70 'hash_value': hashlib.sha512(build_content).hexdigest(), |
71 "size": len(build_content) | 71 'size': len(build_content) |
72 } | 72 } |
73 }) | 73 }) |
74 | 74 |
75 | 75 |
76 def _get_update_manifest(current_build_id, builds_dir, builds_url): | 76 def _get_update_manifest(current_build_id, builds_dir, builds_url): |
77 if not os.path.isdir(builds_dir): | 77 if not os.path.isdir(builds_dir): |
78 return _render_manifest() | 78 return _render_manifest() |
79 | 79 |
80 latest_build = _get_latest_build(builds_dir) | 80 latest_build = _get_latest_build(builds_dir) |
81 if not latest_build or current_build_id >= latest_build["id"]: | 81 if not latest_build or current_build_id >= latest_build['id']: |
82 return _render_manifest() | 82 return _render_manifest() |
83 return _render_manifest(latest_build, builds_url) | 83 return _render_manifest(latest_build, builds_url) |
84 | 84 |
85 | 85 |
86 def _handle_request(environ, start_response, builds_dir, builds_url): | 86 def _handle_request(environ, start_response, builds_dir, builds_url): |
87 params = parse_qs(environ.get("QUERY_STRING", "")) | 87 params = parse_qs(environ.get('QUERY_STRING', '')) |
88 try: | 88 try: |
89 version = params.get("addonVersion", [""])[0] | 89 version = params.get('addonVersion', [''])[0] |
90 build_id = int(re.search(r"(\d+)$", version).group(1)) | 90 build_id = int(re.search(r'(\d+)$', version).group(1)) |
91 except: | 91 except: |
92 start_response("400 Processing Error", [("Content-Type", "text/plain")]) | 92 start_response('400 Processing Error', [('Content-Type', 'text/plain')]) |
93 return ["Failed to parse addonVersion."] | 93 return ['Failed to parse addonVersion.'] |
94 manifest = _get_update_manifest(build_id, builds_dir, builds_url) | 94 manifest = _get_update_manifest(build_id, builds_dir, builds_url) |
95 response = manifest.encode("utf-8") | 95 response = manifest.encode('utf-8') |
96 start_response("200 OK", [("Content-Type", "application/xml; charset=utf-8")
]) | 96 start_response('200 OK', [('Content-Type', 'application/xml; charset=utf-8')
]) |
97 return [response] | 97 return [response] |
98 | 98 |
99 | 99 |
100 @url_handler("/adblockbrowser/updates.xml") | 100 @url_handler('/adblockbrowser/updates.xml') |
101 def adblockbrowser_updates(environ, start_response): | 101 def adblockbrowser_updates(environ, start_response): |
102 config = get_config() | 102 config = get_config() |
103 | 103 |
104 builds_dir = config.get("extensions", "downloadsDirectory") | 104 builds_dir = config.get('extensions', 'downloadsDirectory') |
105 builds_url = config.get("extensions", "downloadsURL").rstrip("/") | 105 builds_url = config.get('extensions', 'downloadsURL').rstrip('/') |
106 | 106 |
107 return _handle_request(environ, start_response, builds_dir, builds_url) | 107 return _handle_request(environ, start_response, builds_dir, builds_url) |
108 | 108 |
109 | 109 |
110 @url_handler("/devbuilds/adblockbrowser/updates.xml") | 110 @url_handler('/devbuilds/adblockbrowser/updates.xml') |
111 def adblockbrowser_devbuild_updates(environ, start_response): | 111 def adblockbrowser_devbuild_updates(environ, start_response): |
112 config = get_config() | 112 config = get_config() |
113 | 113 |
114 nightlies_dir = config.get("extensions", "nightliesDirectory") | 114 nightlies_dir = config.get('extensions', 'nightliesDirectory') |
115 builds_dir = os.path.join(nightlies_dir, "adblockbrowser") | 115 builds_dir = os.path.join(nightlies_dir, 'adblockbrowser') |
116 | 116 |
117 nightlies_url = config.get("extensions", "nightliesURL").rstrip("/") | 117 nightlies_url = config.get('extensions', 'nightliesURL').rstrip('/') |
118 builds_url = "%s/adblockbrowser" % nightlies_url | 118 builds_url = '%s/adblockbrowser' % nightlies_url |
119 | 119 |
120 return _handle_request(environ, start_response, builds_dir, builds_url) | 120 return _handle_request(environ, start_response, builds_dir, builds_url) |
OLD | NEW |