""" Accesses the storage of a component. """
import json
import os
import sys
from typing import List, Optional
from common import run_ffx_command
class ComponentStorage:
""" Manages the storage of a component on a target.
Unlike other OSes, the |remote_path| is impacted by the |capacity|,
e.g. if a file needs to be copied to the /cache which is provided by the
capability cache, the |remote_path| should be / or . instead of /cache.
Unless otherwise mentioned, functions assert on any failures.
"""
def __init__(self,
component: str,
target: str = os.getenv('FUCHSIA_NODENAME')):
self._instance_id = json.loads(
run_ffx_command(cmd=('--machine', 'json', 'component', 'show',
component),
target_id=target,
capture_output=True).stdout.strip())["instance_id"]
self._target = target
def push(self,
local_path: str,
remote_path: str,
capability: Optional[str] = None) -> None:
""" Copies a file from |local_path| to |remote_path| on the target. """
cmd = ['component', 'storage']
if capability:
cmd.extend(['--capability', capability])
cmd.extend(
['copy', local_path, self._instance_id + '::' + remote_path])
run_ffx_command(cmd=cmd, target_id=self._target)
def pull(self,
remote_path: str,
local_path: str,
capability: Optional[str] = None) -> None:
""" Copies a file from |remote_path| to |local_path| from the target.
"""
cmd = ['component', 'storage']
if capability:
cmd.extend(['--capability', capability])
cmd.extend(
['copy', self._instance_id + '::' + remote_path, local_path])
run_ffx_command(cmd=cmd, target_id=self._target)
def delete(self,
remote_path: str,
capability: Optional[str] = None) -> None:
""" Deletes a file from |remote_path| on the target. """
cmd = ['component', 'storage']
if capability:
cmd.extend(['--capability', capability])
cmd.extend(['delete', self._instance_id + '::' + remote_path])
run_ffx_command(cmd=cmd, target_id=self._target)
def list(self,
remote_path: str,
capability: Optional[str] = None) -> List[str]:
""" Lists files in |remote_path| on the target. """
cmd = ['component', 'storage']
if capability:
cmd.extend(['--capability', capability])
cmd.extend(['list', self._instance_id + '::' + remote_path])
return run_ffx_command(
cmd=cmd, target_id=self._target,
capture_output=True).stdout.strip().splitlines()
def list_to_stdout(self,
remote_path: str,
capability: Optional[str] = None) -> None:
""" Lists files in |remote_path| on the target and prints out to stdout.
"""
for file in self.list(remote_path, capability):
print(file)
if __name__ == "__main__":
args = sys.argv
getattr(ComponentStorage(args[1], args[2]), args[3])(*args[4:])
