78 lines
3.0 KiB
Python
78 lines
3.0 KiB
Python
"""
|
|
Installs and configures ssh keys
|
|
"""
|
|
|
|
import glob
|
|
import logging
|
|
import os
|
|
import tempfile
|
|
|
|
import packstack.installer.engine_processors as process
|
|
import packstack.installer.engine_validators as validate
|
|
from packstack.installer import basedefs
|
|
import packstack.installer.common_utils as utils
|
|
|
|
from packstack.modules.ospluginutils import gethostlist
|
|
|
|
# Controller object will be initialized from main flow
|
|
controller = None
|
|
|
|
# Plugin name
|
|
PLUGIN_NAME = "OS-SSHKEYS"
|
|
PLUGIN_NAME_COLORED = utils.getColoredText(PLUGIN_NAME, basedefs.BLUE)
|
|
|
|
logging.debug("plugin %s loaded", __name__)
|
|
|
|
def initConfig(controllerObject):
|
|
global controller
|
|
controller = controllerObject
|
|
logging.debug("Adding SSH KEY configuration")
|
|
paramsList = [
|
|
{"CMD_OPTION" : "ssh-public-key",
|
|
"USAGE" : "Path to a Public key to install on servers. If a usable key has not been installed on the remote servers the user will be prompted for a password and this key will be installed so the password will not be required again",
|
|
"PROMPT" : "Enter the path to your ssh Public key to install on servers",
|
|
"OPTION_LIST" : [],
|
|
"VALIDATION_FUNC" : validate.validateFile,
|
|
"PROCESSOR_FUNC" : process.processSSHKey,
|
|
"DEFAULT_VALUE" : (glob.glob(os.path.join(os.environ["HOME"], ".ssh/*.pub"))+[""])[0],
|
|
"MASK_INPUT" : False,
|
|
"LOOSE_VALIDATION": False,
|
|
"CONF_NAME" : "CONFIG_SSH_KEY",
|
|
"USE_DEFAULT" : False,
|
|
"NEED_CONFIRM" : False,
|
|
"CONDITION" : False },
|
|
]
|
|
|
|
groupDict = { "GROUP_NAME" : "SSHKEY",
|
|
"DESCRIPTION" : "SSH Configs ",
|
|
"PRE_CONDITION" : utils.returnYes,
|
|
"PRE_CONDITION_MATCH" : "yes",
|
|
"POST_CONDITION" : False,
|
|
"POST_CONDITION_MATCH" : True}
|
|
|
|
controller.addGroup(groupDict, paramsList)
|
|
|
|
|
|
def initSequences(controller):
|
|
puppetsteps = [
|
|
{'title': 'Setting up ssh keys',
|
|
'functions':[installKeys]}
|
|
]
|
|
controller.addSequence("Setting up ssh keys", [], [], puppetsteps)
|
|
|
|
|
|
def installKeys():
|
|
with open(controller.CONF["CONFIG_SSH_KEY"]) as fp:
|
|
sshkeydata = fp.read().strip()
|
|
for hostname in gethostlist(controller.CONF):
|
|
if '/' in hostname:
|
|
hostname = hostname.split('/')[0]
|
|
server = utils.ScriptRunner(hostname)
|
|
# TODO replace all that with ssh-copy-id
|
|
server.append("mkdir -p ~/.ssh")
|
|
server.append("chmod 500 ~/.ssh")
|
|
server.append("grep '%s' ~/.ssh/authorized_keys > /dev/null 2>&1 || echo %s >> ~/.ssh/authorized_keys" % (sshkeydata, sshkeydata))
|
|
server.append("chmod 400 ~/.ssh/authorized_keys")
|
|
server.append("restorecon -r ~/.ssh")
|
|
server.execute()
|