From 6a00d7fcaab120912963059bb36513fc82322c0a Mon Sep 17 00:00:00 2001
From: Nataliia Solomko <natalirs1985@gmail.com>
Date: Wed, 12 Jun 2024 17:56:52 +0300
Subject: op_mode: T6227:  Rewrite show conntrack-sync cache internal to use
 tabulate output

---
 data/templates/conntrackd/conntrackd.op-mode.j2 | 13 -------------
 src/op_mode/conntrack_sync.py                   | 25 +++++++++++++++++++++++--
 2 files changed, 23 insertions(+), 15 deletions(-)
 delete mode 100644 data/templates/conntrackd/conntrackd.op-mode.j2

diff --git a/data/templates/conntrackd/conntrackd.op-mode.j2 b/data/templates/conntrackd/conntrackd.op-mode.j2
deleted file mode 100644
index 82f7e2859..000000000
--- a/data/templates/conntrackd/conntrackd.op-mode.j2
+++ /dev/null
@@ -1,13 +0,0 @@
-Source                                           Destination                                      Protocol
-{% for parsed in data if parsed.flow.meta is vyos_defined %}
-{%     for key in parsed.flow.meta %}
-{%         if key['@direction'] == 'original' %}
-{%             set saddr    = key.layer3.src | bracketize_ipv6 %}
-{%             set sport    = key.layer4.sport %}
-{%             set daddr    = key.layer3.dst | bracketize_ipv6 %}
-{%             set dport    = key.layer4.dport %}
-{%             set protocol = key.layer4['@protoname'] %}
-{{ "%-48s" | format(saddr ~ ':' ~ sport) }} {{ "%-48s" | format(daddr ~ ':' ~ dport) }} {{ protocol }}
-{%         endif %}
-{%     endfor %}
-{% endfor %}
diff --git a/src/op_mode/conntrack_sync.py b/src/op_mode/conntrack_sync.py
index 6c86ff492..f3b09b452 100755
--- a/src/op_mode/conntrack_sync.py
+++ b/src/op_mode/conntrack_sync.py
@@ -19,6 +19,8 @@ import sys
 import syslog
 import xmltodict
 
+from tabulate import tabulate
+
 import vyos.opmode
 
 from vyos.configquery import CliShellApiConfigQuery
@@ -27,7 +29,6 @@ from vyos.utils.commit import commit_in_progress
 from vyos.utils.process import call
 from vyos.utils.process import cmd
 from vyos.utils.process import run
-from vyos.template import render_to_string
 
 conntrackd_bin = '/usr/sbin/conntrackd'
 conntrackd_config = '/run/conntrackd/conntrackd.conf'
@@ -59,6 +60,26 @@ def flush_cache(direction):
     if tmp > 0:
         raise vyos.opmode.Error('Failed to clear {direction} cache')
 
+def get_formatted_output(data):
+    data_entries = []
+    for parsed in data:
+        for meta in parsed.get('flow', {}).get('meta', []):
+            direction = meta['@direction']
+            if direction == 'original':
+                src = meta['layer3']['src']
+                dst = meta['layer3']['dst']
+                sport = meta['layer4'].get('sport')
+                dport = meta['layer4'].get('dport')
+                protocol = meta['layer4'].get('@protoname')
+                orig_src = f'{src}:{sport}' if sport else src
+                orig_dst = f'{dst}:{dport}' if dport else dst
+
+                data_entries.append([orig_src, orig_dst, protocol])
+
+    headers = ["Source", "Destination", "Protocol"]
+    output = tabulate(data_entries, headers, tablefmt="simple")
+    return output
+
 def from_xml(raw, xml):
     out = []
     for line in xml.splitlines():
@@ -70,7 +91,7 @@ def from_xml(raw, xml):
     if raw:
         return out
     else:
-        return render_to_string('conntrackd/conntrackd.op-mode.j2', {'data' : out})
+        return get_formatted_output(out)
 
 def restart():
     is_configured()
-- 
cgit v1.2.3