From 748283cced543c486145d5f3f38684becdfe3e1b Mon Sep 17 00:00:00 2001
From: Justin Traglia <95511699+jtraglia@users.noreply.github.com>
Date: Wed, 18 Oct 2023 21:00:40 -0500
Subject: [PATCH] Add script for converting trusted setups (#359)

---
 .github/workflows/trusted-setup-tests.yml | 24 +++++++++++++
 scripts/convert_trusted_setup.py          | 44 +++++++++++++++++++++++
 2 files changed, 68 insertions(+)
 create mode 100644 .github/workflows/trusted-setup-tests.yml
 create mode 100755 scripts/convert_trusted_setup.py

diff --git a/.github/workflows/trusted-setup-tests.yml b/.github/workflows/trusted-setup-tests.yml
new file mode 100644
index 0000000..d5e3b19
--- /dev/null
+++ b/.github/workflows/trusted-setup-tests.yml
@@ -0,0 +1,24 @@
+name: Trusted Setup
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  tests:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-python@v4
+        with:
+          python-version: '3.x'
+
+      - name: Download
+        run: wget -O trusted_setup.json https://github.com/ethereum/consensus-specs/raw/dev/presets/mainnet/trusted_setups/trusted_setup_4096.json
+      - name: Convert
+        run: python3 ./scripts/convert_trusted_setup.py --input trusted_setup.json --output trusted_setup.txt
+      - name: Compare
+        run: cmp src/trusted_setup.txt trusted_setup.txt
diff --git a/scripts/convert_trusted_setup.py b/scripts/convert_trusted_setup.py
new file mode 100755
index 0000000..e1586d5
--- /dev/null
+++ b/scripts/convert_trusted_setup.py
@@ -0,0 +1,44 @@
+#!/usr/bin/python3
+
+import argparse
+import json
+from typing import TextIO
+
+
+def convert(ts_json: TextIO, ts_text: TextIO) -> None:
+    """Convert trusted setup to text format."""
+    trusted_setup = json.load(ts_json)
+    g1_values = trusted_setup["g1_lagrange"]
+    g2_values = trusted_setup["g2_monomial"]
+
+    print(len(g1_values), file=ts_text)
+    print(len(g2_values), file=ts_text)
+    for g1 in g1_values:
+        print(g1.replace("0x", ""), file=ts_text)
+    for g2 in g2_values:
+        print(g2.replace("0x", ""), file=ts_text)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+        description="Convert trusted setup from JSON to text format.",
+    )
+    parser.add_argument(
+        "--input",
+        required=True,
+        type=argparse.FileType("r"),
+        help="the trusted setup in JSON format",
+    )
+    parser.add_argument(
+        "--output",
+        required=True,
+        type=argparse.FileType("w"),
+        help="the trusted setup in text format",
+    )
+    args = parser.parse_args()
+
+    try:
+        convert(args.input, args.output)
+    finally:
+        args.input.close()
+        args.output.close()