#!/usr/bin/python3
'''

make sure pexpect is installed:
$ sudo yum install python3-pexpect

You might need to install pexpect-serial using pip:
$ pip3 install pexpect-serial

./tos_plus_auto.py
'''


import sys
if sys.version_info[0] != 3:
    print("This script requires Python 3")
    exit()

import os
import re
import logging
import time
from time import sleep
import pprint
import telnetlib
import argparse
import pexpect
import subprocess

ptype="QCA"
FORMAT = '%(asctime)s %(name)s %(levelname)s: %(message)s'

def usage():
   print("$0 used connect to automated a test case using cisco controller and LANforge tos-plus script:")
   print("-p|--ptype:  AP Hardware type")
   print("-h|--help")
   print("-l|--log file: log messages here")

# see https://stackoverflow.com/a/13306095/11014343
class FileAdapter(object):
    def __init__(self, logger):
        self.logger = logger
    def write(self, data):
        # NOTE: data can be a partial line, multiple lines
        data = data.strip() # ignore leading/trailing whitespace
        if data: # non-blank
           self.logger.info(data)
    def flush(self):
        pass  # leave it to logging to flush properly

def main():
   global ptype


   parser = argparse.ArgumentParser(description="TOS Plus automation script")
   parser.add_argument("-p", "--ptype",    type=str, help="AP Hardware type")
   parser.add_argument("-l", "--log",     type=str, help="logfile for messages, stdout means output to console")
   
   args = None
   try:
      args = parser.parse_args()
      if (args.ptype != None):
         ptype = args.ptype
      logfile = args.log
      
   except Exception as e:
      logging.exception(e);
      usage()
      exit(2);


   console_handler = logging.StreamHandler()
   formatter = logging.Formatter(FORMAT)
   logg = logging.getLogger(__name__)
   logg.setLevel(logging.DEBUG)
   file_handler = None
   if (logfile is not None):
       if (logfile != "stdout"):
           file_handler = logging.FileHandler(logfile, "w")
           file_handler.setLevel(logging.DEBUG)
           file_handler.setFormatter(formatter)
           logg.addHandler(file_handler)
           logging.basicConfig(format=FORMAT, handlers=[file_handler])
       else:
           # stdout logging
           logging.basicConfig(format=FORMAT, handlers=[console_handler]) 

   # Set up cisco controller.  For now, variables are hard-coded.
   dest = '172.19.27.95'
   port = '2013'
   port_ap = '2014'
   ap = 'AxelMain'
   user = 'cisco'
   passwd = 'Cisco123'
   user_ap = 'cisco'
   passwd_ap = 'Cisco123'
   wlan = 'wlan_open'
   wlanID = '6'

   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-w", wlan, "-i", wlanID,
                  "--action", "wlan"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-w", wlan, "-i", wlanID,
                  "--action", "wlan_security"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-w", wlan, "-i", wlanID,
                  "--action", "wlan_qos", "--value", "platinum"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-w", wlan, "-i", wlanID,
                  "--action", "enable_wlan"], capture_output=True)
   output01 = subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "--action", "show", 
                  "--value", "wlan summary"], capture_output=True)
   #pss = output01.stdout.decode('utf-8', 'ignore');
   #print(pss)
   print(output01)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-b", "b", "--action", 
                  "disable"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-b", "a", "--action", 
                  "disable"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-b", "a", "--action", 
                  "channel", "--value", "149"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-b", "a", "--action", 
                  "bandwidth", "--value", "80"], capture_output=True)
   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-b", "a", "--action", 
                  "enable"], capture_output=True)
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -w %s -i %s --action wlan"%(dest, port, ap, user, passwd, wlan, wlanID))
   #subprocess.run("./cisco_wifi_ctl.py", "-d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -w %s -i %s --action wlan_qos --value platinum"%(dest, port, ap, user, passwd, wlan, wlanID))
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s --action show --value \"wlan summary\""%(dest, port, ap, user, passwd))
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -b b --action disable"%(dest, port, ap, user, passwd))
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -b a --action disable"%(dest, port, ap, user, passwd))
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -b a --action channel --value 149"%(dest, port, ap, user, passwd))
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -b a --action bandwidth --value 80"%(dest, port, ap, user, passwd))
   #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -b a --action enable"%(dest, port, ap, user, passwd))

   #Clear Stormbreaker stats on AP
   subprocess.run(["./cisco_ap.py", "-d", dest, "-o", port_ap, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "--action", "clear_counter"], capture_output=True)
   
   # Run the tos plus script to generate traffic and grab capture files.
   # You may edit this command as needed for different behaviour.
   os.system('python3 ./lf_tos_plus_test.py --dur 1 --lfmgr localhost --ssid wlan_open --radio "1.wiphy0 2 0" --txpkts 9999 --wait_sniffer 1  --cx "1.wiphy0 1.wlan0 anAX 1.eth2 udp 1024 10000 5000000000 184" --sniffer_radios "1.wiphy2"')
   #output02 = subprocess.run(["./lf_tos_plus_test.py", "--dur", "1", "--lfmgr", "localhost", "--ssid", wlan, "--radio", "1.wiphy0 2 0", "--txpkts", "10000", 
   #               "--wait_sniffer", "1", "--cx", "1.wiphy0 1.wlan0 anAX 1.eth2 udp 1024 10000 50000000 184", "--sniffer_radios", "1.wiphy2"], capture_output=True)

   #Read Stormbreaker exported stats after the test
   output02 = subprocess.run(["./cisco_ap.py", "-d", dest, "-o", port_ap, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "--action", "show", 
                  "--value", "interfaces dot11Radio 1 traffic distribution periodic data exported"], capture_output=True)
   print(output02)

   file1 = open('TOS_PLUS.sh', 'r') 
   lines = file1.readlines()


   csv_file = ""
   capture_dir = ""
   # Strips the newline character 
   for line in lines:
       tok_val = line.split("=", 1)
       if tok_val[0] == "CAPTURE_DIR":
           capture_dir = tok_val[1]
       elif tok_val[0] == "CSV_FILE":
           capture_dir = tok_val[1]

   # Remove  third-party tool's tmp file tmp file
   os.unlink("stormbreaker.log")

   # Run third-party tool to process the capture files.
   os.system('python3 sb -p %s -subdir %s'%(ptype, capture_dir))

  
   # Print out one-way latency reported by LANforge
   file2 = open(csv_file, 'r')
   lines = file2.readlines()

   # Strips the newline character 
   for line in lines:
       cols = line.split("\t")
       # Print out endp-name and avg latency
       print("%s\t%s"%(cols[1], cols[15]))

   subprocess.run(["./cisco_wifi_ctl.py", "-d", dest, "-o", port, "-s", "telnet", "-l", "stdout", "-a", ap, "-u", "cisco", "-p", "Cisco123", "-w", wlan, "-i", wlanID,
                  "--action", "delete_wlan"], capture_output=True)
     #subprocess.run("python3 cisco_wifi_ctl.py -d %s  -o %s -s telnet -l stdout -a %s -u %s -p %s -w wlan_open -i 6 --action delete_wlan"%(dest, port, ap, user, passwd))

# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
if __name__ == '__main__':
    main()

####
####
####