# ScaleOutTemplate
#
# CASE1 starts number of nodes specified in param file
#
# cameron@onlab.us

import sys 
import os 


class ScaleOutTemplate:
    def __init__(self):
        self.default = ''
   
    def CASE1(self, main):
        
        global cluster_count 
        cluster_count = 1        

        checkout_branch = main.params['GIT']['checkout']
        git_pull = main.params['GIT']['autopull']
        cell_name = main.params['ENV']['cellName']
        BENCH_ip = main.params['BENCH']['ip1']
        BENCH_user = main.params['BENCH']['user']
        ONOS1_ip = main.params['CTRL']['ip1']
        ONOS2_ip = main.params['CTRL']['ip2']
        ONOS3_ip = main.params['CTRL']['ip3']
        MN1_ip = main.params['MN']['ip1']

        main.log.step("Cleaning Enviornment...")
        main.ONOSbench.onos_uninstall(ONOS1_ip)
        main.ONOSbench.onos_uninstall(ONOS2_ip)
        main.ONOSbench.onos_uninstall(ONOS3_ip)                                     
        
        main.step("Git checkout and pull "+checkout_branch)
        if git_pull == 'on':
            checkout_result = main.ONOSbench.git_checkout(checkout_branch)       
            pull_result = main.ONOSbench.git_pull()
            
        else:
            checkout_result = main.TRUE
            pull_result = main.TRUE
            main.log.info("Skipped git checkout and pull")

        mvn_result = main.ONOSbench.clean_install()
                                                                   
        main.step("Set cell for ONOS cli env")
        main.ONOS1cli.set_cell(cell_name)
        main.ONOS2cli.set_cell(cell_name)
        main.ONOS3cli.set_cell(cell_name)

        main.step("Creating ONOS package")
        package_result = main.ONOSbench.onos_package()                             #no file or directory 

        main.step("Installing ONOS package")
        install1_result = main.ONOSbench.onos_install(node=ONOS1_ip)

        cell_name = main.params['ENV']['cellName']
        main.step("Applying cell file to environment")
        cell_apply_result = main.ONOSbench.set_cell(cell_name)
        main.step("verify cells")
        verify_cell_result = main.ONOSbench.verify_cell()

        main.step("Set cell for ONOS cli env")
        main.ONOS1cli.set_cell(cell_name) 
        cli1 = main.ONOS1cli.start_onos_cli(ONOS1_ip)  
       

    def CASE2(self, main):

        '''  
        Increase number of nodes and initiate CLI
        '''
        import time 
        
        global cluster_count
        
        ONOS1_ip = main.params['CTRL']['ip1']
        ONOS2_ip = main.params['CTRL']['ip2']
        ONOS3_ip = main.params['CTRL']['ip3']
        #ONOS4_ip = main.params['CTRL']['ip4']
        #ONOS5_ip = main.params['CTRL']['ip5']
        #ONOS6_ip = main.params['CTRL']['ip6']
        #ONOS7_ip = main.params['CTRL']['ip7']
        cell_name = main.params['ENV']['cellName']
        scale = int(main.params['SCALE'])
       
        #Cluster size increased everytime the case is defined
        cluster_count += scale
 
        main.log.report("Increasing cluster size to "+
                str(cluster_count))
        install_result = main.FALSE
        
        if scale == 2:
            if cluster_count == 3:
                main.log.info("Installing nodes 2 and 3")
                install2_result = main.ONOSbench.onos_install(node=ONOS2_ip)
                install3_result = main.ONOSbench.onos_install(node=ONOS3_ip)
                cli2 = main.ONOS1cli.start_onos_cli(ONOS2_ip)
                cli3 = main.ONOS1cli.start_onos_cli(ONOS3_ip)

            '''
            elif cluster_count == 5:

                main.log.info("Installing nodes 4 and 5")
                node4_result = main.ONOSbench.onos_install(node=ONOS4_ip)
                node5_result = main.ONOSbench.onos_install(node=ONOS5_ip)
                install_result = node4_result and node5_result
                time.sleep(5)

                main.ONOS4cli.start_onos_cli(ONOS4_ip)
                main.ONOS5cli.start_onos_cli(ONOS5_ip)

            elif cluster_count == 7:

                main.log.info("Installing nodes 4 and 5")
                node6_result = main.ONOSbench.onos_install(node=ONOS6_ip)
                node7_result = main.ONOSbench.onos_install(node=ONOS7_ip)
                install_result = node6_result and node7_result
                time.sleep(5)

                main.ONOS6cli.start_onos_cli(ONOS6_ip)
                main.ONOS7cli.start_onos_cli(ONOS7_ip)
            '''
        if scale == 1: 
            if cluster_count == 2:
                main.log.info("Installing node 2")
                install2_result = main.ONOSbench.onos_install(node=ONOS2_ip)
                cli2 = main.ONOS1cli.start_onos_cli(ONOS2_ip)

            if cluster_count == 3:
                main.log.info("Installing node 3")
                install3_result = main.ONOSbench.onos_install(node=ONOS3_ip)
                cli3 = main.ONOS1cli.start_onos_cli(ONOS3_ip)