import os import re # The directories to scan for Kconfig files SUBSYSTEM_DIRS = ["04_hal", "05_executors", "05_service", "06_behaviors", "tests"] TEMPLATE_SCAN_DIRS = SUBSYSTEM_DIRS + ["sweekar_demos"] BASE_DIR = os.path.dirname(os.path.abspath(__file__)) OUTPUT_FILE = os.path.join(BASE_DIR, "Kconfig.modules") SOURCE_EXTENSIONS = (".c", ".cc", ".cpp") HEADER_EXTENSIONS = (".h", ".hpp") SKIP_DIR_NAMES = {"build", "__pycache__"} def iter_child_dirs(parent_dir): if not os.path.exists(parent_dir): return [] return sorted( [ name for name in os.listdir(parent_dir) if os.path.isdir(os.path.join(parent_dir, name)) and not name.startswith(".") ] ) def compute_default_config_name(subsystem_dir, module_name): if subsystem_dir == "tests": prefix = "TEST" else: prefix = subsystem_dir.split("_")[1].upper() if prefix == "EXECUTORS": prefix = "EXEC" elif prefix == "BEHAVIORS": prefix = "BEHAV" elif prefix == "SERVICE": prefix = "SVC" return f"CONFIG_ENABLE_{prefix}_{module_name.upper()}" def contains_module_files(module_dir): for _, _, files in os.walk(module_dir): for filename in files: if filename.endswith(SOURCE_EXTENSIONS) or filename.endswith(HEADER_EXTENSIONS): return True return False def collect_existing_module_cmake_files(): module_files = [] for root, _, files in os.walk(BASE_DIR): if "module.cmake" in files: module_files.append(os.path.join(root, "module.cmake")) return sorted(module_files) def collect_module_files(module_dir): srcs = [] include_dirs = {"."} for root, dirs, files in os.walk(module_dir): dirs[:] = [ d for d in dirs if d not in SKIP_DIR_NAMES and not d.startswith(".") ] rel_root = os.path.relpath(root, module_dir).replace(os.sep, "/") rel_root = "." if rel_root == "." else rel_root has_header = False for filename in files: if filename.endswith(SOURCE_EXTENSIONS): rel_path = os.path.normpath(os.path.join(rel_root, filename)).replace(os.sep, "/") if rel_path.startswith("./"): rel_path = rel_path[2:] srcs.append(rel_path) elif filename.endswith(HEADER_EXTENSIONS): has_header = True if has_header: include_dirs.add(rel_root) srcs = sorted(set(srcs)) include_dirs = sorted(set(include_dirs)) return srcs, include_dirs def is_already_aggregated(rel_module_dir, existing_module_files): rel_module_prefix = f'"{rel_module_dir}/' for module_file in existing_module_files: try: with open(module_file, "r", encoding="utf-8") as f: content = f.read() except OSError: continue if rel_module_prefix in content: return True return False def render_module_template(subsystem_dir, module_name, module_dir): srcs, include_dirs = collect_module_files(module_dir) rel_module_dir = os.path.relpath(module_dir, BASE_DIR).replace(os.sep, "/") lines = [ f"# main/{subsystem_dir}/{module_name}/module.cmake", "# 自动生成的模块脚手架，可直接手工修改。", "# 如果后续添加了新文件，可以手工在此处添加，或者删除本文件让脚本重新生成。", "", ] if subsystem_dir != "sweekar_demos": config_name = compute_default_config_name(subsystem_dir, module_name) lines.extend( [ f"if(NOT {config_name} AND NOT EARLY_EXPANSION)", f' message(STATUS " [Skip] {module_name} module disabled by menuconfig.")', " return()", "endif()", "", ] ) lines.append("list(APPEND MODULE_SRCS") for src in srcs: lines.append(f' "{rel_module_dir}/{src}"') lines.append(")") lines.append("") lines.append("list(APPEND MODULE_INCLUDE_DIRS") for inc in include_dirs: if inc == ".": lines.append(f' "{rel_module_dir}"') else: lines.append(f' "{rel_module_dir}/{inc}"') lines.append(")") lines.append("") lines.extend( [ "# 需要公开依赖时，在这里手工补充。例如：", "# list(APPEND MODULE_REQUIRES", "# esp_timer", "# driver", "# )", "", "# 需要私有依赖时，在这里手工补充。例如：", "# list(APPEND MODULE_PRIV_REQUIRES", "# nvs_flash", "# )", "", ] ) return "\n".join(lines) def ensure_module_templates(): created_files = [] existing_module_files = collect_existing_module_cmake_files() for subsystem_dir in TEMPLATE_SCAN_DIRS: dir_path = os.path.join(BASE_DIR, subsystem_dir) if not os.path.exists(dir_path): continue for module_name in iter_child_dirs(dir_path): if subsystem_dir == "sweekar_demos" and not module_name.startswith("sweekar_demo"): continue module_dir = os.path.join(dir_path, module_name) module_cmake = os.path.join(module_dir, "module.cmake") if os.path.exists(module_cmake): continue if not contains_module_files(module_dir): continue rel_module_dir = os.path.relpath(module_dir, BASE_DIR).replace(os.sep, "/") if subsystem_dir != "sweekar_demos" and is_already_aggregated(rel_module_dir, existing_module_files): continue template_content = render_module_template(subsystem_dir, module_name, module_dir) with open(module_cmake, "w", encoding="utf-8") as f: f.write(template_content) created_files.append(module_cmake) return created_files def generate_kconfig_modules(): created_templates = ensure_module_templates() kconfig_content = 'menu "Sweekar Subsystems Configuration"\n\n' for d in SUBSYSTEM_DIRS: dir_path = os.path.join(BASE_DIR, d) if not os.path.exists(dir_path): continue # Create a pretty menu name from the directory name # e.g., "05_executors" -> "05 Executors Modules" parts = d.split('_') menu_name = parts[0] + " " + " ".join([p.capitalize() for p in parts[1:]]) + " Modules" kconfig_content += f' menu "{menu_name}"\n' # Sort subdirectories to ensure consistent ordering subdirs = iter_child_dirs(dir_path) for module_name in subdirs: module_dir = os.path.join(dir_path, module_name) kconfig_file = os.path.join(module_dir, "Kconfig") if os.path.exists(kconfig_file): # If the module has its own Kconfig, just source it # Make sure the source path is relative to the project root or uses proper env vars # Since Kconfig.projbuild is processed relative to project root usually rel_path = os.path.join("main", d, module_name, "Kconfig") kconfig_content += f' source "{rel_path}"\n' else: # If no Kconfig exists, generate a default toggle for it config_name = compute_default_config_name(d, module_name) pretty_name = " ".join([word.capitalize() for word in module_name.split('_')]) # Default to 'n' for tests, 'y' for everything else default_val = 'n' if d == 'tests' else 'y' kconfig_content += f''' config {config_name.replace("CONFIG_", "")} bool "Enable {pretty_name}" default {default_val} help Enable the {module_name} module in {d}. ''' kconfig_content += ' endmenu\n\n' kconfig_content += 'endmenu\n' with open(OUTPUT_FILE, 'w') as f: f.write(kconfig_content) print(f"Generated {OUTPUT_FILE} successfully.") if created_templates: print("Generated module.cmake templates:") for path in created_templates: print(f" - {path}") if __name__ == "__main__": generate_kconfig_modules()