# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright (c) 2015 TrilioData, Inc.
# All Rights Reserved.

"""Global Settings."""


import os
import configparser
from workloadmgr.openstack.common import log as logging
from workloadmgr.db.workloadmgrdb import WorkloadMgrDB
from workloadmgr.utils import encrypt_password, tvault_key_file_name


LOG = logging.getLogger(__name__)
db = WorkloadMgrDB().db

settings_file_dir = '/var/triliovault/settings/'
settings_file_name = 'workloadmgr-settings.conf'


default_settings = {
    'cassandra_discovery_timeout': '120',
    'mongodb_discovery_timeout': '120',
    'smtp_email_enable': 'no',
    'smtp_server_name': 'localhost',
    'smtp_default_recipient': 'administrator@tvault.com',
    'smtp_default_sender': 'administrator@tvault.com',
    'smtp_port': '587',
    'smtp_server_username': '',
    'smtp_server_password': '',
    'smtp_timeout': '10',
}


def get_settings(context=None, get_hidden=False, get_smtp_settings=False):
    """get settings"""
    from workloadmgr import workloads as workloadAPI

    @workloadAPI.api.wrap_check_policy
    def get_setting(workloadAPI, context=None, get_hidden=False,
                    get_smtp_settings=False):
        try:
            copy_settings = {}
            is_admin = context.is_admin
            #If get_smtp_settings is True then changing context to admin
            #because smtp settings belongs to admin project.
            if get_smtp_settings is True:
               context.is_admin = True
               copy_settings = default_settings
            persisted_settings = {}
            persisted_setting_objs = db.setting_get_all(
                context, read_deleted='no', get_hidden=get_hidden)
            context.is_admin = is_admin
            for persisted_setting in persisted_setting_objs:
                if get_smtp_settings is False:
                    #Filtering smtp_enable and default recipient setting for current context
                    if persisted_setting.name in ('smtp_email_enable', 'smtp_default_recipient'):
                        if persisted_setting.project_id == context.project_id:
                            persisted_settings[persisted_setting.name] = persisted_setting.value
                    else:
                        persisted_settings[persisted_setting.name] = persisted_setting.value
                elif get_smtp_settings is True and 'smtp' in persisted_setting.name:
                    persisted_settings[persisted_setting.name] = persisted_setting.value
            for setting, value in copy_settings.items():
                if setting not in persisted_settings:
                    persisted_settings[setting] = value
            return persisted_settings

        except Exception as ex:
            LOG.exception(ex)
            raise ex
    try:
        return get_setting(workloadAPI, context, get_hidden, get_smtp_settings)
    except Exception as ex:
        LOG.exception(ex)
        raise ex


def set_settings(context, new_settings):
    """set settings"""
    from workloadmgr import workloads as workloadAPI

    @workloadAPI.api.upload_settings
    def upload_settings(name, context):
        pass
    try:
        persisted_setting_objs = db.setting_get_all(context)
        for name, value in new_settings.items():
            name_found = False
            for persisted_setting in persisted_setting_objs:
                if persisted_setting.name == name:
                    if name == 'smtp_email_enable':
                        if persisted_setting.project_id == context.project_id:
                            db.setting_update(context, name, {'value': value}, cloud_setting=False)
                            name_found = True
                            break
                    else:
                        if name == 'smtp_server_password':
                            value = encrypt_password(value, tvault_key_file_name) if value else ''
                        db.setting_update(context, name, {'value': value}, cloud_setting=True)
                        name_found = True
                        break
            if not name_found:
                coded_password = None
                if name == 'smtp_server_password':
                    coded_password = encrypt_password(value, tvault_key_file_name) if value else ''

                db.setting_create(context, {'name': name,
                                            'value': coded_password if name == 'smtp_server_password' else value,
                                            'user_id': context.user_id,
                                            'project_id': context.project_id,
                                            'status': 'available',
                                            'type': "email_settings" if "smtp" in name else None})

        upload_settings(name, context)
        return get_settings(context)
    except Exception as ex:
        LOG.exception(ex)
        return default_settings