#!/usr/bin/python2.7
from bottle import route, run, post, request

from gradient import Gradient
from imgrid import Imgrid
from breaker import Breaker
from pattern import Pattern

from config import AWS_SECRET_ACCESS_KEY, AWS_ACCESS_KEY_ID, BUCKET_NAME

import os
import sys
import lib.db as db
import mimetypes
import sha
from subprocess import call, Popen, PIPE
import simplejson as json
BIN_IDENTIFY = "/usr/bin/identify"

from boto.s3.connection import S3Connection
from boto.s3.key import Key
#
#try:
#    DB = db.db ()
#except Exception as e:
#    sys.stderr.write("Could not connect to db:\n{}".format(e))
#    sys.exit(1);
BASE_URL = "http://i.asdf.us"

def hashdir(filename):
    return  sha.new(filename).hexdigest()[:2]

def file_size (filepath):
    try:
        return os.stat(filepath)[6]
    except Exception as e:
        sys.stderr.write(str(e))
        raise;

def bin_identify (filepath):
  ident = Popen([BIN_IDENTIFY, filepath], stdout=PIPE).communicate()[0]
  partz = ident.split(" ")
  width,height = partz[2].split("x")
  return [ width, height ]


def cleanup(filepath):
    try:
        call(['rm', filepath])
    except Exception as e:
        sys.stderr.write(str(e))
        raise 

def s3move(filename,objectname):
    try:
        conn = S3Connection(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, is_secure=False)
        b = conn.get_bucket(BUCKET_NAME)
        k = Key(b)
        k.key = objectname
        k.set_contents_from_filename(filename)
        k.set_acl('public-read')
        k.storage_class = 'REDUCED_REDUNDANCY'
    except Exception as e:
        sys.stderr.write(str(e));
        raise(e)

def insert_cmd (date, remote_addr, username, url, directory, oldfile, newfile, cmd, dataobj, tag):
    try:
        sql = "INSERT INTO im_cmd "
        sql += "(date, remote_addr, name, url, dir, oldfile, newfile, cmd, dataobj, tag) "
        sql += "VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
        #or "NULL"
        args = (date, remote_addr, username, url, directory, oldfile, newfile, cmd, dataobj, tag)
        #args = (now(), os.environ['REMOTE_ADDR'], name, url, dir, oldfile, newfile, " ".join(cmd),dataobj)
        DB.execute(sql, args)
    except Exception as e:
        sys.stderr.write(str(e))
        return

def return_image(im, insert_url="NULL"):
    directory = hashdir(im.filename)
    dimensions = bin_identify(im.filepath)
    size = file_size(im.filepath)
    objectname = "im/{}/{}".format(directory, im.filename)
    try:
        s3move(im.filepath, objectname)
        cleanup(im.filepath)
#    #    return "{}\n{}\n{}\n{}\n{}\n{}\n{}\n{}".format(
        insert_cmd(
            im.now, 
            request.environ.get('REMOTE_ADDR'),
            im.params['username'] or "NULL",
            insert_url,
            directory,
            "NULL",
            im.filename, 
            ";".join(im.commands),
            json.dumps(im.params),
            im.tag,
        )
        return json.dumps({
            'url' : "{}/{}".format(BASE_URL, objectname),
            'size' : size,
            'width' : "{}px".format(dimensions[0]),
            'height' : "{}px".format(dimensions[1]),
        })
    except Exception as e:
        sys.stderr.write(str(e))
        raise;
        

@post('/im/api/imgradient')
def gradient():
    try:
        im = Gradient(**(dict(request.forms)))
        im.create();
        return return_image(im)
    except Exception as e:
        sys.stderr.write("imgradient failure\n")
        sys.stderr.write("params:\n")
        for i in request.forms:
            sys.stderr.write("{}:{}\n".format(i, request.forms[i]))
        raise;
        return json.dumps({ 'error' : 'Request could not be processed' })

@post('/im/api/imgrid')
def imgrid():
    try:
        im = Imgrid(request.forms)
        im.create();
        url= "NULL"
        for elem in [ im.params.imageinstead , im.params.bgimage, im.params.planebgimage ]:
            if elem:
                url = elem['url']
                break
        return return_image(im, url)
    except Exception as e:
        sys.stderr.write(str(e))
        return json.dumps({ 'error' : 'Request could not be processed' })

@post('/im/api/imbreak')
def breaker():
    try:
        im = Breaker(request.forms)
        im.create();
        return return_image(im, im.params['url'])
    except Exception as e:
        sys.stderr.write(str(e))
        return json.dumps({ 'error' : 'Request could not be processed' })

@post('/im/api/impattern')
def pattern():
    try:
        im = Pattern(request.forms)
        im.create();
        return return_image(im, im.params['image_url'])
    except Exception as e:
        sys.stderr.write(str(e))
        sys.stderr.write("imgradient failure\n")
        sys.stderr.write("params:\n")
        for i in request.forms:
            sys.stderr.write("{}:{}\n".format(i, request.forms[i]))
        raise;
        return json.dumps({ 'error' : 'Request could not be processed' })

run(host='0.0.0.0', server='flup', port=8999, debug=True)