import os
import sys
import shutil
import importlib
import importlib.util
import time
import re

from print_helper import print_debug, print_info, print_warn, print_error


def find_soc(src, soc):
 soc_dir = [os.path.basename(src)]
 #soc_dir = ['hk32-common', 'replace', 'HK32F030', 'replace', 'TEMP_HK32F030CBT6']
 
 names = os.listdir(src)
 
 for name in names:
 if name == soc:
 soc_dir.append(name)
 #tools.print_debug(soc_dir)
 return soc_dir

 srcname = os.path.join(src, name)

 if not os.path.isdir(srcname):
 continue
 
 sub_soc_dir = find_soc(srcname, soc)
 if len(sub_soc_dir) > 1:
 soc_dir.extend(sub_soc_dir)
 #tools.print_debug(soc_dir)
 return soc_dir
 
 #tools.print_debug(soc_dir)
 return soc_dir

def copytree(src, dst, ignore, symlinks=False):
 names = os.listdir(src)
 if not os.path.isdir(dst):
 os.makedirs(dst)
 
 errors = []
 for name in names:
 if isinstance(ignore, list):
 if name in ignore:
 continue
 
 elif isinstance(ignore, str):
 if name == ignore:
 continue
 
 srcname = os.path.join(src, name)
 dstname = os.path.join(dst, name)
 
 try:
 if symlinks and os.path.islink(srcname):
 linkto = os.readlink(srcname)
 os.symlink(linkto, dstname)
 elif os.path.isdir(srcname):
 copytree(srcname, dstname, ignore, symlinks)
 else:
 if os.path.isdir(dstname):
 os.rmdir(dstname)
 elif os.path.isfile(dstname):
 os.remove(dstname)
 shutil.copy2(srcname, dstname)
 except (IOError, os.error) as why:
 errors.append((srcname, dstname, str(why)))
 except OSError as err:
 errors.extend(err.args[0])
 
def template(source_path, g):
 target_path = os.path.join(source_path, 'target')

 for item in g:
 if 'TEMP_' in item:
 soc = item

 print_info(source_path)
 print_info(target_path)
 print_info(soc)
 
 if os.path.exists(target_path):
 shutil.rmtree(target_path)
 
 # find soc dir
 soc_dir = find_soc(source_path, soc)
 print_info("soc_dir")
 print(soc_dir)
 #soc_dir = ['hk32-common', 'replace', 'HK32F030', 'replace', 'TEMP_HK32F030CBT6']
 
 if len(soc_dir) < 2:
 print_warn("invalid soc_dir %s" % soc_dir)
 return None
 
 # copy replace files
 replace_path = source_path
 copytree(replace_path, target_path, ['replace', 'generate.py'])
 
 for i in range(1, len(soc_dir)):
 if i % 2 == 1 and soc_dir[i] != 'replace':
 print_warn("invalid soc_dir %s\n"
 "soc_dir[%d] is %s, expect 'replace'"
 % (soc_dir, i, soc_dir[i]))
 return None
 
 replace_path = os.path.join(replace_path, soc_dir[i])
 
 if soc_dir[i] == 'replace':
 continue
 
 copytree(replace_path, target_path, ['replace', 'generate.py'])
 
 return target_path