import re
import sys

import inflection
import requests


with open('prelude.py') as f:
    prelude = f.read()



reserved = dir(__builtins__)


def download(url):
    return requests.get(url).json()


def mkname(method, name, params):
    singular = inflection.singularize(name)
    if method == 'get' and 'id_' not in params:
        return 'list_{}'.format(name)
    elif method == 'get':
        return singular
    elif method == 'post':
        return 'create_{}'.format(singular)
    elif method == 'put' or method == 'PATCH':
        return 'update_{}'.format(singular)
    elif method == 'delete':
        return 'delete_{}'.format(singular)
    return name


def mkparam(param):
    param = inflection.underscore(param)
    if param in reserved:
        return "{}_".format(param)
    return param


def pretty_print(name, data):
    info = data['info']
    r = re.compile('\\/\\{([^\\}]*)\\}')
    r2 = re.compile('\\{([^\\}]*)\\}')
    print(prelude)
    print('class {}(Client):'.format(name))
    print('  """{}\n\n {}"""'.format(info['title'], info['description']))
    for path, fns in data['paths'].items():
        for method, body in fns.items():
            method = method.lower()
            fmt = r2.sub('{}', path)
            params = [mkparam(param) for param in r.findall(path)]
            name = r.sub('', path).replace("/", "_").replace('-', '_')[1:]
            name = mkname(method, name, params)
            params = ", {}".format(", ".join(params)) if params else ''
            print('  def {}(self{}):'.format(name, params))
            print('    """{}"""'.format(', '.join(body['tags'])))
            print('    return self.{}("{}"{})\n'.format(method, fmt, params))


def generate(name, url):
    data = download(url)
    pretty_print(name, data)


if __name__ == '__main__':
    if len(sys.argv) != 3:
        print("usage: {} <class-name> <url>".format(sys.argv[0]))
        sys.exit(1)
    generate(sys.argv[1], sys.argv[2])