slideapi.py

#!/usr/bin/env python
#
# This file is part of the LibreOffice project.
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#

import os, json, time
from cherrypy import wsgiserver
from bottle import get, post, auth_basic, run, static_file, request
from bottle import debug, view, HTTPError, server_names, ServerAdapter
from subprocess import call
from lpod.document import odf_get_document, odf_new_document, ODF_MANIFEST
import tempfile

# config
#
# we understand:
# local_conf["users"], yielding {<username>: {"password": <password>, "description": <description>}, ... }
# local_conf["thumbnails"], yielding {"soffice": <path>, "convert": <path>, "thumbnail_size": <128x128>}
# local_conf["tags"], yielding { <tagname>: <tag_desc, ... }
#
local_conf=json.loads(open('local.json').read()) if os.path.exists('local.json') else {}
root=local_conf["filestore"]
tmp_root=local_conf["tmpstore"]

class Decks:
    slide = None
    deck = None

def validate_auth(user, password):
    if "users" in local_conf:
        user_dict = local_conf["users"]
        if user in user_dict and 'password' in user_dict[user]:
            return password == user_dict[user]['password']
    return False

# -------------------------------------------------------------------

def get_users():
    return {user: data['description'] for user, data in local_conf["users"].items()}

@get('/api/users/')
def users():
    return get_users()

@get('/users/')
@view('users')
def users():
    return dict(users=get_users())

# -------------------------------------------------------------------

def get_tags():
    return local_conf["tags"]

@get('/api/tags/')
def tags():
    return get_tags()

@get('/tags/')
@view('tags')
def tags():
    return dict(tags=get_tags())

# -------------------------------------------------------------------

def get_decks(user):
    path = "%s/%s/decks/" % (root,user)
    if not os.path.isdir(path):
        return []
    return [deck for deck in os.listdir(path) if os.path.isdir(path+deck)]

def get_groups(user):
    path = "%s/%s/groups/" % (root,user)
    if not os.path.isdir(path):
        return []
    return [group for group in os.listdir(path) if os.path.isdir(path+group)]

def get_lastrev(user,part,deck):
    path = "%s/%s/%s/%s/" % (root,user,part,deck)
    if not os.path.isdir(path):
        return 0
    # get last revision
    return int(sorted([rev for rev in os.listdir(path) if os.path.isdir(path+rev) and rev != 'latest'], key=int)[-1])

@get('/api/users/<user>/')
def list_decks(user):
    return json.dumps(['decks/'+deck for deck in get_decks(user)] + ['groups/'+group for group in get_groups(user)])

@get('/api/users/<user>/decks/')
def list_decks(user):
    return json.dumps(get_decks(user))

@get('/api/users/<user>/groups/')
def list_groups(user):
    return json.dumps(get_groups(user))

@get('/users/<user>/')
@view('decksgroups')
def list_decksgroups(user):
    return dict(decks=get_decks(user), groups=get_groups(user))

@get('/users/<user>/decks/')
@view('decks')
def list_decks(user):
    return dict(decks=get_decks(user), user=user)

@get('/users/<user>/groups/')
@view('groups')
def list_groups(user):
    return dict(groups=get_groups(user))

@get('/users/<user>/<part>/<deck>/thumbnail.png')
@get('/api/users/<user>/<part>/<deck>/thumbnail.png')
def get_thumbnail(user, part, deck):
    path = "%s/%s/%s/%s/" % (root,user,part,deck)
    last_rev = get_lastrev(user,part,deck)
    # serve thumbnail from that rev, first slide
    return static_file('thumbnail.png', root=path+str(last_rev)+'/0/', mimetype='image/png')

# -------------------------------------------------------------------

def get_revs(user,part,deck):
    path = "%s/%s/%s/%s/" % (root,user,part,deck)
    if not os.path.isdir(path):
        return []
    return sorted([rev for rev in os.listdir(path) if os.path.isdir(path+rev) and rev != 'latest'], key=int)

@get('/api/users/<user>/<part>/<deck>/')
def list_revs(user,part,deck):
    return json.dumps(get_revs(user,part,deck))

@get('/users/<user>/<part>/<deck>/')
@view('revs')
def list_revs(user,part,deck):
    return dict(revs=get_revs(user,part,deck))

