#from: https://gist.github.com/evz/3849704
from hyde.publisher import Publisher
from boto.s3.connection import S3Connection
from boto.s3.key import Key
import os


class S3CredentialsError(Exception):
    def __init__(self, value):
        self.value = value

    def __str__(self):
        return repr(self.value)


class S3(Publisher):
    def initialize(self, settings):
        self.settings = settings
        self.key = getattr(settings, 'key', None)
        self.secret = getattr(settings, 'secret', None)
        self.bucket = getattr(settings, 'bucket', None)
        reds = getattr(settings, 'redirects', None)
        self.redirects = []
        if reds:
            for redirect in reds:
                f, t = redirect.split(' => ')
                self.redirects.append({'from': f, 'to': t})
        if not self.key or not self.secret:
            try:
                self.key = os.environ['AWS_ACCESS_KEY_ID']
                self.secret = os.environ['AWS_SECRET_ACCESS_KEY']
            except KeyError:
                raise S3CredentialsError('You need to define both an AWS key and secret in your site.yaml or as environment variables AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY')

    def publish(self):
        super(S3, self).publish()
        conn = S3Connection(self.key, self.secret)
        bucket = conn.get_bucket(self.bucket)
        k = Key(bucket)
        root = self.site.config.deploy_root_path
        for item in root.walker.walk_files():
            redirect_meta = None
            if self.redirects:
                for redirect in self.redirects:
                    if redirect['from'] in item.path:
                        redirect_meta = redirect['to']
            k.key = item.path.replace(str(root), '')
            kmd5 = bucket.get_key(item.path.replace(str(root), ''))
            #print('f:', repr(item.path), repr(kmd5.etag[1:-1], k.compute_md5(open(item.path))[0]))
            if kmd5 is None or \
              kmd5.etag[1:-1] != k.compute_md5(open(item.path, 'rb'))[0]:
                print('updating:', repr(item.path.replace(str(root), '')))
                k.set_contents_from_filename(item.path)
            if redirect_meta:
                k.set_metadata('website-redirect-location', redirect_meta)