Merge 5c2911e028
into 454c0b2e7e
This commit is contained in:
commit
4459f2aaa4
|
@ -0,0 +1,43 @@
|
|||
import os, signal, json, requests, subprocess, time
|
||||
from pathlib import Path
|
||||
from configparser import ConfigParser
|
||||
import psutil
|
||||
|
||||
|
||||
def is_namecoin_installed():
|
||||
home = str(Path.home())
|
||||
dot_namecoin = os.path.join(home, '.namecoin')
|
||||
is_dot_namecoin = os.path.exists(dot_namecoin)
|
||||
return is_dot_namecoin
|
||||
|
||||
|
||||
def is_namecoin_running():
|
||||
for proc in psutil.process_iter(['pid', 'name', 'username']):
|
||||
if "namecoin" in proc.name():
|
||||
return proc
|
||||
|
||||
|
||||
def get_namecoin_conf():
|
||||
namecoin_conf = os.path.join(str(Path.home()), '.namecoin', 'namecoin.conf')
|
||||
parser = ConfigParser()
|
||||
with open(namecoin_conf) as stream:
|
||||
parser.read_string("[fake-section]\n" + stream.read())
|
||||
config = dict(parser.items('fake-section'))
|
||||
return config
|
||||
|
||||
|
||||
def check_namecoin_rpc_conf(config):
|
||||
if None in (config['rpcconnect'], config['rpcport'], config['rpcuser'], config['rpcpassword']):
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
|
||||
|
||||
def start_namecoin(path):
|
||||
process = subprocess.Popen(["{}namecoind".format(path)])
|
||||
return process.pid
|
||||
|
||||
|
||||
def kill_namecoin(pid):
|
||||
os.kill(pid, signal.SIGTERM)
|
||||
return True
|
|
@ -4,23 +4,57 @@ from Config import config
|
|||
from Debug import Debug
|
||||
from http.client import HTTPSConnection, HTTPConnection, HTTPException
|
||||
from base64 import b64encode
|
||||
from pathlib import Path
|
||||
from .NamecoinUtils import is_namecoin_installed, is_namecoin_running, get_namecoin_conf, check_namecoin_rpc_conf, start_namecoin, kill_namecoin
|
||||
|
||||
allow_reload = False # No reload supported
|
||||
|
||||
@PluginManager.registerTo("SiteManager")
|
||||
class SiteManagerPlugin(object):
|
||||
def load(self, *args, **kwargs):
|
||||
super(SiteManagerPlugin, self).load(*args, **kwargs)
|
||||
self.log = logging.getLogger("ZeronetLocal Plugin")
|
||||
self.error_message = None
|
||||
if not config.namecoin_host or not config.namecoin_rpcport or not config.namecoin_rpcuser or not config.namecoin_rpcpassword:
|
||||
self.error_message = "Missing parameters"
|
||||
self.log.error("Missing parameters to connect to namecoin node. Please check all the arguments needed with '--help'. Zeronet will continue working without it.")
|
||||
def __del__(self):
|
||||
if self.namecoin_pid:
|
||||
kill_namecoin(self.namecoin_pid)
|
||||
self.log.debug("Kill namecoin pid : {}".format(self.namecoin_pid))
|
||||
return
|
||||
|
||||
url = "%(host)s:%(port)s" % {"host": config.namecoin_host, "port": config.namecoin_rpcport}
|
||||
def load(self, *args, **kwargs):
|
||||
super(SiteManagerPlugin, self).load(*args, **kwargs)
|
||||
self.log = logging.getLogger("ZeronameLocal Plugin")
|
||||
self.error_message = None
|
||||
self.namecoin_pid = None
|
||||
self.config = dict({
|
||||
'rpcconnect': config.namecoin_host,
|
||||
'rpcport': config.namecoin_rpcport,
|
||||
'rpcuser': config.namecoin_rpcuser,
|
||||
'rpcpassword': config.namecoin_rpcpassword,
|
||||
'path': config.namecoin_path
|
||||
})
|
||||
self.log.debug("Namecoin config through CLI ? {}".format(self.config))
|
||||
self.log.debug("Namecoin installed ? {}".format(is_namecoin_installed()))
|
||||
if sys.platform.startswith('linux') and is_namecoin_installed():
|
||||
self.log.debug("Namecoin running ? {}".format(is_namecoin_running()))
|
||||
if is_namecoin_running() is not None:
|
||||
if check_namecoin_rpc_conf(self.config) is not True:
|
||||
self.config = get_namecoin_conf()
|
||||
if check_namecoin_rpc_conf(self.config) is not True:
|
||||
self.error_message = "Missing parameters"
|
||||
self.log.error("Missing parameters to connect to namecoin node. Please check all the arguments needed with '--help' or provide them in the namecoin.conf. Zeronet will continue working without it.")
|
||||
return
|
||||
else:
|
||||
if config.namecoin_path is not None:
|
||||
self.namecoin_pid = start_namecoin(self.config['path'])
|
||||
self.log.debug("Namecoin starting ? {}".format(str(self.namecoin_pid)))
|
||||
else:
|
||||
self.log.error("Please use --namecoin_path to specify namecoind binary path")
|
||||
|
||||
if check_namecoin_rpc_conf(self.config) is not True:
|
||||
self.error_message = "Missing parameters"
|
||||
self.log.error("Namecoin is not installed or parameters are missing to connect to namecoin node. Please check all the arguments needed with '--help'. Zeronet will continue working without it.")
|
||||
return
|
||||
|
||||
url = "%(host)s:%(port)s" % {"host": self.config['rpcconnect'], "port": self.config['rpcport']}
|
||||
self.c = HTTPConnection(url, timeout=3)
|
||||
user_pass = "%(user)s:%(password)s" % {"user": config.namecoin_rpcuser, "password": config.namecoin_rpcpassword}
|
||||
user_pass = "%(user)s:%(password)s" % {"user": self.config['rpcuser'], "password": self.config['rpcpassword']}
|
||||
userAndPass = b64encode(bytes(user_pass, "utf-8")).decode("ascii")
|
||||
self.headers = {"Authorization" : "Basic %s" % userAndPass, "Content-Type": " application/json " }
|
||||
|
||||
|
@ -176,5 +210,6 @@ class ConfigPlugin(object):
|
|||
group.add_argument('--namecoin_rpcport', help="Port to connect (eg. 8336)")
|
||||
group.add_argument('--namecoin_rpcuser', help="RPC user to connect to the namecoin node (eg. nofish)")
|
||||
group.add_argument('--namecoin_rpcpassword', help="RPC password to connect to namecoin node")
|
||||
group.add_argument('--namecoin_path', help="Path to namecoind binary")
|
||||
|
||||
return super(ConfigPlugin, self).createArguments()
|
||||
|
|
Loading…
Reference in New Issue