from urllib.parse import urljoin
import tomllib
import sys
def load_toml(file_path):
with open(file_path, "rb") as f:
return tomllib.load(f)
class Link:
def __init__(self, tag, url):
self.tag = tag
self.url = url
def generate_table_for_chain(chain_name, chain_data, risc0_version):
chain_id = chain_data['id']
links = []
etherscan_url = chain_data['etherscan-url']
md_content = f"### {chain_data['name']} ({chain_id})\n\n"
md_content += "| Contract | Address |\n"
md_content += "| ---------------------------- | --------- |\n"
timelock_addr = chain_data['timelock-controller']
router_addr = chain_data['router']
md_content += f"| [RiscZeroVerifierRouter][router-src] | [`{router_addr}`][router-{chain_id}-etherscan] |\n"
links.append(Link(f"router-{chain_id}-etherscan", urljoin(etherscan_url, f"address/{router_addr}#code")))
if "verifiers" in chain_data:
for verifier in chain_data["verifiers"]:
if verifier.get("version") == risc0_version:
verifier_addr = verifier['verifier']
estop_addr = verifier['estop']
md_content += f"| [RiscZeroGroth16Verifier][verifier-src] | [`{verifier_addr}`][verifier-{chain_id}-etherscan] |\n"
md_content += f"| [RiscZeroVerifierEmergencyStop][estop-src] | [`{estop_addr}`][estop-{chain_id}-etherscan] |\n"
links.append(Link(f"estop-{chain_id}-etherscan", urljoin(etherscan_url, f"address/{estop_addr}#code")))
links.append(Link(f"verifier-{chain_id}-etherscan", urljoin(etherscan_url, f"address/{verifier_addr}#code")))
break
md_content += f"| TimelockController | [`{timelock_addr}`][timelock-{chain_id}-etherscan] |\n\n"
links.append(Link(f"timelock-{chain_id}-etherscan", urljoin(etherscan_url, f"address/{timelock_addr}#code")))
for link in links:
md_content += f"[{link.tag}]: {link.url}\n"
return md_content
def main(toml_file, risc0_version, risc0_eth_version):
config = load_toml(toml_file)
all_md_content = "<!-- GENERATED CONTENT BEGIN -->\n\n"
for chain_key, chain_data in config.get("chains", {}).items():
if chain_key in ("polygon-zkevm-testnet",):
continue
md_content = generate_table_for_chain(chain_key, chain_data, risc0_version)
all_md_content += md_content + "\n<br/>\n\n"
all_md_content += f"[router-src]: https://github.com/risc0/risc0-ethereum/tree/v{risc0_eth_version}/contracts/src/RiscZeroVerifierRouter.sol\n"
all_md_content += f"[verifier-src]: https://github.com/risc0/risc0-ethereum/tree/v{risc0_eth_version}/contracts/src/groth16/RiscZeroGroth16Verifier.sol\n"
all_md_content += f"[estop-src]: https://github.com/risc0/risc0-ethereum/tree/v{risc0_eth_version}/contracts/src/RiscZeroVerifierEmergencyStop.sol\n"
all_md_content += "\n<!-- GENERATED CONTENT END-->"
print(all_md_content)
if __name__ == "__main__":
if len(sys.argv) != 4:
print("Usage: python generate_md.py <path_to_toml_file> <risc0-zkvm-version> <risc0-ethereum-version>")
sys.exit(1)
toml_file = sys.argv[1]
risc0_version = sys.argv[2]
risc0_eth_version = sys.argv[3]
main(toml_file, risc0_version, risc0_eth_version)