mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-10-31 18:47:02 -04:00
104 lines
4.0 KiB
Python
104 lines
4.0 KiB
Python
#!/usr/bin/env python2
|
|
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:fdm=marker:ai
|
|
from __future__ import (unicode_literals, division, absolute_import,
|
|
print_function)
|
|
|
|
__license__ = 'GPL v3'
|
|
__copyright__ = '2012, Kovid Goyal <kovid at kovidgoyal.net>'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
import os, shutil, json
|
|
from io import BytesIO
|
|
from zipfile import ZipFile
|
|
from hashlib import sha1
|
|
from tempfile import mkdtemp
|
|
|
|
|
|
from setup import Command, download_securely
|
|
|
|
|
|
class MathJax(Command):
|
|
|
|
description = 'Create the MathJax bundle'
|
|
MATH_JAX_VERSION = '2.7.5'
|
|
MATH_JAX_URL = 'https://github.com/mathjax/MathJax/archive/%s.zip' % MATH_JAX_VERSION
|
|
FONT_FAMILY = 'TeX'
|
|
|
|
def add_options(self, parser):
|
|
parser.add_option('--path-to-mathjax', help='Path to the MathJax source code')
|
|
parser.add_option('--mathjax-url', default=self.MATH_JAX_URL, help='URL to MathJax source archive in zip format')
|
|
parser.add_option('--system-mathjax', default=False, action='store_true',
|
|
help='Treat MathJax as system copy and symlink instead of copy')
|
|
|
|
def download_mathjax_release(self, tdir, url):
|
|
self.info('Downloading MathJax:', url)
|
|
raw = download_securely(url)
|
|
with ZipFile(BytesIO(raw)) as zf:
|
|
zf.extractall(tdir)
|
|
return os.path.join(tdir, 'MathJax-' + self.MATH_JAX_VERSION)
|
|
|
|
def add_file(self, path, name):
|
|
with open(path, 'rb') as f:
|
|
raw = f.read()
|
|
self.h.update(raw)
|
|
self.mathjax_files[name] = len(raw)
|
|
dest = self.j(self.mathjax_dir, *name.split('/'))
|
|
base = os.path.dirname(dest)
|
|
if not os.path.exists(base):
|
|
os.makedirs(base)
|
|
if self.use_symlinks:
|
|
os.symlink(path, dest)
|
|
else:
|
|
with open(dest, 'wb') as f:
|
|
f.write(raw)
|
|
|
|
def add_tree(self, base, prefix, ignore=lambda n:False):
|
|
for dirpath, dirnames, filenames in os.walk(base):
|
|
for fname in filenames:
|
|
f = os.path.join(dirpath, fname)
|
|
name = prefix + '/' + os.path.relpath(f, base).replace(os.sep, '/')
|
|
if not ignore(name):
|
|
self.add_file(f, name)
|
|
|
|
@property
|
|
def mathjax_dir(self):
|
|
return self.j(self.RESOURCES, 'mathjax')
|
|
|
|
def already_present(self):
|
|
manifest = self.j(self.mathjax_dir, 'manifest.json')
|
|
if os.path.exists(manifest):
|
|
with open(manifest, 'rb') as f:
|
|
return json.load(f).get('version') == self.MATH_JAX_VERSION
|
|
return False
|
|
|
|
def clean(self):
|
|
if os.path.exists(self.mathjax_dir):
|
|
shutil.rmtree(self.mathjax_dir)
|
|
|
|
def run(self, opts):
|
|
if not opts.system_mathjax and self.already_present():
|
|
self.info('MathJax already present in the resources directory, not downloading')
|
|
return
|
|
self.use_symlinks = opts.system_mathjax
|
|
self.h = sha1()
|
|
self.mathjax_files = {}
|
|
self.clean()
|
|
os.mkdir(self.mathjax_dir)
|
|
tdir = mkdtemp('calibre-mathjax-build')
|
|
try:
|
|
src = opts.path_to_mathjax or self.download_mathjax_release(tdir, opts.mathjax_url)
|
|
self.info('Adding MathJax...')
|
|
unpacked = 'unpacked' if self.e(self.j(src, 'unpacked')) else ''
|
|
self.add_file(self.j(src, unpacked, 'MathJax.js'), 'MathJax.js')
|
|
self.add_tree(
|
|
self.j(src, 'fonts', 'HTML-CSS', self.FONT_FAMILY, 'woff'),
|
|
'fonts/HTML-CSS/%s/woff' % self.FONT_FAMILY,
|
|
lambda x: not x.endswith('.woff'))
|
|
for d in 'extensions jax/element jax/input jax/output/CommonHTML'.split():
|
|
self.add_tree(self.j(src, unpacked, *d.split('/')), d)
|
|
etag = self.h.hexdigest()
|
|
with open(self.j(self.RESOURCES, 'mathjax', 'manifest.json'), 'wb') as f:
|
|
f.write(json.dumps({'etag': etag, 'files': self.mathjax_files, 'version': self.MATH_JAX_VERSION}, indent=2).encode('utf-8'))
|
|
finally:
|
|
shutil.rmtree(tdir)
|