From c3898576e732fd908a757bdc7b705b0f31327abc Mon Sep 17 00:00:00 2001
From: DingZQ <zqding@hotmail.com>
Date: Wed, 10 May 2023 07:11:14 +0800
Subject: [PATCH] Add API for extension data

---
 main.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 07a872e..8026645 100644
--- a/main.py
+++ b/main.py
@@ -34,7 +34,28 @@ app = FastAPI()
 
 app.add_middleware(GZipMiddleware, minimum_size=1000)
 
-### project
+############################################################
+# extension_data
+############################################################
+@app.get('/getallextensiondatakeys/')
+async def fastapi_get_all_extension_data_keys(network: str) -> list[str]:
+    return get_all_extension_data_keys(network)
+
+@app.get('/getallextensiondata/')
+async def fastapi_get_all_extension_data(network: str) -> dict[str, Any]:
+    return get_all_extension_data(network)
+
+@app.get('/getextensiondata/')
+async def fastapi_get_extension_data(network: str, key: str) -> str | None:
+    return get_extension_data(network, key)
+
+@app.post('/setextensiondata')
+async def fastapi_set_extension_data(network: str, cs: ChangeSet) -> ChangeSet:
+    return set_extension_data(network, cs)
+
+############################################################
+# project
+############################################################
 
 @app.get('/listprojects/')
 async def fastapi_list_projects() -> list[str]: