itamae 1.0.0.beta1

data/lib/itamae/resources/directory.rb

@@ -0,0 +1,26 @@
+require 'itamae'
+
+module Itamae
+  module Resources
+    class Directory < Base
+      define_option :action, default: :create
+      define_option :path, type: String, default_name: true
+      define_option :mode, type: String
+      define_option :owner, type: String
+      define_option :group, type: String
+
+      def create_action
+        if ! backend.check_file_is_directory(path)
+          backend.create_file_as_directory(path)
+        end
+        if options[:mode]
+          backend.change_file_mode(path, options[:mode])
+        end
+        if options[:owner] || options[:group]
+          backend.change_file_owner(path, options[:owner], options[:group])
+        end
+      end
+    end
+  end
+end
+

data/lib/itamae/resources/file.rb

@@ -0,0 +1,36 @@
+require 'itamae'
+
+module Itamae
+  module Resources
+    class File < Base
+      define_option :action, default: :create
+      define_option :path, type: String, default_name: true
+      define_option :content, type: String, default: ''
+      define_option :content_file, type: String
+      define_option :mode, type: String
+      define_option :owner, type: String
+      define_option :group, type: String
+
+      def create_action
+        src = if content_file
+                content_file
+              else
+                Tempfile.open('itamae') do |f|
+                  f.write(content)
+                  f.path
+                end
+              end
+
+        copy_file(src, path)
+
+        if mode
+          backend.change_file_mode(path, mode)
+        end
+        if owner || group
+          backend.change_file_owner(path, owner, group)
+        end
+      end
+    end
+  end
+end
+

data/lib/itamae/resources/package.rb

@@ -0,0 +1,15 @@
+require 'itamae'
+
+module Itamae
+  module Resources
+    class Package < Base
+      define_option :action, default: :install
+      define_option :name, type: String, default_name: true
+
+      def install_action
+        run_specinfra(:install_package, name)
+      end
+    end
+  end
+end
+

data/lib/itamae/resources/remote_file.rb

@@ -0,0 +1,16 @@
+require 'itamae'
+
+module Itamae
+  module Resources
+    class RemoteFile < File
+      define_option :source, type: String, required: true
+
+      def create_action
+        content_file(::File.expand_path(source, ::File.dirname(@recipe.path)))
+
+        super
+      end
+    end
+  end
+end
+

data/lib/itamae/resources/template.rb

@@ -0,0 +1,19 @@
+require 'itamae'
+require 'erb'
+require 'tempfile'
+
+module Itamae
+  module Resources
+    class Template < File
+      define_option :source, type: String, required: true
+
+      def create_action
+        src = ::File.expand_path(source, ::File.dirname(@recipe.path))
+        content(ERB.new(::File.read(src), nil, '-').result(binding))
+
+        super
+      end
+    end
+  end
+end
+

data/lib/itamae/runner.rb

@@ -0,0 +1,49 @@
+require 'itamae'
+
+module Itamae
+  class Runner
+    class << self
+      def run(recipe_files, backend, options)
+        backend = backend_from_options(backend, options)
+        runner = self.new
+        runner.node = node_from_options(options)
+
+        recipe_files.each do |path|
+          recipe = Recipe.new(runner, File.expand_path(path))
+          recipe.run
+        end
+      end
+
+      private
+      def node_from_options(options)
+        if options[:node_json]
+          path = File.expand_path(options[:node_json])
+          Logger.debug "Loading node data from #{path} ..."
+          hash = JSON.load(open(path))
+        else
+          hash = {}
+        end
+
+        Node.new(hash)
+      end
+
+      def backend_from_options(type, options)
+        case type
+        when :local
+          Itamae.create_local_backend
+        when :ssh
+          ssh_options = {}
+          ssh_options[:host] = options[:host]
+          ssh_options[:user] = options[:user] || Etc.getlogin
+          ssh_options[:keys] = [options[:key]] if options[:key]
+          ssh_options[:port] = options[:port] if options[:port]
+
+          Itamae.create_ssh_backend(ssh_options)
+        end
+      end
+    end
+
+    attr_accessor :node
+  end
+end
+

data/lib/itamae/specinfra.rb

