add toml2json

Author: rickeylev

tests/tools/private/toml2json/BUILD.bazel

@@ -0,0 +1,10 @@
+load("@rules_python//python:defs.bzl", "py_test")
+
+py_test(
+    name = "toml2json_test",
+    srcs = ["toml2json_test.py"],
+    main = "toml2json_test.py",
+    deps = [
+        "//tools/private/toml2json",
+    ],
+)

tests/tools/private/toml2json/toml2json_test.py

@@ -0,0 +1,62 @@
+import io
+import json
+import os
+import sys
+import tempfile
+import unittest
+from unittest.mock import patch
+
+from tools.private.toml2json import toml2json
+
+class Toml2JsonTest(unittest.TestCase):
+
+    def setUp(self):
+        self.temp_dir = tempfile.TemporaryDirectory()
+        self.addCleanup(self.temp_dir.cleanup)
+
+    def _create_temp_toml_file(self, content):
+        fd, path = tempfile.mkstemp(suffix=".toml", dir=self.temp_dir.name)
+        with os.fdopen(fd, "wb") as f:
+            f.write(content)
+        return path
+
+    def test_basic_conversion(self):
+        toml_content = b"""
+[owner]
+name = "Tom Preston-Werner"
+dob = 1979-05-27T07:32:00-08:00
+"""
+        expected_json = {
+            "owner": {
+                "name": "Tom Preston-Werner",
+                "dob": "1979-05-27T07:32:00-08:00"
+            }
+        }
+        
+        toml_file_path = self._create_temp_toml_file(toml_content)
+
+        with patch('sys.stdout', new=io.StringIO()) as mock_stdout:
+            with patch('sys.argv', ['toml2json.py', toml_file_path]):
+                toml2json.main()
+                actual_json = json.loads(mock_stdout.getvalue())
+                self.assertEqual(actual_json, expected_json)
+
+    def test_invalid_toml(self):
+        toml_content = b"""
+[owner
+name = "Tom Preston-Werner"
+"""
+
+        toml_file_path = self._create_temp_toml_file(toml_content)
+
+        with patch('sys.stderr', new=io.StringIO()) as mock_stderr:
+            with patch('sys.stdout', new=io.StringIO()): # We don't expect stdout for errors
+                with patch('sys.exit') as mock_exit:
+                    with patch('sys.argv', ['toml2json.py', toml_file_path]):
+                        toml2json.main()
+                        mock_exit.assert_called_with(1)
+                    self.assertIn("Error decoding TOML", mock_stderr.getvalue())
+
+
+if __name__ == '__main__':
+    unittest.main()

tools/private/toml2json/BUILD.bazel

@@ -0,0 +1,9 @@
+load("@rules_python//python:defs.bzl", "py_binary")
+
+exports_files(["toml2json.py"])
+
+py_binary(
+    name = "toml2json",
+    srcs = ["toml2json.py"],
+    visibility = ["//visibility:public"],
+)

tools/private/toml2json/toml2json.py

@@ -0,0 +1,42 @@
+import json
+import sys
+import datetime
+
+try:
+    import tomllib
+except ImportError:
+    try:
+        import tomli as tomllib
+    except ImportError:
+        print("Error: need tomllib (python >=3.11) or tomli installed on host python", file=sys.stderr)
+        sys.exit(1)
+
+
+def json_serializer(obj):
+    if isinstance(obj, datetime.datetime):
+        return obj.isoformat()
+    raise TypeError(f"Object of type {type(obj).__name__} is not JSON serializable")
+
+
+def main():
+    if len(sys.argv) < 2:
+        print("Usage: toml2json <toml_file>", file=sys.stderr)
+        sys.exit(1)
+
+    toml_file_path = sys.argv[1]
+
+    try:
+        with open(toml_file_path, "rb") as f:
+            data = tomllib.load(f)
+            json.dump(data, sys.stdout, indent=2, default=json_serializer)
+            print()
+    except FileNotFoundError:
+        print(f"Error: File not found: {toml_file_path}", file=sys.stderr)
+        sys.exit(1)
+    except tomllib.TOMLDecodeError as e:
+        print(f"Error decoding TOML: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()