automateit 0.70923

data/misc/index_gem_repository.rb

@@ -0,0 +1,303 @@
+#!/usr/bin/env jruby
+# http://svn.codehaus.org/jruby/trunk/jruby/bin/index_gem_repository.rb
+#--
+# Copyright 2006 by Chad Fowler, Rich Kilmer, Jim Weirich and others.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#++
+
+
+# Generate the yaml/yaml.Z index files for a gem server directory.
+#
+# Usage:  generate_yaml_index.rb --dir DIR [--verbose]
+
+$:.unshift '~/rubygems' if File.exist? "~/rubygems"
+
+require 'optparse'
+require 'rubygems'
+require 'zlib'
+require 'digest/sha2'
+begin
+  require 'builder/xchar'
+rescue LoadError
+  fail "index_gem_repository requires that the XML Builder library be installed"
+end
+
+Gem.manage_gems
+
+######################################################################
+# Mixin that provides a +compress+ method for compressing files on
+# disk.
+#
+module Compressor
+  # Compress the given file.
+  def compress(filename, ext="rz")
+    File.open(filename + ".#{ext}", "w") do |file|
+      file.write(zip(File.read(filename)))
+    end
+  end
+
+  # Return a compressed version of the given string.
+  def zip(string)
+    Zlib::Deflate.deflate(string)
+  end
+
+  # Return an uncompressed version of a compressed string.
+  def unzip(string)
+    Zlib::Inflate.inflate(string)
+  end
+end
+
+######################################################################
+# Announcer provides a way of announcing activities to the user.
+#
+module Announcer
+  # Announce +msg+ to the user.
+  def announce(msg)
+    puts msg if @options[:verbose]
+  end
+end
+
+######################################################################
+# Abstract base class for building gem indicies.  Uses the template
+# pattern with subclass specialization in the +begin_index+,
+# +end_index+ and +cleanup+ methods.
+#
+class AbstractIndexBuilder
+  include Compressor
+  include Announcer
+
+  # Build a Gem index.  Yields to block to handle the details of the
+  # actual building.  Calls +begin_index+, # +end_index+ and +cleanup+
+  # at appropriate times to customize basic operations.
+  def build
+    if ! @enabled
+      yield
+    else
+      unless File.exist?(@directory)
+        FileUtils.mkdir_p(@directory)
+      end
+      fail "not a directory: #{@directory}" unless File.directory?(@directory)
+      File.open(File.join(@directory, @filename), "w") do |file|
+        @file = file
+        start_index
+        yield
+        end_index
+      end
+      cleanup
+    end
+  ensure
+    @file = nil
+  end
+
+  # Called immediately before the yield in build.  The index file is
+  # open and availabe as @file.
+  def start_index
+  end
+
+  # Called immediately after the yield in build.  The index file is
+  # still open and available as @file.
+  def end_index
+  end
+
+  # Called from within builder after the index file has been closed.
+  def cleanup
+  end
+end
+
+######################################################################
+# Construct the master Gem index file.
+#
+class MasterIndexBuilder < AbstractIndexBuilder
+  def initialize(filename, options)
+    @filename = filename
+    @options = options
+    @directory = options[:directory]
+    @enabled = true
+  end
+
+  def start_index
+    super
+    @file.puts "--- !ruby/object:Gem::Cache"
+    @file.puts "gems:"
+  end
+
+  def cleanup
+    super
+    index_file_name = File.join(@directory, @filename)
+    compress(index_file_name, "Z")
+    paranoid(index_file_name, "#{index_file_name}.Z")
+  end
+
+  def add(spec)
+    @file.puts "  #{spec.full_name}: #{nest(spec.to_yaml)}"
+  end
+
+  def nest(yaml_string)
+    yaml_string[4..-1].gsub(/\n/, "\n    ")
+  end
+
+  private
+
+  def paranoid(fn, compressed_fn)
+    data = File.read(fn)
+    compressed_data = File.read(compressed_fn)
+    if data != unzip(compressed_data)
+      fail "Compressed file #{compressed_fn} does not match uncompressed file #{fn}"
+    end
+  end
+end
+
+######################################################################
+# Construct a quick index file and all of the individual specs to
+# support incremental loading.
+#
+class QuickIndexBuilder < AbstractIndexBuilder
+  def initialize(filename, options)
+    @filename = filename
+    @options = options
+    @directory = options[:quick_directory]
+    @enabled = options[:quick]
+  end
+
+  def cleanup
+    compress(File.join(@directory, @filename))
+  end
+
+  def add(spec)
+    return unless @enabled
+    @file.puts spec.full_name
+    fn = File.join(@directory, "#{spec.full_name}.gemspec.rz")
+    File.open(fn, "w") do |gsfile|
+      gsfile.write(zip(spec.to_yaml))
+    end
+  end
+end
+
+######################################################################
+# Top level class for building the repository index.  Initialize with
+# an options hash and call +build_index+.
+#
+class Indexer
+  include Compressor
+  include Announcer
+
+  # Create an indexer with the options specified by the options hash.
+  def initialize(options)
+    @options = options.dup
+    @directory = @options[:directory]
+    @options[:quick_directory] = File.join(@directory, "quick")
+    @master_index = MasterIndexBuilder.new("yaml", @options)
+    @quick_index = QuickIndexBuilder.new("index", @options)
+  end
+
+  # Build the index.
+  def build_index
+    announce "Building Server Index"
+    FileUtils.rm_r(@options[:quick_directory]) rescue nil
+    @master_index.build do
+      @quick_index.build do 
+        gem_file_list.each do |gemfile|
+          spec = Gem::Format.from_file_by_path(gemfile).spec
+          abbreviate(spec)
+          sanitize(spec)
+          announce "   ... adding #{spec.full_name}"
+          @master_index.add(spec)
+          @quick_index.add(spec)
+        end
+      end
+    end
+  end
+
+  # List of gem file names to index.
+  def gem_file_list
+    Dir.glob(File.join(@directory, "gems", "*.gem"))
+  end
+
+  # Abbreviate the spec for downloading.  Abbreviated specs are only
+  # used for searching, downloading and related activities and do not
+  # need deployment specific information (e.g. list of files).  So we
+  # abbreviate the spec, making it much smaller for quicker downloads.
+  def abbreviate(spec)
+    spec.files = []
+    spec.test_files = []
+    spec.rdoc_options = []
+    spec.extra_rdoc_files = []
+    spec.cert_chain = []
+    spec
+  end
+
+  # Sanitize the descriptive fields in the spec.  Sometimes non-ASCII
+  # characters will garble the site index.  Non-ASCII characters will
+  # be replaced by their XML entity equivalent.
+  def sanitize(spec)
+    spec.summary = sanitize_string(spec.summary)
+    spec.description = sanitize_string(spec.description)
+    spec.post_install_message = sanitize_string(spec.post_install_message)
+    spec.authors = spec.authors.collect { |a| sanitize_string(a) }
+    spec
+  end
+
+  # Sanitize a single string.
+  def sanitize_string(string)
+    string ? string.to_xs : string
+  end
+end
+
+######################################################################
+# Top Level Functions 
+######################################################################
+
+def handle_options(args)
+  # default options
+  options = {
+    :directory => '.',
+    :verbose => false,
+    :quick => true,
+  }
+  
+  args.options do |opts|
+    opts.on_tail("--help", "show this message") do
+      puts opts
+      exit
+    end
+    opts.on(
+      '-d', '--dir=DIRNAME', '--directory=DIRNAME',
+      "repository base dir containing gems subdir",
+      String) do |value|
+      options[:directory] = value
+    end
+    opts.on('--[no-]quick', "include quick index") do |value|
+      options[:quick] = value
+    end
+    opts.on('-v', '--verbose', "show verbose output") do |value|
+      options[:verbose] = value
+    end
+    opts.on('-V', '--version',
+      "show version") do |value|
+      puts Gem::RubyGemsVersion
+      exit
+    end
+    opts.parse!
+  end
+  
+  if options[:directory].nil?
+    puts "Error, must specify directory name. Use --help"
+    exit
+  elsif ! File.exist?(options[:directory]) ||
+      ! File.directory?(options[:directory])
+    puts "Error, unknown directory name #{directory}."
+    exit
+  end
+  options
+end
+
+# Main program.
+def main_index(args)
+  options = handle_options(args)
+  Indexer.new(options).build_index
+end
+
+if __FILE__ == $0 then
+  main_index(ARGV)
+end

