from functools import wraps

from flask import abort, g

import app.modules.roxywi.auth as roxywi_auth
import app.modules.roxywi.common as roxywi_common


def check_services(fn):
    @wraps(fn)
    def decorated_view(*args, **kwargs):
        service = kwargs['service']
        if service not in ('haproxy', 'nginx', 'apache', 'keepalived', 'cluster', 'waf', 'udp'):
            abort(405, 'Wrong service')
        if not roxywi_auth.is_access_permit_to_service(service):
            abort(403, f'You do not have needed permissions to access to {service.title()} service')
        return fn(*args, **kwargs)
    return decorated_view


def get_user_params(virt=0, disable=0):
    def inner_decorator(fn):
        @wraps(fn)
        def decorated_views(*args, **kwargs):
            try:
                user_params = roxywi_common.get_users_params(virt=virt, disable=disable, service=kwargs.get('service'))
                g.user_params = user_params
            except Exception as e:
                print(e)
                return abort(401)
            return fn(*args, **kwargs)
        return decorated_views
    return inner_decorator


def page_for_admin(level=1):
    def inner_decorator(fn):
        @wraps(fn)
        def decorated_views(*args, **kwargs):
            if not roxywi_auth.is_admin(level=level):
                return abort(403, 'bad permission')
            else:
                return fn(*args, **kwargs)
        return decorated_views
    return inner_decorator


def check_group():
    def inner_decorator(fn):
        @wraps(fn)
        def decorated_views(*args, **kwargs):
            if not roxywi_common.check_user_group_for_flask():
                return roxywi_common.handle_json_exceptions('Wrong group', 'Cannot create user')
            return fn(*args, **kwargs)
        return decorated_views
    return inner_decorator