Rakefile in marionette-0.0.1 vs Rakefile in marionette-0.0.2

@@ -1,2 +1,99 @@
 require 'bundler'
 Bundler::GemHelper.install_tasks
+
+require 'rake'
+
+namespace :marionette do
+
+  # task description
+  desc 'Set up marionette as a service (run this as sudo/root)'
+  
+  # define task:
+  # 1) write to init.d/marionette
+  # 2) set permissions
+  # 3) set ifconfig
+  # 4) start service
+  task :service do
+
+    script = <<CODE
+#!/bin/bash
+
+### BEGIN INIT INFO
+# Provides: marionette
+# Default-Start:  2 3 4 5
+# Default-Stop: 0 1 6
+# Short-Description: start and stop marionette
+# Description: 0mq connection for puppet and master.
+# chkconfig:   - 85 15 
+### END INIT INFO
+
+# source function library
+. /etc/rc.d/init.d/functions
+
+RETVAL=0
+prog="marionette"
+
+set -e
+
+PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
+DESC="marionette daemon"
+NAME=marionette
+DAEMON=/usr/local/bin/$NAME
+PIDFILE=/var/run/$NAME.pid
+LOGFILE=/var/log/$NAME.log
+SCRIPTNAME=/etc/init.d/$NAME
+IP=`ifconfig eth0 | grep 'inet addr:' | cut -d: -f2 | awk '{ print $1}'`
+TCP=tcp://$IP:5555
+
+# Gracefully exit if the package has been removed.
+# test -x $DAEMON || exit 0
+
+d_start() {
+  $DAEMON start $TCP || echo -en "\n already running"
+}
+
+d_stop() {
+  kill -9 `cat $PIDFILE` || echo -en "\n not running"
+}
+
+case "$1" in
+  start)
+    echo -n "Starting $DESC: $NAME"
+    d_start
+        echo "."
+  ;;
+  stop)
+    echo -n "Stopping $DESC: $NAME"
+    d_stop
+        echo "."
+  ;;
+  restart)
+    echo -n "Restarting $DESC: $NAME"
+    d_stop
+    sleep 5
+    d_start
+    echo "."
+  ;;
+  *)
+    echo "Usage: $SCRIPTNAME {start|stop|restart}" >&2
+    exit 3
+  ;;
+esac
+
+exit 0
+
+CODE
+    file = File.open('/etc/init.d/marionette','w')
+    file.write script
+    file.close
+    
+    system "chmod 755 /etc/init.d/marionette"
+    system "chkconfig marionette on"
+    system "service marionette start"
+  end
+  
+  # set "service" as the default task
+  task :default => [:service]
+
+end
+