unleashed-firmware/scripts/runfap.py
Sergey Gavrilov 645a7c5989
[FL-3386] Fast FAP Loader (#2790)
* FBT: build and add FastFAP(tm) sections
* Elf file: fast loading fap files. Really fast, like x15 times faster.
* fastfap.py: cleanup unused imports
* Toolchain: 23 version
* Elf File: remove log messages
* Scripts: fix file permissions
* FBT: explicit interpreter for fastfap invocation

Co-authored-by: あく <alleteam@gmail.com>
2023-06-28 17:19:10 +09:00

85 lines
2.6 KiB
Python
Executable file

#!/usr/bin/env python3
import operator
from functools import reduce
from flipper.app import App
from flipper.storage import FlipperStorage, FlipperStorageOperations
from flipper.utils.cdc import resolve_port
class Main(App):
def init(self):
self.parser.add_argument("-p", "--port", help="CDC Port", default="auto")
self.parser.add_argument(
"--sources",
"-s",
nargs="+",
action="append",
default=[],
help="Files to send",
)
self.parser.add_argument(
"--targets",
"-t",
nargs="+",
action="append",
default=[],
help="File destinations (must be same length as -s)",
)
self.parser.add_argument(
"--host-app",
"-a",
help="Host app to launch",
)
self.parser.set_defaults(func=self.install)
@staticmethod
def flatten(item_list):
return reduce(operator.concat, item_list, [])
def install(self):
self.args.sources = self.flatten(self.args.sources)
self.args.targets = self.flatten(self.args.targets)
if len(self.args.sources) != len(self.args.targets):
self.logger.error(
f"Error: sources ({self.args.sources}) and targets ({self.args.targets}) must be same length"
)
return 1
if not (port := resolve_port(self.logger, self.args.port)):
return 2
try:
with FlipperStorage(port) as storage:
storage_ops = FlipperStorageOperations(storage)
for fap_local_path, fap_dst_path in zip(
self.args.sources, self.args.targets
):
self.logger.info(f'Installing "{fap_local_path}" to {fap_dst_path}')
storage_ops.recursive_send(fap_dst_path, fap_local_path, False)
fap_host_app = self.args.targets[0]
startup_command = f"{fap_host_app}"
if self.args.host_app:
startup_command = self.args.host_app
self.logger.info(f"Launching app: {startup_command}")
storage.send_and_wait_eol(f"loader open {startup_command}\r")
if len(result := storage.read.until(storage.CLI_EOL)):
self.logger.error(f"Unexpected response: {result.decode('ascii')}")
return 3
return 0
except Exception as e:
self.logger.error(f"Error: {e}")
# raise
return 4
if __name__ == "__main__":
Main()()