From 3a8e360e812e305ad4248900d3266408792f07ef Mon Sep 17 00:00:00 2001
From: Aditya Nagesh <adityanagesh@microsoft.com>
Date: Tue, 12 Nov 2024 15:15:51 +0530
Subject: [PATCH] Add B4 Tool

---
 lisa/tools/b4.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 lisa/tools/b4.py

diff --git a/lisa/tools/b4.py b/lisa/tools/b4.py
new file mode 100644
index 0000000000..a97a6c8f73
--- /dev/null
+++ b/lisa/tools/b4.py
@@ -0,0 +1,29 @@
+import pathlib
+import re
+from lisa.executable import Tool
+from lisa.operating_system import Posix
+from lisa.util import get_matched_str
+
+
+class B4(Tool):
+    _output_file_pattern = re.compile(r"[\w-]+\.mbx")
+
+    @property
+    def command(self) -> str:
+        return "b4"
+
+    @property
+    def can_install(self) -> bool:
+        return True
+
+    def _install(self) -> bool:
+        if isinstance(self.node.os, Posix):
+            self.node.os.install_packages("b4")
+        return self._check_exists()
+
+    def am(self, message_id: str, output_dir: pathlib.PurePath) -> pathlib.PurePath:
+        result = self.run(
+            f"am -o '{output_dir}' '{message_id}'", force_run=True, expected_exit_code=0
+        )
+        path_str = get_matched_str(result.stdout, self._output_file_pattern)
+        return pathlib.PurePath(output_dir, path_str)