"""Adapter for the Littlehand robot (vacuum gripper)."""

from __future__ import annotations

from typing import Any, Dict, List

from ...core.robot import RobotAdapter
from ...core.state import RobotStep


class VacuumGripperAdapter(RobotAdapter):
    """Adapter for robots with vacuum gripper (suction)."""

    def grab(self) -> List[RobotStep]:
        return [RobotStep(action="vacuum_on", payload={})]

    def release(self) -> List[RobotStep]:
        return [RobotStep(action="vacuum_off", payload={})]

    def move(self, payload: Dict[str, Any]) -> List[RobotStep]:
        return [RobotStep(action="move_to_pose", payload=payload)]

    def reset_tool(self) -> List[RobotStep]:
        return [RobotStep(action="vacuum_off", payload={})]


NAME = "littlehand"
ALIASES = {"vacuum", "littlehand"}
