#
# Brian C. Lane <bcl@redhat.com>
#
# Copyright 2014 Red Hat, Inc.
#
# This copyrighted material is made available to anyone wishing to use, modify,
# copy, or redistribute it subject to the terms and conditions of the GNU
# General Public License v.2. This program is distributed in the hope that it
# will be useful, but WITHOUT ANY WARRANTY expressed or implied, including the
# implied warranties 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. Any Red Hat
# trademarks that are incorporated in the source code or documentation are not
# subject to the GNU General Public License and may only be used or replicated
# with the express permission of Red Hat, Inc.
#
from pykickstart.version import F22
from pykickstart.base import BaseData, KickstartCommand
from pykickstart.errors import KickstartParseError, KickstartParseWarning
from pykickstart.options import KSOptionParser
import warnings
from pykickstart.i18n import _
[docs]
class F22_SshKeyData(BaseData):
removedKeywords = BaseData.removedKeywords
removedAttrs = BaseData.removedAttrs
def __init__(self, *args, **kwargs):
BaseData.__init__(self, *args, **kwargs)
self.username = kwargs.get("username", None)
self.key = kwargs.get("key", "")
def __eq__(self, y):
if not y:
return False
return self.username == y.username
def __ne__(self, y):
return not self == y
def __str__(self):
retval = BaseData.__str__(self)
retval += "sshkey"
retval += self._getArgsAsStr() + '\n'
return retval
def _getArgsAsStr(self):
retval = ""
retval += " --username=%s" % self.username
retval += ' "%s"' % self.key
return retval
[docs]
class F38_SshKeyData(F22_SshKeyData):
removedKeywords = BaseData.removedKeywords
removedAttrs = BaseData.removedAttrs
def _getArgsAsStr(self):
retval = ""
retval += " --username=%s" % self.username
retval += ' "%s"' % self.key.replace('"', r'\"')
return retval
[docs]
class RHEL9_SshKeyData(F38_SshKeyData):
pass
[docs]
class F22_SshKey(KickstartCommand):
removedKeywords = KickstartCommand.removedKeywords
removedAttrs = KickstartCommand.removedAttrs
def __init__(self, writePriority=0, *args, **kwargs):
KickstartCommand.__init__(self, writePriority, *args, **kwargs)
self.op = self._getParser()
self.sshUserList = kwargs.get("sshUserList", [])
def __str__(self):
retval = ""
for user in self.sshUserList:
retval += user.__str__()
return retval
def _getParser(self):
op = KSOptionParser(prog="sshkey", description="""
This installs a ssh key to the authorized_keys file
of the specified user on the installed system.""",
epilog="""
Note that the key should be quoted, if it contains
spaces and the user should exist (or be root)
either via creation by a package install or the
kickstart ``user`` command.""", version=F22)
op.add_argument("--username", required=True, metavar="<user>",
version=F22, help="""
User for which to install the specified key. This option is required.""")
op.add_argument('sshkey', metavar='"ssh key"', nargs=1,
version=F22, help="""
The content of the ssh key to install.""")
return op
[docs]
def parse(self, args):
ud = self.dataClass() # pylint: disable=not-callable
(ns, extra) = self.op.parse_known_args(args=args, lineno=self.lineno)
assert len(ns.sshkey) == 1
if extra:
mapping = {"command": "sshkey", "options": extra}
raise KickstartParseError(_("Unexpected arguments to %(command)s command: %(options)s") % mapping, lineno=self.lineno)
self.set_to_obj(ns, ud)
ud.key = ns.sshkey[0]
ud.lineno = self.lineno
if ud in self.dataList():
warnings.warn(_("An ssh user with the name %s has already been defined.") % ud.username, KickstartParseWarning)
return ud
[docs]
def dataList(self):
return self.sshUserList
@property
def dataClass(self):
return self.handler.SshKeyData