From aec1d1eecc9c2f9da2c10e4639135d89c3a6b774 Mon Sep 17 00:00:00 2001
From: Brian Warner <warner@lothar.com>
Date: Tue, 5 Dec 2006 12:25:23 -0700
Subject: [PATCH] add bin/allmydata to create/stop/start nodes

---
 allmydata/scripts/__init__.py |   0
 allmydata/scripts/runner.py   | 160 ++++++++++++++++++++++++++++++++++
 bin/allmydata                 |   4 +
 setup.py                      |   3 +-
 4 files changed, 166 insertions(+), 1 deletion(-)
 create mode 100644 allmydata/scripts/__init__.py
 create mode 100644 allmydata/scripts/runner.py
 create mode 100644 bin/allmydata

diff --git a/allmydata/scripts/__init__.py b/allmydata/scripts/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/allmydata/scripts/runner.py b/allmydata/scripts/runner.py
new file mode 100644
index 00000000..ceea9c51
--- /dev/null
+++ b/allmydata/scripts/runner.py
@@ -0,0 +1,160 @@
+#! /usr/bin/python
+
+import os, sys, signal, time
+from twisted.python import usage
+
+class StartOptions(usage.Options):
+    optParameters = [
+        ["basedir", "C", ".", "which directory to start the node in"],
+        ]
+
+class StopOptions(usage.Options):
+    optParameters = [
+        ["basedir", "C", ".", "which directory to stop the node in"],
+        ]
+
+class RestartOptions(usage.Options):
+    optParameters = [
+        ["basedir", "C", ".", "which directory to restart the node in"],
+        ]
+
+class CreateClientOptions(usage.Options):
+    optParameters = [
+        ["basedir", "C", ".", "which directory to create the client in"],
+        ]
+class CreateQueenOptions(usage.Options):
+    optParameters = [
+        ["basedir", "C", ".", "which directory to create the queen in"],
+        ]
+
+client_tac = """
+# -*- python -*-
+
+from allmydata import client
+from twisted.application import service
+
+c = client.Client()
+
+application = service.Application("allmydata_client")
+c.setServiceParent(application)
+"""
+
+queen_tac = """
+# -*- python -*-
+
+from allmydata import queen
+from twisted.application import service
+
+c = queen.Queen()
+
+application = service.Application("allmydata_queen")
+c.setServiceParent(application)
+"""
+
+class Options(usage.Options):
+    synopsis = "Usage:  allmydata <command> [command options]"
+
+    subcommands = [
+        ["create-client", None, CreateClientOptions],
+        ["create-queen", None, CreateQueenOptions],
+        ["start", None, StartOptions],
+        ["stop", None, StopOptions],
+        ["restart", None, RestartOptions],
+        ]
+
+    def postOptions(self):
+        if not hasattr(self, 'subOptions'):
+            raise usage.UsageError("must specify a command")
+
+def run():
+    config = Options()
+    try:
+        config.parseOptions()
+    except usage.error, e:
+        print "%s:  %s" % (sys.argv[0], e)
+        print
+        c = getattr(config, 'subOptions', config)
+        print str(c)
+        sys.exit(1)
+
+    command = config.subCommand
+    so = config.subOptions
+
+    if command == "create-client":
+        rc = create_client(so)
+    elif command == "create-queen":
+        rc = create_queen(so)
+    elif command == "start":
+        rc = start(so)
+    elif command == "stop":
+        rc = stop(so)
+    elif command == "restart":
+        rc = restart(so)
+    rc = rc or 0
+    sys.exit(rc)
+
+def create_client(config):
+    basedir = config['basedir']
+    os.mkdir(basedir)
+    f = open(os.path.join(basedir, "client.tac"), "w")
+    f.write(client_tac)
+    f.close()
+    print "client created, please copy roster_pburl into the directory"
+
+def create_queen(config):
+    basedir = config['basedir']
+    os.mkdir(basedir)
+    f = open(os.path.join(basedir, "queen.tac"), "w")
+    f.write(queen_tac)
+    f.close()
+    print "queen created"
+
+def start(config):
+    basedir = config['basedir']
+    if os.path.exists(os.path.join(basedir, "client.tac")):
+        tac = "client.tac"
+        type = "client"
+    elif os.path.exists(os.path.join(basedir, "queen.tac")):
+        tac = "queen.tac"
+        type = "queen"
+    else:
+        print "%s does not look like a node directory" % basedir
+        sys.exit(1)
+    os.chdir(basedir)
+    rc = os.system("twistd -y %s" % tac)
+    if rc == 0:
+        print "node probably started"
+    else:
+        print "node probably not started"
+    return 1
+
+def stop(config):
+    basedir = config['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.TERM)
+    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
+
+def restart(config):
+    rc = stop(config)
+    if rc:
+        print "not restarting"
+        return rc
+    return start(config)
diff --git a/bin/allmydata b/bin/allmydata
new file mode 100644
index 00000000..57d4ebc6
--- /dev/null
+++ b/bin/allmydata
@@ -0,0 +1,4 @@
+#!/usr/bin/python
+
+from allmydata.scripts import runner
+runner.run()
diff --git a/setup.py b/setup.py
index c6e55d27..bb384b24 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,8 @@ setup(
     name="AllMyData",
     version="0.0.1",
     #packages=find_packages('.'),
-    packages=["allmydata", "allmydata/test", "allmydata/util"],
+    packages=["allmydata", "allmydata/test", "allmydata/util", "allmydata/scripts"],
+    scripts = ["bin/allmydata"],
     package_data={ 'allmydata': ['web/*.xhtml'] },
     description="AllMyData (tahoe2)",
     )
-- 
2.45.2