"""
Description:
    This file is used to setup the running environment
    Include Download code,setup environment variable
            Set onos running config
            Set user name/password
            Onos-push-keys and so on
    lanqinglong@huawei.com

#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
#
"""

import os
import time
import pexpect
import re
import sys
import pxssh
from connection import connection

class environment( connection ):

    def __init__( self ):
        connection.__init__( self )
        self.loginfo = connection( )
        self.masterhandle = ''
        self.home = ''

    def DownLoadCode( self, handle, codeurl ):
        """
        Download Code use 'git clone'
        parameters:
        handle:  current working handle
        codeurl: clone code url
        """
        print "Now loading test codes! Please wait in patient..."
        originalfolder = sys.path[0]
        print originalfolder
        gitclone = handle
        gitclone.sendline( "git clone " + codeurl )
        index = 0
        increment = 0
        while index != 1 or index != 4:
            index = gitclone.expect ( ['already exists', 'esolving deltas: 100%', \
                                    'eceiving objects', 'Already up-to-date', \
                                    'npacking objects: 100%', pexpect.EOF] )

            filefolder = self.home + '/' + codeurl.split('/')[-1].split('.')[0]
            if index == 0 :
                os.chdir( filefolder )
                os.system( 'git pull' )
                os.chdir( originalfolder )
                self.loginfo.log( 'Download code success!' )
                break
            elif index == 1 or index == 4:
                self.loginfo.log( 'Download code success!' )
                gitclone.sendline( "mkdir onos" )
                gitclone.prompt( )
                gitclone.sendline( "cp -rf " + filefolder+ "/tools onos/" )
                gitclone.prompt( )
                break
            elif index == 2 :
                os.write(1, gitclone.before)
                sys.stdout.flush()
            else :
                self.loginfo.log( 'Download code failed!' )
                self.loginfo.log( 'Information before' + gitclone.before )
                break
        gitclone.prompt( )

    def InstallDefaultSoftware( self, handle ):
        """
        Install default software
        parameters:
        handle(input): current working handle
        """
        print "Now Cleaning test environment"
        handle.sendline("sudo apt-get install -y mininet")
        handle.prompt( )
        handle.sendline("sudo pip install configobj")
        handle.prompt( )
        handle.sendline("sudo apt-get install -y sshpass")
        handle.prompt( )
        handle.sendline("OnosSystemTest/TestON/bin/cleanup.sh")
        handle.prompt( )
        time.sleep(5)
        self.loginfo.log( 'Clean environment success!' )

    def OnosPushKeys(self, handle, cmd, password):
        """
        Using onos-push-keys to make ssh device without password
        parameters:
        handle(input): working handle
        cmd(input): onos-push-keys xxx(xxx is device)
        password(input): login in password
        """
        print "Now Pushing Onos Keys:"+cmd
        Pushkeys = handle
        Pushkeys.sendline( cmd )
        Result = 0
        while Result != 2:
            Result = Pushkeys.expect( ["(yes/no)", "assword:", "PEXPECT]#", \
                                       pexpect.EOF, pexpect.TIMEOUT])
            if ( Result == 0 ):
                Pushkeys.sendline( "yes" )
            if ( Result == 1 ):
                Pushkeys.sendline( password )
            if ( Result == 2 ):
                self.loginfo.log( "ONOS Push keys Success!" )
                break
            if ( Result == 3 ):
                self.loginfo.log( "ONOS Push keys Error!" )
                break
            time.sleep(2)
        Pushkeys.prompt( )
        print "Done!"

    def SetOnosEnvVar( self, handle, masterpass, agentpass):
        """
        Setup onos pushkeys to all devices(3+2)
        parameters:
        handle(input): current working handle
        masterpass: scripts running server's password
        agentpass: onos cluster&compute node password
        """
        print "Now Setting test environment"
        for host in self.hosts:
            print "try to connect " + str(host)
            result = self.CheckSshNoPasswd(host)
            if not result:
                print "ssh lgin failed,try to copy master publickey to agent " + str(host)
                self.CopyPublicKey(host)
        self.OnosPushKeys( handle, "onos-push-keys " + self.OCT, masterpass)
        self.OnosPushKeys( handle, "onos-push-keys " + self.OC1, agentpass)
        self.OnosPushKeys( handle, "onos-push-keys " + self.OC2, agentpass)
        self.OnosPushKeys( handle, "onos-push-keys " + self.OC3, agentpass)
        self.OnosPushKeys( handle, "onos-push-keys " + self.OCN, agentpass)
        self.OnosPushKeys( handle, "onos-push-keys " + self.OCN2, agentpass)

    def CheckSshNoPasswd( self, host):
        """
        Check master can connect agent with no password
        """
        login = pexpect.spawn( "ssh " + str(host))
        index = 4
        while index == 4:
            index = login.expect(['(yes/no)','>|#|\$', \
                                   pexpect.EOF, pexpect.TIMEOUT] )
            if index == 0:
                login.sendline( "yes" )
                index = 4
            if index == 1:
                self.loginfo.log("ssh connect to " + str(host) + " success,no need to copy ssh public key" )
                return True
        login.interact()
        return False

    def ChangeOnosName( self, user, password):
        """
        Change onos name in envDefault file
        Because some command depend on this
        parameters:
        user: onos&compute node user
        password: onos&compute node password
        """
        print "Now Changing ONOS name&password"
        filepath = self.home + '/onos/tools/build/envDefaults'
        line = open(filepath, 'r').readlines()
        lenall = len(line) - 1
        for i in range(lenall):
           if "ONOS_USER=" in line[i]:
               line[i]=line[i].replace("sdn",user)
           if "ONOS_GROUP" in line[i]:
               line[i]=line[i].replace("sdn",user)
           if "ONOS_PWD" in line[i]:
               line[i]=line[i].replace("rocks",password)
        NewFile = open(filepath, 'w')
        NewFile.writelines(line)
        NewFile.close
        print "Done!"

    def ChangeTestCasePara(self, testcase, user, password):
        """
        When running test script, there's something need \
        to change in every test folder's *.param & *.topo files
        user: onos&compute node user
        password: onos&compute node password
        """
        print "Now Changing " + testcase +  " name&password"
        if self.masterusername == 'root':
            filepath = '/root/'
        else :
            filepath = '/home/' + self.masterusername + '/'
        filepath = filepath +"OnosSystemTest/TestON/tests/" + testcase + "/" + \
                   testcase + ".topo"
        line = open(filepath,'r').readlines()
        lenall = len(line)-1
        for i in range(lenall-2):
           if ("localhost" in line[i]) or ("OCT" in line[i]):
               line[i+1]=re.sub(">\w+",">"+user,line[i+1])
               line[i+2]=re.sub(">\w+",">"+password,line[i+2])
           if "OC1" in line [i] \
              or "OC2" in line [i] \
              or "OC3" in line [i] \
              or "OCN" in line [i] \
              or "OCN2" in line[i]:
               line[i+1]=re.sub(">\w+",">root",line[i+1])
               line[i+2]=re.sub(">\w+",">root",line[i+2])
        NewFile = open(filepath,'w')
        NewFile.writelines(line)
        NewFile.close

    def SSHlogin ( self, ipaddr, username, password ) :
        """
        SSH login provide a connection to destination.
        parameters:
        ipaddr:   ip address
        username: login user name
        password: login password
        return: handle
        """
        login = pxssh.pxssh( )
        login.login ( ipaddr, username, password, original_prompt='[$#>]')
        #send command ls -l
        login.sendline ('ls -l')
        #match prompt
        login.prompt()
        print ("SSH login " + ipaddr + " success!")
        return login

    def SSHRelease( self, handle ):
        #Release ssh
        handle.logout()

    def CopyOnostoTestbin( self ):
        sourcefile = self.cipath + '/dependencies/onos'
        destifile = self.home + '/onos/tools/test/bin/'
        os.system( 'pwd' )
        runcommand = 'cp ' + sourcefile + ' ' + destifile
        os.system( runcommand )

    def CopyPublicKey( self, host ):
        output = os.popen( 'cat /root/.ssh/id_rsa.pub' )
        publickey = output.read().strip('\n')
        tmphandle = self.SSHlogin( self.installer_master, self.installer_master_username, self.installer_master_password )
        tmphandle.sendline("ssh "+ host + " -T \'echo " + str(publickey) + ">>/root/.ssh/authorized_keys\'" )
        tmphandle.prompt()
        self.SSHRelease(tmphandle)
        print "Add OCT PublicKey to " + host + " success"

    def OnosEnvSetup( self, handle ):
        """
        Onos Environment Setup function
        """
        self.Gensshkey( handle )
        self.home = self.GetEnvValue( handle, 'HOME' )
        self.AddKnownHost( handle, self.OC1, "karaf", "karaf" )
        self.AddKnownHost( handle, self.OC2, "karaf", "karaf" )
        self.AddKnownHost( handle, self.OC3, "karaf", "karaf" )
        self.DownLoadCode( handle, 'https://github.com/sunyulin/OnosSystemTest.git' )
        #self.DownLoadCode( handle, 'https://gerrit.onosproject.org/onos' )
        if self.masterusername == 'root':
            filepath = '/root/'
        else :
            filepath = '/home/' + self.masterusername + '/'
        self.OnosRootPathChange( filepath )
        self.CopyOnostoTestbin()
        self.ChangeOnosName(self.agentusername,self.agentpassword)
        self.InstallDefaultSoftware( handle )
        self.SetOnosEnvVar(handle, self.masterpassword,self.agentpassword)