@post('/users/<user>/decks/<deck>')
@post('/users/<user>/decks/')
@post('/api/users/<user>/decks/<deck>')
@auth_basic(validate_auth, realm='upload')
def upload_deck(user,deck = None):
    if deck is None and request.forms.get('name') is None:
        raise HTTPError(body='No name for the deck to upload')
    name = deck
    if deck is None:
        name = request.forms.get('name')
    path = "%s/%s/decks/%s/" % (root,user,name)
    new_rev = get_lastrev(user,'decks',name) + 1
    new_path = path+str(new_rev)

    if os.path.isdir(new_path):
        raise HTTPError(body='inconsistent repo, bailing out')

    if request.auth[0] != user:
        raise HTTPError(body='invalid user or insufficient rights, bailing out')

    os.makedirs(new_path)
    upload_path = new_path

    # TODO: Add all required parameters here
    tag     = request.forms.get('tag')
    content = request.files.get('file')
    # TODO: Create proper paths for uploads & Check for upload errors
    out = open(upload_path+'/deck.odp', 'wb')
    while True:
        bits = content.file.read(10240)
        if not bits:
            break
        out.write(bits)
    out.close()

    out = open(upload_path+'/meta.json', 'wb')
    json.dump({'tag': tag,
               'server_version': '1',
               'upload_filename': content.filename},
              out)

    # update link to latest rev
    if os.path.exists(path+'latest'):
        os.unlink(path+'latest')
    os.symlink(str(new_rev), path+'latest')

    # thumbnail generation happens asynchronously via updatedeck.py
    return 'Success:'+tag+':'+content.filename

@post('/users/<user>/groups/<deck>')
@post('/api/users/<user>/groups/<deck>')
@auth_basic(validate_auth, realm='upload')
def upload_group(user,deck):
    path = "%s/%s/groups/%s/" % (root,user,deck)
    new_rev = get_lastrev(user,'groups',deck) + 1
    new_path = path+str(new_rev)

    if os.path.isdir(new_path):
        raise HTTPError(body='inconsistent repo, bailing out')

    if request.auth[0] != user:
        raise HTTPError(body='invalid user or insufficient rights, bailing out')

    properties = request.json
    if properties[u'name'] != deck:
        raise HTTPError(body='Failue: url / name mismatch')

    os.makedirs(new_path)
    upload_path = new_path

    # link slides/decks
    for index, slide in enumerate(properties['slides']):
        if not os.path.isdir(root+'/'+slide):
            raise HTTPError(body='non-existing slide %d at path %s, bailing out' % (index, root+slide)) # todo: cleanup!!
        os.symlink('../../../../'+slide, upload_path+'/'+str(index))

    out = open(upload_path+'/meta.json', 'wb')
    json.dump({'server_version': '1',
               'payload': properties},
              out)

    return 'Success'


@get('/users/<user>/<part>/<deck>/<rev:int>/thumbnail.png')
@get('/api/users/<user>/<part>/<deck>/<rev:int>/thumbnail.png')
def get_thumbnail(user, part, deck, rev):
    path = "%s/%s/%s/%s/%d/" % (root,user,part,deck,rev)
    # serve thumbnail from that rev, first slide
    return static_file('thumbnail.png', root=path+'/0/', mimetype='image/png')

# get the (chosen) slides from Decks as one single deck
def get_slides_deck(decks_and_slides, output):
    new_deck = odf_new_document('presentation')
    new_deck.delete_styles()
    new_deck_body = new_deck.get_body()
    new_deck_manifest = new_deck.get_part(ODF_MANIFEST)

    for deck_and_slides in decks_and_slides:
        deck = odf_get_document(deck_and_slides.deck)
        if deck == None:
            continue
        new_deck.merge_styles_from(deck)
        deck_body = deck.get_body()
        deck_manifest = deck.get_part(ODF_MANIFEST)
        for slide in deck_and_slides.slides:
            try:
                slide_position = int(slide) - 1 # Assuming index starts with 1
                source_slide = deck_body.get_draw_page(position = slide_position)
            except:
                continue
            if source_slide == None:
                continue

            # Copy Images
            for image in source_slide.get_images():
                uri = image.get_url()
                media_type = deck_manifest.get_media_type(uri)
                new_deck_manifest.add_full_path(uri, media_type)
                new_deck.set_part(uri, deck.get_part(uri))

            new_deck_body.append(source_slide.clone())

    new_deck.save(target=output, pretty=True)

