angry_mob_common_targets 0.1.0

data/lib/common_mob/template.rb

@@ -0,0 +1,23 @@
+require 'common_mob/erb'
+require 'mustache'
+
+module CommonMob
+  module Mustache
+    def render_mustache(src,variables)
+      ::Mustache.render(src.read, variables)
+    end
+  end
+
+  module Template
+    include CommonMob::Mustache
+    include CommonMob::Erb
+    
+    def render_template(src,variables)
+      if src.to_s[/\.mustache$/]
+        render_mustache(src,variables)
+      else
+        render_erb(src,variables)
+      end
+    end
+  end    
+end

data/lib/common_mob/version.rb

@@ -0,0 +1,3 @@
+module CommonMob
+  VERSION = '0.1.0'
+end

data/targets/crontab_patch.rb

@@ -0,0 +1,37 @@
+require 'common_mob'
+
+class Crontab < AngryMob::Target
+  include CommonMob::ShellHelper
+  include CommonMob::PatchHelper
+
+  default_action
+  def patch
+    if before_state[:changed]
+      sh("crontab -u #{user} -", :input => new_crontab).run
+    end
+  end
+
+  def validate!
+    problem!("please specify a patch key as the default_object (crontab('key-something-unique'))") if     default_object.blank?
+    problem!("please specify a crontab (:crontab => '* * * * * /bin/echo yay')"                  ) unless args.crontab?
+  end
+
+  def state
+    {
+      :changed => (new_crontab != crontab)
+    }
+  end
+
+  protected
+  def new_crontab
+    @new_crontab ||= patch_string(crontab, args.crontab, :key => default_object)
+  end
+
+  def crontab
+    sh("crontab -l -u #{user}").to_s
+  end
+
+  def user
+    args.user || 'root'
+  end
+end

data/targets/extract.rb

@@ -0,0 +1,40 @@
+require 'common_mob'
+
+class Tarball < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  default_action
+  def extract
+    cmd = "tar #{compression_opt}xf #{default_object}"
+
+    if !dest.blank?
+      sh("#{cmd} --strip 1 -C #{dest}").run
+    else
+      sh("#{cmd}").run
+    end
+  end
+
+  protected
+
+  def compression_opt
+    ext = default_object.to_s[/\.([^\.]+)$/,1].downcase
+    log "ext=#{ext}"
+    case ext
+    when 'gz'
+      'z'
+    when 'bz2'
+      'j'
+    end
+  end
+
+  def dest
+    args.dest || args.extract_to || args.to
+  end
+
+  # no really good way to tell if we need to untar again...
+  def state
+    {
+      :rand => rand
+    }
+  end
+end

data/targets/fetch.rb

@@ -0,0 +1,40 @@
+require 'common_mob'
+
+class Fetch < AngryMob::Target
+  include CommonMob::ShellHelper
+  include CommonMob::DigestHelper
+
+  default_action
+  def fetch
+    log "expected_sha=#{expected_sha}"
+
+    return if before_state[:exists] && (expected_sha && before_state[:sha] == expected_sha)
+
+    sh("curl #{args.src} -L -o #{default_object}".tapp, :cwd => args.cwd).run
+
+    if expected_sha && state[:sha] != expected_sha
+      raise "downloaded file's sha didn't match expected sha #{expected_sha} != #{state[:sha]}"
+    end
+  end
+
+  protected
+
+  def expected_sha
+    [ args.sha, args.sha256, args.sha512 ].find {|s| !s.blank?}
+  end
+
+  def state
+    new_sha = if !args.sha.blank?
+                sha(default_object)
+              elsif !args.sha256.blank?
+                sha256(default_object)
+              elsif !args.sha512.blank?
+                sha512(default_object)
+              end
+
+    {
+      :exists => exist?,
+      :sha    => new_sha
+    }
+  end
+end

data/targets/files.rb