data/misc/setup_egg.rb

@@ -0,0 +1,12 @@
+#!/usr/bin/env automateit
+
+# Install Python's easy_install package manager for 'egg' files
+
+SOURCE = 'http://peak.telecommunity.com/dist/ez_setup.py'
+
+require 'open-uri'
+
+mktemp do |t|
+  File.open(t, "w+") {|h| h.write(open(SOURCE).read)}
+  sh "python #{t}"
+end

data/misc/setup_gem_dependencies.sh

@@ -0,0 +1,7 @@
+#!/bin/sh
+
+cmd = 'gem install -y rake open4 activesupport rspec ruby-breakpoint ruby-debug --no-rdoc --no-ri'
+
+# Run it twice because the first one will usually fail. Oh how I hate gem. :/
+$cmd
+$cmd

data/misc/setup_rubygems.sh

@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+
+# RELEASES: http://rubyforge.org/frs/?group_id=126
+
+PACKAGE="rubygems-0.9.3"
+URL="http://rubyforge.org/frs/download.php/20585/rubygems-0.9.3.tgz"
+
+pushd "/tmp"
+
+CACHE=`mktemp -d install_rubygems.XXXXXXXXXX`
+pushd "$CACHE"
+
+wget -c -T20 -q "$URL"
+tar xfz "$PACKAGE.tgz"
+cd "$PACKAGE"
+sudo ruby setup.rb
+
+popd
+rm -rf "$CACHE"
+
+popd

