launch-agent 0.4.1 → 0.5.0

data/bin/launchagent

@@ -15,33 +15,17 @@ Load/Unload a launchd agent
 EOS
   opt :env, "additional environmental variables to be set before running the job. can specify multiple times. e.g. RAILS_ENV=development", :type => String, :multi => true
   opt :daemon, "load as daemon. if it is set, --interval option is ignored", :default => false
+  opt :wdir, "specify a directory to chdir(2) to before running the job", :type => String
   opt :interval, "causes the job to be started every N seconds", :type => Integer
 
   run do |command|
-    abort 'full command must be supplised' if command.argv.empty?
-
-    daemon  = command.opts[:daemon]
-    interval = command.opts[:interval]
-    env     = command.opts[:env]
-    agent   = nil
-
-    if daemon
-      agent = LaunchAgent::Daemon.new(*command.argv)
-    elsif command.opts.interval
-      agent = LaunchAgent::Periodic.new(interval, *command.argv)
-    else
-      abort 'at least one of --daemon and --interval must be set'
-    end
-
-    agent['EnvironmentVariables'] = env.inject({}) do |memo, e|
-      k, v = e.split('=')
-      memo[k] = v
-      memo
+    begin
+      agent = LaunchAgent::CLI::OptionParser.new(command.opts, command.argv).agent
+      action = agent.loaded? ? :unload : :load
+      agent.send(action)
+      puts '%s "%s"' % [action, command.argv.join(' ')]
+    rescue => e
+      abort e.message
     end
-
-    action = agent.loaded? ? :unload : :load
-    agent.send(action)
-
-    puts '%s "%s"' % [action, command.argv.join(' ')]
   end
 end

data/lib/launch_agent.rb

@@ -7,3 +7,4 @@ require 'launch_agent/version'
 require 'launch_agent/base'
 require 'launch_agent/daemon'
 require 'launch_agent/periodic'
+require 'launch_agent/cli'

data/lib/launch_agent/cli.rb

@@ -0,0 +1,40 @@
+module LaunchAgent
+  module CLI
+    class OptionParser
+      def initialize(opts, argv)
+        @opts = opts
+        @argv = argv
+      end
+
+      def agent
+        raise 'full command must be supplied' if @argv.empty?
+
+        daemon   = @opts[:daemon]
+        interval = @opts[:interval]
+        env      = @opts[:env]
+        wdir     = @opts[:wdir]
+        agent    = nil
+
+        if daemon
+          agent = LaunchAgent::Daemon.new(*@argv)
+        elsif @opts[:interval]
+          agent = LaunchAgent::Periodic.new(interval, *@argv)
+        else
+          raise 'at least one of --daemon and --interval must be set'
+        end
+
+        agent['EnvironmentVariables'] = env.inject({}) do |memo, e|
+          k, v = e.split('=')
+          memo[k] = v
+          memo
+        end
+
+        if wdir
+          agent['WorkingDirectory'] = File.expand_path(wdir)
+        end
+
+        agent
+      end
+    end
+  end
+end

data/lib/launch_agent/version.rb

@@ -1,4 +1,4 @@
 module LaunchAgent
-  VERSION = '0.4.1'
+  VERSION = '0.5.0'
 end
 

data/spec/cli_spec.rb

@@ -0,0 +1,91 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe 'CLI' do
+  before do
+    @plist_filename = File.expand_path('~/Library/LaunchAgents/com.buycheapviagraonlinenow.ruby__foo_rb.plist')
+  end
+
+  after do
+    `launchctl unload -w #{@plist_filename}` if `launchctl list | grep ruby__foo_rb` =~ /ruby__foo.rb/
+    File.unlink(@plist_filename) if File.exists?(@plist_filename)
+  end
+
+  it 'should load/unload daemon-like agent' do
+    command = File.expand_path(File.dirname(__FILE__) + '/../bin/launchagent --daemon --env FOO=BAR ruby foo.rb')
+
+    system(command)
+
+    File.exists?(@plist_filename).should be_true
+    open(@plist_filename).read.should eql(<<PLIST)
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>EnvironmentVariables</key>
+	<dict>
+		<key>FOO</key>
+		<string>BAR</string>
+	</dict>
+	<key>KeepAlive</key>
+	<dict>
+		<key>SuccessfulExit</key>
+		<false/>
+	</dict>
+	<key>Label</key>
+	<string>com.buycheapviagraonlinenow.ruby__foo_rb</string>
+	<key>ProgramArguments</key>
+	<array>
+		<string>ruby</string>
+		<string>foo.rb</string>
+	</array>
+	<key>RunAtLoad</key>
+	<true/>
+</dict>
+</plist>
+PLIST
+
+    (`launchctl list | grep ruby__foo_rb` =~ /ruby__foo.rb/).should_not be_nil
+
+    system(command)
+
+    File.exists?(@plist_filename).should be_false
+    (`launchctl list | grep ruby__foo_rb` =~ /ruby__foo.rb/).should be_nil
+  end
+
+  it 'should load/unload periodic agent' do
+    command = File.expand_path(File.dirname(__FILE__) + '/../bin/launchagent --interval 120 --env FOO=BAR ruby foo.rb')
+
+    system(command)
+
+    File.exists?(@plist_filename).should be_true
+    open(@plist_filename).read.should eql(<<PLIST)
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>EnvironmentVariables</key>
+	<dict>
+		<key>FOO</key>
+		<string>BAR</string>
+	</dict>
+	<key>Label</key>
+	<string>com.buycheapviagraonlinenow.ruby__foo_rb</string>
+	<key>ProgramArguments</key>
+	<array>
+		<string>ruby</string>
+		<string>foo.rb</string>
+	</array>
+	<key>StartInterval</key>
+	<integer>120</integer>
+</dict>
+</plist>
+PLIST
+
+    (`launchctl list | grep ruby__foo_rb` =~ /ruby__foo.rb/).should_not be_nil
+
+    system(command)
+
+    File.exists?(@plist_filename).should be_false
+    (`launchctl list | grep ruby__foo_rb` =~ /ruby__foo.rb/).should be_nil
+  end
+end