@get('/users/<user>/<part>/<deck>/<rev:int>/deck.odp')
@get('/api/users/<user>/<part>/<deck>/<rev:int>/deck.odp')
def send_deck(user, part, deck, rev):
    path = "%s/%s/%s/%s/%d/" % (root,user,part,deck,rev)
    slides = None
    try:
        slides=request.query['slides']
        print slides
    except:
        pass
    if slides != None:
        mode, tmp_file = tempfile.mkstemp(suffix='.odp', prefix='temp', dir=tmp_root)
        d = Decks()
        d.slides = slides.split(',')
        d.deck = path+'/deck.odp'
        get_slides_deck([d],tmp_file)
        filename = tmp_file
        if filename.rfind('/') >= 0:
            filename = filename[filename.rfind('/')+1:]
        print filename
        return static_file(filename, root=tmp_root, download='deck.odp')
    return static_file('deck.odp', root=path, mimetype='text/xml', download='deck.odp')

# -------------------------------------------------------------------

def get_slides(user,part,deck,rev):
    path = "%s/%s/%s/%s/%d/" % (root,user,part,deck,rev)
    if not os.path.isdir(path):
        return []
    return sorted([slide for slide in os.listdir(path) if os.path.isdir(path+slide)], key=int)

def get_revmeta(user,part,deck,rev):
    path = "%s/%s/%s/%s/%d/" % (root,user,part,deck,rev)
    comment = open(path+'comment').read() if os.path.exists(path+'comment') else ''
    meta = json.loads(open(path+'meta.json').read()) if os.path.exists(path+'meta.json') else {}
    return {'CreationDate': time.strftime("%a, %d %b %Y %H:%M:%S GMT",
                                          time.gmtime(os.stat(path).st_ctime)),
            'CommitComment': comment,
            'Meta': meta}

@get('/api/users/<user>/<part>/<deck>/<rev:int>/')
def list_slides(user, part, deck, rev):
    return json.dumps(get_slides(user,part,deck,rev))

@get('/users/<user>/<part>/<deck>/<rev:int>/')
@view('slides')
def list_slides(user, part, deck, rev):
    return dict(slides=get_slides(user,part,deck,rev), revmeta=get_revmeta(user,part,deck,rev))

@get('/users/<user>/<part>/<deck>/<rev:int>/<slide:int>/thumbnail.png')
@get('/api/users/<user>/<part>/<deck>/<rev:int>/<slide:int>/thumbnail.png')
def get_thumbnail(user, part, deck, rev, slide):
    path = "%s/%s/%s/%s/%d/%d/" % (root,user,part,deck,rev,slide)
    return static_file('thumbnail.png', root=path, mimetype='image/png')

@get('/api/users/<user>/<part>/<deck>/<rev:int>/meta.json')
def list_revmeta(user,part,deck,rev):
    return get_revmeta(user,part,deck,rev)

# -------------------------------------------------------------------

def get_slidemeta(user,part,deck,rev,slide):
    path = "%s/%s/%s/%s/%d/%d/" % (root,user,part,deck,rev,slide)
    return json.loads(open(path+'meta.json').read()) if os.path.exists(path+'meta.json') else {}

@get('/api/users/<user>/<part>/<deck>/<rev:int>/<slide:int>/meta.json')
def list_slidemeta(user, part, deck, rev, slide):
    return get_slidemeta(user, part, deck, rev, slide)

@get('/users/<user>/<part>/<deck>/<rev:int>/<slide:int>/slide.odp')
@get('/api/users/<user>/<part>/<deck>/<rev:int>/<slide:int>/slide.odp')
def send_slide(user, part, deck, rev, slide):
    path = "%s/%s/%s/%s/%d/%d/" % (root,user,part,deck,rev,slide)
    return static_file(str(slide)+'.odp', root=path, mimetype='text/xml')

@get('/')
def home_page():
    return '<html><body>Welcome</body></html>'

@get('/users/<username>/edit')
@view('app')
def app_page(username):
    return dict(user=username)

# -------------------------------------------------------------------

class SSLInterface(ServerAdapter):
    def run(self, handler):
        server = wsgiserver.CherryPyWSGIServer((self.host, self.port), handler)
        #cert = 'server.pem'
        #server.ssl_certificate = cert
        #server.ssl_private_key = cert
        try:
            server.start()
        finally:
            server.stop()

server_names['sslinterface'] = SSLInterface

def main():
    debug(True)
    run(host='localhost', port=8080, reloader=True, quiet=False, server='sslinterface')

if __name__ == "__main__":
    main()