itamae-spec 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 32c0a5e4298c6d97786f447e64e6edea3071ed874f8f347532e33bfa92330fdf
+  data.tar.gz: d69af9290654e7d104d9decb50ff9f6c8afb80587af0189ab1e933264abec152
+SHA512:
+  metadata.gz: 1234cfda80bca59a5e53251046f7c743587d692097f80a63033c9bbe150eb18dcc161b703bd3019d416527c327703e02ea8065b486f0f7ab44beca652f46343a
+  data.tar.gz: a3a82b93122ab5b95eddb874f09b14fb0c919e888008b834d32e501a6bb7ad070cefb88bdc11fb2cee752c75b5cda54174d7ec394cc8824cce32442de312c4c3

data/.gitignore

@@ -0,0 +1,14 @@
+*.gem
+*.rbc
+.bundle
+.config
+Gemfile.lock
+Gemfile.local
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.vagrant
+logs

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-spec.gemspec
+gemspec
+
+path = Pathname.new("Gemfile.local")
+eval(path.read) if path.exist?

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2018 kammy1231
+
+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,42 @@
+[![Gem Version](https://badge.fury.io/rb/itamae-spec.svg)](http://badge.fury.io/rb/itamae-spec)
+
+Customized version of Itamae
+
+## Concept
+
+- Integration with Serverspec
+- Attributes are defined by: Nodes, Environments, and Recipes
+- Support for some AWS Resources
+- Running on the RakeTask
+
+## Installation
+
+```
+$ gem install itamae-spec
+$ itamae-spec init your-project-name
+```
+
+## If you want to use the AWS Resources
+```
+$ aws configure
+```
+
+## Getting Started
+- [Getting Started](https://github.com/kammy1231/itamae-spec/wiki/Getting-Started)
+
+## Usage AWS Resource
+- [Customized Resources](https://github.com/kammy1231/itamae-spec/wiki/Customized-Resources)
+
+## Reference
+- [itamae](https://github.com/itamae-kitchen/itamae)
+- [Serverspec](https://github.com/mizzy/serverspec)
+
+## Contributing
+
+If you have a problem, please [create an issue](https://github.com/kammy1231/itamae-spec) 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,47 @@
+# coding: utf-8
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+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-spec/version.rb'
+def version
+  ItamaeSpec::VERSION
+end
+
+Rake::ExtensionTask.new do |ext|
+  ext.name = 'itamae-spec'
+  ext.ext_dir = 'ext/'
+  ext.lib_dir = 'lib/'
+end
+
+Rake::PackageTask.new('itamae-spec', "#{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-spec/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-spec

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+require "itamae-spec/cli"
+ItamaeSpec::CLI.start

data/itamae-spec.gemspec

@@ -0,0 +1,36 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'itamae-spec/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "itamae-spec"
+  spec.version       = ItamaeSpec::VERSION
+  spec.authors       = ["kammy1231"]
+  spec.email         = ["akihiro.vamps@gmail.com"]
+  spec.summary       = %q{Customized version of Itamae.
+Integration with Serverspec.
+It can be provisioning using itamae's some AWS resources.}
+  spec.homepage      = "https://github.com/kammy1231/itamae-spec"
+  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.required_ruby_version = Gem::Requirement.new(">= 2.0")
+
+  spec.add_runtime_dependency "aws-sdk", "~> 2"
+  spec.add_runtime_dependency "serverspec", "~> 2.1"
+  spec.add_runtime_dependency "itamae", "1.9.11"
+  spec.add_runtime_dependency "rake"
+  spec.add_runtime_dependency "multi_json"
+  spec.add_runtime_dependency "bundler"
+  spec.add_runtime_dependency "highline"
+
+  spec.add_development_dependency "rake-compiler"
+  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"
+end

data/lib/itamae-spec.rb

@@ -0,0 +1,10 @@
+require 'itamae'
+require 'itamae-spec/cli'
+require 'itamae-spec/version'
+require 'itamae-spec/resource'
+require 'itamae-spec/logger'
+require 'itamae-spec/generators'
+
+module ItamaeSpec
+  # Your code goes here
+end

data/lib/itamae-spec/cli.rb

@@ -0,0 +1,19 @@
+require 'itamae-spec'
+
+module ItamaeSpec
+  class CLI < Itamae::CLI
+    # Support for password authentication
+    option :password, type: :string, desc: 'input ssh password'
+
+    def initialize(*)
+      super
+
+      Itamae.logger.level = ::Logger.const_get(options[:log_level].upcase) if options[:log_level]
+      Itamae.logger.formatter.colored = options[:color] if options[:color]
+    end
+
+    def version
+      puts "Itamae-Spec v#{ItamaeSpec::VERSION}"
+    end
+  end
+end

data/lib/itamae-spec/generators.rb

@@ -0,0 +1,20 @@
+require "itamae-spec/generators/project"
+require "itamae-spec/generators/cookbook"
+
+module Itamae
+  module Generators
+    def self.find(target)
+      case target
+      when 'cookbook'
+        Cookbook
+      when 'project'
+        Project
+      when 'role'
+        puts 'Not support generate role. Do nothing.'
+        exit 1
+      else
+        raise "Unexpected target: #{target}"
+      end
+    end
+  end
+end

data/lib/itamae-spec/generators/cookbook.rb

@@ -0,0 +1,10 @@
+
+module Itamae
+  module Generators
+    class Cookbook
+      def self.source_root
+        File.expand_path('../templates/cookbook', __FILE__)
+      end
+    end
+  end
+end

data/lib/itamae-spec/generators/project.rb

@@ -0,0 +1,10 @@
+
+module Itamae
+  module Generators
+    class Project
+      def self.source_root
+        File.dirname(__FILE__) + '/templates/project'
+      end
+    end
+  end
+end

data/lib/itamae-spec/generators/templates/project/.rspec

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

data/lib/itamae-spec/generators/templates/project/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gem 'itamae-spec'

data/lib/itamae-spec/generators/templates/project/Project.json

@@ -0,0 +1 @@
+{"project":"."}

data/lib/itamae-spec/generators/templates/project/Rakefile

@@ -0,0 +1,9 @@
+### Itamae task
+require 'itamae-spec/task/itamae_task'
+
+### Serverspec task
+require 'itamae-spec/task/serverspec_task'
+
+### Local Mode
+# require 'itamae-spec/task/local_itamae_task'
+# require 'itamae-spec/task/local_serverspec_task'

data/lib/itamae-spec/generators/templates/project/cookbooks/sample/attributes/default.json

@@ -0,0 +1,5 @@
+{
+  "sample": {
+
+  }
+}

data/lib/itamae-spec/generators/templates/project/cookbooks/sample/recipes/default.rb

@@ -0,0 +1,7 @@
+
+file "/tmp/sample" do
+  user "root"
+  group "root"
+  mode "755"
+  content "This is a sample."
+end

data/lib/itamae-spec/generators/templates/project/cookbooks/sample/spec/default_spec.rb

@@ -0,0 +1,9 @@
+require 'spec_helper'
+
+describe file('/tmp/sample') do
+  it { should be_file }
+  it { should be_mode 755 }
+  it { should be_owned_by 'root' }
+  it { should be_grouped_into 'root' }
+  its(:content) { should match /This is a sample./ }
+end

data/lib/itamae-spec/generators/templates/project/environments/sample.json

@@ -0,0 +1,7 @@
+{
+  "environments": {
+    "ssh_port": "22",
+    "ssh_user": "ec2-user",
+    "ssh_key":  "sample.pem"
+  }
+}

data/lib/itamae-spec/generators/templates/project/nodes/sample.json

@@ -0,0 +1,10 @@
+{
+  "run_list": [
+    "recipe[sample]",
+    "role[sample]"
+    ],
+  "environments": {
+    "set": "sample",
+    "hostname": "sample"
+  }
+}

data/lib/itamae-spec/generators/templates/project/roles/sample.json

@@ -0,0 +1,5 @@
+{
+  "run_list": [
+    "recipe[sample]"
+  ]
+}

data/lib/itamae-spec/generators/templates/project/spec/spec_helper.rb

@@ -0,0 +1,41 @@
+require 'serverspec'
+require 'net/ssh'
+require 'specinfra/helper/set'
+require 'json'
+include Specinfra::Helper::Set
+
+if ENV['LOCAL_MODE']
+  set :backend, :exec
+else
+  set :backend, :ssh
+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']
+node_file = ENV['NODE_FILE']
+attributes = JSON.parse(File.read(node_file), symbolize_names: true)
+set_property attributes
+
+unless ENV['LOCAL_MODE']
+  options = Net::SSH::Config.for(host)
+  options[:user] = ENV['SSH_USER']
+  options[:password] = ENV['SSH_PASSWORD']
+  options[:keys] = ENV['SSH_KEY']
+  options[:port] = ENV['SSH_PORT']
+
+  set :host, options[:host_name] || host
+  set :shell, '/bin/bash'
+  set :ssh_options, options
+end
+
+set :request_pty, true

data/lib/itamae-spec/logger.rb

@@ -0,0 +1,76 @@
+
+module Itamae
+  module Logger
+    class FileFormatter < Formatter
+      def call(severity, datetime, progname, msg)
+        log = "%s : %s" % ["%5s" % severity, msg2str(msg)]
+        Time.now.strftime('%F %T %z').to_s + log + "\n"
+      end
+    end
+
+    def self.broadcast(logger)
+      Module.new do
+        define_method(:add) do |*args, &block|
+          logger.add(*args, &block)
+          super(*args, &block)
+        end
+
+        define_method(:<<) do |x|
+          logger << x
+          super(x)
+        end
+
+        define_method(:close) do
+          logger.close
+          super()
+        end
+
+        define_method(:progname=) do |name|
+          logger.progname = name
+          super(name)
+        end
+
+        define_method(:formatter=) do |formatter|
+          logger.formatter = formatter
+          super(formatter)
+        end
+
+        define_method(:level=) do |level|
+          logger.level = level
+          super(level)
+        end
+
+        define_method(:local_level=) do |level|
+          logger.local_level = level if logger.respond_to?(:local_level=)
+          super(level) if respond_to?(:local_level=)
+        end
+
+        define_method(:silence) do |level = Logger::ERROR, &block|
+          if logger.respond_to?(:silence)
+            logger.silence(level) do
+              if defined?(super)
+                super(level, &block)
+              else
+                block.call(self)
+              end
+            end
+          else
+            if defined?(super)
+              super(level, &block)
+            else
+              block.call(self)
+            end
+          end
+        end
+      end
+    end
+  end
+
+  if Dir.exist?('logs')
+    @file_logger = ::Logger.new('logs/itamae.log', 'daily').tap do |l|
+      l.formatter = Itamae::Logger::FileFormatter.new
+    end.extend(Itamae::Logger::Helper)
+
+    @logger.extend Itamae::Logger.broadcast(@file_logger)
+  end
+end