Import("env")
import shutil

def get_build_flag_value(flag_name):
    build_flags = env.ParseFlags(env['BUILD_FLAGS'])
    flags_with_value_list = [build_flag for build_flag in build_flags.get('CPPDEFINES') if type(build_flag) == list]
    defines = {k: v for (k, v) in flags_with_value_list}
    return defines.get(flag_name)

def after_build(source, target, env):
    print( "Executing custom step " )
    dir    = env.GetLaunchDir()
    name   = env.get( "PIOENV" )
    source = dir + "\\.pio\\build\\" + name + "\\firmware.bin"
    if name == "gravity-debug" :
        target = dir + "\\bin\\firmware-debug.bin"
    if name == "gravity-release" :
        target = dir + "\\bin\\firmware.bin"
    if name == "gravity-perf" :
        target = dir + "\\bin\\firmware-perf.bin"
    print( "Copy file : " + source + " -> " + target )
    shutil.copyfile( source, target )


print( "Adding custom build step (copy firmware): ")
env.AddPostAction("buildprog", after_build)