#!/usr/bin/python3

from gi.repository import GLib
from pydbus import SystemBus
import raw_write
import logging
import threading

DEAMON_ORG = 'org.mageia.Magiback'
DEAMON_INTERFACE = DEAMON_ORG
ISODUMPER_INTERFACE = DEAMON_ORG + ".Isodumper"
logger = logging.getLogger('magiback.system')


class Isodumper(raw_write.Dumper):
    """
		<node>
			<interface name='org.mageia.Magiback.Isodumper'>
				<method name='do_write'>
					<arg type='s' name='source' direction='in'/>
					<arg type='s' name='target' direction='in'/>
				</method>
				<method name='do_unmount'>
					<arg type='s' name='device' direction='in'/>
					<arg type='b' name='success' direction='out'/>
					<arg type='s' name='message' direction='out'/>
				</method>
				<method name='do_format'>
					<arg type='s' name='device' direction='in'/>
					<arg type='s' name='format' direction='in'/>
					<arg type='b' name='success' direction='out'/>
					<arg type='s' name='message' direction='out'/>
				</method>
				<method name='check_write'>
					<arg type='s' name='target' direction='in'/>
					<arg type='s' name='source' direction='in'/>
				</method>
				<method name='end'>
					<arg type='b' name='success' direction='out'/>
					<arg type='s' name='message' direction='out'/>
				</method>
                <property name="done" type="b" access="read"/>
                <property name="progress" type="i" access="read"/>
                <property name="message" type="s" access="read"/>
                <property name="state" type="b" access="read"/>
			</interface>
		</node>
    """
    def __init__(self):
        super().__init__()
        self.finished = threading.Event()
        self._progress = 0
        self.loop = GLib.MainLoop()
        self.return_state = False
        self.return_message = ""
        self.authorized_sender_read = set()
        self.authorized_sender_write = set()

    def do_write(self,source, target):
        print("Writing")
        self.finished.clear()
        self.thread = threading.Thread(target=self._do_write, args=(source, target, ))
        self.thread.start()

    @property
    def done(self):
        return self.finished.wait(1)
        
    @property
    def message(self):
        return self.return_message

    @property
    def state(self):
        return self.return_state

    def end(self):
        self.thread.join()
        self._progress = 0
        self.finished.clear()
        return self.return_state, self.return_message

    @property
    def progress(self):
        return self._progress

    def do_unmount(self, device):
        print("Unmounting")
        return self._do_unmount([device])

    def check_write(self,target, source):
        self.thread = threading.Thread(target=self._check_write, args=(target, source,))
        self.thread.start()
        
    def run(self):
       self.loop.run()

    def check_permission_write(self, sender):
        """ Check for senders permission to update system packages"""
        if sender in self.authorized_sender_write:
            return
        else:
            if self._check_permission(sender, 'org.baseurl.DnfSystem.write'):
                self.authorized_sender_write.add(sender)

    def _check_permission(self, sender, action):
        """ Check senders permissions using PolicyKit1
        """
        if not sender:
            raise ValueError('sender == None')

        obj = SystemBus().get(
            'org.freedesktop.PolicyKit1',
            '/org/freedesktop/PolicyKit1/Authority')
        iface = obj['org.freedesktop.PolicyKit1.Authority']
        (granted, _, details) = iface.CheckAuthorization(
            ('system-bus-name', {'name': sender}), action, {},
            1, '', timeout=600)
        if not granted:
            return False
        else:
            return True

  
class ConfFile(object):
    """
		<node>
			<interface name='org.mageia.Magiback'>
				<method name='setName'>
					<arg type='s' name='a' direction='in'/>
				</method>
				<method name='getFile'>
					<arg type='s' name='content' direction='out'/>
				</method>
				<method name='saveFile'>
					<arg type='s' name='content' direction='in'/>
				</method>
			</interface>
		</node>
    """
    def __init__(self):
        super().__init__()
    
    def setName(self,file_name):
        self.file_name = file_name

    def getFile(self):
        with open(self.file_name,'r') as tcf :
            # read the file
            content=""
            while 1:
                line = tcf.readline()
                if not line:
                    break
                content += line
        return content
    
    def saveFile(self, tc):
        try:
            with open(self.file_name,'w') as tcf :
                tcf.write(tc)
        except:
            return False
        return True

if __name__ == '__main__':
    print("Running Magiback service.")

    bus = SystemBus()
    conf_file =  ConfFile()
    bus.publish(DEAMON_ORG, conf_file,
        ("Isodumper", Isodumper()))
    loop = GLib.MainLoop()
    loop.run()