From ec0dacb0421f736e311d501a8114de00e5ed6128 Mon Sep 17 00:00:00 2001
From: Joshua Boniface <joshua@boniface.me>
Date: Wed, 6 Jun 2018 01:20:09 -0400
Subject: [PATCH] Make initialization function nicer

---
 pvc.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/pvc.py b/pvc.py
index 8642c56f..13192121 100755
--- a/pvc.py
+++ b/pvc.py
@@ -478,16 +478,26 @@ def search(dom_name, dom_uuid, long_output):
 # pvc init
 ###############################################################################
 @click.command(name='init', short_help='Initialize a new cluster')
+@click.option('--yes', is_flag=True,
+              expose_value=False,
+              prompt='DANGER: This command will destroy any existing cluster data. Do you want to continue?')
 def init_cluster():
     """
     Perform initialization of Zookeeper to act as a PVC cluster
     """
 
-    click.echo('{}'.format(zk_host))
+    click.echo('Initializing a new cluster with Zookeeper address "{}".'.format(zk_host))
 
     # Open a Zookeeper connection
     zk = pvcf.startZKConnection(zk_host)
 
+    # Destroy the existing data
+    try:
+        zk.delete('/domains', recursive=True)
+        zk.delete('nodes', recursive=True)
+    except:
+        pass
+
     # Create the root keys
     transaction = zk.transaction()
     transaction.create('/domains', ''.encode('ascii'))
@@ -511,6 +521,7 @@ def cli(_zk_host):
     Parallel Virtual Cluster CLI management tool
     """
 
+    global zk_host
     zk_host = _zk_host
 
 
@@ -519,7 +530,6 @@ def cli(_zk_host):
 #
 node.add_command(flush_host)
 node.add_command(ready_host)
-#node.add_command(get_details)
 
 vm.add_command(define_vm)
 vm.add_command(start_vm)
@@ -527,11 +537,11 @@ vm.add_command(shutdown_vm)
 vm.add_command(stop_vm)
 vm.add_command(migrate_vm)
 vm.add_command(unmigrate_vm)
-#vm.add_command(get_details)
 
 cli.add_command(node)
 cli.add_command(vm)
 cli.add_command(search)
+cli.add_command(init_cluster)
 
 #
 # Main entry point