angry_mob_common_targets 0.1.0

data/targets/group.rb

@@ -0,0 +1,33 @@
+require 'common_mob'
+require 'etc'
+
+class Group < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  default_action
+  def ensure
+    unless before_state[:exists]
+      create
+    end
+  end
+
+  def create
+    log "creating #{default_object}"
+    sh("groupadd #{default_object}").run
+  end
+
+  protected
+  def state
+    begin
+      group = Etc.getgrnam(default_object)
+      {
+        :exists => true
+      }
+    rescue ArgumentError
+      {
+        :exists => false
+      }
+    end
+  end
+  
+end

data/targets/packages.rb

@@ -0,0 +1,94 @@
+require 'common_mob'
+
+class Apt < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  default_action
+  def install
+    if args.version && !before_state[:version_matches]
+      # TODO - install with version
+      sh("apt-get install -y #{default_object}").run
+    elsif !before_state[:installed]
+      sh("apt-get install -y #{default_object}").run
+    else
+      log "no need to install #{default_object}"
+    end
+  end
+
+  def upgrade
+  end
+
+  def uninstall
+    if before_state[:installed]
+      sh("apt-get remove -y #{default_object}").run
+    end
+  end
+
+  protected
+
+  def state
+    version = sh("apt-cache policy #{default_object}").to_s[/^\s+Installed: (.*)$/, 1]
+
+    {
+      :installed       => (version != '(none)'),
+      :version_matches => (version == args.version),
+      :installable     => !version.nil?
+    }
+  end
+
+  protected
+  def before_call
+    skip! unless before_state[:installable]
+  end
+
+end
+
+class AptSource < AngryMob::Target
+  include CommonMob::ShellHelper
+  default_action
+  def install
+    act.in_sub_act { apt 'python-software-properties' }
+    # TODO make null op less expensive
+    sh("add-apt-repository #{default_object} && apt-get update").run
+  end
+end
+
+class GemPackage < AngryMob::Target
+  nickname 'gem'
+
+  include CommonMob::ShellHelper
+
+  default_action
+  def install
+    gemsh("install #{default_object} #{gem_version}").run unless before_state[:installed]
+  end
+
+  def upgrade
+    gemsh("update #{default_object}").run
+  end
+
+  def uninstall
+  end
+
+  protected
+
+  def state
+    {
+      :installed => installed?
+    }
+  end
+
+  def gem_version
+    args.version.blank? ? '' : " -v '#{args.version}'"
+  end
+
+  # works around bundler being a bit pushy
+  def gemsh(*args)
+    args[0] = "gem #{args[0]}"
+    sh(*args)
+  end
+
+  def installed?
+    gemsh("list -i #{gem_version} #{default_object}").to_s.strip == 'true'
+  end
+end

data/targets/ruby.rb

@@ -0,0 +1,13 @@
+class Block < AngryMob::Target
+  default_action
+  def run
+    log "calling block..."
+    instance_eval &default_object
+  end
+
+  protected
+
+  def default_object
+    args.default_object
+  end
+end

data/targets/services.rb

@@ -0,0 +1,184 @@
+require 'common_mob'
+
+class Service < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  def self.instance_key(args)
+    "service:#{nickname}"
+  end
+
+  default_action
+  def enable
+    unless before_state[:enabled]
+      begin
+        sh("/usr/sbin/update-rc.d #{name} defaults").run
+      rescue CommonMob::ShellError
+        # can't really be expected to enable before init.d exists
+        if ! $!.result.stderr[/file does not exist$/]
+          raise $!
+        end
+      end
+      log "enabled service #{nickname}"
+    end
+  end
+
+  def disable
+    if before_state[:enabled]
+      sh("/usr/sbin/update-rc.d -f #{name} remove").run
+      log "disabled service #{nickname}"
+    end
+  end
+
+  def start
+    initd('start')
+    ensure_running!
+  end
+
+  def stop
+    initd('stop')
+  end
+
+  def restart
+    if is_running?
+      ui.log "restarting."
+      initd('restart', true)
+    else
+      ui.log "restart requested, but not running: starting."
+      initd('start')
+    end
+
+    ensure_running!
+  end
+
+  def reload
+    if is_running?
+      ui.log "reloading."
+      initd('reload', true)
+    else
+      ui.log "reload requested, but not running: starting."
+      initd('start')
+    end
+
+    ensure_running!
+  end
+
+  def be_running
+    start
+  end
+
+
+  def to_s
+    "#{nickname}()"
+  end
+
+  protected
+  def validate!
+  end
+
+  def state
+    enabled = sh("/usr/sbin/update-rc.d -n -f #{name} remove").to_s[%r[/etc/rc\d+.d/]]
+
+    {
+      :enabled => enabled
+    }
+  end
+
+  def name
+    nickname
+  end
+
+  def self.service_name(name)
+    self.class_eval "def name; '#{name}' end"
+  end
+
+  def initd(command,should_raise=false)
+    should_raise = raise_on_failed_initd?
+    begin
+      sh("/etc/init.d/#{name} #{command}").run
+    rescue CommonMob::ShellError
+      if should_raise
+        raise $!
+      else
+        log "/etc/init.d/#{name} #{command} failed (but swallowing exception)"
+        log "(out=#{$!.result.stdout})"
+        log "(err=#{$!.result.stderr})"
+      end
+    end
+  end
+
+
+  def raise_on_failed_initd?
+    false
+  end
+
+  def self.pidfile(pidfile)
+    self.class_eval %{
+      def pidfile; "#{pidfile}" end
+      def process_approach; :pid end
+    }
+  end
+
+  def process_approach
+    :initd
+  end
+
+  def pidfile
+    raise "Not implemented. To use the pid based process approach, please override pidfile in #{self.class}."
+  end
+
+  def ensure_running!
+    case process_approach
+    when :initd
+      ensure_running_with_initd!
+    when :pid
+      ensure_running_with_pid!
+    else
+      raise ArgumentError, "Unknown process_approach '#{process_approach}'\nSet the process approach to :initd or :pid\ndef process_approach\n\t:initd\nend"
+    end
+  end
+
+  def is_running?
+    case process_approach
+    when :initd
+      is_running_initd?
+    when :pid
+      is_running_pid?
+    else
+      raise ArgumentError, "Unknown process_approach '#{process_approach}'\nSet the process approach to :initd or :pid\ndef process_approach\n\t:initd\nend"
+    end
+  end
+
+  def is_running_initd?
+    sh("/etc/init.d/#{name} status").ok?
+  end
+
+  def is_running_pid?
+    pid = pidfile.pathname.read.chomp.to_i
+    Process.kill(0,pid)
+    true
+  rescue Errno::ENOENT,Errno::ESRCH
+    false
+  end
+
+  def ensure_running_with_initd!
+    unless sh("/etc/init.d/#{name} status").ok?
+      raise "#{name} should be running but isn't"
+    end
+    log "#{name} is running"
+  end
+
+  def ensure_running_with_pid!
+    pid = pidfile.pathname.read.chomp.to_i
+    Process.kill(0,pid)
+    true
+  rescue Errno::ENOENT
+    raise "#{name} not running: no pidfile found at #{pidfile}"
+  rescue Errno::ESRCH
+    raise "#{name} not running: no process with pid #{pid} found (pidfile at #{pidfile})"
+  end
+
+  def at_least_lucid?
+    issue = "/etc/issue".pathname
+    issue.exist? && issue.read[/ubuntu.+10\.04/i]
+  end
+end

