#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import sys
import json
import requests
import pprint
import warnings
warnings.filterwarnings('ignore', category=requests.packages.urllib3.exceptions.InsecurePlatformWarning)

import argparse


# create the top-level parser
parser = argparse.ArgumentParser(description='Google Apps Script ExImporter')
parser.add_argument('--token', required=True)
parser.add_argument('-d', dest='debug', action='store_true', help='debug')

subparsers = parser.add_subparsers(dest='command')

# list projects
parser_im = subparsers.add_parser('list')

# import
parser_im = subparsers.add_parser('import')
parser_im.add_argument('project')
parser_im.add_argument('directory')
parser_im.add_argument('--exported-file', required=False)

# export
parser_ex = subparsers.add_parser('export')
parser_ex.add_argument('project')
parser_ex.add_argument('file')

options = parser.parse_args()

if options.debug:
    print options


def request_get(url, headers, timeout=10):
    s = requests.Session()
    req = requests.Request(
        'GET',
        url,
        headers=headers if headers else dict(),
    )

    return s.send(
        req.prepare(),
        timeout=timeout,
    )


def request_put(url, headers, data, timeout=10):
    s = requests.Session()
    req = requests.Request(
        'PUT',
        url,
        data=data,
        headers=headers if headers else dict(),
    )

    return s.send(
        req.prepare(),
        timeout=timeout,
    )


def list_projects(token):
    url = 'https://www.googleapis.com/drive/v2/files?q=mimeType%3D\'application%2Fvnd.google-apps.script\'+and+\'me\'+in+owners'

    resp = request_get(
        url,
        headers={
            'Authorization': 'Bearer %s' % token,
        },
    )

    if resp.status_code not in (200,):
        print '[ee] failed to get list: %s' % resp.text
        sys.exit(1)

    projects = dict()
    data = json.loads(resp.text)
    for i in data.get('items'):
        print '>', i.get('title')
        print '  id: %s: ' % i.get('id')
        print 'link: %s' % i.get('selfLink')
        projects[i.get('title')] = i

    return projects


def get_project(project, token):
    projects = list_projects(token)

    if options.debug:
        print 'current projects'
        pprint.pprint(projects)

    project = project.decode('utf-8')
    if project not in projects:
        print '[ee] project not found: `%s`' % project
        sys.exit(1)

    return projects.get(project)


def export_project(project, token, filename=None):
    project = get_project(project, token)

    url = 'https://script.google.com/feeds/download/export?id=%(id)s&format=json'
    resp = request_get(
        url % project,
        headers={
            'Authorization': 'Bearer %s' % token,
        },
    )

    # save file
    if resp.status_code not in (200,):
        print '[ee] failed to export project: %s' % resp.text
        sys.exit(1)

    content = json.loads(resp.text),
    if not filename:
        return content

    with file(filename, 'wb') as f:
        json.dump(
            content,
            f,
            indent=2,
        )

    print 'exported to `%s`' % filename

    return


def import_project(project, token, directory, exported=None):
    if not os.path.exists(directory):
        print '[ee] file not found, `%s`' % directory
        sys.exit(1)

    if not exported:
        exported = export_project(project, token)
    else:
        exported = json.loads(file(exported).read())

    print 'current exported file are: %s' % exported

    file_ids = dict()
    if len(exported) > 0:
        for i in exported[0].get('files'):
            file_ids[i.get('name')] = i.get('id')

    files = list()
    for i in os.listdir(directory):
        name, ext = os.path.splitext(i)
        if ext not in ('.gs', '.html'):
            continue

        if ext in ('.gs',):
            file_type = 'server_js'
        elif ext in ('.html'):
            file_type = 'html'
        else:
            print '[ee] unknown file, `%s`' % i
            sys.exit(1)

        f = dict(
            source=file(os.path.join(directory, i)).read(),
            type=file_type,
            name=name,
        )
        if name in file_ids:
            f['id'] = file_ids.get(name)

        files.append(f)

    project = get_project(project, token)

    url = 'https://www.googleapis.com/upload/drive/v2/files/%(id)s'

    resp = request_put(
        url % project,
        data=json.dumps(dict(files=files)),
        headers={
            'Authorization': 'Bearer %s' % token,
            'Content-Type': 'application/vnd.google-apps.script+json',
        },
    )

    if resp.status_code not in (200,):
        print '[ee] failed to export project: %s' % resp.text
        sys.exit(1)

    print 'imported.'

    return


if options.command == 'list':
    pprint.pprint(list_projects(options.token))
elif options.command == 'export':
    export_project(options.project, options.token, options.file)
elif options.command == 'import':
    import_project(options.project, options.token, options.directory, exported=options.exported_file)


sys.exit(0)