249 lines
		
	
	
		
			8.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			249 lines
		
	
	
		
			8.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from dataclasses import dataclass, field
 | 
						|
from typing import List, Optional
 | 
						|
from enum import Enum
 | 
						|
import os
 | 
						|
 | 
						|
 | 
						|
class FlipperManifestException(Exception):
 | 
						|
    pass
 | 
						|
 | 
						|
 | 
						|
class FlipperAppType(Enum):
 | 
						|
    SERVICE = "Service"
 | 
						|
    SYSTEM = "System"
 | 
						|
    APP = "App"
 | 
						|
    PLUGIN = "Plugin"
 | 
						|
    DEBUG = "Debug"
 | 
						|
    ARCHIVE = "Archive"
 | 
						|
    SETTINGS = "Settings"
 | 
						|
    STARTUP = "StartupHook"
 | 
						|
    EXTERNAL = "External"
 | 
						|
    METAPACKAGE = "Package"
 | 
						|
 | 
						|
 | 
						|
@dataclass
 | 
						|
class FlipperApplication:
 | 
						|
    appid: str
 | 
						|
    apptype: FlipperAppType
 | 
						|
    name: Optional[str] = None
 | 
						|
    entry_point: Optional[str] = None
 | 
						|
    flags: List[str] = field(default_factory=lambda: ["Default"])
 | 
						|
    cdefines: List[str] = field(default_factory=list)
 | 
						|
    requires: List[str] = field(default_factory=list)
 | 
						|
    conflicts: List[str] = field(default_factory=list)
 | 
						|
    provides: List[str] = field(default_factory=list)
 | 
						|
    stack_size: int = 2048
 | 
						|
    icon: Optional[str] = None
 | 
						|
    order: int = 0
 | 
						|
    _appdir: Optional[str] = None
 | 
						|
 | 
						|
 | 
						|
class AppManager:
 | 
						|
    def __init__(self):
 | 
						|
        self.known_apps = {}
 | 
						|
 | 
						|
    def get(self, appname: str):
 | 
						|
        try:
 | 
						|
            return self.known_apps[appname]
 | 
						|
        except KeyError as _:
 | 
						|
            raise FlipperManifestException(
 | 
						|
                f"Missing application manifest for '{appname}'"
 | 
						|
            )
 | 
						|
 | 
						|
    def load_manifest(self, app_manifest_path: str, app_dir_name: str):
 | 
						|
        if not os.path.exists(app_manifest_path):
 | 
						|
            raise FlipperManifestException(
 | 
						|
                f"App manifest not found at path {app_manifest_path}"
 | 
						|
            )
 | 
						|
        # print("Loading", app_manifest_path)
 | 
						|
 | 
						|
        app_manifests = []
 | 
						|
 | 
						|
        def App(*args, **kw):
 | 
						|
            nonlocal app_manifests
 | 
						|
            app_manifests.append(FlipperApplication(*args, **kw, _appdir=app_dir_name))
 | 
						|
 | 
						|
        try:
 | 
						|
            with open(app_manifest_path, "rt") as manifest_file:
 | 
						|
                exec(manifest_file.read())
 | 
						|
        except Exception as e:
 | 
						|
            raise FlipperManifestException(
 | 
						|
                f"Failed parsing manifest '{app_manifest_path}' : {e}"
 | 
						|
            )
 | 
						|
 | 
						|
        if len(app_manifests) == 0:
 | 
						|
            raise FlipperManifestException(
 | 
						|
                f"App manifest '{app_manifest_path}' is malformed"
 | 
						|
            )
 | 
						|
 | 
						|
        # print("Built", app_manifests)
 | 
						|
        for app in app_manifests:
 | 
						|
            self._add_known_app(app)
 | 
						|
 | 
						|
    def _add_known_app(self, app: FlipperApplication):
 | 
						|
        if self.known_apps.get(app.appid, None):
 | 
						|
            raise FlipperManifestException(f"Duplicate app declaration: {app.appid}")
 | 
						|
        self.known_apps[app.appid] = app
 | 
						|
 | 
						|
    def filter_apps(self, applist: List[str]):
 | 
						|
        return AppBuildset(self, applist)
 | 
						|
 | 
						|
 | 
						|