@@ -0,0 +1,44 @@
+require 'specinfra'
+ 
+# TODO: move to specinfra
+ 
+module Itamae
+  def self.backend=(backend)
+    @backend = backend
+  end
+
+  def self.backend
+    @backend
+  end
+
+  def self.create_local_backend
+    create_backend(:exec)
+  end
+
+  def self.create_ssh_backend(options)
+    Specinfra.configuration.request_pty = true
+
+    Specinfra.configuration.host = options.delete(:host)
+    Specinfra.configuration.ssh_options = options
+    create_backend(:ssh)
+  end
+
+  private
+  def self.create_backend(type)
+    Specinfra.configuration.backend = type
+    Itamae.backend = Specinfra.backend
+  end
+
+  module SpecinfraHelpers
+    module RunCommand
+      def backend
+        Itamae.backend
+      end
+
+      def run_command(cmd)
+        backend.run_command(cmd)
+      end
+    end
+  end
+end
+

data/lib/itamae/version.rb

@@ -0,0 +1,3 @@
+module Itamae
+  VERSION = "1.0.0.beta1"
+end

data/spec/integration/Vagrantfile

@@ -0,0 +1,19 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+# Vagrantfile API/syntax version. Don't touch unless you know what you're doing!
+VAGRANTFILE_API_VERSION = "2"
+
+Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
+  config.vm.define :trusty do |c|
+    c.vm.box = "ubuntu/trusty64"
+
+    c.vm.provision :shell, inline: <<-EOC
+cat /etc/apt/sources.list | sed -e 's|http://[^ ]*|mirror://mirrors.ubuntu.com/mirrors.txt|g' > /tmp/sources.list
+if !(diff -q /etc/apt/sources.list /tmp/sources.list); then
+  mv /tmp/sources.list /etc/apt/sources.list
+  apt-get update
+fi
+    EOC
+  end
+end

data/spec/integration/default_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe package('dstat') do
+  it { should be_installed }
+end
+
+describe package('sl') do
+  it { should be_installed }
+end
+
+describe file('/tmp/remote_file') do
+  it { should be_file }
+  its(:content) { should match(/Hello Itamae/) }
+end
+
+describe file('/tmp/directory') do
+  it { should be_directory }
+  it { should be_mode 700 }
+  it { should be_owned_by "vagrant" }
+  it { should be_grouped_into "vagrant" }
+end
+
+describe file('/tmp/template') do
+  it { should be_file }
+  its(:content) { should match(/Hello/) }
+end
+
+describe file('/tmp/file') do
+  it { should be_file }
+  its(:content) { should match(/Hello World/) }
+end
+
+

data/spec/integration/recipes/default.rb

@@ -0,0 +1,26 @@
+package 'dstat' do
+  action :install
+end
+
+package 'sl'
+
+remote_file "/tmp/remote_file" do
+  source "hello.txt"
+end
+
+directory "/tmp/directory" do
+  mode "0700"
+  owner "vagrant"
+  group "vagrant"
+end
+
+template "/tmp/template" do
+  source "hello.erb"
+end
+
+file "/tmp/file" do
+  content "Hello World"
+end
+
+
+

data/spec/integration/recipes/hello.erb

@@ -0,0 +1 @@
+<%= node['greeting'] %>

data/spec/integration/recipes/hello.txt

@@ -0,0 +1 @@
+Hello Itamae

data/spec/integration/recipes/node.json

@@ -0,0 +1,3 @@
+{
+  "greeting": "Hello"
+}

data/spec/integration/spec_helper.rb

@@ -0,0 +1,45 @@
+require 'serverspec'
+require 'net/ssh'
+require 'tempfile'
+
+set :backend, :ssh
+
+def vagrant(cmd)
+  Bundler.with_clean_env do
+    env = {"VAGRANT_CWD" => File.dirname(__FILE__)}
+    system env, "/usr/bin/vagrant #{cmd}"
+  end
+end
+
+if ENV['ASK_SUDO_PASSWORD']
+  begin
+    require 'highline/import'
+  rescue LoadError
+    fail "highline is not available. Try installing it."
+  end
+  set :sudo_password, ask("Enter sudo password: ") { |q| q.echo = false }
+else
+  set :sudo_password, ENV['SUDO_PASSWORD']
+end
+
+host = ENV['TARGET_HOST']
+
+config = Tempfile.new('', Dir.tmpdir)
+vagrant "ssh-config #{host} > #{config.path}"
+
+options = Net::SSH::Config.for(host, [config.path])
+
+options[:user] ||= Etc.getlogin
+
+set :host,        options[:host_name] || host
+set :ssh_options, options
+
+# Disable sudo
+# set :disable_sudo, true
+
+# Set environment variables
+# set :env, :LANG => 'C', :LC_MESSAGES => 'C' 
+
+# Set PATH
+# set :path, '/sbin:/usr/local/sbin:$PATH'
+

