#!/usr/bin/env python
# Copyright (C) 2007 Platform Computing Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of version 2 of the GNU General Public License as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty 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

import subprocess
from kusu.core import rcplugin

class KusuRC(rcplugin.Plugin):
    def __init__(self):
        rcplugin.Plugin.__init__(self)
        self.name = 'osmsetup'
        self.desc = 'generating OpenSM conf and restarting daemon'
        self.ngtypes = ['installer' , 'compute']
        self.delete = True 

    def run(self):
        p = subprocess.Popen("chkconfig --level 35 opensmd on" ,
                             shell=True,
                             stdout=subprocess.PIPE,
                             stderr=subprocess.PIPE)
        out, err = p.communicate()
        if err:
            raise Exception, "Couldn't run chkconfig"

        p = subprocess.Popen("/usr/sbin/opensm -c /etc/opensm/opensm.conf" ,
                             shell=True,
                             stdout=subprocess.PIPE,
                             stderr=subprocess.PIPE)
        out, err = p.communicate()
        if err:
            raise Exception, "Couldn't generate the configuration file (/etc/opensm/opensm.conf)"

        p = subprocess.Popen("service opensmd restart" ,
                             shell=True,
                             stdout=subprocess.PIPE,
                             stderr=subprocess.PIPE)
        out, err = p.communicate()

        if err:
            raise Exception, "Couldn't restart the opensmd service"
        return True