# Copyright 2014, Tresys Technology, LLC
#
# This file is part of SETools.
#
# SETools is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 2.1 of
# the License, or (at your option) any later version.
#
# SETools 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with SETools.  If not, see
# <http://www.gnu.org/licenses/>.
#

import string

import setools.qpol as qpol

import role
import mls
import symbol


class User(symbol.PolicySymbol):

    """A user."""

    @property
    def roles(self):
        """The user's set of roles."""

        r = set()

        aiter = self.qpol_symbol.get_role_iter(self.policy)
        while not aiter.end():
            item = role.Role(
                self.policy, qpol.qpol_role_from_void(aiter.get_item()))

            # object_r is implicitly added to all roles by the compiler.
            # technically it is incorrect to skip it, but policy writers
            # and analysts don't expect to see it in results, and it
            # will confuse, especially for set equality user queries.
            if item != "object_r":
                r.add(item)

            aiter.next()

        return r

    @property
    def mls_default(self):
        """The user's default MLS level."""
        return mls.MLSRange(self.policy, self.qpol_symbol.get_range(self.policy))

    @property
    def mls_range(self):
        """The user's MLS range."""
        return mls.MLSLevel(self.policy, self.qpol_symbol.get_dfltlevel(self.policy))

    def statement(self):
        roles = list(self.roles)
        stmt = "user {0} ".format(self)
        if (len(roles) > 1):
            stmt += "{{ {0} }}".format(string.join(str(r) for r in roles))
        else:
            stmt += str(roles[0])

        try:
            stmt += " level {0.mls_default} range {0.mls_range};".format(self)
        except AttributeError:
            stmt += ";"

        return stmt

# Copyright 2014, Tresys Technology, LLC
#
# This file is part of SETools.
#
# SETools is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 2.1 of
# the License, or (at your option) any later version.
#
# SETools 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with SETools.  If not, see
# <http://www.gnu.org/licenses/>.
#

import string

import setools.qpol as qpol

import role
import mls
import symbol


class User(symbol.PolicySymbol):

    """A user."""

    @property
    def roles(self):
        """The user's set of roles."""

        r = set()

        aiter = self.qpol_symbol.get_role_iter(self.policy)
        while not aiter.end():
            item = role.Role(
                self.policy, qpol.qpol_role_from_void(aiter.get_item()))

            # object_r is implicitly added to all roles by the compiler.
            # technically it is incorrect to skip it, but policy writers
            # and analysts don't expect to see it in results, and it
            # will confuse, especially for set equality user queries.
            if item != "object_r":
                r.add(item)

            aiter.next()

        return r

    @property
    def mls_level(self):
        """The user's default MLS level."""
        return mls.MLSLevel(self.policy, self.qpol_symbol.get_dfltlevel(self.policy))

    @property
    def mls_range(self):
        """The user's MLS range."""
        return mls.MLSRange(self.policy, self.qpol_symbol.get_range(self.policy))


    def statement(self):
        roles = list(str(r) for r in self.roles)
        stmt = "user {0} roles ".format(self)
        if (len(roles) > 1):
            stmt += "{{ {0} }}".format(string.join(roles))
        else:
            stmt += roles[0]

        try:
            stmt += " level {0.mls_level} range {0.mls_range};".format(self)
        except AttributeError:
            stmt += ";"

        return stmt

Fix user statement rendering issues.