#!/usr/bin/env python
|
|
import time
|
|
Import("env")
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Utils
|
|
# ------------------------------------------------------------------------------
|
|
|
|
class Color(object):
|
|
BLACK = '\x1b[1;30m'
|
|
RED = '\x1b[1;31m'
|
|
GREEN = '\x1b[1;32m'
|
|
YELLOW = '\x1b[1;33m'
|
|
BLUE = '\x1b[1;34m'
|
|
MAGENTA = '\x1b[1;35m'
|
|
CYAN = '\x1b[1;36m'
|
|
WHITE = '\x1b[1;37m'
|
|
LIGHT_GREY = '\x1b[0;30m'
|
|
LIGHT_RED = '\x1b[0;31m'
|
|
LIGHT_GREEN = '\x1b[0;32m'
|
|
LIGHT_YELLOW = '\x1b[0;33m'
|
|
LIGHT_BLUE = '\x1b[0;34m'
|
|
LIGHT_MAGENTA = '\x1b[0;35m'
|
|
LIGHT_CYAN = '\x1b[0;36m'
|
|
LIGHT_WHITE = '\x1b[0;37m'
|
|
|
|
def clr(color, text):
|
|
return color + str(text) + '\x1b[0m'
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Callbacks
|
|
# ------------------------------------------------------------------------------
|
|
|
|
def check_size(source, target, env):
|
|
time.sleep(2)
|
|
size = target[0].get_size()
|
|
print clr(Color.LIGHT_BLUE, "Binary size: %s bytes" % size)
|
|
#if size > 512000:
|
|
# print clr(Color.LIGHT_RED, "File too large for OTA!")
|
|
# Exit(1)
|
|
|
|
def add_build_flags(source, target, env):
|
|
build_h = "espurna/config/build.h"
|
|
build_flags = env['BUILD_FLAGS'][0]
|
|
lines = open(build_h).readlines()
|
|
with open(build_h, "w") as fh:
|
|
for line in lines:
|
|
if "APP_BUILD_FLAGS" in line:
|
|
fh.write("#define APP_BUILD_FLAGS \"%s\"" % build_flags)
|
|
else:
|
|
fh.write(line)
|
|
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Hooks
|
|
# ------------------------------------------------------------------------------
|
|
|
|
env.AddPreAction("$BUILD_DIR/src/espurna.ino.o", add_build_flags)
|
|
env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", check_size)
|