Prepared buildtools for Safari

packagerSafari.py

+# coding: utf-8
+
+# This file is part of the Adblock Plus build tools,
+# Copyright (C) 2006-2013 Eyeo GmbH
+#
+# Adblock Plus is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 3 as
+# published by the Free Software Foundation.
+#
+# Adblock Plus is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# 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 os
+import re
+import json
+from urlparse import urlparse
+from collections import OrderedDict
+
+from packager import readMetadata, getDefaultFileName, getBuildVersion, getTemplate, Files
+from buildtools.packagerChrome import convertJS, importGeckoLocales, getIgnoredFiles, getPackageFiles, ImageConverter
+
+def createPlist(params, files):
+  template = getTemplate('Info.plist.tmpl')
+  metadata = params['metadata']
+  catalog = json.loads(files['_locales/en_US/messages.json'])
+
+  def parse_section(section, depth=1):
+    rv = {}
+
+    if not metadata.has_section(section):
+      return rv
+
+    for opt in metadata.options(section):
+      bits = opt.split('_', depth)
+      key = bits.pop(-1).replace('_', ' ').title()
+      d = rv
+
+      for x in bits:
+        d = d.setdefault(x, {})
+
+      val = metadata.get(section, opt)
+
+      if val is True:
+        d[key] = '<true/>'
+      elif val is False:
+        d[key] = '<false/>'
+      elif isinstance(val, (int, long)):
+        d[key] = '<real>%d</real>' % val
+      else:
+        d[key] = '<string>%s</string>' % val
+
+    return rv
+
+  allowedDomains = set()
+  allowAllDomains = False
+  allowSecurePages = False
+
+  for perm in re.split(r'\s+', metadata.get('general', 'permissions')):
+    if perm == '<all_urls>':
+      allowAllDomains = True
+      allowSecurePages = True
+      continue
+
+    url = urlparse(perm)
+
+    if url.scheme == 'https':
+      allowSecurePages = True
+    elif url.scheme != 'http':
+      continue
+
+    if '*' in url.hostname:
+      allowAllDomains = True
+      continue
+
+    allowedDomains.add(url.hostname)
+
+  return template.render(
+    author=metadata.get('general', 'author'),
+    version=params['version'],
+    name=catalog['name']['message'],
+    description=catalog['description']['message'],
+    website=metadata.get('general', 'website'),
+    identifier=metadata.get('general', 'identifier'),
+    allowedDomains=allowedDomains,
+    allowAllDomains=allowAllDomains,
+    allowSecurePages=allowSecurePages,
+    contentScripts={
+      'start': metadata.get('contentScripts', 'document_start').split(),
+      'end':   metadata.get('contentScripts', 'document_end'  ).split(),
+    },
+    menus=parse_section('menus', 2),
+    toolbarItems=parse_section('toolbar_items'),
+    popovers=parse_section('popovers')
+  ).encode('utf-8')
+
+def createBackgroundPage(params):
+  template = getTemplate('background.html.tmpl')
+  return template.render(
+    backgroundScripts=re.split(r'\s+', params['metadata'].get(
+      'general', 'backgroundScripts'
+    ))
+  ).encode('utf-8')
+
+def createInfoModule(params):
+  template = getTemplate('safariInfo.js.tmpl')
+  return template.render(params).encode('utf-8');
+
+def createSignedXarArchive(outFile, files, keyFile, certs):
+  import subprocess
+  import tempfile
+  import shutil
+  import errno
+  import M2Crypto
+
+  # write files to temporary directory and create a xar archive
+  dirname = tempfile.mkdtemp()
+  try:
+    for filename, contents in files.iteritems():
+      path = os.path.join(dirname, filename)
+
+      while True:
+        try:
+          file = open(path, 'wb')
+          break
+        except IOError, e:
+          if e.errno != errno.ENOENT:
+            raise
+          os.makedirs(os.path.dirname(path))
+
+      with file:
+        file.write(contents)
+
+    subprocess.check_output(
+      ['xar', '-czf', os.path.abspath(outFile), '--distribution'] + os.listdir(dirname),
+      cwd=dirname
+    )
+  finally:
+    shutil.rmtree(dirname)
+
+  # add placeholder signature and certificates to the xar archive, and get data to sign
+  key = M2Crypto.RSA.load_key(keyFile)
+  digestinfo_filename = tempfile.mktemp()
+  try:
+    subprocess.check_call(
+      [
+        'xar', '--sign', '-f', outFile,
+        '--digestinfo-to-sign', digestinfo_filename,
+        '--sig-size', str(len(key.sign('')))
+      ] + [
+        arg for cert in certs for arg in ('--cert-loc', cert)
+      ]
+    )
+
+    with open(digestinfo_filename, 'rb') as file:
+      digestinfo = file.read()
+  finally:
+    try:
+      os.unlink(digestinfo_filename)
+    except OSError, e:
+      if e.errno != errno.ENOENT:
+        raise
+
+  # sign data and inject signature into xar archive
+  fd, signature_filename = tempfile.mkstemp()
+  try:
+    try:
+      os.write(fd, key.private_encrypt(
+        digestinfo,
+        M2Crypto.RSA.pkcs1_padding
+      ))
+    finally:
+      os.close(fd)
+
+    subprocess.check_call(['xar', '--inject-sig', signature_filename, '-f', outFile])
+  finally:
+    os.unlink(signature_filename)
+
+def createBuild(baseDir, type, outFile=None, buildNum=None, releaseBuild=False, keyFile=None, certs=()):
+  metadata = readMetadata(baseDir, type)
+  version = getBuildVersion(baseDir, metadata, releaseBuild, buildNum)
+
+  if not outFile:
+    outFile = getDefaultFileName(baseDir, metadata, version, 'safariextz' if keyFile else 'zip')
+
+  params = {
+    'type': type,
+    'baseDir': baseDir,
+    'releaseBuild': releaseBuild,
+    'version': version,
+    'devenv': False,
+    'metadata': metadata,
+  }
+
+  files = Files(getPackageFiles(params), getIgnoredFiles(params),
+                process=lambda path, data: data)
+  if metadata.has_section('mapping'):
+    files.readMappedFiles(metadata.items('mapping'))
+  files.read(baseDir)
+
+  if metadata.has_section('convert_js'):
+    convertJS(params, files)
+
+  if metadata.has_section('convert_img'):
+    ImageConverter().convert(params, files)
+
+  if metadata.has_section('import_locales'):
+    importGeckoLocales(params, files)
+
+  files['lib/info.js'] = createInfoModule(params)
+  files['background.html'] = createBackgroundPage(params)
+  files['Info.plist'] = createPlist(params, files)
+
+  dirname = getDefaultFileName('', metadata, version, 'safariextension')
+  for filename in files.keys():
+    files[os.path.join(dirname, filename)] = files.pop(filename)
+
+  if keyFile:
+    createSignedXarArchive(outFile, files, keyFile, certs)
+  else:
+    files.zip(outFile)