add action send_goal prototype completer

Signed-off-by: artivis <jeremie.deray@canonical.com>

ros2action/ros2action/api/__init__.py

@@ -22,6 +22,10 @@
 from rclpy.expand_topic_name import expand_topic_name
 from rclpy.validate_full_topic_name import validate_full_topic_name
 from ros2cli.node.direct import DirectNode
+from rosidl_parser.definition import AbstractNestedType
+from rosidl_parser.definition import NamespacedType
+from rosidl_runtime_py.convert import message_to_yaml
+from rosidl_runtime_py.import_message import import_message_from_namespaced_type
 
 
 def _is_action_status_topic(topic_name, action_name):
@@ -139,3 +143,23 @@ def __call__(self, prefix, parsed_args, **kwargs):
                 if n == action_name:
                     return t
         return []
+
+
+class ActionGoalPrototypeCompleter:
+    """Callable returning an action goal prototype."""
+
+    def __init__(self, *, action_type_key=None):
+        self.action_type_key = action_type_key
+
+    def __call__(self, prefix, parsed_args, **kwargs):
+        full_action_type = getattr(parsed_args, self.action_type_key)
+        try:
+            parts = full_action_type.split('/')
+            package_name = parts[0]
+            interface_type = parts[1]
+            action_type = parts[-1]
+        except LookupError as e:
+            return None
+        action = import_message_from_namespaced_type(
+            AbstractNestedType(NamespacedType([package_name, interface_type], action_type)))
+        return [message_to_yaml(action.Goal())]

ros2action/ros2action/verb/send_goal.py

@@ -19,6 +19,7 @@
 from rclpy.action import ActionClient
 from ros2action.api import action_name_completer
 from ros2action.api import ActionTypeCompleter
+from ros2action.api import ActionGoalPrototypeCompleter
 from ros2action.verb import VerbExtension
 from ros2cli.node import NODE_NAME_PREFIX
 from rosidl_runtime_py import message_to_yaml
@@ -39,9 +40,10 @@ def add_arguments(self, parser, cli_name):
             'action_type',
             help="Type of the ROS action (e.g. 'example_interfaces/action/Fibonacci')")
         arg.completer = ActionTypeCompleter(action_name_key='action_name')
-        parser.add_argument(
+        arg = parser.add_argument(
             'goal',
             help="Goal request values in YAML format (e.g. '{order: 10}')")
+        arg.completer = ActionGoalPrototypeCompleter(action_type_key='action_type')
         parser.add_argument(
             '-f', '--feedback', action='store_true',
             help='Echo feedback messages for the goal')