# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
# GPL 2009
from __future__ import division
import os
import hashlib
import sys
import urllib
import urllib2
import cookielib
import time
import subprocess
import tempfile
import shutil

try:
    import json
except:
    import simplejson as json

import oxlib
from oxlib import MultiPartForm
from firefogg import Firefogg

class Backend:
    url = 'http://127.0.0.1:8000/api/'
    files = {}
    def __init__(self, name):
        self.cj = cookielib.CookieJar()
        self.opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(self.cj))
        urllib2.install_opener(self.opener)

        self.name = name

    def load_files(self, files_json='files.json'):
        with open(files_json) as f:
            self.files = json.loads(f.read())
    
    def json_request(self, url, form):
        try:
            request = urllib2.Request(url)
            request.add_header('User-agent', 'oxclient')
            body = str(form)
            request.add_header('Content-type', form.get_content_type())
            request.add_header('Content-length', len(body))
            request.add_data(body)
            result = urllib2.urlopen(request).read().strip()
            return json.loads(result)
        except urllib2.HTTPError, e:
            if e.code >= 500:
                with open('/tmp/error.html', 'w') as f:
                    f.write(e.read())
                os.system('firefox /tmp/error.html')
            result = e.read()
            try:
                result = json.loads(result)
            except:
                result = {'status':{}}
            result['status']['code'] = e.code
            result['status']['text'] = str(e)
            return result

    def api(self, function, data, files={}):
        form = MultiPartForm()
        form.add_field('action', function)
        form.add_field('data', json.dumps(data))
        if files:
            for key in files:
                value = files[key]
                form.add_file(key, os.path.basename(value.name), value)
        return self.json_request(self.url, form)

    def login(self, username, password):
        r = self.api('login', {'username': username, 'password': password})
        return r['status']['code'] == 200

    def update(self):
        data = {
            'files': self.files,
            'archive': self.name,
        }
        result = self.api('update', data)
        print result

    def upload(self, oshash, filename):
        url = "%supload/" % self.url
        ogg = Firefogg(cj=self.cj, debug=True)
        ogg.upload(url, filename, {"oshash": oshash})

    def upload_frames(self, oshash, filename):
        info = oxlib.avinfo(filename)
        pos = info['duration'] / 2
        def get_frame(filename, output, pos):
            cmd = ['oxframe', '-i', filename, '-o', output, '-p', str(pos)]
            p = subprocess.Popen(cmd)
            p.wait()
            return p.returncode == 0
        tmp = tempfile.mkdtemp()
        for p in [pos/2, pos, pos+pos/2]:
            output = os.path.join(tmp, '%s.png' % oxlib.formatDuration(p*1000).replace(':', '.'))
            if get_frame(filename, output, p):
                print self.api('upload', {
                    'item': 'frame',
                    'oshash': oshash,
                    'position': p},
                    files={'file': open(output)})
        shutil.rmtree(tmp)

if __name__ == "__main__":
    c = Backend('TextArchive')
    #c.url = 'http://oxdev.local/api/'
    c.login('test', 'test2')
    c.api('addArchive', {'name': 'TextArchive'})
    c.load_files()
    c.update()

    c.upload('6000aeaa8c90c092', '/home/j/Downloads/Elgincinema_03.ogv')
    c.upload_frames('6000aeaa8c90c092', '/home/j/Downloads/Elgincinema_03.ogv')