openmedialibrary/oml/commands.py

# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import division

from flask.ext.script import Command


class Setup(Command):
        """
            setup new node
        """
        def run(self):
            import setup
            import settings
            setup.create_default_lists()
            settings.db.session.connection().execute("PRAGMA journal_mode=WAL")
            settings.db.session.commit()

class UpdateStatic(Command):
        """
            setup new node
        """
        def run(self):
            import subprocess
            import os
            import settings

            def r(*cmd):
                print ' '.join(cmd)
                return subprocess.call(cmd)

            oxjs = os.path.join(settings.static_path, 'oxjs')
            if not os.path.exists(oxjs):
                r('git', 'clone', 'https://git.0x2620.org/oxjs.git', oxjs)
            r('python2', os.path.join(oxjs, 'tools', 'build', 'build.py'))
            r('python2', os.path.join(settings.static_path, 'py', 'build.py'))

class Release(Command):
        """
            release new version
        """
        def run(self):
            print 'checking...'
            import settings
            import os
            import subprocess
            import json
            import hashlib
            import ed25519
            from os.path import join, exists, dirname

            root_dir = dirname(settings.base_dir)
            os.chdir(root_dir)

            def run(*cmd):
                p = subprocess.Popen(cmd)
                p.wait()
                return p.returncode

            def get(*cmd):
                p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
                stdout, error = p.communicate()
                return stdout

            def version(module):
                os.chdir(join(root_dir, module))
                version = get('git', 'log', '-1', '--format=%cd', '--date=iso').split(' ')[0].replace('-', '')
                version += '-' + get('git', 'rev-list', 'HEAD', '--count').strip()
                version += '-' + get('git', 'describe', '--always').strip()
                os.chdir(root_dir)
                return version

            with open(os.path.expanduser('~/Private/openmedialibrary_release.key')) as fd:
                SIG_KEY=ed25519.SigningKey(fd.read())
            SIG_ENCODING='base64'

            def sign(release):
                value = []
                for module in sorted(release['modules']):
                    value += ['%s/%s' % (release['modules'][module]['version'], release['modules'][module]['sha1'])]
                value = '\n'.join(value)
                sig = SIG_KEY.sign(value, encoding=SIG_ENCODING)
                release['signature'] = sig

            def sha1sum(path):
                h = hashlib.sha1()
                with open(path) as fd:
                    for chunk in iter(lambda: fd.read(128*h.block_size), ''):
                        h.update(chunk)
                return h.hexdigest()

            MODULES = ['platform', 'openmedialibrary']
            VERSIONS = {module:version(module) for module in MODULES}

            EXCLUDE=[
                '--exclude', '.git', '--exclude', '.bzr',
                '--exclude', '.*.swp', '--exclude', '._*', '--exclude', '.DS_Store'
            ]

            #run('./ctl', 'update_static')
            for module in MODULES:
                tar = join('updates', '%s-%s.tar.bz2' % (module, VERSIONS[module]))
                if not exists(tar):
                    cmd = ['tar', 'cvjf', tar, '%s/' % module] + EXCLUDE
                    if module in ('openmedialibrary', ):
                        cmd += ['--exclude', '*.pyc']
                    if module == 'openmedialibrary':
                        cmd += ['--exclude', 'oxjs/examples', '--exclude', 'gunicorn.pid']
                    run(*cmd)
            release = {}
            release['modules'] = {module: {
                'name': '%s-%s.tar.bz2' % (module, VERSIONS[module]),
                'version': VERSIONS[module],
                'sha1': sha1sum(join('updates', '%s-%s.tar.bz2' % (module, VERSIONS[module])))
            } for module in MODULES}
            sign(release)
            with open('updates/release.json', 'w') as fd:
                json.dump(release, fd, indent=2)
            print 'signed latest release in updates/release.json'