data/targets/shell.rb

@@ -0,0 +1,43 @@
+require 'common_mob'
+
+class Sh < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  default_action
+  def execute
+    if ! before_state[:created]
+      begin
+        sh(default_object, opts).run
+      rescue CommonMob::ShellError
+        if args.swallow_error?
+          log "command failed, but ignoring"
+        else
+          raise $!
+        end
+      end
+    end
+  end
+
+  protected
+
+  def opts
+    opts = AngryHash[ args ]
+
+    opts.delete_all_of(%w{notify action default_object creates})
+
+    opts
+  end
+
+  def state
+    if args.creates
+      {
+        :created => args.creates.pathname.exist?
+      }
+    else # force state changed if we don't have any state comparison to go on
+      {
+        :rand => rand
+      }
+    end
+
+  end
+end

data/targets/user.rb

@@ -0,0 +1,108 @@
+require 'common_mob'
+require 'etc'
+
+class User < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  default_action
+  def ensure
+    if before_state[:exists] then update else create end
+  end
+
+  def create
+    opt_string = opts
+
+    unless opt_string.blank?
+      log "creating #{default_object}"
+      sh("useradd #{opts} #{default_object}").run
+    end
+  end
+
+  def update
+    opt_string = opts
+
+    unless opt_string.blank?
+      log "updating #{default_object}"
+      sh("usermod #{opts} #{default_object}").run
+    end
+  end
+
+  def delete
+  end
+
+  def lock
+  end
+
+  def unlock
+  end
+
+  protected
+
+  def state
+    begin
+      user = Etc.getpwnam(default_object)
+
+      groups = []
+
+      username = default_object
+      Etc.group do |g|
+        groups << g.name if g.mem.include?(username)
+      end
+
+      extra_groups = groups.uniq - [ Etc.getgrgid(user.gid).name ]
+
+      {
+        :exists  => true,
+        :uid     => user.uid,
+        :gid     => user.gid,
+        :comment => user.gecos,
+        :home    => user.dir,
+        :shell   => user.shell,
+        :extra_groups => extra_groups
+      }
+    rescue ArgumentError => e
+      {
+        :exists => false
+      }
+    end
+  end
+
+
+  def default_object
+    super.to_s
+  end
+
+  def validate!
+    super
+    problem!("username is blank") if default_object.blank?
+  end
+
+  protected
+  def opts
+    opts = ''
+    switches = {
+      :comment => '-c',
+      :gid     => '-g',
+      :uid     => '-u',
+      :shell   => '-s',
+      :set_home => lambda {|value| "-d '#{value}'"   },
+      :home     => lambda {|value| "-d '#{value}' -m"},
+      :extra_groups => lambda {|value| "-G #{value * ','}"}
+    }
+
+    switches.each do |(key,switch)|
+      to_value = args.__send__(key)
+
+    if !to_value.nil? && to_value != before_state[key]
+      if switch.respond_to?(:call)
+        opts << switch.call(to_value)
+      else
+        opts << " #{switch} '#{to_value}'"
+      end
+      opts << ' '
+    end
+    end
+
+    opts
+  end
+end