# -*- coding: utf-8 -*-

# Copyright 2013 Evgeniy Alekseev <esalexeev@gmail.com>
# 
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyKDE4.plasma import Plasma
import config
import ptmnotify



class NewPlasmaLabel(Plasma.Label):
    """new Label with defined clicked() event"""
    def __init__(self, applet, parent):
        """class definition"""
        Plasma.Label.__init__(self, applet)
        self.parent = parent
        self.notify = ptmnotify.PTMNotify(self)
    
    
    def mousePressEvent(self, event):
        """mouse click event"""
        if (event.button() == Qt.LeftButton):
            self.notify.init()



class Reinit():
    def __init__(self, parent):
        """class definition"""
        self.parent = parent
    
    
    def reinit(self, confAccept=False):
        """function to reinitializate widget"""
        settings = config.Config(self.parent)
        self.parent.interval = settings.get('interval', 2000).toInt()[0]
        self.parent.font_family = str(settings.get('font_family', 'Terminus'))
        self.parent.font_size = settings.get('font_size', 12).toInt()[0]
        self.parent.font_color = str(settings.get('font_color', '#000000'))
        self.parent.font_style = str(settings.get('font_style', 'normal'))
        self.parent.font_weight = settings.get('font_weight', 400).toInt()[0]
        self.parent.formatLine = "<pre><p align=\"center\"><span style=\" font-family:'" + self.parent.font_family + "'; font-style:" + self.parent.font_style
        self.parent.formatLine = self.parent.formatLine + "; font-size:" + str(self.parent.font_size) + "pt; font-weight:" + str(self.parent.font_weight)
        self.parent.formatLine = self.parent.formatLine + "; color:" + self.parent.font_color + ";\">$LINE</span></p></pre>"
        self.parent.label_order = str(settings.get('label_order', '1345'))
        for label in self.parent.dict_orders.values():
            if ((label == 'cpu') or (label == 'mem') or (label == 'swap') or (label == 'net')):
                exec ('self.parent.' + label + 'Bool = int(settings.get("' + label + 'Bool",  2))')
            else:
                exec ('self.parent.' + label + 'Bool = int(settings.get("' + label + 'Bool",  0))')
        
        # labels
        for order in self.parent.label_order:
            if (order == "1"):
                if (self.parent.cpuBool > 0):
                    self.parent.cpuFormat = str(settings.get('cpuFormat', '[cpu: $cpu%]'))
                    if (self.parent.cpuFormat.split('$ccpu')[0] != self.parent.cpuFormat):
                        self.parent.label_cpu = NewPlasmaLabel(self.parent.applet, self.parent)
                        self.parent.label_cpu1 = Plasma.Label(self.parent.applet)
                        if (self.parent.cpuFormat.split('$ccpu')[0].split('$cpu')[0] != self.parent.cpuFormat.split('$ccpu')[0]):
                            line = self.parent.cpuFormat.split('$ccpu')[0].split('$cpu')[0] + '-----' + self.parent.cpuFormat.split('$ccpu')[0].split('$cpu')[1]
                        else:
                            line = self.parent.cpuFormat.split('$ccpu')[0]
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_cpu.setText(text)
                        self.parent.layout.addItem(self.parent.label_cpu)
                        text = self.parent.formatLine.split('$LINE')[0] + "-----" + self.parent.formatLine.split('$LINE')[1]
                        for core in range(self.parent.numCores):
                            exec ('self.parent.label_coreCpu' + str(core) + ' = Plasma.Label(self.parent.applet)')
                            exec ('self.parent.label_coreCpu' + str(core) + '.setText(text)')
                            exec ('self.parent.layout.addItem(self.parent.label_coreCpu' + str(core) + ')')
                        if (self.parent.cpuFormat.split('$ccpu')[1].split('$cpu')[0] != self.parent.cpuFormat.split('$ccpu')[1]):
                            line = self.parent.cpuFormat.split('$ccpu')[1].split('$cpu')[0] + '-----' + self.parent.cpuFormat.split('$ccpu')[1].split('$cpu')[1]
                        else:
                            line = self.parent.cpuFormat.split('$ccpu')[1]
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_cpu1.setText(text)
                        self.parent.layout.addItem(self.parent.label_cpu1)
                    else:
                        self.parent.label_cpu = NewPlasmaLabel(self.parent.applet, self.parent)
                        if (self.parent.cpuFormat.split('$cpu')[0] != self.parent.cpuFormat):
                            line = self.parent.cpuFormat.split('$cpu')[0] + '-----' + self.parent.cpuFormat.split('$cpu')[1]
                        else:
                            line = self.parent.cpuFormat
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_cpu.setText(text)
                        self.parent.layout.addItem(self.parent.label_cpu)
            elif (order == "2"):
                if (self.parent.tempBool > 0):
                    self.parent.tempdev = str(settings.get('temp_device', '<select device>'))
                    self.parent.tempFormat = str(settings.get('tempFormat', '[temp: $temp&deg;C]'))
                    self.parent.label_temp = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.tempFormat.split('$temp')[0] != self.parent.tempFormat):
                        line = self.parent.tempFormat.split('$temp')[0] + '----' + self.parent.tempFormat.split('$temp')[1]
                    else:
                        line = self.parent.tempFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_temp.setText(text)
                    self.parent.layout.addItem(self.parent.label_temp)
            elif (order == "3"):
                if (self.parent.memBool > 0):
                    self.parent.memFormat = str(settings.get('memFormat', '[mem: $mem%]'))
                    if (self.parent.memFormat.split('$memmb')[0] != self.parent.memFormat):
                        self.parent.memInMb = True
                        line = self.parent.memFormat.split('$memmb')[0] + '-----' + self.parent.memFormat.split('$memmb')[1]
                    elif (self.parent.memFormat.split('$mem')[0] != self.parent.memFormat):
                        self.parent.memInMb = False
                        self.parent.mem_used = 0.0
                        self.parent.mem_free = 1.0
                        self.parent.mem_uf = 0.0
                        line = self.parent.memFormat.split('$mem')[0] + '-----' + self.parent.memFormat.split('$mem')[1]
                    else:
                        line = self.parent.memFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_mem = NewPlasmaLabel(self.parent.applet, self.parent)
                    self.parent.label_mem.setText(text)
                    self.parent.layout.addItem(self.parent.label_mem)
            elif (order == "4"):
                if (self.parent.swapBool > 0):
                    self.parent.swapFormat = str(settings.get('swapFormat', '[swap: $swap%]'))
                    if (self.parent.swapFormat.split('$swapmb')[0] != self.parent.swapFormat):
                        self.parent.swapInMb = True
                        line = self.parent.swapFormat.split('$swapmb')[0] + '-----' + self.parent.swapFormat.split('$swapmb')[1]
                    elif (self.parent.swapFormat.split('$swap')[0] != self.parent.swapFormat):
                        self.parent.swapInMb = False
                        self.parent.swap_free = 1.0
                        self.parent.swap_used = 0.0
                        line = self.parent.swapFormat.split('$swap')[0] + '-----' + self.parent.swapFormat.split('$swap')[1]
                    else:
                        line = self.parent.swapFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_swap = NewPlasmaLabel(self.parent.applet, self.parent)
                    self.parent.label_swap.setText(text)
                    self.parent.layout.addItem(self.parent.label_swap)
            elif (order == "5"):
                if (self.parent.netBool > 0):
                    self.parent.netNonFormat = str(settings.get('netNonFormat', '[net: $netKB/s]'))
                    if (self.parent.netNonFormat.split('@@')[0] != self.parent.netNonFormat):
                        self.parent.netdev = self.parent.netNonFormat.split('@@')[1]
                        self.parent.netNonFormat = self.parent.netNonFormat.split('@@')[0] + self.parent.netNonFormat.split('@@')[2]
                    else:
                        self.parent.netdev = self.parent.setupNetdev()
                    if (self.parent.netNonFormat.split('$netdev')[0] != self.parent.netNonFormat):
                        self.parent.netFormat = self.parent.netNonFormat.split('$netdev')[0] + self.parent.netdev + self.parent.netNonFormat.split('$netdev')[1]
                    else:
                        self.parent.netFormat = self.parent.netNonFormat
                    self.parent.label_netDown = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.netFormat.split('$net')[0] != self.parent.netFormat):
                        line = self.parent.netFormat.split('$net')[0] + '----'
                    else:
                        line = self.parent.netFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_netDown.setText(text)
                    self.parent.layout.addItem(self.parent.label_netDown)
                    self.parent.label_netUp = Plasma.Label(self.parent.applet)
                    if (self.parent.netFormat.split('$net')[0] != self.parent.netFormat):
                        line = '/----' + self.parent.netFormat.split('$net')[1]
                    else:
                        line = ''
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_netUp.setText(text)
                    self.parent.layout.addItem(self.parent.label_netUp)
            elif (order == "6"):
                if (self.parent.batBool > 0):
                    self.parent.batFormat = str(settings.get('batFormat', '[bat: $bat%$ac]'))
                    self.parent.battery_device= str(settings.get('battery_device', '/sys/class/power_supply/BAT0/capacity'))
                    self.parent.ac_device = str(settings.get('ac_device', '/sys/class/power_supply/AC/online'))
                    self.parent.label_bat = NewPlasmaLabel(self.parent.applet, self.parent)
                    line = self.parent.batFormat
                    if (line.split('$ac')[0] != line):
                        line = line.split('$ac')[0] + '(-)' + line.split('$ac')[1]
                    if (line.split('$bat')[0] != line):
                        line = line.split('$bat')[0] + '---' + line.split('$bat')[1]
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_bat.setText(text)
                    self.parent.layout.addItem(self.parent.label_bat)
            elif (order == "7"):
                if (self.parent.cpuclockBool > 0):
                    self.parent.cpuclockFormat = str(settings.get('cpuclockFormat', '[mhz: $cpucl]'))
                    if (self.parent.cpuclockFormat.split('$ccpucl')[0] != self.parent.cpuclockFormat):
                        self.parent.label_cpuclock = NewPlasmaLabel(self.parent.applet, self.parent)
                        self.parent.label_cpuclock1 = Plasma.Label(self.parent.applet)
                        if (self.parent.cpuclockFormat.split('$ccpucl')[0].split('$cpucl')[0] != self.parent.cpuclockFormat.split('$ccpucl')[0]):
                            line = self.parent.cpuclockFormat.split('$ccpucl')[0].split('$cpucl')[0] + '----' + self.parent.cpuclockFormat.split('$ccpucl')[0].split('$cpucl')[1]
                        else:
                            line = self.parent.cpuclockFormat.split('$ccpucl')[0]
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_cpuclock.setText(text)
                        self.parent.layout.addItem(self.parent.label_cpuclock)
                        text = self.parent.formatLine.split('$LINE')[0] + "----" + self.parent.formatLine.split('$LINE')[1]
                        for core in range(self.parent.numCores):
                            exec ('self.parent.label_coreCpuclock' + str(core) + ' = Plasma.Label(self.parent.applet)')
                            exec ('self.parent.label_coreCpuclock' + str(core) + '.setText(text)')
                            exec ('self.parent.layout.addItem(self.parent.label_coreCpuclock' + str(core) + ')')
                        if (self.parent.cpuclockFormat.split('$ccpucl')[1].split('$cpucl')[0] != self.parent.cpuclockFormat.split('$ccpucl')[1]):
                            line = self.parent.cpuclockFormat.split('$ccpucl')[1].split('$cpucl')[0] + '----' + self.parent.cpuclockFormat.split('$ccpucl')[1].split('$cpucl')[1]
                        else:
                            line = self.parent.cpuclockFormat.split('$ccpucl')[1]
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_cpuclock1.setText(text)
                        self.parent.layout.addItem(self.parent.label_cpuclock1)
                    else:
                        self.parent.label_cpuclock = NewPlasmaLabel(self.parent.applet, self.parent)
                        if (self.parent.cpuclockFormat.split('$cpucl')[0] != self.parent.cpuclockFormat):
                            line = self.parent.cpuclockFormat.split('$cpucl')[0] + '----' + self.parent.cpuclockFormat.split('$cpucl')[1]
                        else:
                            line = self.parent.cpuclockFormat
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_cpuclock.setText(text)
                        self.parent.layout.addItem(self.parent.label_cpuclock)
            elif (order == "8"):
                if (self.parent.uptimeBool > 0):
                    self.parent.uptimeFormat = str(settings.get('uptimeFormat', '[uptime: $uptime]'))
                    self.parent.label_uptime = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.uptimeFormat.split('$uptime')[0] != self.parent.uptimeFormat):
                        line = self.parent.uptimeFormat.split('$uptime')[0] + '---d--h--m' + self.parent.uptimeFormat.split('$uptime')[1]
                    else:
                        line = self.parent.uptimeFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_uptime.setText(text)
                    self.parent.layout.addItem(self.parent.label_uptime)
            elif (order == "9"):
                if (self.parent.gpuBool > 0):
                    self.parent.gpuFormat = str(settings.get('gpuFormat', '[gpu: $gpu%]'))
                    self.parent.label_gpu = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.gpuFormat.split('$gpu')[0] != self.parent.gpuFormat):
                        line = self.parent.gpuFormat.split('$gpu')[0] + '-----' + self.parent.gpuFormat.split('$gpu')[1]
                    else:
                        line = self.parent.gpuFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_gpu.setText(text)
                    self.parent.layout.addItem(self.parent.label_gpu)
            elif (order == "a"):
                if (self.parent.gputempBool > 0):
                    self.parent.gputempFormat = str(settings.get('gputempFormat', '[gpu temp: $gputemp&deg;C]'))
                    self.parent.label_gputemp = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.gputempFormat.split('$gputemp')[0] != self.parent.gputempFormat):
                        line = self.parent.gputempFormat.split('$gputemp')[0] + '----' + self.parent.gputempFormat.split('$gputemp')[1]
                    else:
                        line = self.parent.gputempFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_gputemp.setText(text)
                    self.parent.layout.addItem(self.parent.label_gputemp)
            elif (order == "b"):
                if (self.parent.hddBool > 0):
                    self.parent.hddFormat = str(settings.get('hddFormat', '[hdd: @@/@@%]'))
                    if (self.parent.hddFormat.split('@@')[0] != self.parent.hddFormat):
                        self.parent.mountPoints = self.parent.hddFormat.split('@@')[1].split(';')
                        line = self.parent.hddFormat.split('@@')[0]
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_hdd0 = NewPlasmaLabel(self.parent.applet, self.parent)
                        self.parent.label_hdd0.setText(text)
                        self.parent.layout.addItem(self.parent.label_hdd0)
                        text = self.parent.formatLine.split('$LINE')[0] + "-----" + self.parent.formatLine.split('$LINE')[1]
                        for mount in self.parent.mountPoints:
                            exec ('self.parent.label_hdd_' + ''.join(mount.split('/')) + ' = Plasma.Label(self.parent.applet)')
                            exec ('self.parent.label_hdd_' + ''.join(mount.split('/')) + '.setText(text)')
                            exec ('self.parent.layout.addItem(self.parent.label_hdd_' + ''.join(mount.split('/')) + ')')
                        line = self.parent.hddFormat.split('@@')[2]
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_hdd1 = Plasma.Label(self.parent.applet)
                        self.parent.label_hdd1.setText(text)
                        self.parent.layout.addItem(self.parent.label_hdd1)
                    else:
                        line = self.parent.hddFormat
                        text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                        self.parent.label_hdd0.setText(text)
                        self.parent.layout.addItem(self.parent.label_hdd0)
            elif (order == "c"):
                if (self.parent.hddtempBool > 0):
                    self.parent.hddtempFormat = str(settings.get('hddtempFormat', '[hdd temp: @@/dev/sda@@&deg;C]'))
                    self.parent.label_hddtemp = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.hddtempFormat.split('@@')[0] != self.parent.hddtempFormat):
                        line = self.parent.hddtempFormat.split('@@')[0] + '----' + self.parent.hddtempFormat.split('@@')[2]
                    else:
                        line = self.parent.hddtempFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_hddtemp.setText(text)
                    self.parent.layout.addItem(self.parent.label_hddtemp)
            elif (order == "d"):
                if (self.parent.playerBool > 0):
                    self.parent.playerFormat = str(settings.get('playerFormat', '[$artist - $title]'))
                    self.parent.player_name = settings.get('player_name', 0).toInt()[0]
                    self.parent.label_player = NewPlasmaLabel(self.parent.applet, self.parent)
                    line = self.parent.playerFormat
                    if (line.split('$album')[0] != line):
                        line = line.split('$album')[0] + 'N\\A' + line.split('$album')[1]
                    if (line.split('$artist')[0] != line):
                        line = line.split('$artist')[0] + 'N\\A' + line.split('$artist')[1]
                    if (line.split('$progress')[0] != line):
                        line = line.split('$progress')[0] + '00:00' + line.split('$progress')[1]
                    if (line.split('$title') != line):
                        line = line.split('$title')[0] + 'N\\A' + line.split('$title')[1]
                    if (line.split('$time') != line):
                        line = line.split('$time')[0] + '00:00' + line.split('$time')[1]
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_player.setText(text)
                    self.parent.layout.addItem(self.parent.label_player)
            elif (order == "e"):
                if (self.parent.timeBool > 0):
                    self.parent.timeFormat = str(settings.get('timeFormat', '[$time]'))
                    self.parent.label_time = NewPlasmaLabel(self.parent.applet, self.parent)
                    if (self.parent.timeFormat.split('$time')[0] != self.parent.timeFormat):
                        line = self.parent.timeFormat.split('$time')[0] + '----------------------' + self.parent.timeFormat.split('$time')[1]
                    else:
                        line = self.parent.timeFormat
                    text = self.parent.formatLine.split('$LINE')[0] + line + self.parent.formatLine.split('$LINE')[1]
                    self.parent.label_time.setText(text)
                    self.parent.layout.addItem(self.parent.label_time)
        if not confAccept:
            self.parent.layout.setContentsMargins(1, 1, 1, 1)
            self.parent.applet.setLayout(self.parent.layout)
            self.parent.theme = Plasma.Svg(self.parent)
            self.parent.theme.setImagePath("widgets/background")
            self.parent.setBackgroundHints(Plasma.Applet.DefaultBackground)
        self.parent.resize(10, 10)
        
        # create dataengines
        self.parent.thread().wait(60000)
        self.parent.dataengine.connectToEngine()
        
        self.parent.timer.setInterval(self.parent.interval)
        self.parent.startPolling()