#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Copyright (c) 2020 Huawei Technologies Co., Ltd.
# A-Tune is licensed under the Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#     http://license.coscl.org.cn/MulanPSL2
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
# PURPOSE.
# See the Mulan PSL v2 for more details.
# Create: 2022-6-25

"""
Routers for /v1/UI/tuning url.
"""

import logging
import json
from flask import abort
from flask_restful import Resource

from analysis.ui.parser import UI_TUNING_GET_PARSER
from analysis.ui.config import UiConfig
from analysis.engine import transfer_web
from analysis.ui.util import authenticate

LOGGER = logging.getLogger(__name__)
CORS = [('Access-Control-Allow-Origin', '*')]


class UiTuning(Resource):
    """restful api for web ui tuning page"""
    method_decorators = [authenticate]

    def get(self, cmd):
        """restful api get"""
        if not cmd:
            abort(404, 'does not get command')

        args = UI_TUNING_GET_PARSER.parse_args()
        if not UiConfig.db_enable:
            if cmd == 'initialPage':
                status = args.get('status')
                return transfer_web.get_type_list(status)

            if cmd == 'rename':
                name = args.get('name')
                new_name = args.get('newName')
                return transfer_web.rename_tuning_file(name, new_name)

            if cmd == 'chooseFile':
                name = args.get('name')
                status = args.get('status')
                return json.dumps({'isExist': transfer_web.tuning_exist(status, name)}), 200, CORS

            if cmd == 'initialChart':
                name = args.get('name')
                status = args.get('status')
                if not transfer_web.tuning_exist(status, name):
                    return json.dumps({'isExist': False}), 200, CORS
                return json.dumps(transfer_web.get_file_info(status, name)), 200, CORS

            if cmd == 'compareWith':
                name = args.get('name')
                line = args.get('line')
                response_obj = {}
                response_obj['status'] = 'finished'
                if not transfer_web.tuning_exist('finished', name):
                    response_obj['isExist'] = False
                    return json.dumps(response_obj), 200, CORS
                return json.dumps(transfer_web.get_file_data('finished', name, line, \
                        response_obj)), 200, CORS

            if cmd == 'getTuningData':
                name = args.get('name')
                status = args.get('status')
                line = args.get('line')
                response_obj = {}
                response_obj['status'] = status
                if not transfer_web.tuning_exist(status, name):
                    response_obj['isExist'] = False
                    return json.dumps(response_obj), 200, CORS
                return json.dumps(transfer_web.get_file_data(status, name, line, \
                        response_obj)), 200, CORS

            if cmd == 'getTuningStatus':
                name = args.get('name')
                return json.dumps(transfer_web.find_file_dir(name)), 200, CORS
            return '', 200, CORS

        from analysis.ui.database import trigger_tuning
        if cmd == 'initialPage':
            uid = args.get('uid')
            res = trigger_tuning.count_tuning_list(int(uid))
            return json.dumps({'count': res}), 200, CORS

        if cmd == 'getList':
            uid = int(args.get('uid'))
            page_num = int(args.get('pageNum'))
            page_size = int(args.get('pageSize'))
            list = trigger_tuning.get_tuning_list(uid, page_num, page_size)
            return json.dumps({'data': list}), 200, CORS

        if cmd == 'rename':
            name = args.get('name')
            new_name = args.get('newName')
            res, reason = trigger_tuning.rename_tuning(name, new_name)
            return json.dumps({'rename': res, 'reason': reason}), 200, CORS

        if cmd == 'chooseFile':
            name = args.get('name')
            res = trigger_tuning.tuning_exist(name)
            return json.dumps({'isExist': res}), 200, CORS

        if cmd == 'initialChart':
            name = args.get('name')
            status = args.get('status')
            if not trigger_tuning.tuning_exist(name):
                return json.dumps({'isExist': False}), 200, CORS
            response_obj = trigger_tuning.get_tuning_info(status, name)
            return json.dumps(response_obj), 200, CORS

        if cmd == 'compareWith':
            name = args.get('name')
            line = args.get('line')
            response_obj = trigger_tuning.get_tuning_data('finished', name, line)
            return json.dumps(response_obj), 200, CORS

        if cmd == 'getTuningData':
            name = args.get('name')
            status = args.get('status')
            line = args.get('line')
            response_obj = trigger_tuning.get_tuning_data(status, name, line)
            return json.dumps(response_obj), 200, CORS

        if cmd == 'getTuningStatus':
            name = args.get('name')
            res = trigger_tuning.get_tuning_status(name)
            return json.dumps({'status': res}), 200, CORS
        
        if cmd == 'updateDescription':
            tid = args.get('tid')
            description = args.get('description')
            res = trigger_tuning.update_tuning_description(tid, description)
            return json.dumps({'status': res}), 200, CORS
        return '', 200, CORS