ansible-server/src/main/scripts/RestServer.py

'''
/*-
* ============LICENSE_START=======================================================
* ONAP : APPC
* ================================================================================
* Copyright (C) 2017-2019 AT&T Intellectual Property.  All rights reserved.
* ================================================================================
* Copyright (C) 2017 Amdocs
* =============================================================================
* 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.
* 
* ECOMP is a trademark and service mark of AT&T Intellectual Property.
* ============LICENSE_END=========================================================
*/
'''

import time, datetime, json, os, sys, subprocess, re
import uuid
import tarfile
import shutil
import glob
import crypt

import requests

import cherrypy
from cherrypy.lib.httputil import parse_query_string
from cherrypy.lib import auth_basic

from multiprocessing import Process, Manager

from AnsibleModule import ansibleSysCall
from BuildHostFile import buildHostsSysCall

from os import listdir
from os.path import isfile, join

TestRecord = Manager().dict()
ActiveProcess = {}

def validate_password(realm, username, password):
    comp = crypt.crypt(password, salt)
    if username in userpassdict and userpassdict[username] == comp:
       return True
    return False

def sys_call (cmd):
    p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    output = p.stdout.readlines()
    retval = p.wait()
    if len (output) > 0:
        for i in range(len(output)):
            output[i] = output[i].strip()
    return retval, output

def callback (Id, Result, Output, Log, returncode):
    
    print "***> in RestServer.callback"

    if Id in TestRecord:
        time_now = datetime.datetime.utcnow()
        delta_time = (time_now - TestRecord[Id]['Time']).total_seconds()
        Result['PlaybookName'] = TestRecord[Id]['PlaybookName']
        Result['Version'] = TestRecord[Id]['Version']
        if returncode == 137:
            Result['StatusCode'] = 500
            Result['StatusMessage'] = "TERMINATED"
        else:
            Result['StatusCode'] = 200
            Result['StatusMessage'] = "FINISHED"

        # Need to update the whole data structure for key=Id otherwise Manager is not updated
        TestRecord[Id] = {'PlaybookName': TestRecord[Id]['PlaybookName'],
                          'Version': TestRecord[Id]['Version'],
                          'NodeList': TestRecord[Id]['NodeList'],
                          'HostGroupList': TestRecord[Id]['HostGroupList'],
                          'HostNameList': TestRecord[Id]['HostNameList'],
                          'Time': TestRecord[Id]['Time'],
                          'Timeout': TestRecord[Id]['Timeout'],
                          'Duration': str(delta_time),
                          'EnvParameters': TestRecord[Id]['EnvParameters'],
                          'LocalParameters': TestRecord[Id]['LocalParameters'],
                          'FileParameters': TestRecord[Id]['FileParameters'],
                          'CallBack': TestRecord[Id]['CallBack'],
                          'Result': Result,
                          'Log': Log, 
                          'Output': Output, 
                          'Path': TestRecord[Id]['Path'],
                          'Mandatory': TestRecord[Id]['Path']}

        if not TestRecord[Id]['CallBack'] == None:
            
            # Posting results to callback server

            data = {"StatusCode": 200,
                    "StatusMessage": "FINISHED",
                    "PlaybookName": TestRecord[Id]["PlaybookName"],
                    "Version": TestRecord[Id]["Version"],
                    "Duration": TestRecord[Id]["Duration"],
                    "Results": TestRecord[Id]['Result']['Results']}

            cherrypy.log("CALLBACK: TestRecord[Id]['Output']['Output']:", str(TestRecord[Id]['Output']['Output']))
            cherrypy.log("CALLBACK: Results:", str(data["Results"]))

            if not TestRecord[Id]['Output']['Output'] == {}:
                for key in data["Results"]:
                    if key in TestRecord[Id]['Output']['Output']:
                        data["Results"][key]["Output"] = TestRecord[Id]['Output']['Output'][key]

            print "     Posting to", TestRecord[Id]['CallBack']
            
            s = requests.Session()
            r = s.post(TestRecord[Id]['CallBack'], data = json.dumps(data),
                       headers = {'content-type': 'application/json'})
            print  "     Response", r.status_code, r.text

def RunAnsible_Playbook (callback, Id, Inventory, Playbook, NodeList, TestRecord,
                         Path, ArchiveFlag):

    print "***> in RestServer.RunAnsible_Playbook"

    # Run test in playbook for given target
    Result = ''

    retval, log, returncode = ansibleSysCall (Inventory, Playbook, NodeList,
                                              TestRecord[Id]['Mandatory'],
                                              TestRecord[Id]['EnvParameters'],
                                              TestRecord[Id]['LocalParameters'],
                                              TestRecord[Id]['Timeout'],
                                              Path)


    cherrypy.log("Return code:" + str(returncode))
    cherrypy.log("Return val:" +  str(retval))
    
    Log = ''.join(log)
    #Output = {'Output': {}}
    Output = {}
    
    onlyfiles = [f for f in listdir(Path)
                 if isfile(join(Path, f))]

    cherrypy.log("Checking for results.txt files: ")
    for file in onlyfiles:
        if "results.txt" in file:
