diff --git a/public_info.py b/public_info.py
index 54012da..61412f3 100644
--- a/public_info.py
+++ b/public_info.py
@@ -1,6 +1,7 @@
import json
import account
import requests
+import os
# Plugin Data
info = {
@@ -27,9 +28,17 @@ functions = {
}
}
-def main(params, authentication):
- info = account.hsd.getInfo()
+if not os.path.exists("user_data/public_info.json"):
+ with open("user_data/public_info.json", "w") as f:
+ json.dump({"enabled": False}, f)
+def main(params, authentication):
+ with open("user_data/public_info.json", "r") as f:
+ multiwallet = json.load(f)
+ if not multiwallet['enabled']:
+ return {"status": None}
+
+ info = account.hsd.getInfo()
status = f"Version: {info['version']}
Inbound Connections: {info['pool']['inbound']}
Outbound Connections: {info['pool']['outbound']}
"
if info['pool']['public']['listen']:
status += f"Public Node: Yes
Host: {info['pool']['public']['host']}
Port: {info['pool']['public']['port']}
"
@@ -38,4 +47,19 @@ def main(params, authentication):
status += f"Agent: {info['pool']['agent']}
Services: {info['pool']['services']}
"
return {"status": status}
-
\ No newline at end of file
+
+def enable(params, authentication):
+ with open("user_data/public_info.json", "r") as f:
+ multiwallet = json.load(f)
+ multiwallet['enabled'] = True
+ with open("user_data/public_info.json", "w") as f:
+ json.dump(multiwallet, f)
+ return {"result": "Success"}
+
+def disable(params, authentication):
+ with open("user_data/public_info.json", "r") as f:
+ multiwallet = json.load(f)
+ multiwallet['enabled'] = False
+ with open("user_data/public_info.json", "w") as f:
+ json.dump(multiwallet, f)
+ return {"result": "Success"}
\ No newline at end of file