#!/usr/bin/env python3

import sys, os
import mod_globals
import mod_db_manager
import mod_utils
from mod_optfile   import *
from mod_scan_ecus import families as families

os.chdir(os.path.dirname(os.path.realpath(sys.argv[0])))

import pickle
from   mod_acf_func     import ACE

mod_globals.os = os.name
 
if mod_globals.os == 'nt':
  import pip
  
  try:
    import serial
  except ImportError:
    pip.main(['install','pyserial'])

  try:
    import colorama
  except ImportError:
    pip.main(['install','colorama'])
    try:
      import colorama
    except ImportError:
      print("\n\n\n\t\t\tGive me access to the Internet for download modules\n\n\n")
      sys.exit()
  colorama.init()
else:
  # let's try android
  try:
    import androidhelper as android
    mod_globals.os = 'android'
  except:
    try:
      import android
      mod_globals.os = 'android'
    except:
      pass
  
if mod_globals.os != 'android':    
  try:
    import serial
    from serial.tools  import list_ports
    #import ply
  except ImportError:
    print("\n\n\n\tPleas install additional modules")
    print("\t\t>sudo easy_install pyserial")
    sys.exit()
    
from mod_elm       import ELM
from mod_scan_ecus import ScanEcus
from mod_utils     import *
from mod_mtc       import acf_getMTC
from mod_acf_func  import ACE
from mod_acf_func  import acf_loadModules
from mod_acf_proc  import acf_MTC_generateDefaults
from mod_acf_proc  import acf_MTC_findDiff
from mod_acf_proc  import acf_MTC_optionsExplorer


def optParser():
  '''Parsing of command line parameters. User should define at least com port name'''
  
  import argparse

  parser = argparse.ArgumentParser(
    #usage = "%prog -p <port> [options]",
    version="acfg Version 1.0",
    description = "acfg - auto configuration tool"
  )
  
  parser.add_argument('-p',
      help="ELM327 com port name",
      dest="port",
      default="")

  parser.add_argument("-s",
      help="com port speed configured on ELM {38400[default],57600,115200,230400,500000} DEPRECATED",
      dest="speed",
      default="38400")

  parser.add_argument("-r",
      help="com port rate during diagnostic session {38400[default],57600,115200,230400,500000}",
      dest="rate",
      default="38400",)

  parser.add_argument("--si",
      help="try SlowInit first",
      dest="si",
      default=False,
      action="store_true")

  parser.add_argument("-L",
      help="language option {RU[default],GB,FR,IT,...}",
      dest="lang",
      default="RU")

  parser.add_argument("--cfc",
      help="turn off automatic FC and do it by script",
      dest="cfc",
      default=False,
      action="store_true")

  parser.add_argument("--n1c",
      help="turn off L1 cache",
      dest="n1c",
      default=False,
      action="store_true")

  parser.add_argument("--log",
      help="log file name",
      dest="logfile",
      default="")

  parser.add_argument("--scan",
      help="scan ECUs even if savedEcus.p file exists",
      dest="scan",
      default=False,
      action="store_true")

  parser.add_argument("--demo",
      help="for debuging purpose. Work without car and ELM",
      dest="demo",
      default=False,
      action="store_true")

  parser.add_argument("--dev",
      help="swith to Development Session for commands from DevList, you should define alternative command for opening the session, like a 1086",
      dest="dev",
      default='')

  parser.add_argument("--can2",
      help="CAN network connected to pin 13 and pin 12",
      dest="can2",
      default=False,
      action="store_true")

  parser.add_argument("--vin",
      help="vin number",
      dest="vinnum",
      default="")

  parser.add_argument("-vv", "--verbose",
      help="show parameter explanations",
      dest="verbose",
      default=False,
      action="store_true")

  parser.add_argument("-vvv",
      help="show parameter explanations",
      dest="verbose2",
      default=False,
      action="store_true")



  options = parser.parse_args()
  
  if not options.port and mod_globals.os != 'android':
    parser.print_help()
    iterator = sorted(list(list_ports.comports()))
    print("")
    print("Available COM ports:")
    for port, desc, hwid in iterator:
      print("%-30s \n\tdesc: %s \n\thwid: %s" % (port,desc,hwid))
    print("")
    exit(2)
  else:
    mod_globals.opt_port      = options.port
    mod_globals.opt_speed     = int(options.speed)
    mod_globals.opt_rate      = int(options.rate)
    mod_globals.opt_lang      = options.lang
    mod_globals.opt_log       = options.logfile
    mod_globals.opt_demo      = options.demo
    mod_globals.opt_scan      = options.scan
    mod_globals.opt_si        = options.si
    mod_globals.opt_cfc0      = options.cfc
    mod_globals.opt_n1c       = options.n1c
    mod_globals.opt_can2      = options.can2
    mod_globals.vin           = options.vinnum
    mod_globals.opt_verbose   = options.verbose
    mod_globals.opt_verbose2  = options.verbose2

    if options.dev=='' or len(options.dev)!=4 or options.dev[0:2]!='10':
      mod_globals.opt_dev       = False
      mod_globals.opt_devses    = '1086'   
    else:
      print("Development MODE")
      mod_globals.opt_dev       = True
      mod_globals.opt_devses    = options.dev   
    