data/spec/option_parser_spec.rb

@@ -0,0 +1,83 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+include LaunchAgent
+
+describe CLI::OptionParser do
+  before do
+    @argv = %w/ruby foo.rb/
+    @opts = {
+      :env => []
+    }
+
+    @agent = CLI::OptionParser.new(opts, @argv).agent
+    @plist = agent2plist(@agent)
+  end
+
+  shared_examples_for 'valid agent' do
+    it "should have necessary attributes" do
+      @plist['Label'].should eql('com.buycheapviagraonlinenow.ruby__foo_rb')
+      @plist['ProgramArguments'].should eql(@argv)
+    end
+  end
+
+  describe 'damon' do
+    let(:opts) do
+      @opts.merge(
+        :daemon => true)
+    end
+
+    it_should_behave_like 'valid agent'
+
+    it 'should parse damon option' do
+      @plist['KeepAlive'].should eql({
+          'SuccessfulExit' => false
+        })
+      @plist['RunAtLoad'].should be_true
+    end
+  end
+
+  describe 'wdir' do
+    let(:opts) do
+      @opts.merge(
+        :daemon => true,
+        :wdir => '/foo/bar')
+    end
+
+    it_should_behave_like 'valid agent'
+
+    it 'should parse wdir option' do
+      @plist['WorkingDirectory'].should eql('/foo/bar')
+    end
+  end
+
+  describe 'env' do
+    let(:opts) do
+      @opts.merge(
+        :daemon => true,
+        :env => ['FOO=BAR','BAR=BAZ'])
+    end
+
+    it_should_behave_like 'valid agent'
+
+    it 'should parse env option' do
+      @plist['EnvironmentVariables'].should eql({ 'FOO' => 'BAR', 'BAR' => 'BAZ' })
+    end
+  end
+
+  describe 'interval' do
+    let(:opts) do
+      @opts.merge(
+        :interval => 120)
+    end
+
+    it_should_behave_like 'valid agent'
+
+    it 'should parse env option' do
+      @plist['StartInterval'].should eql(120)
+    end
+  end
+
+  def agent2plist(agent)
+    Plist.parse_xml(agent.plist_content)
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: launch-agent
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-04-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: plist
-  requirement: &70252751560340 !ruby/object:Gem::Requirement
+  requirement: &70200854706280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70252751560340
+  version_requirements: *70200854706280
 - !ruby/object:Gem::Dependency
   name: git-style-binaries
-  requirement: &70252751559820 !ruby/object:Gem::Requirement
+  requirement: &70200854705860 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70252751559820
+  version_requirements: *70200854705860
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70252751559180 !ruby/object:Gem::Requirement
+  requirement: &70200854705340 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -43,10 +43,10 @@ dependencies:
         version: 2.8.0
   type: :development
   prerelease: false
-  version_requirements: *70252751559180
+  version_requirements: *70200854705340
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &70252751558680 !ruby/object:Gem::Requirement
+  requirement: &70200854704920 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,7 +54,7 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70252751558680
+  version_requirements: *70200854704920
 description: A library to use launchd easily
 email:
 - youpy@buycheapviagraonlinenow.com
@@ -73,10 +73,13 @@ files:
 - launch-agent.gemspec
 - lib/launch_agent.rb
 - lib/launch_agent/base.rb
+- lib/launch_agent/cli.rb
 - lib/launch_agent/daemon.rb
 - lib/launch_agent/periodic.rb
 - lib/launch_agent/version.rb
+- spec/cli_spec.rb
 - spec/daemon_spec.rb
+- spec/option_parser_spec.rb
 - spec/periodic_spec.rb
 - spec/spec_helper.rb
 homepage: ''
@@ -93,7 +96,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 375448672285476255
+      hash: 1175971151883054079
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -102,7 +105,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 375448672285476255
+      hash: 1175971151883054079
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.10
@@ -110,6 +113,8 @@ signing_key:
 specification_version: 3
 summary: A library to use launchd easily
 test_files:
+- spec/cli_spec.rb
 - spec/daemon_spec.rb
+- spec/option_parser_spec.rb
 - spec/periodic_spec.rb
 - spec/spec_helper.rb