#
# $Id$
# $URL$
#
# NodeManager plugin - first step of handling omf_controlled slices

"""
Overwrites the 'resctl' tag of slivers controlled by OMF so sm.py does the right thing
"""

import os
import glob
import subprocess

import tools
import logger

priority = 50

def start(options, conf):
    logger.log("omf_resctl: plugin starting up...")

def GetSlivers(data, conf = None, plc = None):
    if 'accounts' not in data: 
        logger.log_missing_data("omf_resctl.GetSlivers",'accounts')
        return

    try:
        xmpp_server=data['xmpp']['server']
    except:
        # disabled feature - bailing out
        # xxx might need to clean up more deeply..
        return

    # as hrn is set only at AddNode-time, upgraded myplcs might still miss this
    # clue: just overwrite the hostname of all nodes
    # for node in GetNodes(): UpdateNode(node['node_id'],{'hostname':node['hostname']})
    try:
        node_hrn = data['hrn']
        if not hrn: raise Exception,"Empty hrn"
    except:
        node_hrn='default   # Failed to read hrn from GetSlivers, please upgrade PLCAPI'

    for sliver in data['slivers']:
        name=sliver['name']
        for chunk in sliver['attributes']:
            if chunk['tagname']=='omf_control':
                # scan all versions of omf-resctl
                etc_path="/vservers/%s/etc/"%name
                pattern = etc_path + "omf-resctl-*/omf-resctl.yaml.in"
                templates = glob.glob (pattern)
                if not templates:
                    logger.log("WARNING: omf_resctl plugin, no template found for slice %s using pattern %s"\
                                   %(name,pattern))
                    continue
                for template in templates:
                    # remove the .in extension
                    yaml=template[:-3]
                    # figure service name as subdir under etc/
                    service_name=os.path.split(template.replace(etc_path,''))[0]
                    # read template and replace
                    template_contents=file(template).read()
                    yaml_contents=template_contents\
                        .replace('@XMPP_SERVER@',xmpp_server)\
                        .replace('@NODE_HRN@',node_hrn)\
                        .replace('@SLICE_NAME@',name)
                    changes=tools.replace_file_with_string(yaml,yaml_contents)
                    if changes:
                        sp=subprocess.Popen(['vserver',name,'exec','service',service_name,'restart'],
                                            stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
                        (output,retcod)=sp.communicate()
                        logger.log("omf_resctl: %s: restarted resource controller (retcod=%r)"%(name,retcod))
                        logger.log("omf_resctl: got output\n%s"%output)
                    else:
                        logger.log("omf_resctl: %s: omf_control'ed sliver has no change" % name)