data/spec/unit/lib/itamae/logger_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+module Itamae
+  describe Logger do
+    let(:io) { StringIO.new }
+
+    before do
+      Logger.logger = ::Logger.new(io)
+    end
+
+    [:fatal, :error, :warn, :info, :debug].each do |level|
+      describe "##{level}" do
+        it "puts #{level} log" do
+          Logger.public_send(level, "CONTENT")
+          expect(io.string).to include('CONTENT')
+        end
+      end
+    end
+  end
+end

data/spec/unit/lib/itamae/node_spec.rb

@@ -0,0 +1,6 @@
+require 'spec_helper'
+
+module Itamae
+  describe Node do
+  end
+end

data/spec/unit/lib/itamae/recipe_spec.rb

@@ -0,0 +1,6 @@
+require 'spec_helper'
+
+module Itamae
+  describe Recipe do
+  end
+end

data/spec/unit/lib/itamae/resources/base_spec.rb

@@ -0,0 +1,135 @@
+require 'itamae'
+
+class DefineOptionTestResource < Itamae::Resources::Base
+  define_option :action, default: :create
+  define_option :default_option, default: :something
+  define_option :required_option, required: true
+  define_option :typed_option, type: Numeric
+  define_option :default_name_option, default_name: true
+end
+
+describe DefineOptionTestResource do
+  describe "define_option" do
+    describe "default" do
+      subject do
+        described_class.new(double(:recipe), 'resource name') do
+          required_option :required_value
+        end
+      end
+      it "returns the default value" do
+        expect(subject.options[:default_option]).to eq(:something)
+      end
+    end
+
+    describe "required" do
+      subject do
+        described_class.new(double(:recipe), 'resource name') do
+          #required_option :required_value
+        end
+      end
+      context "without setting required option" do
+        it "raises an error" do
+          expect do
+            subject
+          end.to raise_error(Itamae::Resources::OptionMissingError)
+        end
+      end
+    end
+
+    describe "type" do
+      context "with correct type value" do
+        subject do
+          described_class.new(double(:recipe), 'resource name') do
+            required_option :required_value
+            typed_option 10
+          end
+        end
+        it "returns the value" do
+          expect(subject.options[:typed_option]).to eq(10)
+        end
+      end
+
+      context "with incorrect type value" do
+        subject do
+          described_class.new(double(:recipe), 'resource name') do
+            required_option :required_value
+            typed_option "string"
+          end
+        end
+        it "raises an error" do
+          expect do
+            subject
+          end.to raise_error(Itamae::Resources::InvalidTypeError)
+        end
+      end
+    end
+
+    describe "default_name" do
+      context "without setting the value" do
+        subject do
+          described_class.new(double(:recipe), 'resource name') do
+            required_option :required_value
+          end
+        end
+        it "returns the resource name" do
+          expect(subject.options[:default_name_option]).
+            to eq("resource name")
+        end
+      end
+    end
+  end
+end
+
+class TestResource < Itamae::Resources::Base
+  define_option :action, default: :create
+  define_option :option_key, required: false
+end
+
+describe TestResource do
+  let(:commands) { double(:commands) }
+  let(:runner) do
+    double(:runner)
+  end
+  let(:recipe) do
+    double(:recipe).tap do |r|
+      r.stub(:runner).and_return(runner)
+    end
+  end
+
+  subject(:resource) { described_class.new(recipe, "name") }
+
+  before do
+    Itamae.backend = double(:backend).tap do |b|
+      b.stub(:commands).and_return(commands)
+    end
+  end
+
+  describe "#run" do
+    before do
+      subject.action :action_name
+    end
+    it "executes <ACTION_NAME>_action method" do
+      expect(subject).to receive(:action_name_action)
+      subject.run
+    end
+  end
+
+  describe "#run_specinfra" do
+    it "runs specinfra's command by specinfra's backend" do
+      expect(Specinfra.command).to receive(:cmd).and_return("command")
+      expect(Itamae.backend).to receive(:run_command).with("command").
+        and_return(Specinfra::CommandResult.new(exit_status: 0))
+      subject.send(:run_specinfra, :cmd)
+    end
+    context "when the command execution failed" do
+      it "raises CommandExecutionError" do
+        expect(Specinfra.command).to receive(:cmd).and_return("command")
+        expect(Itamae.backend).to receive(:run_command).with("command").
+          and_return(Specinfra::CommandResult.new(exit_status: 1))
+        expect do
+          subject.send(:run_specinfra, :cmd)
+        end.to raise_error(Itamae::Resources::CommandExecutionError)
+      end
+    end
+  end
+end