#!/usr/bin/env python
# coding=utf8

import re

from flask import Blueprint, current_app, render_template, json, url_for

from huansi_utils.webapi import ApiController

CDN_HOST = 'cdn.staticfile.org'
ELEMENT_VERSION = '2.3.8'
VUE_VERSION = '2.5.17-beta.0'
MARKED_VERSION = '0.3.19'
FILE_SAVER_VERSION = '2014-11-29'


class CDN(object):
    def get_resource_url(self, filename):
        raise NotImplementedError


class StaticCDN(object):
    def __init__(self, static_endpoint='static', rev=False):
        self.static_endpoint = static_endpoint
        self.rev = rev

    def get_resource_url(self, filename):
        extra_args = {}

        return url_for(self.static_endpoint, filename=filename, **extra_args)


class WebCDN(object):
    def __init__(self, baseurl):
        self.baseurl = baseurl

    def get_resource_url(self, filename):
        return self.baseurl + filename


class ConditionalCDN(object):
    def __init__(self, confvar, primary, fallback):
        self.confvar = confvar
        self.primary = primary
        self.fallback = fallback

    def get_resource_url(self, filename):
        if not current_app.config[self.confvar]:
            return self.primary.get_resource_url(filename)
        return self.fallback.get_resource_url(filename)


def find_resource(filename, cdn, use_minified=True, local=True):
    if use_minified:
        filename = '%s.min.%s' % tuple(filename.rsplit('.', 1))

    cdns = current_app.extensions['api_doc']['cdns']
    resource_url = cdns[cdn].get_resource_url(filename)

    if resource_url.startswith('//'):
        resource_url = 'https:%s' % resource_url

    return resource_url


def get_all_subclasses(cls):
    all_subclasses = []
    for subclass in cls.__subclasses__():
        all_subclasses.append(subclass)
        tmp = get_all_subclasses(subclass)
        all_subclasses.extend(tmp)

    return all_subclasses


