From be6d77aceba6b2be2f99b2e8687041cb818b41a9 Mon Sep 17 00:00:00 2001
From: TheNain38 <TheNain38@users.noreply.github.com>
Date: Tue, 22 Mar 2016 17:12:22 +0100
Subject: [PATCH] Add arg to change resolving site

---
 plugins/Zeroname/SiteManagerPlugin.py | 8 ++++----
 plugins/Zeroname/UiRequestPlugin.py   | 9 +++++++++
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/plugins/Zeroname/SiteManagerPlugin.py b/plugins/Zeroname/SiteManagerPlugin.py
index d2b82c37..f6d98725 100644
--- a/plugins/Zeroname/SiteManagerPlugin.py
+++ b/plugins/Zeroname/SiteManagerPlugin.py
@@ -1,6 +1,7 @@
 import logging
 import re
 
+from Config import config
 from Plugin import PluginManager
 
 allow_reload = False  # No reload supported
@@ -10,13 +11,12 @@ log = logging.getLogger("ZeronamePlugin")
 
 @PluginManager.registerTo("SiteManager")
 class SiteManagerPlugin(object):
-    zeroname_address = "1Name2NXVi1RDPDgf5617UoW7xA6YrhM9F"
     site_zeroname = None
 
     def load(self):
         super(SiteManagerPlugin, self).load()
-        if not self.get(self.zeroname_address):
-            self.need(self.zeroname_address)  # Need ZeroName site
+        if not self.get(config.bit_resolver):
+            self.need(config.bit_resolver)  # Need ZeroName site
 
     # Checks if its a valid address
     def isAddress(self, address):
@@ -34,7 +34,7 @@ class SiteManagerPlugin(object):
     def resolveDomain(self, domain):
         domain = domain.lower()
         if not self.site_zeroname:
-            self.site_zeroname = self.need(self.zeroname_address)
+            self.site_zeroname = self.need(config.bit_resolver)
         self.site_zeroname.needFile("data/names.json", priority=10)
         db = self.site_zeroname.storage.loadJson("data/names.json")
         return db.get(domain)
diff --git a/plugins/Zeroname/UiRequestPlugin.py b/plugins/Zeroname/UiRequestPlugin.py
index d080a312..513b8e1b 100644
--- a/plugins/Zeroname/UiRequestPlugin.py
+++ b/plugins/Zeroname/UiRequestPlugin.py
@@ -1,4 +1,5 @@
 import re
+
 from Plugin import PluginManager
 
 
@@ -40,3 +41,11 @@ class UiRequestPlugin(object):
             return True
         else:  # Invalid referer
             return False
+
+@PluginManager.registerTo("ConfigPlugin")
+class ConfigPlugin(object):
+    def createArguments(self):
+        group = self.parser.add_argument_group("Zeroname plugin")
+        group.add_argument('--bit_resolver', help='ZeroNet site to resolve .bit domains', default="1Name2NXVi1RDPDgf5617UoW7xA6YrhM9F", metavar="address")
+
+        return super(ConfigPlugin, self).createArguments()