launchr 1.1.0

data/ext/extconf.rb

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require 'mkmf'
+create_makefile("launchd-socket-listener-unload")
+
+makefile = File.read("Makefile")
+makefile.gsub!(/-bundle ?/,"")
+makefile.gsub!(/\$\(TARGET\)\.bundle/,"$(TARGET)")
+
+File.open("Makefile","w") do |f|
+  f.puts makefile
+end
+
+

data/ext/launchd-socket-listener-unload.cpp

@@ -0,0 +1,160 @@
+// 
+// launchd-socket-listener-unload.cpp
+// http://fazekasmiklos.blogspot.com/2007/04/howto-unload-launchd-deamon-from-itself.html
+// 
+// Instructions by Dreamcat4 (dreamcat4@gmail.com)
+// 
+// To compile:
+// $ g++ launchd-socket-listener-unload.cpp -o launchd-socket-listener-unload
+// 
+// A small program which unloads the launchd job that started it
+// Use with the below configuration to trigger the real job to load
+// 
+// Create a plist file for this daemon called "launchd.myjob.label-loader.plist"
+// Put the listener socket definitions here
+// 
+// 
+// <key>Program</key>
+// <string>/path/to/launchd-socket-listener-unload</string>
+// <key>ServiceIPC</key>
+// <true/>
+// <key>Sockets</key>
+// <dict>
+//     <key>Listeners</key>
+//     <dict>
+//         <!-- Service name (/etc/services), or a tcp port number -->
+//         <!-- or can be anything else launchd supports (udp,etc) -->
+//         <key>SockServiceName</key>
+//         <string>8080</string>
+//     </dict>
+// </dict>
+// 
+// You probably already have made a 2nd plist file called "launchd_myjob_label.plist"
+// Add to this file a KeepAlive directive, which depends on the first job not running
+// 
+// keep alive -> other job enabled -> false
+// 
+// <key>KeepAlive</key>
+// <dict>
+//     <key>OtherJobEnabled</key>
+//     <dict>
+//         <key>launchd.myjob.label-loader</key>
+//         <false/>
+//     </dict>
+// </dict>
+// 
+// 
+// This job will then start when the first job is unloaded, which is
+// whenever the first job is triggered by one of the listener sockets
+// 
+// Bear in mind, 
+// The very first http request / connection attempt will be dropped
+// So a connecting client will have to try again in order to establish
+// a connection to the real job.
+
+#include <stdlib.h>
+#include <errno.h>
+#include <syslog.h>
+
+#include "launch.h"
+
+class LaunchD
+{
+public:
+ LaunchD()
+ {
+  startedWithLaunchD = false;
+  me = 0;
+  CheckIn();
+ }
+
+ ~LaunchD()
+ {
+  if (me) {
+   launch_data_free(me);
+  }
+ }
+
+ bool CheckIn(bool allowRunWithoutLaunchd = true)
+ {
+  launch_data_t msg, resp;
+  msg = launch_data_new_string(LAUNCH_KEY_CHECKIN);
+  if ((resp = launch_msg(msg)) == NULL) {
+   if (allowRunWithoutLaunchd) {
+    startedWithLaunchD = false;
+    return false;
+   }
+   syslog(LOG_ERR,"Checkin with launchd failed: %m");
+   exit(EXIT_FAILURE);
+  }
+  launch_data_free(msg);
+  if (LAUNCH_DATA_ERRNO == launch_data_get_type(resp)) { 
+   errno = launch_data_get_errno(resp); 
+   if (errno == EACCES) { 
+    syslog(LOG_ERR, "Check-in failed. Did you forget to set" 
+        "ServiceIPC == true in your plist?"); 
+   } else { 
+    syslog(LOG_ERR, "Check-in failed: %m"); 
+   } 
+   exit(EXIT_FAILURE); 
+  }
+  launch_data_t tmp = launch_data_dict_lookup(resp, LAUNCH_JOBKEY_LABEL); 
+  me = launch_data_copy(tmp);
+  if(me)
+  {
+      syslog(LOG_ERR, "%s triggered\n",job_label()); 
+  }
+  startedWithLaunchD = true;
+  return true;
+ }
+ 
+ const char* job_label()
+ {
+     return launch_data_get_string(me);
+ }
+
+ bool Stop()
+ {
+  if (startedWithLaunchD) {
+   launch_data_t resp;
+   launch_data_t msg = launch_data_alloc(LAUNCH_DATA_DICTIONARY);
+   if (! launch_data_dict_insert(msg,me,LAUNCH_KEY_REMOVEJOB)) {
+    syslog(LOG_ERR, "launch_data_dict_insert failed!\n"); 
+    return false;
+   }
+   if (! launch_data_dict_insert(msg,me,LAUNCH_KEY_STOPJOB)) {
+    syslog(LOG_ERR, "launch_data_dict_insert failed!\n"); 
+    return false;
+   }
+   syslog(LOG_ERR, "%s unloaded\n", job_label()); 
+   resp = launch_msg(msg);
+   syslog(LOG_ERR, "%s ...not unloaded. Failure when trying to unload %s.\n", job_label(), job_label()); 
+   if (resp == NULL) {
+    syslog(LOG_ERR, "launch_msg() LAUNCH_KEY_STOPJOB failed!\n"); 
+    return false;
+   }
+   if (LAUNCH_DATA_ERRNO == launch_data_get_type(resp)) { 
+    errno = launch_data_get_errno(resp); 
+    if (errno == EACCES) { 
+     syslog(LOG_ERR, "Stop request failed EACCESS!"); 
+    } else { 
+     syslog(LOG_ERR, "Check-in failed: %m"); 
+    } 
+    exit(EXIT_FAILURE);
+   }
+   launch_data_free(msg);
+  }
+  return true;
+ }
+private:
+ bool   startedWithLaunchD;
+ launch_data_t me;
+};
+
+
+int main()
+{
+    LaunchD ld;
+    ld.Stop(); 
+}
+

