itamae-mitsurin 0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: ebdf4efc9c889e355f3e693436b1dac36b24f313
+  data.tar.gz: b9494d2c5291046feb71801dde6582fc20137020
+SHA512:
+  metadata.gz: 26ca1bc353e49f4bba91aace86ec64faf2444f08236f12fff5d5c5fa302c44eb49f424c850b5f00b9fcbb5f14cef448513f0bafdf5076f49861d4334c99bc179
+  data.tar.gz: e5cd9b57cd29db325d8a7ef4739664a2913008d53a35c01bf0e1eac48675a8c486ec85fb57412103f36954c278e5931edcb572ee57c72717635cef9d0992f251

data/.gitignore

@@ -0,0 +1,19 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.vagrant
+Gemfile.local

data/.rspec

@@ -0,0 +1,2 @@
+--color
+-fd

data/Gemfile

@@ -0,0 +1,7 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in itamae-mitsurin.gemspec
+gemspec
+
+path = Pathname.new("Gemfile.local")
+eval(path.read) if path.exist?

data/LICENSE.txt

@@ -0,0 +1,26 @@
+Copyright (c) 2016 Akihiro Kamiyama
+
+Ripped out of Itamae 1.9.3, Itamae-plugin-resource-aws 0.1 and modified.
+Copyright (c) 2013-2015 Ryota Arai
+Copyright (c) 2015 Gosuke Miyashita
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,44 @@
+[![Gem Version](https://badge.fury.io/rb/itamae-mitsurin.svg)](http://badge.fury.io/rb/itamae-mitsurin)
+
+Customized version of Itamae and plugin
+
+## Concept
+
+- More like Chef
+- Minimum attributes
+- Support AWS Resource
+- Require Rake
+
+## Installation
+
+```
+$ gem install itamae-mitsurin
+$ itamae-mitsurin init [project_name]
+```
+
+## Usage AWS Resource
+
+```ruby
+# recipe
+
+Aws.config[:region] = 'ap-northeast-1'
+
+aws_ebs_volume 'db_ebs_volume' do
+  action [:create, :attach]
+  availability_zone 'ap-northeast-1b'
+  device '/dev/xvdf'
+  volume_type 'standard'
+  size 30
+  instance_id node['ec2']['instance-id']
+end
+```
+
+## Contributing
+
+If you have a problem, please [create an issue](https://github.com/kammy1231/itamae-mitsurin) or a pull request.
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,55 @@
+# coding: utf-8
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+require 'rake/testtask'
+require 'rake/extensiontask'
+require 'rake/packagetask'
+begin
+  require 'bundler/gem_tasks'
+rescue LoadError
+  puts 'If you want to create gem, You must install Bundler'
+end
+
+require './lib/itamae/mitsurin/version.rb'
+def version
+  Itamae::Mitsurin::VERSION
+end
+
+task :default => :test
+task :test => :compile
+Rake::TestTask.new do |t|
+  t.libs << 'lib' << 'test'
+  t.test_files = FileList['test/test_*.rb'].exclude('test/test_assoccoords.rb')
+end
+
+Rake::ExtensionTask.new do |ext|
+  ext.name = 'itamae-mitsurin'
+  ext.ext_dir = 'ext/'
+  ext.lib_dir = 'lib/'
+end
+
+Rake::PackageTask.new('itamae-mitsurin', "#{version}") do |t|
+  t.need_tar_gz = true
+  t.package_files.include `git ls-files`.split("\n")
+end
+
+namespace :release do
+  desc "Bump up version and commit"
+  task :version_up do
+    version_file = File.expand_path("lib/itamae/mitsurin/version.txt")
+    current_version = File.read(version_file).strip
+
+    if /\A(.+?)(\d+)\z/ =~ current_version
+      next_version = "#{$1}#{$2.to_i + 1}"
+    else
+      raise "Invalid version"
+    end
+
+    open(version_file, "w") do |f|
+      f.write next_version
+    end
+    system "git add #{version_file}"
+    system "git commit -m 'Bump up version'"
+  end
+end

data/bin/itamae

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'itamae/cli'
+Itamae::CLI.start
+

data/bin/itamae-mitsurin

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'itamae/mitsurin/cli'
+Itamae::Mitsurin::CLI.start
+

data/itamae-mitsurin.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'itamae/mitsurin/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "itamae-mitsurin"
+  spec.version       = Itamae::Mitsurin::VERSION
+  spec.authors       = ["Akihiro Kamiyama"]
+  spec.email         = ["kamiyama@qualitia.co.jp"]
+  spec.summary       = %q{Customized version of Itamae and Itamae plugin}
+  spec.homepage      = "https://github.com/kammy1231/itamae-mitsurin"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "thor"
+  spec.add_runtime_dependency "specinfra", [">= 2.51.2", "< 3.0.0"]
+  spec.add_runtime_dependency "hashie"
+  spec.add_runtime_dependency "ansi"
+  spec.add_runtime_dependency "schash", "~> 0.1.0"
+  spec.add_runtime_dependency "aws-sdk", "~> 2"
+  spec.add_runtime_dependency "serverspec", [">= 2.30", "< 3.0.0"]
+
+  spec.add_development_dependency "bundler", "~> 1.10"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "pry-byebug"
+  spec.add_development_dependency "docker-api", "~> 1.20"
+  spec.add_development_dependency "fakefs"
+  spec.add_development_dependency "fluent-logger"
+end

data/lib/itamae.rb

@@ -0,0 +1,18 @@
+require "itamae/runner"
+require "itamae/cli"
+require "itamae/recipe"
+require "itamae/resource"
+require "itamae/handler"
+require "itamae/handler_proxy"
+require "itamae/recipe_children"
+require "itamae/logger"
+require "itamae/node"
+require "itamae/backend"
+require "itamae/notification"
+require "itamae/definition"
+require "itamae/ext"
+require "itamae/generators"
+
+module Itamae
+  # Your code goes here...
+end

data/lib/itamae/backend.rb

@@ -0,0 +1,293 @@
+require 'specinfra/core'
+require 'singleton'
+require 'io/console'
+require 'net/ssh'
+
+Specinfra::Configuration.error_on_missing_backend_type = true
+
+module Specinfra
+  module Configuration
+    def self.sudo_password
+      return ENV['SUDO_PASSWORD'] if ENV['SUDO_PASSWORD']
+      return @sudo_password if @sudo_password
+
+      # TODO: Fix this dirty hack
+      return nil unless caller.any? {|call| call.include?('channel_data') }
+
+      print "sudo password: "
+      @sudo_password = STDIN.noecho(&:gets).strip
+      print "\n"
+      @sudo_password
+    end
+  end
+end
+
+module Itamae
+  module Backend
+    UnknownBackendTypeError = Class.new(StandardError)
+    CommandExecutionError = Class.new(StandardError)
+    SourceNotExistError = Class.new(StandardError)
+
+    class << self
+      def create(type, opts = {})
+        self.const_get(type.capitalize).new(opts)
+      end
+    end
+
+    class Base
+      attr_reader :executed_commands
+
+      def initialize(options)
+        @options = options
+        @backend = create_specinfra_backend
+        @executed_commands = []
+      end
+
+      def run_command(commands, options = {})
+        options = {error: true}.merge(options)
+
+        command = build_command(commands, options)
+        Itamae.logger.debug "Executing `#{command}`..."
+
+        result = nil
+
+        Itamae.logger.with_indent do
+          reset_output_handler
+
+          result = run_command_with_profiling(command)
+
+          flush_output_handler_buffer
+
+          if result.exit_status == 0 || !options[:error]
+            method = :debug
+            message = "exited with #{result.exit_status}"
+          else
+            method = :error
+            message = "Command `#{command}` failed. (exit status: #{result.exit_status})"
+
+            unless Itamae.logger.level == ::Logger::DEBUG
+              result.stdout.each_line do |l|
+                log_output_line("stdout", l, :error)
+              end
+              result.stderr.each_line do |l|
+                log_output_line("stderr", l, :error)
+              end
+            end
+          end
+
+          Itamae.logger.public_send(method, message)
+        end
+
+        if options[:error] && result.exit_status != 0
+          raise CommandExecutionError
+        end
+
+        result
+      end
+
+      def get_command(*args)
+        @backend.command.get(*args)
+      end
+
+      def receive_file(src, dst = nil)
+        if dst
+          Itamae.logger.debug "Receiving a file from '#{src}' to '#{dst}'..."
+        else
+          Itamae.logger.debug "Receiving a file from '#{src}'..."
+        end
+        @backend.receive_file(src, dst)
+      end
+
+      def send_file(src, dst)
+        Itamae.logger.debug "Sending a file from '#{src}' to '#{dst}'..."
+        unless ::File.exist?(src)
+          raise SourceNotExistError, "The file '#{src}' doesn't exist."
+        end
+        unless ::File.file?(src)
+          raise SourceNotExistError, "'#{src}' is not a file."
+        end
+        @backend.send_file(src, dst)
+      end
+
+      def send_directory(src, dst)
+        Itamae.logger.debug "Sending a directory from '#{src}' to '#{dst}'..."
+        unless ::File.exist?(src)
+          raise SourceNotExistError, "The directory '#{src}' doesn't exist."
+        end
+        unless ::File.directory?(src)
+          raise SourceNotExistError, "'#{src}' is not a directory."
+        end
+        @backend.send_directory(src, dst)
+      end
+
+      def host_inventory
+        @backend.host_inventory
+      end
+
+      def finalize
+        # pass
+      end
+
+      private
+
+      def create_specinfra_backend
+        raise NotImplementedError
+      end
+
+      def reset_output_handler
+        @buf = {}
+        %w!stdout stderr!.each do |output_name|
+          @buf[output_name] = ""
+          handler = lambda do |str|
+            lines = str.split(/\r?\n/, -1)
+            @buf[output_name] += lines.pop
+            unless lines.empty?
+              lines[0] = @buf[output_name] + lines[0]
+              @buf[output_name] = ""
+              lines.each do |l|
+                log_output_line(output_name, l)
+              end
+            end
+          end
+          @backend.public_send("#{output_name}_handler=", handler)
+        end
+      end
+
+      def flush_output_handler_buffer
+        @buf.each do |output_name, line|
+          next if line.empty?
+          log_output_line(output_name, line)
+        end
+      end
+
+      def log_output_line(output_name, line, severity = :debug)
+        line = line.gsub(/[[:cntrl:]]/, '')
+        Itamae.logger.public_send(severity, "#{output_name} | #{line}")
+      end
+
+      def build_command(commands, options)
+        if commands.is_a?(Array)
+          command = commands.map do |cmd|
+            cmd.shellescape
+          end.join(' ')
+        else
+          command = commands
+        end
+
+        cwd = options[:cwd]
+        if cwd
+          command = "cd #{cwd.shellescape} && #{command}"
+        end
+
+        user = options[:user]
+        if user
+          command = "cd ~#{user.shellescape} ; #{command}"
+          command = "sudo -H -u #{user.shellescape} -- #{shell.shellescape} -c #{command.shellescape}"
+        end
+
+        command
+      end
+
+      def shell
+        @options[:shell] || '/bin/sh'
+      end
+
+      def run_command_with_profiling(command)
+        start_at = Time.now
+        result = @backend.run_command(command)
+        duration = Time.now.to_f - start_at.to_f
+
+        @executed_commands << {command: command, duration: duration}
+
+        result
+      end
+    end
+
+    class Local < Base
+      private
+      def create_specinfra_backend
+        Specinfra::Backend::Exec.new(
+          shell: @options[:shell],
+        )
+      end
+    end
+
+    class Ssh < Base
+      private
+      def create_specinfra_backend
+        Specinfra::Backend::Ssh.new(
+          request_pty: true,
+          host: ssh_options[:host_name],
+          disable_sudo: disable_sudo?,
+          ssh_options: ssh_options,
+          shell: @options[:shell],
+        )
+      end
+
+      def ssh_options
+        opts = {}
+
+        opts[:host_name] = @options[:host]
+
+        # from ssh-config
+        opts.merge!(Net::SSH::Config.for(@options[:host]))
+        opts[:user] = @options[:user] || opts[:user] || Etc.getlogin
+        opts[:keys] = [@options[:key]] if @options[:key]
+        opts[:port] = @options[:port] if @options[:port]
+
+        if @options[:vagrant]
+          config = Tempfile.new('', Dir.tmpdir)
+          hostname = opts[:host_name] || 'default'
+          vagrant_cmd = "vagrant ssh-config #{hostname} > #{config.path}"
+          if defined?(Bundler)
+            Bundler.with_clean_env do
+              `#{vagrant_cmd}`
+            end
+          else
+            `#{vagrant_cmd}`
+          end
+          opts.merge!(Net::SSH::Config.for(hostname, [config.path]))
+        end
+
+        if @options[:ask_password]
+          print "password: "
+          password = STDIN.noecho(&:gets).strip
+          print "#{ENV['SSH_PASSWORD']}\n"
+          opts.merge!(password: password)
+        end
+
+        opts
+      end
+
+      def disable_sudo?
+        !@options[:sudo]
+      end
+    end
+
+    class Docker < Base
+      def finalize
+        image = @backend.commit_container
+        Itamae.logger.info "Image created: #{image.id}"
+      end
+
+      private
+      def create_specinfra_backend
+        begin
+          require 'docker'
+        rescue LoadError
+          Itamae.logger.fatal "To use docker backend, please install 'docker-api' gem"
+        end
+
+        # TODO: Move to Specinfra?
+        Excon.defaults[:ssl_verify_peer] = @options[:tls_verify_peer]
+        ::Docker.logger = Itamae.logger
+
+        Specinfra::Backend::Docker.new(
+          docker_image: @options[:image],
+          docker_container: @options[:container],
+          shell: @options[:shell],
+        )
+      end
+    end
+  end
+end