import os
import rtconfig
import subprocess
from building import *

group   = []
cwd     = GetCurrentDir()
CPPPATH = [cwd, cwd + "/kernel"]
list    = os.listdir(cwd)
src     = Glob('kernel/*.c') + Glob('kernel/*.S')

if not GetDepend(['RT_USING_VDSO']):
    Return('group')

if rtconfig.ARCH != "aarch64" and rtconfig.ARCH != "risc-v":
    # not supported arch
    src = []
else:
    if not hasattr(rtconfig, 'CPP') or rtconfig.CPP is None:
        rtconfig.CPP = rtconfig.PREFIX + 'cpp'
    if not hasattr(rtconfig, 'CPPFLAGS') or rtconfig.CPPFLAGS is None:
        rtconfig.CPPFLAGS = ' -E -P -x assembler-with-cpp'

    if not os.path.exists(cwd + "/user" + "/arch" +"/" + rtconfig.ARCH + "/vdso.lds"):
        Preprocessing("user/arch/" + rtconfig.ARCH + "/vdso.lds.S", ".lds", CPPPATH=[cwd])

    vdso_arch = os.path.join(cwd, 'user',"arch", rtconfig.ARCH)

    process_env = os.environ.copy()
    if hasattr(rtconfig, 'EXEC_PATH') and rtconfig.EXEC_PATH is not None:
        process_env['RTT_EXEC_PATH'] = rtconfig.EXEC_PATH
    if hasattr(rtconfig, 'PREFIX') and rtconfig.PREFIX is not None:
        process_env['RTT_CC_PREFIX'] = rtconfig.PREFIX
    if hasattr(rtconfig, 'DEVICE') and rtconfig.DEVICE is not None:
        process_env['RTT_DEVICE'] = rtconfig.DEVICE

    command = ["scons", "-C", vdso_arch]
    clean_command = ["scons", "-C", vdso_arch, "--clean"] 

    if not GetOption('clean'):    
        result = subprocess.run(command, env=process_env, check=True)
    else:
        result = subprocess.run(clean_command, env=process_env, check=True)

    if result.returncode == 0:
        print("Command executed successfully")
    else:
        print("Command failed with exit code:", result.returncode)
        exit(1)

group = DefineGroup('vDSO', src, depend = ['RT_USING_SMART','RT_USING_VDSO'], CPPPATH = CPPPATH)
Return('group')