sync-s3.py

#!/usr/bin/env python3

# Copyright 2018 Tyler Caraza-Harter
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import boto3, uuid, os, io, json, mimetypes, git, sys

# https://s3.us-east-2.amazonaws.com/caraza-harter-4dcf7c05-8564-11e8-a86d-6a00020017a0/index.html
#
# to invalidate SDN, do this:
# aws cloudfront create-invalidation --distribution-id E2MK8CNX2PPPIY --paths "/*"

s3 = boto3.client('s3')

class Syncer:
    def __init__(self):
        self.subdomain = 'tyler' # will be the root directory locally
        self.bucket = self.get_web_bucket('%s.caraza-harter.com' % self.subdomain)
        self.dry = False # dry run
    
    def get_web_bucket(self, search):
        for bucket in s3.list_buckets()['Buckets']:
            name = bucket['Name']
            try:
                tagging = s3.get_bucket_tagging(Bucket=name)
            except Exception as e:
                continue

            tagging = {row['Key']: row['Value'] for row in tagging['TagSet']}
            website = tagging.get('website', None)
            if website:
                if website == search:
                    return name

        return None

    def get_s3_path(self, local_path):
        return os.path.relpath(local_path, self.subdomain)

    def get_last_commit(self):
        response = s3.get_object(Bucket=self.bucket,
                                     Key='commit.txt')
        return response['Body'].read().decode('utf-8')

    def set_last_commit(self, commit):
        s3.put_object(Bucket=self.bucket,
                      Key='commit.txt',
                      Body=commit.encode('utf-8'),
                      ContentType='text/plain')

    def delete_path(self, local_path):
        s3_path = self.get_s3_path(local_path)
        if s3_path.startswith('..'):
            print('SKIP ' + local_path)
            return
        print('DELETE %s' % (s3_path))

        # do upload
        if not self.dry:
            s3.delete_object(Bucket=self.bucket, Key=s3_path)

    def sync_path(self, local_path, ttl=3600):
        s3_path = self.get_s3_path(local_path)
        if s3_path.startswith('..'):
            print('SKIP ' + local_path)
            return
        print('%s => %s' % (local_path, s3_path))

        # do upload
        with open(local_path, 'rb') as f:
            ContentType = mimetypes.guess_type(local_path)[0]
            if ContentType == None:
                ContentType = 'string'
            if local_path.endswith(".ipynb"):
                ContentType = "application/octet-stream"
            if not self.dry:
                s3.put_object(Bucket=self.bucket,
                              Key=s3_path,
                              Body=f,
                              ContentType=ContentType,
                              CacheControl='max-age=%d'%ttl)

    def sync_all(self):
        for dirpath,_,filenames in os.walk(self.subdomain):
            for filename in filenames:
                local_path = os.path.join(dirpath, filename)
                sync_path(local_path)
        self.set_last_commit(git.Repo('.').commit().hexsha)

    def sync(self):
        repo = git.Repo('.')
        prev = repo.commit(self.get_last_commit())
        curr = repo.commit()
        for d in prev.diff(curr):
            change_type = d.change_type
            print('GIT: %s %s' % (change_type, d.b_path))
            if change_type in ['A', 'M']:
                self.sync_path(d.b_path)
            elif change_type in ['D']:
                self.delete_path(d.b_path)
            elif change_type in ['R']:
                self.delete_path(d.a_path)
                self.sync_path(d.b_path)
            else:
                print('Cannot sync mutations of type "%s" yet, please resync all.' % change_type)
        if not self.dry:
            self.set_last_commit(curr.hexsha)

def main():
    syncer = Syncer()
    paths = sys.argv[1:]
    if len(paths) == 0:
        syncer.sync()
    else:
        for path in paths:
            # use short cache timeout for these since we're debugging
            print('sync %s' % path)
            syncer.sync_path(path, ttl=3)

if __name__ == '__main__':
    main()