#       if file.endswith("results.txt"):
            cherrypy.log("results file: " + file)
            f = open(Path + "/" + file, "r")
            resultsData =  f.read()  # Not to pass vnf instance name
            OutputP = json.loads(resultsData)
            Output['Output'] =  OutputP 
	    cherrypy.log("Output = " + str(Output['Output']))
            #Output['Output'][key] = f.read() # To pass vnf instance name
            f.close()

    if Output == {}:
      Output = {'Output': {}}

    Result = {'Results': {}}
    if 'could not be found' in Log:
        Result['Results'] = {"StatusCode": 101,
                             "StatusMessage": "PLAYBOOK NOT FOUND"}
    if returncode == 137:
        Result['Results'] = {"StatusCode": 500,
                             "StatusMessage": "TERMINATED"}

    elif TestRecord[Id]['NodeList'] == []:
        
        host_index = None
        
        if 'TargetNode' in TestRecord[Id]['EnvParameters']:
            targetlist = TestRecord[Id]['EnvParameters']['TargetNode'].split(' ')
        else:
            targetlist = ["localhost"]
            
        for key in retval:
            for i in range (len(targetlist)):
                if key in targetlist[i]:
                    host_index = i
    
            if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and \
                   int(retval[key][3]) == 0:

                if host_index:
                    Result['Results'][targetlist[host_index]] = \
                             {"GroupName": 'na', "StatusCode": 200, \
                              "StatusMessage": "SUCCESS"}
                else:
                    Result['Results'][key] = \
                             {"GroupName": 'na', "StatusCode": 200, \
                              "StatusMessage": "SUCCESS"}                    
            elif int(retval[key][2]) > 0:
                if host_index:
                    Result['Results'][targetlist[host_index]] = \
                       {"GroupName": 'na', "StatusCode": 400, \
                        "StatusMessage": "NOT REACHABLE"}
                else:
                    Result['Results'][key] = \
                       {"GroupName": 'na', "StatusCode": 400, \
                        "StatusMessage": "NOT REACHABLE"}                    
            elif int(retval[key][3]) > 0:
                if host_index:                
                    Result['Results'][targetlist[host_index]] = \
                       {"GroupName": 'na', "StatusCode": 400, \
                        "StatusMessage": "FAILURE"}
                else:
                    Result['Results'][key] = \
                       {"GroupName": 'na', "StatusCode": 400, \
                        "StatusMessage": "FAILURE"}                    
    else:
        
        for key in retval:

            if len(TestRecord[Id]['HostNameList']) > 0:

                host_index = []
                for i in range (len(TestRecord[Id]['HostNameList'])):
                    if key in TestRecord[Id]['HostNameList'][i]:
                        host_index.append(i)

                if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and \
                       int(retval[key][3]) == 0:

                    if len(host_index) > 0:
                        Result['Results'][TestRecord[Id]['HostNameList'][host_index[0]]] = \
                          {"GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
                           "StatusCode": 200, "StatusMessage": "SUCCESS"}
                    
                        for i in range (1, len(host_index)):
                            Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"]+=\
                             "," + TestRecord[Id]['HostGroupList'][host_index[i]]
                    else:
                       Result['Results'][key] = \
                          {"GroupName": key,
                           "StatusCode": 200, "StatusMessage": "SUCCESS"}  

                elif int(retval[key][2]) > 0:

                    if len(host_index) > 0:
                        Result['Results'][TestRecord[Id]['HostNameList'][host_index[0]]] = \
                          {"GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
                           "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
                    
                        for i in range (1, len(host_index)):
                            Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"]+=\
                             "," + TestRecord[Id]['HostGroupList'][host_index[i]]
                    else:
                       Result['Results'][key] = \
                          {"GroupName": key,
                           "StatusCode": 200, "StatusMessage": "NOT REACHABLE"}  
                    
                elif int(retval[key][3]) > 0:

                    if len(host_index) > 0:
                        Result['Results'][TestRecord[Id]['HostNameList'][host_index[0]]] = \
                          {"GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
                           "StatusCode": 400, "StatusMessage": "FAILURE"}
                    
                        for i in range (1, len(host_index)):
                            Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"]+=\
                             "," + TestRecord[Id]['HostGroupList'][host_index[i]]
                    else:
                       Result['Results'][key] = \
                          {"GroupName": key,
                           "StatusCode": 200, "StatusMessage": "FAILURE"}                          
            else:
                host_index = None
                for i in range (len(TestRecord[Id]['NodeList'])):
                    if key in TestRecord[Id]['NodeList'][i]:
                        host_index = i
    
                if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and \
                       int(retval[key][3]) == 0:
                    Result['Results'][TestRecord[Id]['NodeList'][host_index]] = \
                             {"GroupName": 'na', "StatusCode": 200, \
                             "StatusMessage": "SUCCESS"}
                elif int(retval[key][2]) > 0:
                    Result['Results'][TestRecord[Id]['NodeList'][host_index]] = \
                       {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
                elif int(retval[key][3]) > 0:
                    Result['Results'][TestRecord[Id]['NodeList'][host_index]] = \
                       {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "FAILURE"}
    
    cherrypy.log("TESTRECORD: " + str(TestRecord[Id]))
    #cherrypy.log("Output: " + str(Output))
    callback (Id, Result, Output, Log, returncode)

class TestManager (object):

    @cherrypy.expose
    @cherrypy.tools.json_out()
    @cherrypy.tools.json_in()
    @cherrypy.tools.allow(methods=['POST', 'GET', 'DELETE'])

    def Dispatch(self, **kwargs):

        # Let cherrypy error handler deal with malformed requests
        # No need for explicit error handler, we use default ones

        time_now = datetime.datetime.utcnow()

        # Erase old test results (2x timeout)
        # Do cleanup too of ActiveProcess list and old Records - PAP
        if TestRecord:
            for key in TestRecord.copy():
                cherrypy.log( "LOOKING AT ALL TestRecords: " + str(key))
                if key in ActiveProcess:
                   if not ActiveProcess[key].is_alive(): # Just to cleanup defunct processes
                      cherrypy.log( "Not ActiveProcess for ID: " + str(key))
                delta_time = (time_now - TestRecord[key]['Time']).seconds
                if delta_time > 2*TestRecord[key]['Timeout']:
                    cherrypy.log( "DELETED HISTORY for ID: " + str(key))
                    if key in ActiveProcess:
                      if not ActiveProcess[key].is_alive():
                          ActiveProcess.pop (key)
                          cherrypy.log( "DELETED ActiveProcess for ID: " + str(key))
                    #if os.path.exists(TestRecord[key]['Path']):
                        # don't remove run dirrectory
                        #shutil.rmtree (TestRecord[key]['Path'])
                    del TestRecord[key]
                    
        cherrypy.log("RestServer.Dispatch: " + cherrypy.request.method)


        if 'POST' in cherrypy.request.method:
            
            input_json = cherrypy.request.json
            cherrypy.log("Payload: " + str(input_json))

            if 'Id' in input_json and 'PlaybookName' in input_json and 'EnvParameters' in input_json:

                if True:

                    if not input_json['Id'] in TestRecord:
			# check if Id exists in previous run dirctory
                        # if so retun error
                        s_cmd = 'ls ' + ansible_temp + '/*_' + input_json['Id']
                        #if subprocess.check_output([s_cmd, ]):
                        Id = input_json['Id']
                        if glob.glob(  ansible_temp + '/*_' + input_json['Id']):
                            cherrypy.log("Old directory found for ID: " + Id)
                            return {"StatusCode": 101, "StatusMessage": "TEST ID FILE ALREADY DEFINED"}
                    
                        PlaybookName = input_json['PlaybookName']
                        # if required it should be passed as an argument
                        EnvParameters = input_json['EnvParameters']

                        # The lines below are to test multiple EnvParameters being passed
                        #for i in EnvParameters:
                        #  cherrypy.log("EnvParameter object: " + i)
                        #  cherrypy.log("  EnvParameter Value: " + EnvParameters[ i ])

			# Now get things out of EnvParameters
                        VNF_instance   = None
                        VNF_instance = EnvParameters.get('vnf_instance')

                        # Get Version if present
                        version = None
                        if 'Version' in input_json:
                            version = input_json['Version']

                        # GetInventoryNames
                        HaveNodeList         = False
                        HaveInventoryNames   = False 
                        inventory_names = None
                        if 'InventoryNames' in input_json:
                           inventory_names = input_json['InventoryNames']
                           HaveInventoryNames   = True 
                    
                        #AnsibleInvFail = True
                        AnsiblePlaybookFail = True

                        LocalNodeList = None

                        str_uuid = str (uuid.uuid4())
                        

                        VnfType= PlaybookName.split("/")[0] 
                        if auth:
                           cherrypy.log( "Request USER  :                  " + cherrypy.request.login)
                        cherrypy.log( "Request Decode: ID               " + Id)
                        cherrypy.log( "Request Decode: VnfType          " + VnfType)
                        cherrypy.log( "Request Decode: EnvParameters    " + json.dumps(EnvParameters))
                        
                        # Verify VNF_instance was passed in EnvParameters
                        if VNF_instance != None:
			  cherrypy.log( "Request Decode: VnfInstance      " + VNF_instance)
                        else:
			  cherrypy.log( "StatusCode: 107, StatusMessage: VNF_instance NOT PROVIDED" )
                          return {"StatusCode": 107,
                                    "StatusMessage": "VNF_instance NOT PROVIDED"}

                        if inventory_names != None:
			  cherrypy.log( "Request Decode: Inventory Names  " + inventory_names)
                        else:
			  cherrypy.log( "Request Decode: Inventory Names  " + "Not provided")

			cherrypy.log( "Request Decode: PlaybookName     " + PlaybookName)
			PlayBookFunction = PlaybookName.rsplit("/",2)[1]
			PlayBookFile = PlayBookFunction + "/site.yml"
			cherrypy.log( "Request Decode: PlaybookFunction " + PlayBookFunction)
			cherrypy.log( "Request Decode: Playbook file    " + PlayBookFile)
			
                        BaseDir = ansible_path + "/" + PlaybookName.rsplit("/",1)[0]
                        CopyDir = ansible_path + "/" + PlaybookName.rsplit("/",2)[0]
			cherrypy.log( "Request Decode: Basedir          " + BaseDir)
			cherrypy.log( "Request Decode: Copydir          " + CopyDir)
			

                        PlaybookDir = ansible_temp + "/" + \
                                      VNF_instance + "_" + str_uuid + "_" + str(Id)

			# AnsibleInv is the directory where the host file to be run exsists
                        AnsibleInv = ansible_path + "/" + VnfType + "/latest/ansible/inventory/" + VNF_instance
                        ArchiveFlag = False

			# Create base run directory if it doesn't exist
                        if not os.path.exists(ansible_temp):
                            cherrypy.log( "Creating Base Run Directory: "  + ansible_temp)
                            os.makedirs(ansible_temp)

			if not os.path.exists( CopyDir ):
                            cherrypy.log("Playbook Not Found")
                            return {"StatusCode": 101,
                                    "StatusMessage": "PLAYBOOK NOT FOUND"}

                        # copy static playbook dir to run dir
                        cherrypy.log("Copying from " + CopyDir + " to " + PlaybookDir)
                        shutil.copytree(CopyDir, PlaybookDir)
                        cmd="/usr/bin/find " + PlaybookDir + " -exec /usr/bin/touch {} \;"
                        cmd="/usr/bin/find " + PlaybookDir + " -exec chmod +rx  {} \;"
                        sys_call(cmd)
                        cherrypy.log(cmd)

                        cherrypy.log( "PlaybookDir:    " + PlaybookDir)
                        cherrypy.log( "AnsibleInv:     " + AnsibleInv)

                        # Process inventory file for target
                    
                        hostgrouplist = []
                        hostnamelist = []

                        NodeList = []
                        if 'NodeList' in input_json:
                            NodeList = input_json['NodeList']

                        cherrypy.log("NodeList: " + str(NodeList));

                        # if NodeList empty 
                        if NodeList == []:
                                cherrypy.log( "*** NodeList - Empty ***")

                        else:
                                HaveNodeList = True

			###############################################################################
			##### Host file processing                          ###########################
			##### 1. Use file delivered with playbook           ###########################
			##### 2. If HostNames + NodeList generate and use   ###########################
			###############################################################################

                        #Verify inventory directory exists
                        path = PlaybookDir + "/inventory/"
                        if not os.path.isdir(path):
                            cherrypy.log ("Inventory directory %s does not exist - create it" % path)
                            try:
                                os.mkdir(path)
                            except OSError:
                                cherrypy.log ("Creation of the directory %s failed" % path)
                            else:
                                cherrypy.log ("Successfully created the directory %s " % path)

			#location of host file - Default
                        HostFile = PlaybookDir + "/inventory/" + VNF_instance + "hosts"
                        cherrypy.log("HostFile: " +  HostFile)

                        # if NodeList and InventoryNames need to build host file
                        if HaveInventoryNames & HaveNodeList:
                           cherrypy.log("Build host file from NodeList")
                           ret = buildHostsSysCall (input_json, PlaybookDir, inventory_names)
                           if (ret < 0):
                              cherrypy.log("Returning Error: Not running Playbook")
                              return {"StatusCode": 105,
                                    "StatusMessage": "NodeList: Missing vnfc-type field"}

                           # Having been built now copy new file to correct file
                           shutil.copy(PlaybookDir + "/host_file.txt", HostFile)
                           cherrypy.log("Copying Generated host file to: " + HostFile)

                        timeout = timeout_seconds
                        if 'Timeout' in input_json:
                            timeout = int (input_json['Timeout'])
                            cherrypy.log("Timeout from API: " + str(timeout))

                        else:
                            cherrypy.log("Timeout not passed from API using default: " + str(timeout))

                        EnvParam = {}
                        if 'EnvParameters' in input_json:
                            EnvParam = input_json['EnvParameters']

                        LocalParam = {}
                        if 'LocalParameters' in input_json:
                            LocalParam = input_json['LocalParameters']

                        FileParam = {}
                        if 'FileParameters' in input_json:
                            FileParam = input_json['FileParameters']
                    
                        callback_flag = None
                        if 'CallBack' in input_json:
                            callback_flag = input_json['CallBack']

			# if AnsibleServer is not set to 'na'  don't send AnsibleServer in PENDING responce.
			if AnsibleServer != 'na':
                        	TestRecord[Id] = {'PlaybookName': PlaybookName,
                                          'Version': version,
                                          'NodeList': NodeList,
                                          'HostGroupList': hostgrouplist,
                                          'HostNameList': hostnamelist,
                                          'Time': time_now,
                                          'Duration': timeout,
                                          'Timeout': timeout,
                                          'EnvParameters': EnvParam,
                                          'LocalParameters': LocalParam,
                                          'FileParameters': FileParam,
                                          'CallBack': callback_flag,
                                          'Result': {"StatusCode": 100,
                                                     "StatusMessage": 'PENDING',
                                                     "AnsibleServer": str(AnsibleServer),
                                                     "ExpectedDuration": str(timeout) + "sec"},
                                          'Log': '',
                                          'Output': {},
                                          'Path': PlaybookDir,
                                          'Mandatory': None}
			else:
                        	TestRecord[Id] = {'PlaybookName': PlaybookName,
                                          'Version': version,
                                          'NodeList': NodeList,
                                          'HostGroupList': hostgrouplist,
                                          'HostNameList': hostnamelist,
                                          'Time': time_now,
                                          'Duration': timeout,
                                          'Timeout': timeout,
                                          'EnvParameters': EnvParam,
                                          'LocalParameters': LocalParam,
                                          'FileParameters': FileParam,
                                          'CallBack': callback_flag,
                                          'Result': {"StatusCode": 100,
                                                     "StatusMessage": 'PENDING',
                                                     "ExpectedDuration": str(timeout) + "sec"},
                                          'Log': '',
                                          'Output': {},
                                          'Path': PlaybookDir,
                                          'Mandatory': None}

                        cherrypy.log("Test_Record: " +  str(TestRecord[Id]))
                        # Write files
                        
                        if not TestRecord[Id]['FileParameters'] == {}:
                            for key in TestRecord[Id]['FileParameters']:
                                filename = key
                                filecontent = TestRecord[Id]['FileParameters'][key]
                                f = open(PlaybookDir + "/" + filename, "w")
                                f.write(filecontent)
                                f.close()
                                
                        
                        # Process playbook
			if os.path.exists( ansible_path + '/' + PlaybookName):
                            AnsiblePlaybookFail = False
                                    
                        if AnsiblePlaybookFail:
                            #if os.path.exists(PlaybookDir):
                                #shutil.rmtree (PlaybookDir)
                            del TestRecord[Id]
                            return {"StatusCode": 101,
                                    "StatusMessage": "PLAYBOOK NOT FOUND"}
                        else:

                            # Test EnvParameters
                            playbook_path = PlaybookDir

                            # Store local vars
                            if not os.path.exists(playbook_path + "/vars"):
                                os.mkdir(playbook_path + "/vars")
                            if not os.path.isfile(playbook_path + "/vars/defaults.yml"):
                                os.mknod(playbook_path + "/vars/defaults.yml")

                            ###################################################
                            # PAP
                            #write local parameters passed into defaults.yml
                            # PAP
                            f = open(playbook_path + "/vars/defaults.yml","a")
                            #for id, record in TestRecord.items():
                            print TestRecord[Id]['LocalParameters']
                            local_parms = TestRecord[Id]['LocalParameters']
                            for key, value in local_parms.items():
                                f.write(key +"=" + value + "\n");
                            f.close()
                            ###################################################
                                 
                            for key in TestRecord[Id]['LocalParameters']:
                                host_index = []
                                for i in range(len(TestRecord[Id]['HostNameList'])):
                                    if key in TestRecord[Id]['HostNameList'][i]:
                                        host_index.append(i)
                                if len(host_index) == 0:
                                    for i in range(len(TestRecord[Id]['HostGroupList'])):
                                        if key in TestRecord[Id]['HostGroupList'][i]:
                                            host_index.append(i)
                                if len(host_index) > 0:
                                    for i in range(len(host_index)):
                                        f = open(playbook_path + "/vars/" +
                                                 TestRecord[Id]['HostNameList'][host_index[i]] +
                                                 ".yml", "a")
                                        for param in TestRecord[Id]['LocalParameters'][key]:
                                            f.write(param + ": " +
                                             str (TestRecord[Id]['LocalParameters'][key][param]) +
                                                  "\n")
                                        f.close()

    
                            # write some info out to files before running
                            if auth:
                               f = open(playbook_path + "/User.txt", "a")
                               f.write(cherrypy.request.login)
                               f.close()
                            f = open(playbook_path + "/PlaybookName.txt", "a")
                            f.write(PlaybookName)
                            f.close()
                            f = open(playbook_path + "/PlaybookExDir.txt", "a")
                            f.write(PlaybookDir + "/" + PlayBookFunction)
                            f.close()
                            f = open(playbook_path + "/JsonRequest.txt", "w")
                            #f.write(str(input_json))
                            #print( json.dumps(input_json, indent=4, sort_keys=True))
                            f.write( json.dumps(input_json, indent=4, sort_keys=True))
                            f.close()


                            # Check that HostFile exists
                            if not os.path.isfile(HostFile):
                              cherrypy.log("Inventory file Not Found: " + HostFile)
                              return {"StatusCode": 101,
                                   "StatusMessage": "PLAYBOOK INVENTORY FILE NOT FOUND"}
   
                            # Cannot use thread because ansible module uses
                            # signals which are only supported in main thread.
                            # So use multiprocess with shared object
                                       # args = (callback, Id,  PlaybookDir + "/" + AnsibleInv,

                            p = Process(target = RunAnsible_Playbook,
                                        args = (callback, Id,  HostFile,
                                                PlaybookDir + '/' + PlayBookFile,
                                                NodeList, TestRecord, PlaybookDir + "/" + PlayBookFunction,
                                                ArchiveFlag))
                            p.start()
                            ActiveProcess[Id] = p
                            return TestRecord[Id]['Result']
                    else:
                        cherrypy.log("TEST ID ALREADY DEFINED")
                        return {"StatusCode": 101, "StatusMessage": "TEST ID ALREADY DEFINED"}

                else:
                    return {"StatusCode": 500, "StatusMessage": "REQUEST MUST INCLUDE: NODELIST"}
                
            else:
                return {"StatusCode": 500, "StatusMessage": "JSON OBJECT MUST INCLUDE: ID, PLAYBOOKNAME, EnvParameters"}

        elif 'GET' in cherrypy.request.method:
            
            # Lets pause for a second just incase the resquest was just kicked off
            time.sleep(1)

            input_data = parse_query_string(cherrypy.request.query_string)
             
            # Verify we have a Type passed in GET request
            if not ( 'Type' in input_data):
                return {"StatusCode": 500, "StatusMessage": "RESULTS TYPE UNDEFINED"}

            if auth:
               cherrypy.log( "Request USER:             " + cherrypy.request.login)
            cherrypy.log("Payload: " + str(input_data) + " Type " + input_data['Type'])

            if 'LogRest' in input_data['Type']:
		sys.stdout.close()
		sys.stdout = open("/var/log/RestServer.log", "w")

            # Just a debug to dump any records
            if 'GetStatus' in input_data['Type']:
                cherrypy.log( "******** Dump Records **********")
                if TestRecord.items():
                  for id, record in TestRecord.items():
                    cherrypy.log( "    Id: " + id)
                    cherrypy.log( "Record: " + str(record))
                else:
                  cherrypy.log(" No Records to dump")
            
            if 'Id' in input_data and 'Type' in input_data:
                if not ('GetResult' in input_data['Type'] or 'GetOutputLog' in input_data['Type'] or'GetTheOutput' in input_data['Type'] or 'GetLog' in input_data['Type']):
                    return {"StatusCode": 500, "StatusMessage": "RESULTS TYPE UNDEFINED"}
                if input_data['Id'] in TestRecord:
                    
                    if 'GetResult' in input_data['Type']:
                        
                        cherrypy.log( " ** GetResult for: " + str (input_data['Id']))

                        if 'StatusMessage' in TestRecord[input_data['Id']]['Result'] and getresults_block:


                            #check if playbook is still running
                            while ActiveProcess[input_data['Id']].is_alive():
                                cherrypy.log( "*** Playbook running returning PENDING for " + str(input_data['Id']))
                                ##
                                ## If still running return PENDING response
                                ##
                                if AnsibleServer != 'na':
                                   return {"StatusCode": 100,
                                                     "StatusMessage": 'PENDING',
                                                     "AnsibleServer": str(AnsibleServer)}
                                else: 
                                   return {"StatusCode": 100,
                                                     "StatusMessage": 'PENDING'}
                                #time.sleep(5)

                            #cherrypy.log( "*** Request released " + input_data['Id'])

                        cherrypy.log(str( TestRecord[input_data['Id']]['Result']))
                        cherrypy.log("Output: " + str( TestRecord[input_data['Id']]['Output']))
                        cherrypy.log("StatusCode: " + str( TestRecord[input_data['Id']]['Result']['StatusCode']))
                        cherrypy.log("StatusMessage: " + str( TestRecord[input_data['Id']]['Result']['StatusMessage']))

                        #out_obj gets returned to GET request
                        if TestRecord[input_data['Id']]['Result']['StatusCode'] == 500:
                            out_obj = TestRecord[input_data['Id']]['Result']['Results']
                        else:
                            out_obj = {"StatusCode": 200,
                                   "StatusMessage": "FINISHED",
                                   "PlaybookName": TestRecord[input_data['Id']]["PlaybookName"],
                                   "Version": TestRecord[input_data['Id']]["Version"],
                                   "Duration": TestRecord[input_data['Id']]["Duration"],
                                   "Output": TestRecord[input_data['Id']]["Output"]["Output"],
                                   "Results": TestRecord[input_data['Id']]['Result']['Results']}
                        if not TestRecord[input_data['Id']]['Output']['Output'] == {}:
                            cherrypy.log("TestRecord has Output:" + str(TestRecord[input_data['Id']]['Output']['Output']))
                            # PAP                
                            for key in out_obj["Results"]:
                                cherrypy.log("Output key: " + str(key))
                                if key in TestRecord[input_data['Id']]['Output']['Output']:
                                    out_obj["Results"][key]["Output"] = TestRecord[input_data['Id']]['Output']['Output'][key]

                        cherrypy.log("***** GET RETURNING RESULTS Back ****")
                        cherrypy.log(str(out_obj))
                        return out_obj

                    elif 'GetStatus' in input_data['Type']:
                        print " Dump Records"
                        for id, record in TestRecord,items():
                           print " id: " + id
                           print "   Record:" + str(reecord)

                    elif 'GetTheOutput' in input_data['Type']:

                        if TestRecord[input_data['Id']]['Output'] == {} and \
                               getresults_block:

                            cherrypy.log( "*** Request blocked " + input_data['Id'])
                            
                            while TestRecord[input_data['Id']]['Output'] == {} \
                                      or 'StatusMessage' in TestRecord[input_data['Id']]['Result']:
                                time.sleep(5)

                            cherrypy.log( "*** Request released " + input_data['Id'])
                        
                        cherrypy.log( "Output: " + str(TestRecord[input_data['Id']]['Output']))
                        return {"Output": TestRecord[input_data['Id']]['Output']['Output']}

                    elif 'GetOutputLog' in input_data['Type']:
                       cherrypy.log("GetOutputLog: processing.")
                       if glob.glob(  ansible_temp + '/*_' + input_data['Id']):
                          id = input_data['Id']
                          cherrypy.log("Old directory found for ID: " + id)
                          run_dir = glob.glob(  ansible_temp + '/*_' + input_data['Id'])
                          for dir in run_dir:
                              rdir=dir
                          if os.path.exists (rdir + "/PlaybookExDir.txt"):
                               cherrypy.log("Found PlaybookExDir.txt file")
                               f = open( rdir + '/PlaybookExDir.txt', 'r')
                               playbookexdir =  f.readline()
                               rdir = playbookexdir
                               f.close()
                          cherrypy.log("Id:     " + id)
                          cherrypy.log("RunDir: " + rdir)
                          if os.path.exists( rdir + "/output.log"):
                               cherrypy.log("Found output.log file")
                               f = open( rdir + '/output.log', 'r')
                               output_log =  f.readline()
                               f.close()
                               return output_log
                       else:
                         cherrypy.log("Globglob failed:")
                         return

                    else:
                        # GetLog

                        if TestRecord[input_data['Id']]['Log'] == '' and \
                               getresults_block:

                            cherrypy.log( "*** Request blocked " + input_data['Id'])
                            
                            while TestRecord[input_data['Id']]['Log'] == '' \
                                      or 'StatusMessage' in TestRecord[input_data['Id']]['Result']:
                                time.sleep(5)

                            cherrypy.log( "*** Request released " + input_data['Id'])
                            
                        cherrypy.log( "Log:" + str(TestRecord[input_data['Id']]['Log']))
                        return {"Log": TestRecord[input_data['Id']]['Log']}
                else:
                   # Not in memory check for a file 
                   if glob.glob(  ansible_temp + '/*_' + input_data['Id']):
                       id = input_data['Id']
                       cherrypy.log("Old directory found for ID: " + id)
                       run_dir = glob.glob(  ansible_temp + '/*_' + input_data['Id'])
                       for dir in run_dir:
                           rdir=dir
                       if os.path.exists (rdir + "/PlaybookExDir.txt"):
                            cherrypy.log("Found PlaybookExDir.txt file")
                            f = open( rdir + '/PlaybookExDir.txt', 'r')
                            playbookexdir =  f.readline()
                            rdir = playbookexdir
                            f.close()
                       cherrypy.log("Id:     " + id)
                       cherrypy.log("RunDir: " + rdir)
                       if 'GetLog' in input_data['Type']:
                         if os.path.exists( rdir + "/output.log"):
                            cherrypy.log("Found output.log file")
                            f = open( rdir + '/output.log', 'r')
                            output_log =  f.readline()
                            f.close()
                            return output_log
                       elif 'GetOutputLog' in input_data['Type']:
                            if os.path.exists( rdir + "/output.log"):
                               cherrypy.log("Found output.log file")
                               f = open( rdir + '/output.log', 'r')
                               output_log =  f.readline()
                               f.close()
                               return output_log
                       elif 'GetResult' in input_data['Type']:
                            if os.path.exists (rdir + "/PlaybookName.txt"):
                               cherrypy.log("Found PlaybookName.txt file")
                               f = open( rdir + '/PlaybookName.txt', 'r')
                               playbooknametxt =  f.readline()
                               f.close()
                            else:
                               playbooknametxt = "NA"

			    # Add code to get other items not just output.log from files
                            if os.path.exists( rdir + "/log.file"):
                               cherrypy.log("Found log.file")
                               out_results = "NA:"
                               f = open( rdir + '/log.file', 'r')
                               
                               line =  f.readline()
                               while line :
				 if "fatal" in line:
                                   out_results = out_results +  line
                                 elif "RECAP" in line:
                                   out_results = out_results +  line
                                   recap_line =  f.readline()
                                   while recap_line :
                                     out_results = out_results +  recap_line
                                     recap_line =  f.readline()
                                 line = f.readline()
                               f.close()
                            out_obj = {"StatusCode": 200,
                                     "StatusMessage": "FINISHED",
                                     "PlaybookName": playbooknametxt,
                                     "Version": "Version",
                                     "Duration": 200,
                                     "Results": out_results}
                            return out_obj
                       else:
                          return {"StatusCode": 500, "StatusMessage": "PLAYBOOK FAILED "}
                           

                   return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"}
            else:
                return {"StatusCode": 500, "StatusMessage": "MALFORMED REQUEST"}
        elif 'DELETE' in cherrypy.request.method:
            input_data = parse_query_string(cherrypy.request.query_string)
            
            cherrypy.log( "***> in RestServer.DELETE")
            cherrypy.log("Payload: " + str(input_data))
            
            if input_data['Id'] in TestRecord:
                if not 'PENDING' in TestRecord[input_data['Id']]['Result']:
                    cherrypy.log(" Path: " + str(TestRecord[input_data['Id']]['Path']))
                    TestRecord.pop (input_data['Id'])
                    if input_data['Id'] in ActiveProcess:
                        ActiveProcess.pop (input_data['Id'])

                    return {"StatusCode": 200, "StatusMessage": "PLAYBOOK EXECUTION RECORDS DELETED"}
                else:
                    return {"StatusCode": 200, "StatusMessage": "PENDING"}
            else:
                return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"}


if __name__ == '__main__':

    # Read configuration

    config_file_path = "RestServer_config"

    if not os.path.exists(config_file_path):
        print '[INFO] The config file does not exist'
        sys.exit(0)

    ip = 'na'
    AnsibleServer = 'na'
    port = 'na'
    tls = False
    auth = False
    pub = 'na'
    priv = 'na'
    timeout_seconds = 'na'
    ansible_path = 'na'
    ansible_temp = 'na'    
    host = 'na'
    users= 'na'
    getresults_block = False
    from_files = False
    
    file = open(config_file_path, 'r')
    for line in file.readlines():
        if '#' not in line:
            if 'ip:' in line:
                ip = line.split(':')[1].strip()
            elif 'AnsibleServer:' in line:
                AnsibleServer = line.split(':')[1].strip()
            elif 'port:' in line:
                port = line.split(':')[1].strip()
            elif 'ksalt:' in line:
                salt = line.split(':')[1].strip()
            elif 'tls:' in line:
                tls = 'YES' in line.split(':')[1].strip().upper()
            elif 'auth:' in line:
                auth = 'YES' in line.split(':')[1].strip().upper()
            if tls and 'priv:' in line:
                priv = line.split(':')[1].strip()
            if tls and 'pub:' in line:
                pub = line.split(':')[1].strip()
            if tls and 'inter_cert:' in line:
                intermediate = line.split(':')[1].strip()
            if 'timeout_seconds' in line:
                timeout_seconds = int (line.split(':')[1].strip())
            if 'ansible_path' in line:
                ansible_path = line.split(':')[1].strip()
            if 'ansible_temp' in line:
                ansible_temp = line.split(':')[1].strip()
            if 'host' in line:
                host = line.split(':')[1].strip()
            if 'users' in line:
                users = line.split(':')[1].strip()
            if 'getresults_block' in line:
                getresults_block = 'YES' in line.split(':')[1].strip().upper()
            if 'from_files' in line:
                from_files = 'YES' in line.split(':')[1].strip().upper()
    file.close()

    # Initialization
    
    global_conf = {
        'global': {
            'log.screen': True,
            'response.timeout': 5400,
            'server.socket_host': ip,
            'server.socket_port': int(port),
            'server.protocol_version': 'HTTP/1.1'
            }
        }

    if tls:
        # Use pythons built-in SSL
        cherrypy.server.ssl_module = 'builtin'

        # Point to certificate files

        if not os.path.exists(pub):
            print '[INFO] The public certificate does not exist'
            sys.exit(0)

        if not os.path.exists(priv):
            print '[INFO] The private key does not exist'
            sys.exit(0)            

        if not os.path.exists(intermediate):
            print '[INFO] The intermediate certificate does not exist'
            sys.exit(0)

        
        cherrypy.server.ssl_certificate = pub
        cherrypy.server.ssl_certificate_chain = intermediate
        cherrypy.server.ssl_private_key = priv

    if auth:
        # Read in and build user dictionary
        if not os.path.exists(users):
          print '[INFO] The users file does not exist: ' + users
          sys.exit(0)
        userpassdict = {}
        user_file = open(users, 'r')
        for line in user_file.readlines():
          if '#' not in line:
            id = line.split(':')[0].strip()
            pw = line.split(':')[1].strip()
            userpassdict[id] = pw
            #print str(userpassdict)

        app_conf = {'/':
                    {'tools.auth_basic.on': True,
                     'tools.auth_basic.realm': 'earth',
                     'tools.auth_basic.checkpassword': validate_password,
                     }
                    }

        application = cherrypy.tree.mount(TestManager(), '/', app_conf)
    else:
        application = cherrypy.tree.mount(TestManager(), '/')
        
    cherrypy.config.update({
        'log.access_file': "/var/log/RestServer.access"
    })
    accessLogName = "/var/log/RestServer.access"
    applicationLogName = "/var/log/RestServer.log"
    cherrypy.config.update(global_conf)

    log = application.log
    log.error_file = ""
    log.access_file = ""
    from logging import handlers
    applicationLogFileHandler = handlers.RotatingFileHandler(applicationLogName, 'a', 1000000, 5000)
    accessLogFileHandler = handlers.RotatingFileHandler(accessLogName, 'a', 1000000, 5000)
    import logging
    applicationLogFileHandler.setLevel(logging.DEBUG)
    log.error_log.addHandler(applicationLogFileHandler)
    log.access_log.addHandler(accessLogFileHandler)

    # Start server
    
    cherrypy.engine.start()
    cherrypy.engine.block()