class ApiDoc(object):
    def __init__(self, app=None):
        if app is not None:
            self.init_app(app)
            # app_code获取
            self.app_code = app.config.get('APP_CODE', 'APP_CODE 未定义')

    def init_app(self, app):
        app.config.setdefault('API_DOC_MEMBER', [])
        app.config.setdefault('API_DOC_ENABLE', True)
        app.config.setdefault('API_DOC_CDN', False)
        app.config.setdefault('RESTFUL_API_DOC_EXCLUDE', [])

        with app.app_context():
            if current_app.config['API_DOC_ENABLE']:
                api_doc = Blueprint(
                    'api_doc',
                    __name__,
                    template_folder='templates',
                    static_folder='static',
                    static_url_path=app.static_url_path + '/api_doc')

                # api_doc = Blueprint(
                #     'api_doc',
                #     __name__,
                #     template_folder='templates',
                #     static_folder='static',
                #     static_url_path=app.static_url_path + '/api_doc',
                #     url_prefix='/docs/api')

                app.jinja_env.globals['find_resource'] = \
                    find_resource
                app.jinja_env.add_extension('jinja2.ext.do')

                local = StaticCDN('api_doc.static', rev=True)
                static = StaticCDN()

                def lwrap(cdn, primary=static):
                    return ConditionalCDN('API_DOC_CDN', primary, cdn)

                elementJs = lwrap(
                    WebCDN('//%s/element-ui/%s/' % (CDN_HOST,
                                                    ELEMENT_VERSION)), local)

                elementCss = lwrap(
                    WebCDN('//%s/element-ui/%s/theme-chalk/' %
                           (CDN_HOST, ELEMENT_VERSION)), local)

                vue = lwrap(
                    WebCDN('//%s/vue/%s/' % (CDN_HOST, VUE_VERSION)), local)

                marked = lwrap(
                    WebCDN('//%s/marked/%s/' % (CDN_HOST, MARKED_VERSION)),
                    local)

                fileSaver = lwrap(
                    WebCDN('//%s/FileSaver.js/%s/' %
                           (CDN_HOST, FILE_SAVER_VERSION)), local)

                app.extensions['api_doc'] = {
                    'cdns': {
                        'local': local,
                        'static': static,
                        'elementJs': elementJs,
                        'elementCss': elementCss,
                        'vue': vue,
                        'marked': marked,
                        'fileSaver': fileSaver
                    },
                }

                @api_doc.route('/', methods=['GET'])
                def index():

                    dataDict = {}
                    existsRule = []

                    for rule in app.url_map.iter_rules():

                        dataList = []
                        api_group_name = str(rule).split('/')[1]

                        if api_group_name == "":
                            continue

                        if re.search(r'^/{}/.+'.format(api_group_name), str(rule) + '//') \
                                and api_group_name not in existsRule:

                            existsRule.append(api_group_name)
                            apiController = ApiController()
                            dataDict[api_group_name.lower()] = {'children': []}

                            if not hasattr(app.view_functions[rule.endpoint], 'view_class'):
                                api = {
                                    'name': '',
                                    'name_extra': '',
                                    'url': '',
                                    'method': [],
                                    'doc': '',
                                    'doc_md': '',
                                    'router': api_group_name.lower(),
                                    'api_type': 'api'
                                }
                                try:
                                    func = app.view_functions[rule.endpoint]

                                    api['name'] = func.__name__
                                    api['url'] = str(rule)
                                    api['method'] = ' '.join([
                                        r.upper() for r in rule.methods if r in
                                                                           ['GET', 'POST', 'PUT', 'DELETE']
                                    ])

                                    doc = self.get_api_doc(func)
                                    try:
                                        api['doc'] = doc.split('@@@')[0]
                                    except Exception as e:
                                        api['doc'] = 'No doc found for this Api'

                                    if api['doc'] != 'No doc found for this Api':
                                        api['name_extra'] = api['doc'].split(
                                            '\n\n')[0].split('\n')[0].strip(
                                            ' ').strip('\n\n').strip(
                                            ' ').strip('\n').strip(' ')

                                    api['doc'] = api['doc'].replace(
                                        api['name_extra'], '',
                                        1).rstrip(' ').strip('\n\n').rstrip(
                                        ' ').strip('\n').rstrip(' ')

                                    if api['doc'] == '':
                                        api['doc'] = 'No doc found for this Api'

                                    try:
                                        api['doc_md'] = doc.split('@@@')[
                                            1].strip(' ')
                                        try:
                                            space_count = doc.split('@@@')[
                                                0].split('\n')[-1].count(' ')
                                        except Exception as e:
                                            space_count = 0
                                        api['doc_md'] = '\n'.join(
                                            api['doc_md'].split(
                                                '\n' + ' ' * space_count))
                                    except Exception as e:
                                        api['doc_md'] = ''

                                except Exception as e:
                                    pass
                                else:
                                    dataList.append(api)
                            else:
                                for api_url, api_name_list in apiController._ApiController__rules[
                                    app.view_functions[rule.endpoint].view_class].items():
                                    for api_name in api_name_list:
                                        api = {
                                            'name': '',
                                            'name_extra': '',
                                            'url': '',
                                            'method': [],
                                            'doc': '',
                                            'doc_md': '',
                                            'router': api_group_name.lower(),
                                            'api_type': 'api'
                                        }
                                        try:
                                            func = getattr(app.view_functions[rule.endpoint].view_class, api_name)

                                            api['name'] = api_group_name + ' ' + api_name
                                            api['url'] = api_url[3:]
                                            api['method'] = api_name.split('_')[0].upper()

                                            doc = self.get_api_doc(func)
                                            try:
                                                api['doc'] = doc.split('@@@')[0]
                                            except Exception as e:
                                                api['doc'] = 'No doc found for this Api'

                                            if api['doc'] != 'No doc found for this Api':
                                                api['name_extra'] = api['doc'].split(
                                                    '\n\n')[0].split('\n')[0].strip(
                                                    ' ').strip('\n\n').strip(
                                                    ' ').strip('\n').strip(' ')

                                            api['doc'] = api['doc'].replace(
                                                api['name_extra'], '',
                                                1).rstrip(' ').strip('\n\n').rstrip(
                                                ' ').strip('\n').rstrip(' ')

                                            if api['doc'] == '':
                                                api['doc'] = 'No doc found for this Api'

                                            try:
                                                api['doc_md'] = doc.split('@@@')[
                                                    1].strip(' ')
                                                try:
                                                    space_count = doc.split('@@@')[
                                                        0].split('\n')[-1].count(' ')
                                                except Exception as e:
                                                    space_count = 0
                                                api['doc_md'] = '\n'.join(
                                                    api['doc_md'].split(
                                                        '\n' + ' ' * space_count))
                                            except Exception as e:
                                                api['doc_md'] = ''

                                        except Exception as e:
                                            pass
                                        else:
                                            dataList.append(api)
                        else:
                            continue
                        if dataList != []:
                            dataList.sort(key=lambda x: x['name'])
                            dataDict[api_group_name.lower()]['children'] = dataList

                    dataDict = {'data': dataDict}
                    return render_template(
                        'apidoc/api_doc.html', data=json.dumps(dataDict), app_code=self.app_code)

                app.register_blueprint(api_doc)

    def get_api_name(self, func):
        """e.g. Convert 'do_work' to 'Do Work'"""
        words = func.__name__.split('_')
        words = [w.capitalize() for w in words]
        return ' '.join(words)

    def get_api_doc(self, func):
        if func.__doc__:
            return func.__doc__
        else:
            return 'No doc found for this Api'