#!/usr/bin/python3 -Es

# Authors: Petr Lautrbach <plautrba@redhat.com>
#
# Copyright (C) 2020 Red Hat, Inc.

# 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.

import signal
from gi.repository import GLib
from dasbus.connection import SystemMessageBus
import setroubleshoot.util

loop = GLib.MainLoop()


def alarm_handler(signum, frame):
    loop.quit()


class Privileged():
    __dbus_xml__ = """
    <node>
        <interface name='org.fedoraproject.SetroubleshootPrivileged'>
            <method name='get_rpm_nvr_by_scontext'>
                <arg type='s' name='scontext' direction='in'/>
                <arg type='s' name='rpmnvr' direction='out'/>
            </method>
            <method name='finish'/>
        </interface>
    </node>
    """

    def __init__(self, timeout=10):
        self.timeout = timeout
        signal.signal(signal.SIGALRM, alarm_handler)
        self.alarm(self.timeout)

    def alarm(self, timeout=10):
        signal.alarm(timeout)

    def get_rpm_nvr_by_scontext(self, scontext):
        """Finds an SELinux module which defines given SELinux context"""
        signal.alarm(self.timeout)
        rpmnvr = setroubleshoot.util.get_rpm_nvr_by_scontext(scontext)
        if rpmnvr is None:
            return ""

        return rpmnvr

    def finish(self):
        loop.quit()


if __name__ == "__main__":
    bus = SystemMessageBus()
    try:
        bus.publish_object("/org/fedoraproject/SetroubleshootPrivileged", Privileged())
        bus.register_service("org.fedoraproject.SetroubleshootPrivileged")
        loop.run()
    finally:
        bus.disconnect()