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)
 | 
