1 files changed, 43 insertions, 2 deletions
diff --git a/src/op_mode/route.py b/src/op_mode/route.py
index d11b00ba0..4aa57dbf4 100755
--- a/src/op_mode/route.py
+++ b/src/op_mode/route.py
@@ -54,8 +54,49 @@ frr_command_template = Template("""
 {% endif %}
 """)
 
+ArgFamily = typing.Literal['inet', 'inet6']
+
+def show_summary(raw: bool, family: ArgFamily, table: typing.Optional[int], vrf: typing.Optional[str]):
+    from vyos.utils.process import cmd
+
+    if family == 'inet':
+        family_cmd = 'ip'
+    elif family == 'inet6':
+        family_cmd = 'ipv6'
+    else:
+        raise ValueError(f"Unsupported address family {family}")
+
+    if (table is not None) and (vrf is not None):
+        raise ValueError("table and vrf options are mutually exclusive")
+
+    # Replace with Jinja if it ever starts growing
+    if table:
+        table_cmd = f"table {table}"
+    else:
+        table_cmd = ""
+
+    if vrf:
+        vrf_cmd = f"vrf {vrf}"
+    else:
+        vrf_cmd = ""
+
+    if raw:
+        from json import loads
+
+        output = cmd(f"vtysh -c 'show {family_cmd} route {vrf_cmd} summary {table_cmd} json'").strip()
+
+        # If there are no routes in a table, its "JSON" output is an empty string,
+        # as of FRR 8.4.1
+        if output:
+            return loads(output)
+        else:
+            return {}
+    else:
+        output = cmd(f"vtysh -c 'show {family_cmd} route {vrf_cmd} summary {table_cmd}'")
+        return output
+
 def show(raw: bool,
-         family: str,
+         family: ArgFamily,
          net: typing.Optional[str],
          table: typing.Optional[int],
          protocol: typing.Optional[str],
@@ -78,7 +119,7 @@ def show(raw: bool,
         frr_command = frr_command_template.render(kwargs)
         frr_command = re.sub(r'\s+', ' ', frr_command)
 
-        from vyos.util import cmd
+        from vyos.utils.process import cmd
         output = cmd(f"vtysh -c '{frr_command}'")
 
         if raw: