mh
/
mhsw-espurna


								# 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_REVISION=os.environ.get("ESPURNA_RELEASE_REVISION", ""),

								        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)