mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-29 01:08:54 +00:00
278 lines
8.3 KiB
Python
278 lines
8.3 KiB
Python
#! /usr/bin/env python
|
|
|
|
import os, subprocess, sys, signal, time
|
|
from twisted.python import usage
|
|
|
|
from twisted.python.procutils import which
|
|
|
|
def testtwistd(loc):
|
|
try:
|
|
return subprocess.call(["python", loc,], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
except:
|
|
return -1
|
|
|
|
twistd = None
|
|
if not twistd:
|
|
for maybetwistd in which("twistd"):
|
|
ret = testtwistd(maybetwistd)
|
|
if ret == 0:
|
|
twistd = maybetwistd
|
|
break
|
|
|
|
if not twistd:
|
|
for maybetwistd in which("twistd.py"):
|
|
ret = testtwistd(maybetwistd)
|
|
if ret == 0:
|
|
twistd = maybetwistd
|
|
break
|
|
|
|
if not twistd:
|
|
maybetwistd = os.path.join(sys.prefix, 'Scripts', 'twistd')
|
|
ret = testtwistd(maybetwistd)
|
|
if ret == 0:
|
|
twistd = maybetwistd
|
|
|
|
if not twistd:
|
|
maybetwistd = os.path.join(sys.prefix, 'Scripts', 'twistd.py')
|
|
ret = testtwistd(maybetwistd)
|
|
if ret == 0:
|
|
twistd = maybetwistd
|
|
|
|
if not twistd:
|
|
print "Can't find twistd (it comes with Twisted). Aborting."
|
|
sys.exit(1)
|
|
|
|
class BasedirMixin:
|
|
optFlags = [
|
|
["multiple", "m", "allow multiple basedirs to be specified at once"],
|
|
]
|
|
|
|
def postOptions(self):
|
|
if not self.basedirs:
|
|
raise usage.UsageError("<basedir> parameter is required")
|
|
if self['basedir']:
|
|
del self['basedir']
|
|
self['basedirs'] = [os.path.abspath(os.path.expanduser(b))
|
|
for b in self.basedirs]
|
|
|
|
def parseArgs(self, *args):
|
|
self.basedirs = []
|
|
if self['basedir']:
|
|
self.basedirs.append(self['basedir'])
|
|
if self['multiple']:
|
|
self.basedirs.extend(args)
|
|
else:
|
|
if len(args) == 0 and not self.basedirs:
|
|
self.basedirs.append(".")
|
|
if len(args) > 0:
|
|
self.basedirs.append(args[0])
|
|
if len(args) > 1:
|
|
raise usage.UsageError("I wasn't expecting so many arguments")
|
|
|
|
class NoDefaultBasedirMixin(BasedirMixin):
|
|
def parseArgs(self, *args):
|
|
# create-client won't default to --basedir=.
|
|
self.basedirs = []
|
|
if self['basedir']:
|
|
self.basedirs.append(self['basedir'])
|
|
if self['multiple']:
|
|
self.basedirs.extend(args)
|
|
else:
|
|
if len(args) > 0:
|
|
self.basedirs.append(args[0])
|
|
if len(args) > 1:
|
|
raise usage.UsageError("I wasn't expecting so many arguments")
|
|
if not self.basedirs:
|
|
raise usage.UsageError("--basedir must be provided")
|
|
|
|
class StartOptions(BasedirMixin, usage.Options):
|
|
optParameters = [
|
|
["basedir", "C", None, "which directory to start the node in"],
|
|
]
|
|
|
|
class StopOptions(BasedirMixin, usage.Options):
|
|
optParameters = [
|
|
["basedir", "C", None, "which directory to stop the node in"],
|
|
]
|
|
|
|
class RestartOptions(BasedirMixin, usage.Options):
|
|
optParameters = [
|
|
["basedir", "C", None, "which directory to restart the node in"],
|
|
]
|
|
|
|
class CreateClientOptions(NoDefaultBasedirMixin, usage.Options):
|
|
optParameters = [
|
|
["basedir", "C", None, "which directory to create the client in"],
|
|
]
|
|
optFlags = [
|
|
["quiet", "q", "operate silently"],
|
|
]
|
|
|
|
class CreateIntroducerOptions(NoDefaultBasedirMixin, usage.Options):
|
|
optParameters = [
|
|
["basedir", "C", None, "which directory to create the introducer in"],
|
|
]
|
|
optFlags = [
|
|
["quiet", "q", "operate silently"],
|
|
]
|
|
|
|
client_tac = """
|
|
# -*- python -*-
|
|
|
|
from allmydata import client
|
|
from twisted.application import service
|
|
|
|
c = client.Client()
|
|
|
|
application = service.Application("allmydata_client")
|
|
c.setServiceParent(application)
|
|
"""
|
|
|
|
introducer_tac = """
|
|
# -*- python -*-
|
|
|
|
from allmydata import introducer_and_vdrive
|
|
from twisted.application import service
|
|
|
|
c = introducer_and_vdrive.IntroducerAndVdrive()
|
|
|
|
application = service.Application("allmydata_introducer")
|
|
c.setServiceParent(application)
|
|
"""
|
|
|
|
class Options(usage.Options):
|
|
synopsis = "Usage: allmydata <command> [command options]"
|
|
|
|
subCommands = [
|
|
["create-client", None, CreateClientOptions, "Create a client node."],
|
|
["create-introducer", None, CreateIntroducerOptions, "Create a introducer node."],
|
|
["start", None, StartOptions, "Start a node (of any type)."],
|
|
["stop", None, StopOptions, "Stop a node."],
|
|
["restart", None, RestartOptions, "Restart a node."],
|
|
]
|
|
|
|
def postOptions(self):
|
|
if not hasattr(self, 'subOptions'):
|
|
raise usage.UsageError("must specify a command")
|
|
|
|
def runner(argv, run_by_human=True):
|
|
config = Options()
|
|
try:
|
|
config.parseOptions(argv)
|
|
except usage.error, e:
|
|
if not run_by_human:
|
|
raise
|
|
print "%s: %s" % (sys.argv[0], e)
|
|
print
|
|
c = getattr(config, 'subOptions', config)
|
|
print str(c)
|
|
return 1
|
|
|
|
command = config.subCommand
|
|
so = config.subOptions
|
|
|
|
rc = 0
|
|
if command == "create-client":
|
|
for basedir in so.basedirs:
|
|
rc = create_client(basedir, so) or rc
|
|
elif command == "create-introducer":
|
|
for basedir in so.basedirs:
|
|
rc = create_introducer(basedir, so) or rc
|
|
elif command == "start":
|
|
for basedir in so.basedirs:
|
|
rc = start(basedir, so) or rc
|
|
elif command == "stop":
|
|
for basedir in so.basedirs:
|
|
rc = stop(basedir, so) or rc
|
|
elif command == "restart":
|
|
for basedir in so.basedirs:
|
|
rc = stop(basedir, so) or rc
|
|
if rc:
|
|
print "not restarting"
|
|
return rc
|
|
for basedir in so.basedirs:
|
|
rc = start(basedir, so) or rc
|
|
return rc
|
|
|
|
def run():
|
|
rc = runner(sys.argv[1:])
|
|
sys.exit(rc)
|
|
|
|
def create_client(basedir, config):
|
|
if os.path.exists(basedir):
|
|
if os.listdir(basedir):
|
|
print "The base directory already exists: %s" % basedir
|
|
print "To avoid clobbering anything, I am going to quit now"
|
|
print "Please use a different directory, or delete this one"
|
|
return -1
|
|
# we're willing to use an empty directory
|
|
else:
|
|
os.mkdir(basedir)
|
|
f = open(os.path.join(basedir, "client.tac"), "w")
|
|
f.write(client_tac)
|
|
f.close()
|
|
if not config['quiet']:
|
|
print "client created in %s" % basedir
|
|
print " please copy introducer.furl and vdrive.furl into the directory"
|
|
|
|
def create_introducer(basedir, config):
|
|
if os.path.exists(basedir):
|
|
if os.listdir(basedir):
|
|
print "The base directory already exists: %s" % basedir
|
|
print "To avoid clobbering anything, I am going to quit now"
|
|
print "Please use a different directory, or delete this one"
|
|
return -1
|
|
# we're willing to use an empty directory
|
|
else:
|
|
os.mkdir(basedir)
|
|
f = open(os.path.join(basedir, "introducer.tac"), "w")
|
|
f.write(introducer_tac)
|
|
f.close()
|
|
if not config['quiet']:
|
|
print "introducer created in %s" % basedir
|
|
|
|
def start(basedir, config):
|
|
print "STARTING", basedir
|
|
if os.path.exists(os.path.join(basedir, "client.tac")):
|
|
tac = "client.tac"
|
|
type = "client"
|
|
elif os.path.exists(os.path.join(basedir, "introducer.tac")):
|
|
tac = "introducer.tac"
|
|
type = "introducer"
|
|
else:
|
|
print "%s does not look like a node directory" % basedir
|
|
if not os.path.isdir(basedir):
|
|
print " in fact, it doesn't look like a directory at all!"
|
|
sys.exit(1)
|
|
rc = subprocess.call(["python", twistd, "-y", tac,], cwd=basedir)
|
|
if rc == 0:
|
|
print "%s node probably started" % type
|
|
return 0
|
|
else:
|
|
print "%s node probably not started" % type
|
|
return 1
|
|
|
|
def stop(basedir, config):
|
|
print "STOPPING", basedir
|
|
pidfile = os.path.join(basedir, "twistd.pid")
|
|
if not os.path.exists(pidfile):
|
|
print "%s does not look like a running node directory (no twistd.pid)" % basedir
|
|
return 1
|
|
pid = open(pidfile, "r").read()
|
|
pid = int(pid)
|
|
|
|
timer = 0
|
|
os.kill(pid, signal.SIGTERM)
|
|
time.sleep(0.1)
|
|
while timer < 5:
|
|
# poll once per second until twistd.pid goes away, up to 5 seconds
|
|
try:
|
|
os.kill(pid, 0)
|
|
except OSError:
|
|
print "process %d is dead" % pid
|
|
return
|
|
timer += 1
|
|
time.sleep(1)
|
|
print "never saw process go away"
|
|
return 1
|