class AppBuilderException(Exception):
 | 
						|
    pass
 | 
						|
 | 
						|
 | 
						|
class AppBuildset:
 | 
						|
    BUILTIN_APP_TYPES = (
 | 
						|
        FlipperAppType.SERVICE,
 | 
						|
        FlipperAppType.SYSTEM,
 | 
						|
        FlipperAppType.APP,
 | 
						|
        FlipperAppType.PLUGIN,
 | 
						|
        FlipperAppType.DEBUG,
 | 
						|
        FlipperAppType.ARCHIVE,
 | 
						|
        FlipperAppType.SETTINGS,
 | 
						|
        FlipperAppType.STARTUP,
 | 
						|
    )
 | 
						|
 | 
						|
    def __init__(self, appmgr: AppManager, appnames: List[str]):
 | 
						|
        self.appmgr = appmgr
 | 
						|
        self.appnames = set(appnames)
 | 
						|
        self._orig_appnames = appnames
 | 
						|
        self._process_deps()
 | 
						|
        self._check_conflicts()
 | 
						|
        self._check_unsatisfied()  # unneeded?
 | 
						|
        self.apps = sorted(
 | 
						|
            list(map(self.appmgr.get, self.appnames)),
 | 
						|
            key=lambda app: app.appid,
 | 
						|
        )
 | 
						|
 | 
						|
    def _is_missing_dep(self, dep_name: str):
 | 
						|
        return dep_name not in self.appnames
 | 
						|
 | 
						|
    def _process_deps(self):
 | 
						|
        while True:
 | 
						|
            provided = []
 | 
						|
            for app in self.appnames:
 | 
						|
                # print(app)
 | 
						|
                provided.extend(
 | 
						|
                    filter(
 | 
						|
                        self._is_missing_dep,
 | 
						|
                        self.appmgr.get(app).provides + self.appmgr.get(app).requires,
 | 
						|
                    )
 | 
						|
                )
 | 
						|
            # print("provides round", provided)
 | 
						|
            if len(provided) == 0:
 | 
						|
                break
 | 
						|
            self.appnames.update(provided)
 | 
						|
 | 
						|
    def _check_conflicts(self):
 | 
						|
        conflicts = []
 | 
						|
        for app in self.appnames:
 | 
						|
            # print(app)
 | 
						|
            if conflict_app_name := list(
 | 
						|
                filter(
 | 
						|
                    lambda dep_name: dep_name in self.appnames,
 | 
						|
                    self.appmgr.get(app).conflicts,
 | 
						|
                )
 | 
						|
            ):
 | 
						|
                conflicts.append((app, conflict_app_name))
 | 
						|
 | 
						|
        if len(conflicts):
 | 
						|
            raise AppBuilderException(
 | 
						|
                f"App conflicts for {', '.join(f'{conflict_dep[0]}: {conflict_dep[1]}' for conflict_dep in conflicts)}"
 | 
						|
            )
 | 
						|
 | 
						|
    def _check_unsatisfied(self):
 | 
						|
        unsatisfied = []
 | 
						|
        for app in self.appnames:
 | 
						|
            if missing_dep := list(
 | 
						|
                filter(self._is_missing_dep, self.appmgr.get(app).requires)
 | 
						|
            ):
 | 
						|
                unsatisfied.append((app, missing_dep))
 | 
						|
 | 
						|
        if len(unsatisfied):
 | 
						|
            raise AppBuilderException(
 | 
						|
                f"Unsatisfied dependencies for {', '.join(f'{missing_dep[0]}: {missing_dep[1]}' for missing_dep in unsatisfied)}"
 | 
						|
            )
 | 
						|
 | 
						|
    def get_apps_cdefs(self):
 | 
						|
        cdefs = set()
 | 
						|
        for app in self.apps:
 | 
						|
            cdefs.update(app.cdefines)
 | 
						|
        return sorted(list(cdefs))
 | 
						|
 | 
						|
    def get_apps_of_type(self, apptype: FlipperAppType):
 | 
						|
        return sorted(
 | 
						|
            filter(lambda app: app.apptype == apptype, self.apps),
 | 
						|
            key=lambda app: app.order,
 | 
						|
        )
 | 
						|
 | 
						|
    def get_builtin_app_folders(self):
 | 
						|
        return sorted(
 | 
						|
            set(
 | 
						|
                app._appdir
 | 
						|
                for app in filter(
 | 
						|
                    lambda app: app.apptype in self.BUILTIN_APP_TYPES, self.apps
 | 
						|
                )
 | 
						|
            )
 | 
						|
        )
 | 
						|
 | 
						|
 | 
						|
