mirror of
https://github.com/FUjr/gl-infra-builder.git
synced 2025-12-17 17:44:06 +00:00
235 lines
7.7 KiB
Python
Executable File
235 lines
7.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
|
||
from os import getenv
|
||
from pathlib import Path
|
||
from shutil import rmtree
|
||
from subprocess import run
|
||
from subprocess import call
|
||
import time
|
||
import sys
|
||
import yaml
|
||
|
||
profile_folder = Path(getenv("PROFILES", "./profiles")).absolute()
|
||
|
||
|
||
def die(msg: str):
|
||
"""Quit script with error message
|
||
|
||
msg (str): Error message to print
|
||
"""
|
||
print('\033[1;31m' + msg + '\033[0m')
|
||
sys.exit(1)
|
||
|
||
def warning(msg: str):
|
||
"""Quit script with warning message
|
||
|
||
msg (str): Error message to print
|
||
"""
|
||
print('\033[1;33m' + msg + '\033[0m')
|
||
|
||
|
||
def usage(code: int = 0):
|
||
"""Print script usage
|
||
|
||
Args:
|
||
code (int): exit code
|
||
"""
|
||
print(f"""Usage: {sys.argv[0]} <profile> [options...]
|
||
|
||
clean Remove feeds before setup
|
||
list List available profiles
|
||
help Print this message
|
||
""")
|
||
sys.exit(code)
|
||
|
||
def load_metadata():
|
||
try:
|
||
with open("gl_metadata.yaml", "r") as stream:
|
||
metadata=yaml.safe_load(stream)
|
||
version = metadata["version"]
|
||
call("echo %s > %s" % (version, "files/etc/glversion"), shell=True)
|
||
call("echo %s > %s" % (version, "release"), shell=True)
|
||
version_type = metadata["type"]
|
||
call("echo %s > %s" % (version_type, "files/etc/version.type"), shell=True)
|
||
print("firmware version: " +version)
|
||
print("firmware type: " +version_type)
|
||
except:
|
||
pass
|
||
#print("load gl metadata failed")
|
||
|
||
|
||
def load_yaml(fname: str, profile: dict):
|
||
profile_file = (profile_folder / fname).with_suffix(".yml")
|
||
|
||
if not profile_file.is_file():
|
||
die(f"Profile {fname} not found")
|
||
|
||
new = yaml.safe_load(profile_file.read_text())
|
||
for n in new:
|
||
if n in {"profile", "target", "subtarget", "external_target"}:
|
||
if profile.get(n):
|
||
die(f"Duplicate tag found {n}")
|
||
profile.update({n: new.get(n)})
|
||
elif n in {"description"}:
|
||
profile["description"].append(new.get(n))
|
||
elif n in {"packages"}:
|
||
profile["packages"].extend(new.get(n))
|
||
elif n in {"diffconfig"}:
|
||
profile["diffconfig"] += new.get(n)
|
||
elif n in {"metadata"}:
|
||
profile["metadata"] = new.get(n)
|
||
elif n in {"feeds"}:
|
||
for f in new.get(n):
|
||
if f.get("name", "") == "" or (f.get("uri", "") == "" and f.get("path", "") == ""):
|
||
die(f"Found bad feed {f}")
|
||
profile["feeds"][f.get("name")] = f
|
||
return profile
|
||
|
||
|
||
def clean_tree():
|
||
print("Cleaning tree")
|
||
rmtree("./tmp", ignore_errors=True)
|
||
rmtree("./packages/feeds/", ignore_errors=True)
|
||
rmtree("./tmp/", ignore_errors=True)
|
||
rmtree(".git/rebase-apply/", ignore_errors=True)
|
||
if Path("./feeds.conf").is_file():
|
||
Path("./feeds.conf").unlink()
|
||
if Path("./.config").is_file():
|
||
Path("./.config").unlink()
|
||
|
||
|
||
def merge_profiles(profiles):
|
||
profile = {"packages": [], "description": [], "metadata":{}, "diffconfig": "", "feeds": {}}
|
||
|
||
for p in profiles:
|
||
profile = load_yaml(p, profile)
|
||
|
||
return profile
|
||
|
||
|
||
def setup_feeds(profile):
|
||
feeds_conf = Path("feeds.conf")
|
||
if feeds_conf.is_file():
|
||
feeds_conf.unlink()
|
||
|
||
feeds = []
|
||
for p in profile.get("feeds", []):
|
||
try:
|
||
f = profile["feeds"].get(p)
|
||
if all(k in f for k in ("branch", "revision", "path")):
|
||
die(f"Please specify either a branch, a revision or a path: {f}")
|
||
if "path" in f:
|
||
feeds.append(f'{f.get("method", "src-link")},{f["name"]},{f["path"]}')
|
||
elif "revision" in f:
|
||
feeds.append(f'{f.get("method", "src-git")},{f["name"]},{f["uri"]}^{f.get("revision")}')
|
||
else:
|
||
feeds.append(f'{f.get("method", "src-git")},{f["name"]},{f["uri"]};{f.get("branch", "master")}')
|
||
|
||
except:
|
||
print(f"Badly configured feed: {f}")
|
||
|
||
if run(["./scripts/feeds", "uninstall", "-a"]).returncode:
|
||
die(f"Error uninstall feeds")
|
||
|
||
if run(["./scripts/feeds", "setup", "-b", *feeds]).returncode:
|
||
die(f"Error setting up feeds")
|
||
|
||
if run(["./scripts/feeds", "update"]).returncode:
|
||
warning(f"Error updating feeds")
|
||
|
||
print("Install all feeds...")
|
||
#lte_data_oss这个feeds与我们自己的ui包有冲突，先注释掉，后续有问题可以打开
|
||
call("sed -i 's/^src-link lte_data_oss/#src-link lte_data_oss/' ./feeds.conf.default",shell=True)
|
||
if run(["./scripts/feeds", "install", "-a", "-f"]).returncode:
|
||
die(f"Error installing")
|
||
#for p in profile.get("feeds", []):
|
||
# f = profile["feeds"].get(p)
|
||
# if run(["./scripts/feeds", "install", "-a", "-f", "-p", f.get("name")]).returncode:
|
||
# die(f"Error installing {feed}")
|
||
|
||
packages = ["./scripts/feeds", "install"]
|
||
for package in profile.get("packages", []):
|
||
packages.append(package)
|
||
if len(packages) > 2:
|
||
if run(packages).returncode:
|
||
die(f"Error installing packages")
|
||
|
||
if profile.get("external_target", False):
|
||
if run(["./scripts/feeds", "install", profile["target"]]).returncode:
|
||
die(f"Error installing external target {profile['target']}")
|
||
|
||
def setup_all_feeds():
|
||
print("Install all feeds...")
|
||
if run(["./scripts/feeds", "install", "-a", "-f"]).returncode:
|
||
die(f"Error installing")
|
||
|
||
def generate_config(profile):
|
||
config_output = f"""CONFIG_TARGET_{profile["target"]}=y
|
||
CONFIG_TARGET_{profile["target"]}_{profile["subtarget"]}=y
|
||
CONFIG_TARGET_{profile["target"]}_{profile["subtarget"]}_DEVICE_{profile["profile"]}=y
|
||
CONFIG_TARGET_{profile["target"]}_{profile["subtarget"]}_{profile["profile"]}=y
|
||
"""
|
||
|
||
config_output += f"{profile.get('diffconfig', '')}"
|
||
|
||
for package in profile.get("packages", []):
|
||
print(f"Add package to .config: {package}")
|
||
config_output += f"CONFIG_PACKAGE_{package}=y\n"
|
||
|
||
Path(".config").write_text(config_output)
|
||
print("Configuration written to .config")
|
||
|
||
|
||
def generate_files(profile):
|
||
if run(["mkdir", "-p", "files/etc"]).returncode:
|
||
die(f"Error create files")
|
||
|
||
compile_time = time.strftime('%Y-%m-%d %k:%M:%S', time.localtime(time.time()))
|
||
call("echo %s > %s" % (compile_time, "files/etc/version.date"), shell=True)
|
||
load_metadata()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if "list" in sys.argv:
|
||
print(f"Profiles in {profile_folder}")
|
||
print("Target Profiles:")
|
||
print("\n".join(sorted(map(lambda p: str(p.stem), profile_folder.glob("target_*.yml")))))
|
||
print("\nFunction Profiles:")
|
||
print("\n".join(sorted(map(lambda p: str(p.stem), profile_folder.glob('[!t]*.yml')))))
|
||
quit(0)
|
||
|
||
if "help" in sys.argv:
|
||
usage()
|
||
|
||
if len(sys.argv) < 2:
|
||
usage(1)
|
||
|
||
if "clean" in sys.argv:
|
||
clean_tree()
|
||
print("Tree is now clean")
|
||
quit(0)
|
||
|
||
profile = merge_profiles(sys.argv[1:])
|
||
|
||
print("Using the following profiles:")
|
||
for d in profile.get("description"):
|
||
print(f" - {d}")
|
||
|
||
clean_tree()
|
||
setup_feeds(profile)
|
||
#setup_all_feeds()
|
||
generate_config(profile)
|
||
generate_files(profile)
|
||
|
||
print("Running make defconfig")
|
||
if run(["make", "defconfig"]).returncode:
|
||
die(f"Error running make defconfig")
|
||
|
||
if call("grep 'CONFIG_TARGET_PROFILE' .config|grep %s" % (profile['profile']), shell=True) != 0:
|
||
if call("grep 'CONFIG_TARGET_%s_%s_%s' .config" % (profile['target'], profile['subtarget'], profile['profile']), shell=True) != 0:
|
||
die(f"Error: Cant not find the profile '%s'! Please check again!" % (profile['profile']))
|
||
|
||
for package in profile['packages']:
|
||
if call("grep 'CONFIG_PACKAGE_%s=y' .config" % (package), shell=True) != 0:
|
||
die(f"Error: the package '%s' is not found or is not configured!" % (package))
|