# -*- python -*-
from common import *

class MyCG(Entry):
    def __init__(self, title, command, cg_class, *args, priority = None, **kwargs):
        self.title = title
        self.cg_class = cg_class
        rule = '::'.join(cg_class.split('::')[2:])
        self.start_command = [command, '-R', rule]
        if priority is not None:
            self.start_command.extend(['-p', str(priority)])
        self.stop_command = [command, '-R', rule, '-x']
        Entry.__init__(self, *args, **kwargs)
    
    def action(self, col, button, x, y):
        proc = subprocess.Popen(['cg-query', '-c', '?'], stdout = subprocess.PIPE)
        output = []
        while True:
            chunk = proc.stdout.read(128)
            if not chunk:
                break
            output.append(chunk)
        proc.stdout.close()
        proc.wait()
        output = b''.join(output).decode('utf-8', 'replace')
        crtcs = output.split('\n')
        output = []
        for crtc in crtcs:
            if not crtc:
                continue
            proc = subprocess.Popen(['cg-query', '-c', crtc], stdout = subprocess.PIPE)
            while True:
                chunk = proc.stdout.read(128)
                if not chunk:
                    break
                output.append(chunk)
            proc.stdout.close()
            proc.wait()
        output = b''.join(output).decode('utf-8', 'replace')
        active = ('\n    Class: %s\n' % self.cg_class) in output
        del output
        print(active)
        print(repr(self.stop_command if active else self.start_command))
        subprocess.Popen(self.stop_command if active else self.start_command).wait()
    
    def function(self):
        return self.title

class MyCGNegative(MyCG):
    def __init__(self, *args, title = 'Negative', rule = 'xpybar', priority = None, **kwargs):
        MyCG.__init__(self, title, 'cg-negative', 'cg-tools::cg-negative::' + rule, *args, priority = priority, **kwargs)