data/features/launchr.feature

@@ -0,0 +1,9 @@
+Feature: something something
+  In order to something something
+  A user something something
+  something something something
+
+  Scenario: something something
+    Given inspiration
+    When I create a sweet new gem
+    Then everyone should see how awesome I am

data/features/support/env.rb

@@ -0,0 +1,14 @@
+require 'bundler'
+begin
+  Bundler.setup(:runtime, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+
+$LOAD_PATH.unshift(File.dirname(__FILE__) + '/../../lib')
+require 'launchr'
+
+require 'spec/expectations'
+

data/lib/launchr.rb

@@ -0,0 +1,105 @@
+
+dir = File.dirname(__FILE__)
+$LOAD_PATH.unshift dir unless $LOAD_PATH.include?(dir)
+
+require 'launchr/path'
+
+module Launchr
+  class << self
+    def config
+      @config ||= {}
+    end
+
+    def load_default
+      if Launchr::Path.launchr_default_boot.exist?
+        config[:boot] = true
+      else
+        config[:boot] = nil
+      end
+    end
+
+    def import_args
+      if config[:args][:user]
+        config[:boot] = nil
+
+      elsif config[:args][:boot]
+        config[:boot] = true
+      end
+    end
+
+    def label
+      # "com.github.homebrew.launchr"
+      "com.github.launchr"
+    end
+
+    def launchctl_timeout
+      5
+    end
+
+    def real_user
+      Etc.getpwuid(Process.uid).name
+    end
+
+    def real_group
+      Etc.getgrgid(Process.gid).name
+    end
+
+    def superuser?
+      real_user == "root"
+    end
+
+    def sudo_user
+      if ENV["SUDO_USER"]
+        ENV["SUDO_USER"]
+      elsif ENV["SUDO_UID"]
+        Etc.getpwuid(ENV["SUDO_UID"].to_i).name
+      else
+        nil
+      end
+    end
+
+    def sudo_group
+      if ENV["SUDO_GROUP"]
+        ENV["SUDO_GROUP"]
+      elsif ENV["SUDO_GID"]
+        Etc.getgrgid(ENV["SUDO_GID"].to_i).name
+      else
+        nil
+      end
+    end
+
+    def user
+      if superuser?
+        sudo_user || real_user
+      else
+        real_user
+      end
+    end
+
+    def group
+      if superuser?
+        sudo_group || real_group
+      else
+        real_group
+      end
+    end
+
+    def uid
+      Etc.getpwnam(user).uid
+    end
+
+    def gid
+      Etc.getgrnam(group).gid
+    end
+
+    def version
+      if Launchr::Path.launchr_version.exist?
+        Launchr::Path.launchr_version.read.strip
+      else
+        "0.0.0"
+      end
+    end
+  end
+end
+
+

data/lib/launchr/application.rb

@@ -0,0 +1,24 @@
+
+require 'launchr/cli'
+require 'launchr/commands'
+
+module Launchr
+  # The Launchr Application Object. Instantiated for command-line mode
+  # @see Launchr::CLI
+  class Application
+
+    def initialize *args, &blk
+      @cli = Launchr::CLI.new
+
+      Launchr.load_default
+
+      Launchr.config[:args] = @cli.parse
+      Launchr.import_args
+
+      @commands = Launchr::Commands.new
+      @commands.run
+    end
+  end
+end
+
+

data/lib/launchr/cli.rb

@@ -0,0 +1,142 @@
+
+require 'launchr/mixin/mixlib_cli'
+
+module Launchr
+  # Defines options for the +launchr+ command line utility
+  class CLI
+    include Launchr::Mixlib::CLI
+
+    # The Launchr CLI Options
+    def self.launchr_cli_options
+
+      spaced_summary true
+      # banner false
+      summary_indent ""
+      summary_width 29
+
+      header "brew launchd - an extension to start and stop Launchd services."
+      header "              `man brew-launchd` for more information"
+      header ""
+      banner "Usage: brew launchd [options]"
+
+      argument :start,
+        :long  => "start service,(s)", 
+        :type => Array,
+        :description => ["Start launchd service(s)",
+          "Equivalent to launchctl load -w files..."],
+        :example => "start dnsmasq memcached couchdb",
+        :default => nil
+
+      argument :stop,
+        :long  => "stop service,(s)", 
+        :type => Array,
+        :description => ["Stop launchd service(s)",
+        "Equivalent to launchctl unload -w files..."],
+        :example => "stop mamcached dnsmasq",
+        :default => nil
+
+      argument :restart,
+        :long  => "restart service,(s)", 
+        :type => Array,
+        :description => ["Restart launchd service(s)"],
+        :example => "restart couchdb",
+        :default => nil
+
+      option :user,
+        :indent => true,
+        :long  => "--user", 
+        :description => ["At user login.",
+          "Otherwise, the default setting will be used."],
+        :example => "start --user openvpn ddclient",
+        :requires => Proc.new { |args| (args[:boot] ^ args[:user]) && true || raise("--boot|--user") },
+        :default => nil
+
+      option :boot,
+        :indent => true,
+        :long  => "--boot", 
+        :description => ["At boot time. Requires sudo/root privelidges.",
+          "Otherwise, the default setting will be used."],
+        :example => "sudo brew start --boot nginx mysql",
+        :requires => Proc.new { |args| (args[:boot] ^ args[:user]) && true || raise("--boot|--user") },
+        :default => nil
+
+
+      argument :info,
+        :long  => "info [service,(s)]", 
+        :type => Array,
+        :proc => Proc.new { |l| (l == true) ? [] : l },
+        :description => ["Info for launchd service(s)","With no arguments prints info for all services."],
+        :example => "brew launchd info",
+        :default => nil
+
+      argument :clean,
+        :long  => "clean", 
+        :description => ["Clean missing/broken launchd service(s)."],
+        :example => ["brew launchd clean", "sudo brew launchd clean"],
+        :default => nil
+
+      argument :default,
+        :long  => "default [--user|--boot]",
+        :description => [
+          "Set the default target to start launchd services.",
+          "The initial setting, --user will start daemons at",
+          "user login - from the Loginwindow (not over ssh).",
+          " ",
+          "Wheras --boot will set services to start at boot",
+          "time. But be aware that brew should be installed",
+          "to the root filesystem, not on a mounted volume."],
+
+        :example => [
+          "brew launchd default --boot",
+          "brew launchd default --user"
+          ],
+        :requires => Proc.new { |args| (args[:boot] ^ args[:user]) && true || raise("--boot|--user") },
+        :default => nil
+
+      option :help, 
+        :long => "--help",
+        :description => "Show this message",
+        :show_options => true,
+        :exit => 0
+
+      option :version, 
+        :long => "--version",
+        :description => "Print version information",
+        :default => nil
+    end
+    launchr_cli_options
+
+    def parse argv=ARGV
+      parse_options(argv)
+
+      [:start,:stop,:restart].each do |cmd|
+        case config[cmd]
+        when Array
+          [:user, :boot].each do |level|
+            if config[cmd].include? "--#{level}"
+              config[level] = true
+            end
+          end
+          config[cmd] -= ["--user","--boot"]
+        end
+      end
+
+      raise "Please choose one of --user|--boot" if config[:user] && config[:boot]
+
+      unless filtered_argv.empty?
+        start_stop_restart_value = [config[:start],config[:stop],config[:restart],config[:info]].compact!
+
+        if start_stop_restart_value.size == 1
+          services = *start_stop_restart_value
+          extra_services = filtered_argv
+
+          services << extra_services
+          services.flatten!
+        end
+      end
+
+      config
+    end
+
+  end
+end