data/misc/which.cmd

@@ -0,0 +1,6 @@
+@ rem "which" workalike for Windows. Searches current directory and PATH for all extensions, stops after first match, and returns meaningful exit value for "system" or "ERRORLEVEL" checks.
+@setlocal
+@set P2=.;%PATH%
+@for %%e in (%PATHEXT%) do @for %%i in (%1%%e) do @if NOT "%%~$P2:i"=="" echo %%~$P2:i && goto end
+@exit 1
+:end

data/spec/extras/automateit_service_sysv_test

@@ -0,0 +1,50 @@
+#!/bin/bash -e
+set -e
+
+### BEGIN INIT INFO
+# Provides:          automateit_service_test
+# Default-Start:     5
+# Default-Stop:      6
+# Short-Description: automateit_service_test
+# Description:       automateit_service_test
+### END INIT INFO
+
+STATE="/tmp/automateit_service_test.state"
+status() {
+    test -f $STATE;
+}
+case $1 in
+    start)
+        if status; then
+            echo "ERROR: already running"
+            exit 1
+        else
+            echo "started"
+            touch $STATE
+            exit 0
+        fi
+    ;;
+    stop)
+        if status; then
+            echo "stopping"
+            rm $STATE
+            exit 0
+        else
+            echo "ERROR: not running"
+            exit 1
+        fi
+    ;;
+    status)
+        if status; then
+            echo "running"
+            exit 0
+        else
+            echo "not running"
+            exit 1
+        fi
+    ;;
+    *)
+        echo "ERROR: unknown command"
+        exit 1
+    ;;
+esac

data/spec/extras/scratch.rb

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH << "lib"
+
+require "automateit"
+
+puts "==0"
+ai = AutomateIt.new
+#puts "==1"
+#ai.sh "ls"
+#puts "==2"
+#ai.platform_manager.setup(:default => :struct)
+ai.shell_manager.sh "id"
+ai.sh "id"
+ai.eval{sh "id"}

data/spec/extras/simple_recipe.rb

@@ -0,0 +1,8 @@
+#!/usr/bin/env automateit
+
+puts "Hello world!"
+puts "I'm an #{self.class}"
+puts "My tags, let me show them to you: %s" % tags.to_a.inspect
+preview_for("I'm in preview mode") do
+  puts "I'm not in preview mode"
+end

data/spec/integration/account_manager_spec.rb