@@ -0,0 +1,244 @@
+require 'common_mob'
+require 'fileutils'
+
+module FileHelpers
+  include CommonMob::DigestHelper
+  include CommonMob::ShellHelper
+  include CommonMob::PatchHelper
+  include CommonMob::FileHelper
+  include CommonMob::Template
+
+  def set_file_ownership
+    args.owner ||= args.user
+    args.group ||= args.user
+
+    set_file_attrs(default_object, args.owner, args.group, args.mode)
+  end
+end
+
+
+class DirTarget < AngryMob::Target
+  nickname 'dir'
+
+  include FileHelpers
+  default_action
+  def create
+    begin
+      mkpath unless exist?
+      set_file_ownership
+    rescue Errno::EEXIST
+      # *ulp*
+    end
+  end
+
+  def delete
+    FileUtils.rm_rf(default_object) if exist?
+  end
+
+  protected
+
+  def state
+    {
+      :exists => exist?
+    }
+  end
+
+  def default_object
+    Pathname(super)
+  end
+
+  def validate!
+  end
+end
+
+class FileTarget < AngryMob::Target
+  nickname 'file'
+  include FileHelpers
+
+  default_action
+  def create
+    if args.src
+      copy_resource resource(args.src)
+    elsif args.string
+      create_string
+    end
+
+    set_file_ownership
+  end
+
+  def delete
+    if exist?
+      log "deleting #{default_object}"
+      unlink
+    end
+  end
+
+  protected
+
+  def default_object
+    Pathname(super)
+  end
+
+  def state
+    {
+      :sha512 => sha512(default_object)
+    }
+  end
+
+  def src
+    resource(args.src)
+  end
+
+  def validate!
+    if !args.actions.include?('delete')
+      if args.src?
+        problem!(":src #{src} doesn't exist" ) unless src.exist?
+      elsif !args.src? && !args.string?
+        problem!("please specify one of :src or :string")
+      end
+    end
+  end
+
+
+  def copy_resource(src)
+    if sha512(src) != before_state[:sha512]
+      log "input string different from file contents, overwriting"
+      src.cp_to(default_object)
+    end
+  end
+
+  def create_string
+    if sha512(args.string) != before_state[:sha512]
+      log "input string different from file contents, overwriting"
+      default_object.open('w') {|f| f << args.string}
+    end
+  end
+
+end
+
+# symlink("from/path", :to => "to/path")
+#
+class Symlink < AngryMob::Target
+  include FileHelpers
+
+  default_action
+  def create
+    if before_state[:points_to] != to
+      log "linking #{from} -> #{to}"
+      sh("ln -nfs #{to} #{from}").run
+    end
+
+    ensure_points_to_correct_file!
+  end
+
+  def delete
+  end
+
+  protected
+
+  def ensure_points_to_correct_file!
+    raise "symlink doesn't point #{from} -> #{to} after action" unless state[:points_to] == to
+  end
+
+  def from
+    Pathname(default_object).expand_path
+  end
+  def to
+    Pathname(args.to).expand_path.realpath
+  end
+
+  def state
+    points_to = begin
+                  from.readlink
+                rescue Errno::ENOENT,Errno::EINVAL
+                  nil
+                end
+    {
+      :points_to => points_to
+    }
+  end
+
+  def validate!
+    problem!("file we're linking to doesn't exist") unless to.exist? # is this a real problem?
+  end
+end
+
+
+class Template < AngryMob::Target
+  include FileHelpers
+
+  default_action
+  def create
+    new_content = render_template(src,variables)
+
+    if sha512(new_content) != before_state[:sha512]
+      log "template has changed, overwriting"
+
+      backup_file(default_object)
+
+      default_object.open('w') {|f| f << new_content}
+    end
+
+    set_file_ownership
+  end
+
+  protected
+
+  def default_object
+    Pathname(super)
+  end
+
+  def src
+    resource(args.src)
+  end
+
+  def variables
+    (args.variables || args.vars || {}).update(:node => node)
+  end
+
+  def state
+    {
+      :sha512 => sha512(default_object)
+    }
+  end
+
+  def validate!
+    super
+    problem!(":src #{src} doesn't exist" ) unless src.exist?
+    problem!(":src #{src} isn't a file"  ) unless src.file?
+    problem!(":src #{src} isn't readable") unless src.readable?
+  end
+end
+
+class Patch < AngryMob::Target
+  include FileHelpers
+
+  default_action
+  def patch
+    log "patchhing"
+
+    patched = patch_file(default_object)
+
+    if before_state[:sha512] != sha512(patched)
+      log "patch has changed, overwriting"
+
+      backup_file(default_object)
+      default_object.open('w') {|f| f << patched}
+    end
+  end
+
+  def unpatch
+  end
+
+  protected
+
+  def default_object
+    Pathname(super)
+  end
+
+  def state
+    {
+      :sha512 => sha512(default_object)
+    }
+  end
+end

data/targets/git.rb

@@ -0,0 +1,84 @@
+require 'common_mob'
+
+class Git < AngryMob::Target
+  include CommonMob::ShellHelper
+
+  default_action
+  def sync
+    if is_git?
+      update
+    else
+      create
+    end
+  end
+
+  def update
+    log "updating"
+
+    set_repo
+
+    git("fetch").run
+    git("reset --hard #{ref}").run
+  end
+
+  def create
+    log "cloning"
+    git("clone -o #{remote} #{args.repo} #{default_object}", :cwd => default_object.parent).run
+    git("reset --hard #{ref}").run
+  end
+
+  protected
+
+  def changed
+    unless FalseClass === args.enable_submodules
+      git("submodule update --init").ok?
+    end
+  end
+
+  def is_git?
+    exist? && git('rev-parse --is-inside-work-tree').to_s == "true"
+  rescue CommonMob::ShellError
+  end
+
+  def git(*cmd)
+    cmd.options[:cwd] ||= default_object
+    cmd.options[:as] = args.as if args.as?
+    cmd[0] = "git #{cmd.first}"
+    sh(*cmd)
+  end
+
+  def set_repo
+    git("config remote.#{remote}.url #{args.repo}").run
+    # TODO write merge spec
+    # git("config branch.master.remote origin").run
+  end
+
+  def default_object
+    args.default_object.pathname
+  end
+
+  def remote
+    'origin'
+  end
+
+  def ref
+    ref = args.ref || args.branch || 'master'
+    "#{remote}/#{ref}"
+  end
+
+  def revision
+    exist? && git("rev-parse HEAD").to_s
+  rescue CommonMob::ShellError
+  end
+
+  def state
+    {
+      :revision => revision
+    }
+  end
+
+  def validate!
+    problem!("no git repo defined") unless args.repo?
+    problem!("no git path defined") unless args.default_object?
+  end
+end