class ApplicationsCGenerator:
 | 
						|
    APP_TYPE_MAP = {
 | 
						|
        FlipperAppType.SERVICE: ("FlipperApplication", "FLIPPER_SERVICES"),
 | 
						|
        FlipperAppType.SYSTEM: ("FlipperApplication", "FLIPPER_SYSTEM_APPS"),
 | 
						|
        FlipperAppType.APP: ("FlipperApplication", "FLIPPER_APPS"),
 | 
						|
        FlipperAppType.PLUGIN: ("FlipperApplication", "FLIPPER_PLUGINS"),
 | 
						|
        FlipperAppType.DEBUG: ("FlipperApplication", "FLIPPER_DEBUG_APPS"),
 | 
						|
        FlipperAppType.SETTINGS: ("FlipperApplication", "FLIPPER_SETTINGS_APPS"),
 | 
						|
        FlipperAppType.STARTUP: ("FlipperOnStartHook", "FLIPPER_ON_SYSTEM_START"),
 | 
						|
    }
 | 
						|
 | 
						|
    def __init__(self, buildset: AppBuildset):
 | 
						|
        self.buildset = buildset
 | 
						|
 | 
						|
    def get_app_ep_forward(self, app: FlipperApplication):
 | 
						|
        if app.apptype == FlipperAppType.STARTUP:
 | 
						|
            return f"extern void {app.entry_point}();"
 | 
						|
        return f"extern int32_t {app.entry_point}(void* p);"
 | 
						|
 | 
						|
    def get_app_descr(self, app: FlipperApplication):
 | 
						|
        if app.apptype == FlipperAppType.STARTUP:
 | 
						|
            return app.entry_point
 | 
						|
        return f"""
 | 
						|
    {{.app = {app.entry_point},
 | 
						|
     .name = "{app.name}",
 | 
						|
     .stack_size = {app.stack_size},
 | 
						|
     .icon = {f"&{app.icon}" if app.icon else "NULL"},
 | 
						|
     .flags = {'|'.join(f"FlipperApplicationFlag{flag}" for flag in app.flags)} }}"""
 | 
						|
 | 
						|
    def generate(self):
 | 
						|
        contents = ['#include "applications.h"', "#include <assets_icons.h>"]
 | 
						|
        for apptype in self.APP_TYPE_MAP:
 | 
						|
            contents.extend(
 | 
						|
                map(self.get_app_ep_forward, self.buildset.get_apps_of_type(apptype))
 | 
						|
            )
 | 
						|
            entry_type, entry_block = self.APP_TYPE_MAP[apptype]
 | 
						|
            contents.append(f"const {entry_type} {entry_block}[] = {{")
 | 
						|
            contents.append(
 | 
						|
                ",\n".join(
 | 
						|
                    map(self.get_app_descr, self.buildset.get_apps_of_type(apptype))
 | 
						|
                )
 | 
						|
            )
 | 
						|
            contents.append("};")
 | 
						|
            contents.append(
 | 
						|
                f"const size_t {entry_block}_COUNT = COUNT_OF({entry_block});"
 | 
						|
            )
 | 
						|
 | 
						|
        archive_app = self.buildset.get_apps_of_type(FlipperAppType.ARCHIVE)
 | 
						|
        if archive_app:
 | 
						|
            contents.extend(
 | 
						|
                [
 | 
						|
                    self.get_app_ep_forward(archive_app[0]),
 | 
						|
                    f"const FlipperApplication FLIPPER_ARCHIVE = {self.get_app_descr(archive_app[0])};",
 | 
						|
                ]
 | 
						|
            )
 | 
						|
 | 
						|
        return "\n".join(contents)
 |