@@ -0,0 +1,218 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+if not INTERPRETER.euid?
+  puts "NOTE: Can't check 'euid' on this platform, #{__FILE__}"
+elsif not INTERPRETER.superuser?
+  puts "NOTE: Must be root to check #{__FILE__}"
+elsif not INTERPRETER.account_manager.available?(:add_user)
+  puts "NOTE: Can't find AccountManager for this platform, #{__FILE__}"
+else
+  describe "AutomateIt::AccountManager" do
+    before(:all) do
+      @a = AutomateIt.new(:verbosity => Logger::WARN)
+      @m = @a.account_manager
+
+      @username = "automateit_testuser"
+      @groupname = "automateit_testgroup"
+
+      raise "User named '#{@username}' found. If this isn't a real user, delete it so that the test can contineu. If this is a real user, change the spec to test with a user that shouldn't exist." if @m.users[@username]
+      raise "Group named '#{@groupname}' found. If this isn't a real group, delete it so that the test can contineu. If this is a real group, change the spec to test with a group that shouldn't exist." if @m.groups[@groupname]
+    end
+
+    after(:all) do
+      @m.remove_user(@username, :quiet => true)
+      @m.remove_group(@username, :quiet => true)
+      @m.remove_group(@groupname, :quiet => true)
+    end
+
+    it "should find root user" do
+      entry = @m.users["root"]
+      entry.should_not be_nil
+      entry.uid.should == 0
+    end
+
+    it "should not have a user before one is created" do
+      @m.has_user?(@username).should be_false
+    end
+
+    it "should create a user" do
+      entry = @m.add_user(@username, :passwd => "asdf", :shell => "/bin/false")
+
+      entry.should_not be_nil
+      entry.name.should == @username
+      # Leaves behind user for further tests
+    end
+
+    it "should have a user after one is created" do
+      # Depends on user to be created by previous tests
+      @m.has_user?(@username).should be_true
+    end
+
+    it "should query user data by name" do
+      # Depends on user to be created by previous tests
+      entry = @m.users[@username]
+      entry.should_not be_nil
+      entry.name.should == @username
+    end
+
+    it "should query user data by id" do
+      # Depends on user to be created by previous tests
+      uid = @m.users[@username].uid
+
+      entry = @m.users[uid]
+      entry.should_not be_nil
+      entry.name.should == @username
+    end
+
+    it "should not query user data by invalid type" do
+      lambda{ @m.users[false] }.should raise_error(TypeError)
+    end
+
+    it "should create user group" do
+      # Depends on user to be created by previous tests
+      @m.groups[@username].should_not be_nil
+    end
+
+    it "should not re-add an existing user" do
+      # Depends on user to be created by previous tests
+      @m.add_user(@username).should be_false
+    end
+
+    it "should not have a non-existent group" do
+      @m.has_group?(@groupname).should be_false
+    end
+
+    it "should add a group" do
+      entry = @m.add_group(@groupname)
+      entry.should_not be_nil
+      entry.name.should == @groupname
+      # Leaves behind group for further tests
+    end
+
+    it "should not re-add a group" do
+      @m.add_group(@groupname).should be_false
+    end
+
+    it "should query group data by name" do
+      entry = @m.groups[@groupname]
+      entry.should_not be_nil
+      entry.name.should == @groupname
+    end
+
+    it "should query group data by id" do
+      gid = @m.groups[@groupname].gid
+
+      entry = @m.groups[gid]
+      entry.should_not be_nil
+      entry.name.should == @groupname
+    end
+
+    it "should not query group data by invalid type" do
+      lambda{ @m.groups[false] }.should raise_error(TypeError)
+    end
+
+    it "should remove a group" do
+      # Depends on group to be created by previous tests
+      @m.remove_group(@groupname).should be_true
+    end
+
+    it "should not remove a non-existent group" do
+      @m.remove_group(@groupname).should be_false
+    end
+
+    it "should not have members for a non-existent group" do
+      @m.users_for_group(@groupname).should == []
+    end
+
+    it "should add a group with members" do
+      # Depends on user to be created by previous tests
+      @m.add_group(@groupname, :members => @username)
+      # Leaves behind group for further tests
+    end
+
+    it "should query users in a group" do
+      # Depends on group to be created by previous tests
+      @m.users_for_group(@groupname).should == [@username]
+    end
+
+    it "should query groups for a user" do
+      # Depends on user to be created by previous tests
+      # Depends on group to be created by previous tests
+      @m.groups_for_user(@username).should include(@groupname)
+    end
+
+    it "should remove users from a group" do
+      # Depends on user to be created by previous tests
+      # Depends on group to be created by previous tests
+      @m.remove_users_from_group(@username, @groupname).should == [@username]
+    end
+
+    it "should add groups to a user" do
+      # Depends on user to be created by previous tests
+      @m.add_groups_to_user(@groupname, @username).should == [@groupname]
+    end
+
+    it "should remove groups from user" do
+      # Depends on user to be created by previous tests
+      @m.remove_groups_from_user(@groupname, @username).should == [@groupname]
+    end
+
+    it "should remove a group with members" do
+      # Depends on group to be created by previous tests
+      @m.remove_group(@groupname).should be_true
+    end
+
+    it "should not add users to non-existent group" do
+      lambda{ @m.add_users_to_group(@username, @groupname) }.should raise_error(ArgumentError)
+    end
+
+    it "should pretend to add users to non-existent group in preview mode" do
+      begin
+        @a.preview = true
+        @m.add_users_to_group(@username, @groupname).should == [@username]
+      ensure
+        @a.preview = false
+      end
+    end
+
+    it "should not remove users from non-existent group" do
+      lambda{ @m.remove_users_from_group(@username, @groupname) }.should raise_error(ArgumentError)
+    end
+
+    it "should pretend to remove users from non-existent group in preview mode" do
+      begin
+        @a.preview = true
+        @m.remove_users_from_group(@username, @groupname).should == [@username]
+      ensure
+        @a.preview = false
+      end
+    end
+
+    it "should change password" do
+      # Depends on user to be created by previous tests
+      pass = "automateit"
+
+      # TODO This isn't portable
+      def extract_pwent(username)
+        for filename in %w(/etc/shadow /etc/passwd)
+          next unless File.exist?(filename)
+          return File.read(filename).split(/\n/).grep(/^#{username}\b/)
+        end
+      end
+
+      before = extract_pwent(@username)
+      @m.passwd(@username, pass).should be_true
+      after = extract_pwent(@username)
+      before.should_not eql(after)
+    end
+
+    it "should remove a user" do
+      # Depends on user to be created by previous tests
+      @m.remove_user(@username, :quiet => true).should be_true
+    end
+
+    it "should not remove a non-existent user" do
+      @m.remove_user(@username).should be_false
+    end
+  end
+end