import os
import sys
import subprocess
from oneoslib.build_tools import *
Import('OS_ROOT')

objs = []
cwd = PresentDir()
list = os.listdir(cwd)

for item in list:
 if os.path.isfile(os.path.join(cwd, item, 'SConscript')):
 objs = objs + SConscript(os.path.join(item, 'SConscript'))

global qstr_srcs
qstr_srcs=[]
def gen_srcs():
 global qstr_srcs
 for item in objs:
 filename=str(item)
 dlen = len(cwd)
 if filename[:dlen] == cwd:
 if filename[dlen:] != "\\kernel\\lib\\utils\\pyexec.c":
 qstr_srcs.append(str(item))
 qstr_srcs.append(cwd+"/kernel/py/mpconfig.h")
 qstr_srcs.append(cwd+"/user/start/include/mpconfigport.h")

def create_dir_not_exist(path):
 print(path)
 if not os.path.exists(path):
 os.mkdir(path)

def gen_qstr(env, target, source):
 ucwd = "../../components/MicroPython/"
 qstr_files=""
 dlen = len(cwd)
 for item in objs:
 filename=str(item)
 if filename[:dlen] == cwd:
 qstr_files+=ucwd+filename[dlen+1:]+" "
 qstr_files+=ucwd+"kernel/py/mpconfig.h"+" " 
 qstr_files+=ucwd+"user/start/include/mpconfigport.h"

 qenv_path = Env.get('CPPPATH', [''])
 # special cflags
 qstr_flags=' -E -DNO_QSTR'
 qenv_lpath=" "
 vcwd = "../../"
 dlen = len(OS_ROOT)
 for j in qenv_path:
 filename=str(j)
 qenv_lpath+="-I"+vcwd+filename[dlen+1:]+" "
 #need cpppaths after micropython
 qenv_lpath+="-I../../libc/include"+" "
 qenv_lpath+="-I../../libc/include/newlib"+" "

 gendir = cwd+"/user/start/include/genhdr"
 create_dir_not_exist(gendir)

 qstr_cmd="python "+cwd+"/tools/makeversionhdr.py "+cwd+"/user/start/include/genhdr/mpversion.h"
 print("GEN genhdr/mpversion.h")
 subprocess.run(qstr_cmd)

 qstr_cmd="arm-none-eabi-gcc " + qstr_flags + qenv_lpath + qstr_files
 outfile1=cwd+"/user/start/include/genhdr/qstr.i.last"
 outfile2=cwd+"/user/start/include/genhdr/qstr.i.last.err"
 f1=open(outfile1, 'w')
 f2=open(outfile2, 'w')
 print("GEN genhdr/qstr.i.last")
 p=subprocess.Popen(qstr_cmd,stdout=f1,stderr=f2)
 p.wait()
 f1.close()
 f2.close()

 qstr_cmd="python "+cwd+"/tools/makeqstrdefs.py split "+cwd+"/user/start/include/genhdr/qstr.i.last "+cwd+"/user/start/include/genhdr/qstr "+cwd+"/user/start/include/genhdr/qstrdefs.collected.h"
 print("GEN genhdr/qstr.split")
 mqdef = subprocess.run(qstr_cmd)
 if mqdef.returncode != 0:
 sys.exit(1)

 qstr_cmd="touch "+cwd+"/user/start/include/genhdr/qstr.split"
 subprocess.run(qstr_cmd)

 qstr_cmd="python "+cwd+"/tools/makeqstrdefs.py cat "+cwd+"/user/start/include/genhdr/qstr.i.last "+cwd+"/user/start/include/genhdr/qstr "+cwd+"/user/start/include/genhdr/qstrdefs.collected.h"
 print("GEN genhdr/qstrdefs.collected.h")
 subprocess.run(qstr_cmd)


 qstr_cmd=r"cat "+cwd+"/kernel/py/qstrdefs.h "+cwd+r"/user/start/include/genhdr/qstrdefs.collected.h | sed 's/^Q(.*)/\"&\"/' | arm-none-eabi-gcc " + qstr_flags + qenv_lpath + r" - | sed 's/^\"\(Q(.*)\)\"/1円/' > " +cwd+"/user/start/include/genhdr/qstrdefs.preprocessed.h"
 print("GEN genhdr/qstrdefs.preprocessed.h")
 # pipe need os.system, can't use subprocess
 os.system(qstr_cmd)

 qstr_cmd="python "+cwd+"/tools/makeqstrdata.py "+cwd+"/user/start/include/genhdr/qstrdefs.preprocessed.h > " +cwd+"/user/start/include/genhdr/qstrdefs.generated.h"
 print("GEN genhdr/qstrdefs.generated.h")
 # pipe need os.system, can't use subprocess
 os.system(qstr_cmd)

gen_srcs()
qstr_target=cwd+"/user/start/include/genhdr/qstr.i.last"
Export('qstr_target', 'qstr_srcs', 'gen_qstr')

Return('objs')