--- /dev/null
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+import requests
+import sys
+import json
+import time
+from st2actions.runners.pythonrunner import Action
+
+
+class FogAction(Action):
+ """
+ This class talks with the REST web api for the FOG server.
+ """
+ def __init__(self, config=None):
+ self.baseURL = config['fog']['address']
+ self.fogKey = config['fog']['api_key']
+ self.userKey = config['fog']['user_key']
+ self.updateHeader()
+
+ def updateHeader(self):
+ """
+ recreates the http header used to talk to the fog api
+ """
+ self.header = {}
+ self.header['fog-api-token'] = self.fogKey
+ self.header['fog-user-token'] = self.userKey
+
+ def getImageID(self, img=None, os=None, host=None):
+ """
+ returns the numerical id associated with the given img name or
+ operating system. If both are given, the img gets priority.
+ if img is a number, it is assumed to be a valid id
+ """
+ # st2 will promote an empty arg to the str "None" :(
+ if not img or img == "None":
+ return self.getImageIDFromOS(os, host)
+ try:
+ return int(img)
+ except:
+ url = self.baseURL+"image"
+ images = requests.get(url=url, headers=self.header)
+ images = images.json()['images']
+ for image in images:
+ if img == image['name']:
+ return image['id']
+ return -1
+
+ def getImageIDFromOS(self, os, host):
+ enum = {"ubuntu": "ubuntu_image",
+ "centos": "centos_image",
+ "suse": "suse_image"
+ }
+ os = os.lower()
+ if os not in enum.keys():
+ return -1
+ host_dict = json.loads(
+ self.action_service.get_value(name=host, local=False)
+ )
+ return int(host_dict[enum[os]])
+
+ def delTask(self, hostNum):
+ """
+ Tries to delete an existing task for the host
+ with hostNum as a host number
+ """
+ try:
+ url = self.baseURL+'fog/host/'+str(hostNum)+'/cancel'
+ req = requests.delete(url, headers=self.header)
+ if req.status_code == 200:
+ self.logger.info("%s", "successfully deleted image task")
+ except Exception:
+ self.logger.exception("Failed to delete the imaging task!")
+
+ def getHostNumber(self, hostname):
+ """
+ returns the host number of given host
+ """
+ try:
+ req = requests.get(self.baseURL+"host", headers=self.header)
+ hostData = req.json()
+ if hostData is not None:
+ for hostDict in hostData['hosts']:
+ if hostname == hostDict['name']:
+ return hostDict['id']
+ return -1
+ except Exception:
+ self.logger.exception('%s', "Failed to connect to the FOG server")
+
+ def request(self, url, data=None, method="get"):
+ if data is not None:
+ return self.dataRequest(url, data, method=method)
+ try:
+ response = requests.get(url, headers=self.header)
+ return response.json()
+ except Exception:
+ self.logger.exception("Failed to reach FOG at %s", url)
+ sys.exit(1)
+
+ def dataRequest(self, url, data, method="post"):
+ methods = {
+ "post": requests.post,
+ "put": requests.put
+ }
+ try:
+ return methods[method](url, json=data, headers=self.header)
+ except Exception:
+ self.logger.exception("Failed to reach FOG at %s", url)
+ sys.exit(1)
+
+ def getFogHost(self, host):
+ hostData = self.action_service.get_value(host, local=False)
+ return json.loads(hostData)['fog_name']
+
+ def waitForTask(self, taskID):
+ """
+ Watches a task and waits for it to finish (disapear).
+ There may be a smarter way to do this and track errors,
+ but st2 will timeout for me if something goes wrong
+ """
+ task = self.getTask(taskID)
+ while(task):
+ time.sleep(15)
+ task = self.getTask(taskID)
+
+ def getAllTasks(self):
+ try:
+ tasks = requests.get(
+ self.baseURL+'task/current',
+ headers=self.header
+ ).json()['tasks']
+ return tasks
+ except Exception:
+ return []
+
+ def getTask(self, taskID):
+ for task in self.getAllTasks():
+ if task['id'] == taskID:
+ return task
+ return {}
--- /dev/null
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+from fogAction import FogAction
+import requests
+import sys
+
+
+class StartCaptureAction(FogAction):
+ def __init__(self, config=None):
+ super(StartCaptureAction, self).__init__(config=config)
+
+ def run(self, host=None):
+ """
+ Schedules a capture for the given host with its
+ assigned image
+ """
+ host = self.getFogHost(host)
+ num = str(self.getHostNumber(host))
+ url = self.baseURL+'host/'+num+'/task'
+ try:
+ requests.post(url, headers=self.header, json={"taskTypeID": 2})
+ except Exception:
+ sys.exit(1)
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: fog_captureHost
+entry_point: fog_captureHost.py
+runner_type: python-script
+enabled: true
+parameters:
+ host:
+ type: string
+ required: true
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: fog_captureWorkflow
+entry_point: workflows/fog_captureWorkflow.yaml
+runner_type: action-chain
+enabled: true
+parameters:
+ host:
+ type: string
+ required: true
--- /dev/null
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+from fogAction import FogAction
+import requests
+
+
+class ChangeImageAction(FogAction):
+ def __init__(self, config):
+ super(ChangeImageAction, self).__init__(config=config)
+
+ def run(self, host=None, image=None, os=None):
+ """
+ Sets the image to be used during ghosting to the image
+ with id imgNum. host can either be a hostname or number.
+ """
+ imgNum = self.getImageID(img=image, os=os, host=host)
+ if imgNum < 0:
+ return
+ host = self.getFogHost(host)
+ hostnum = self.getHostNumber(host)
+ url = self.baseURL+"host/"+str(hostnum)
+ host_conf = requests.get(url, headers=self.header).json()
+ host_conf['imageID'] = str(imgNum)
+ requests.put(url+"/edit", headers=self.header, json=host_conf)
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: "fog_changeImage"
+enabled: true
+runner_type: python-script
+entry_point: "fog_changeImage.py"
+parameters:
+ host:
+ type: string
+ required: true
+ image:
+ required: false
+ type: string
+ os:
+ required: false
+ type: string
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: "fog_imageWorkflow"
+entry_point: "workflows/fog_imageWorkflow.yaml"
+runner_type: "action-chain"
+enabled: true
+parameters:
+ host:
+ type: "string"
+ required: true
+ image:
+ required: false
+ type: "string"
+ os:
+ required: false
+ type: "string"
--- /dev/null
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+from fogAction import FogAction
+import requests
+import sys
+
+
+class StartImagingAction(FogAction):
+ def __init__(self, config=None):
+ super(StartImagingAction, self).__init__(config=config)
+
+ def run(self, host=None):
+ """
+ Schedules an imaging task for the given host.
+ This automatically uses the "associated" disk image.
+ """
+ host = self.getFogHost(host)
+ num = str(self.getHostNumber(host))
+ url = self.baseURL+'host/'+num+'/task'
+ try:
+ req = requests.post(
+ url,
+ headers=self.header,
+ json={"taskTypeID": 1}
+ )
+ if req.status_code == 200:
+ # self.logger.info("%s", "Scheduled image task for host")
+ pass
+ except Exception:
+ # self.logger.warning("%s", "Failed to schedule host imaging")
+ # self.logger.warning("%s", "Trying to delete existing image task")
+ self.delTask(num)
+ req = requests.post(
+ url,
+ headers=self.header,
+ json={"taskTypeID": 1}
+ )
+ if req.status_code == 200:
+ # self.logger.info("%s", "Scheduled image task for host")
+ pass
+ sys.exit(0)
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: "fog_startImaging"
+description: "This will use FOG to image the given host"
+runner_type: "python-script"
+enabled: true
+entry_point: "fog_startImaging.py"
+parameters:
+ host:
+ type: "string"
+ description: "the hostname of the host (eg pod2)"
+ required: true
--- /dev/null
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+from fogAction import FogAction
+import sys
+
+
+class waitForCaptureAction(FogAction):
+ def __init__(self, config=None):
+ super(waitForCaptureAction, self).__init__(config=config)
+
+ def run(self, host=None):
+ host = self.getFogHost(host)
+ captureTaskID = self.getCaptureTaskID(host)
+ if(captureTaskID < 0):
+ sys.exit(1)
+ self.waitForTask(captureTaskID)
+
+ def getCaptureTaskID(self, host=None):
+ for task in self.getAllTasks():
+ hostname = str(task['host']['name'])
+ if hostname == host and int(task['typeID']) == 2:
+ return task['id']
+ return -1
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: fog_waitForCapture
+enabled: true
+runner_type: python-script
+entry_point: fog_waitForCapture.py
+parameters:
+ host:
+ type: string
+ required: true
--- /dev/null
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+from fogAction import FogAction
+import sys
+
+
+class WaitForImagingAction(FogAction):
+ def __init__(self, config=None):
+ super(WaitForImagingAction, self).__init__(config=config)
+
+ def run(self, host):
+ """
+ tracks the imaging task to completion.
+ """
+ host = self.getFogHost(host)
+ imageTaskID = self.getImagingTaskID(host)
+ if(imageTaskID < 0):
+ sys.exit(1)
+ self.waitForTask(imageTaskID)
+
+ def getImagingTaskID(self, host):
+ """
+ Sorts through all current tasks to find the image task
+ associated with the given host.
+ """
+ for task in self.getAllTasks():
+ hostname = str(task['host']['name'])
+ if hostname == host and int(task['typeID']) == 1:
+ return task['id']
+ return -1
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: "fog_waitForImaging"
+entry_point: "fog_waitForImaging.py"
+runner_type: "python-script"
+enabled: true
+parameters:
+ host:
+ required: true
+ type: "string"
--- /dev/null
+#!/bin/bash
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+ssh -o userknownhostsfile=/dev/null -o stricthostkeychecking=false root@"$1" "shutdown -r now"
+
+exit 0
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: restartHost
+description: restarts given host imediately and returns 0
+runner_type: local-shell-script
+entry_point: restartHost.sh
+parameters:
+ host:
+ type: string
+ required: true
+ description: host to restart
+ position: 0
--- /dev/null
+#!/bin/bash
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+ping -c 1 "$1"
+
+while [ $? -gt 0 ] ; do
+ ping -c 1 "$1" && sleep 5
+done
+
+false
+
+while [ $? -gt 0 ] ; do
+ ping -c 1 "$1" && sleep 5
+done
+exit 0
--- /dev/null
+---
+##############################################################################
+# Copyright 2017 Parker Berberian and Others #
+# #
+# Licensed under the Apache License, Version 2.0 (the "License"); #
+# you may not use this file except in compliance with the License. #
+# You may obtain a copy of the License at #
+# #
+# http://www.apache.org/licenses/LICENSE-2.0 #
+# #
+# Unless required by applicable law or agreed to in writing, software #
+# distributed under the License is distributed on an "AS IS" BASIS, #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and #
+# limitations under the License. #
+##############################################################################
+
+name: waitForBoot
+entry_point: waitForBoot.sh
+runner_type: local-shell-script
+enabled: true
+parameters:
+ host:
+ type: string
+ required: true
+ position: 0
--- /dev/null
+---
+chain:
+ -
+ name: "startCapture"
+ ref: "pharoslaas.fog_captureHost"
+ parameters:
+ host: "{{host}}"
+ on-success: "restartHost"
+ -
+ name: "restartHost"
+ ref: "pharoslaas.restartHost"
+ parameters:
+ host: "{{host}}"
+ on-success: "waitForCapture"
+ -
+ name: "waitForCapture"
+ ref: "pharoslaas.fog_waitForCapture"
+ parameters:
+ host: "{{host}}"
+ on-success: "waitForBoot"
+
+ -
+ name: "waitForBoot"
+ ref: pharoslaas.waitForBoot
+ parameters:
+ host: "{{host}}"
+ timeout: 60
--- /dev/null
+---
+chain:
+ -
+ name: "changeImage"
+ ref: "pharoslaas.fog_changeImage"
+ parameters:
+ image: "{{image}}"
+ host: "{{host}}"
+ os: "{{os}}"
+ on-success: "startImaging"
+
+ -
+ name: "startImaging"
+ ref: pharoslaas.fog_startImaging
+ parameters:
+ host: "{{host}}"
+ on-success: "restartHost"
+
+ -
+ name: "restartHost"
+ ref: pharoslaas.restartHost
+ parameters:
+ host: "{{host}}"
+ on-success: "waitForImaging"
+
+ -
+ name: "waitForImaging"
+ ref: pharoslaas.fog_waitForImaging
+ parameters:
+ host: "{{host}}"
+ timeout: 180
+ on-success: "waitForBoot"
+
+ -
+ name: "waitForBoot"
+ ref: pharoslaas.waitForBoot
+ parameters:
+ host: "{{host}}"
+ timeout: 60
Code Review / pharos-tools.git / commitdiff