You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
177 lines
5.6 KiB
177 lines
5.6 KiB
# coding=utf-8
|
|
# pylint: dummy-variables-rgx='(_+[a-zA-Z0-9]*?$)|dummy|env'
|
|
#
|
|
# Original extra_scripts.py
|
|
# Copyright (C) 2016-2019 by Xose Pérez <xose dot perez at gmail dot com>
|
|
#
|
|
# ldscripts, lwip patching, updated postmortem flags and git support
|
|
# Copyright (C) 2019-2020 by Maxim Prokhorov <prokhorov dot max at outlook dot com>
|
|
|
|
# Run this script every time building an env BEFORE platform-specific code is loaded
|
|
|
|
from __future__ import print_function
|
|
|
|
Import("env")
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
from SCons.Script import ARGUMENTS
|
|
|
|
from espurna_utils.release import merge_cpp
|
|
|
|
|
|
CI = "true" == os.environ.get("CI")
|
|
PIO_PLATFORM = env.PioPlatform()
|
|
CONFIG = env.GetProjectConfig()
|
|
VERBOSE = "1" == ARGUMENTS.get("PIOVERBOSE", "0")
|
|
|
|
|
|
class ExtraScriptError(Exception):
|
|
pass
|
|
|
|
|
|
def log(message, verbose=False, file=sys.stderr):
|
|
if verbose or VERBOSE:
|
|
print(message, file=file)
|
|
|
|
|
|
# Most portable way, without depending on platformio internals
|
|
def subprocess_libdeps(lib_deps, storage=None, verbose=False):
|
|
import subprocess
|
|
|
|
args = [env.subst("$PYTHONEXE"), "-mplatformio", "lib"]
|
|
if not storage:
|
|
args.append("-g")
|
|
else:
|
|
args.extend(["-d", storage])
|
|
args.append("install")
|
|
if not verbose:
|
|
args.append("-s")
|
|
|
|
args.extend(lib_deps)
|
|
|
|
subprocess.check_call(args)
|
|
|
|
|
|
# Avoid spawning pio lib every time, hook into the LibraryManager API (sort-of internal)
|
|
def library_manager_libdeps(lib_deps, storage=None):
|
|
from platformio.managers.lib import LibraryManager
|
|
from platformio.project.helpers import get_project_global_lib_dir
|
|
|
|
if not storage:
|
|
manager = LibraryManager(get_project_global_lib_dir())
|
|
else:
|
|
manager = LibraryManager(storage)
|
|
|
|
for lib in lib_deps:
|
|
if manager.get_package_dir(*manager.parse_pkg_uri(lib)):
|
|
continue
|
|
log("installing: {}".format(lib))
|
|
manager.install(lib)
|
|
|
|
|
|
def get_shared_libdeps_dir(section, name):
|
|
|
|
if not CONFIG.has_option(section, name):
|
|
raise ExtraScriptError("{}.{} is required to be set".format(section, name))
|
|
|
|
opt = CONFIG.get(section, name)
|
|
|
|
if not opt in env.GetProjectOption("lib_extra_dirs"):
|
|
raise ExtraScriptError(
|
|
"lib_extra_dirs must contain {}.{}".format(section, name)
|
|
)
|
|
|
|
return os.path.join(env["PROJECT_DIR"], opt)
|
|
|
|
|
|
def ensure_platform_updated():
|
|
try:
|
|
if PIO_PLATFORM.are_outdated_packages():
|
|
log("updating platform packages")
|
|
PIO_PLATFORM.update_packages()
|
|
except Exception:
|
|
log("Warning: no connection, cannot check for outdated packages", verbose=True)
|
|
|
|
|
|
# handle build flags through os environment.
|
|
# using env instead of ini to avoid platformio ini changing hash on every change
|
|
env.Append(
|
|
ESPURNA_BOARD=os.environ.get("ESPURNA_BOARD", ""),
|
|
ESPURNA_AUTH=os.environ.get("ESPURNA_AUTH", ""),
|
|
ESPURNA_FLAGS=os.environ.get("ESPURNA_FLAGS", ""),
|
|
)
|
|
|
|
ESPURNA_OTA_PORT = os.environ.get("ESPURNA_IP")
|
|
if ESPURNA_OTA_PORT:
|
|
env.Replace(UPLOAD_PROTOCOL="espota")
|
|
env.Replace(UPLOAD_PORT=ESPURNA_OTA_PORT)
|
|
env.Replace(UPLOAD_FLAGS="--auth=$ESPURNA_AUTH")
|
|
else:
|
|
env.Replace(UPLOAD_PROTOCOL="esptool")
|
|
|
|
# handle `-t release` parameters
|
|
if CI:
|
|
env.Append(
|
|
ESPURNA_RELEASE_NAME=os.environ.get("ESPURNA_RELEASE_NAME", ""),
|
|
ESPURNA_RELEASE_VERSION=os.environ.get("ESPURNA_RELEASE_VERSION", ""),
|
|
ESPURNA_RELEASE_DESTINATION=os.environ.get("ESPURNA_RELEASE_DESTINATION", ""),
|
|
)
|
|
|
|
# updates arduino core git to the latest master commit
|
|
if CI:
|
|
package_overrides = env.GetProjectOption("platform_packages")
|
|
for package in package_overrides:
|
|
if "https://github.com/esp8266/Arduino.git" in package:
|
|
ensure_platform_updated()
|
|
break
|
|
|
|
# to speed-up build process, install libraries in either global or local shared storage
|
|
if os.environ.get("ESPURNA_PIO_SHARED_LIBRARIES"):
|
|
if CI:
|
|
storage = None
|
|
log("using global library storage")
|
|
else:
|
|
storage = get_shared_libdeps_dir("common", "shared_libdeps_dir")
|
|
log("using shared library storage: {}".format(storage))
|
|
|
|
subprocess_libdeps(env.GetProjectOption("lib_deps"), storage, verbose=VERBOSE)
|
|
|
|
# tweak build system to ignore espurna.ino, but include user code
|
|
# ref: platformio-core/platformio/tools/piomisc.py::ConvertInoToCpp()
|
|
def ConvertInoToCpp(env):
|
|
pass
|
|
|
|
|
|
ino = env.Glob("$PROJECT_DIR/espurna/*.ino") + env.Glob("$PROJECT_DIR/espurna/*.pde")
|
|
if len(ino) == 1 and ino[0].name == "espurna.ino":
|
|
env.AddMethod(ConvertInoToCpp)
|
|
|
|
# merge every .cpp into a single file and **only** build that single file
|
|
if "1" == os.environ.get("ESPURNA_BUILD_SINGLE_SOURCE", "0"):
|
|
cpp_files = []
|
|
for root, dirs, filenames in os.walk("espurna"):
|
|
for name in filenames:
|
|
if not name.endswith(".cpp"):
|
|
continue
|
|
|
|
abspath = os.path.join(os.path.abspath(root), name)
|
|
env.AddBuildMiddleware(lambda node: None, abspath)
|
|
|
|
relpath = os.path.relpath(abspath, "espurna")
|
|
cpp_files.append(relpath)
|
|
merge_cpp(cpp_files, "espurna/espurna_single_source.cpp")
|
|
|
|
# make sure to register as a valid command. however, it is always called right here
|
|
# (--list-targets is a kind-of inefficient for finding this, though, since it *will* install libs into .pio/ anyways...)
|
|
def install_libs_dummy(target, source, env):
|
|
pass
|
|
env.AddCustomTarget("install-libs", None, install_libs_dummy)
|
|
|
|
from SCons.Script import COMMAND_LINE_TARGETS
|
|
if "install-libs" in COMMAND_LINE_TARGETS:
|
|
storage = get_shared_libdeps_dir("common", "shared_libdeps_dir")
|
|
subprocess_libdeps(env.GetProjectOption("lib_deps"), storage, verbose=VERBOSE)
|
|
sys.exit(0)
|