def main():
  '''Main function'''

  optParser()

  mod_utils.chkDirTree()
  mod_db_manager.find_DBs()

  '''Check direcories'''
  if  not os.path.exists('../BVMEXTRACTION'):
    print("Can't find MTC database. (../BVMEXTRACTION)")
    exit()
    
  print('Opening ELM')
  elm = ELM( mod_globals.opt_port, mod_globals.opt_speed, mod_globals.opt_log )

  #change serial port baud rate 
  if mod_globals.opt_speed<mod_globals.opt_rate and not mod_globals.opt_demo:
    elm.port.soft_boudrate( mod_globals.opt_rate )



  print("Loading language ")
  sys.stdout.flush()
                                         #loading language data
  lang = optfile("Location/DiagOnCAN_"+mod_globals.opt_lang+".bqm",True)
  mod_globals.language_dict = lang.dict
  print("Done")
  
  print('Loading ECUs list')
  se  = ScanEcus(elm)                    #Prepare list of all ecus
 
  SEFname = "savedEcus.p" 
  if mod_globals.opt_can2:
    SEFname = "savedEcus2.p" 

  if mod_globals.opt_demo and len(mod_globals.opt_ecuid)>0:
    # demo mode with predefined ecu list
    se.read_Uces_file( all = True )    
    se.detectedEcus = []
    for i in mod_globals.opt_ecuid.split(','):
      if  i in list(se.allecus.keys()):
        se.allecus[i]['ecuname']=i
        se.allecus[i]['idf']=se.allecus[i]['ModelId'][2:4]
        if se.allecus[i]['idf'][0]=='0': 
          se.allecus[i]['idf'] = se.allecus[i]['idf'][1]
        se.allecus[i]['pin'] = 'can' 
        se.detectedEcus.append( se.allecus[i] )    
  else:
    if not os.path.isfile(SEFname) or mod_globals.opt_scan: 
      # choosing model 
      se.chooseModel( mod_globals.opt_car )  #choose model of car for doing full scan
    
    # Do this check every time
    se.scanAllEcus()                       #First scan of all ecus
 
  de = se.detectedEcus

  if mod_globals.vin=='':
    print('Reading VINs')
    VIN = getVIN( de, elm )
    mod_globals.vin = VIN

  VIN = mod_globals.vin
  
  if len(VIN)!=17:
    print("ERROR!!! Can't find any VIN. Check connection")
    exit()
  else:
    print("\tVIN     :",VIN)
    
  #print 'Finding MTC'
  vindata, mtcdata, refdata, platform = acf_getMTC( VIN )
  
  #print vindata
  
  if vindata=='' or mtcdata=='' or refdata=='':
    print("ERROR!!! Can't find MTC data in database")
    exit()
  
  print("\tPlatform:",platform)
  print("\tvindata:",vindata)
  print("\tmtcdata:",mtcdata)
  print("\trefdata:",refdata)
  
  mtc = mtcdata.replace(' ','').replace('\n','').replace('\r','').replace('\t','').split(';')
  
  print('Loading Modules')
  module_list = acf_loadModules( de, refdata, platform )
  
  print()
  
  for m in module_list:
    if 'sref' not in list(m.keys()) or m['sref']=='': continue
    if families[m['idf']] in list(mod_globals.language_dict.keys()):
      m['fam_txt']=mod_globals.language_dict[families[m['idf']]]
    else:
      m['fam_txt']=m['idf']
    if 'sref' in list(m.keys()):
      print("\n#########   Family : ",m['idf'], ' : ', m['fam_txt'])
      if 'mo' in list(m.keys()) and m['mo']!='':
        print("%2s : %s : %s" % (m['idf'],m['sref'],m['mo'].NOM))
    
        acf_MTC_generateDefaults( m, mtc )
        #acf_MTC_findDiff( m, mtc, elm )
    
      else:
        print("%2s : %s :   " % (m['idf'],m['sref']))      
  
  #for option in mtc:
  #  acf_MTC_optionsExplorer( de, option )
  

if __name__ == '__main__':  
  main()