# !/usr/bin/python
# -*- coding: utf-8 -*-


import json

from base.plugin.abstract_plugin import AbstractPlugin


class Conky(AbstractPlugin):
    def __init__(self, data, context):
        super(Conky, self).__init__()
        self.data = data
        self.context = context
        self.logger = self.get_logger()
        self.conky_config_file = '/etc/conky/conky.conf'
        self.start_command = "su - {} -c 'conky'"
        self.logger.debug('[Conky] Parameters were initialized.')

    def handle_policy(self):
        try:

            try:
                if self.is_installed('conky') is False:
                    self.logger.info('[Conky] Could not found Conky. It will be installed')
                    self.logger.debug('[Conky] Conky installing with using apt-get')
                    self.install_with_apt_get('conky')
                    self.logger.info('[Conky] Could installed')

                self.logger.debug('[Conky] Some processes found which names are conky. They will be killed.')
                self.execute('killall -9 conky')
            except:
                self.logger.error('[Conky] Conky install-kill problem.')
                raise

            self.create_file('/tmp/conky.conf')
            self.logger.debug('[Conky] Temp file was created.')

            self.write_file('/tmp/conky.conf', json.loads(self.data)['message'])
            self.logger.debug('[Conky] Temp file was filled by context.')

            self.copy_file('/tmp/conky.conf', '/etc/conky/conky.conf')
            self.logger.debug('[Conky] Configurated conf file.')
            self.logger.debug('[Conky] Running Conky...')

            if self.context.get('username') != None:
                self.execute(self.start_command.format(self.context.get('username')), result=False)
                self.logger.debug('[Conky] Creating response.')
                self.context.create_response(code=self.get_message_code().POLICY_PROCESSED.value, message='Conky policy executed successfully')
            else:
                raise Exception('Username: None')

        except Exception as e:
            self.logger.error('[Conky] A problem occurred while handling Conky policy. Error Message: {}'.format(str(e)))
            self.context.create_response(code=self.get_message_code().POLICY_ERROR.value, message='A problem occurred while handling Conky policy')


def handle_policy(profile_data, context):
    print('[Conky] Handling...')
    plugin = Conky(profile_data, context)
    plugin.handle_policy()
    print('[Conky] Executed.')