itamae-mitsurin 0.1

data/lib/itamae/cli.rb

@@ -0,0 +1,86 @@
+require 'itamae'
+require 'thor'
+
+module Itamae
+  class CLI < Thor
+
+    class_option :log_level, type: :string, aliases: ['-l'], default: 'info'
+    class_option :color, type: :boolean, default: true
+    class_option :config, type: :string, aliases: ['-c']
+
+    def initialize(*)
+      super
+
+      Itamae.logger.level = ::Logger.const_get(options[:log_level].upcase)
+      Itamae.logger.formatter.colored = options[:color]
+    end
+
+    def self.define_exec_options
+      option :recipe_graph, type: :string, desc: "[EXPERIMENTAL] Write recipe dependency graph in DOT", banner: "PATH"
+      option :node_json, type: :string, aliases: ['-j']
+      option :node_yaml, type: :string, aliases: ['-y']
+      option :dry_run, type: :boolean, aliases: ['-n']
+      option :shell, type: :string, default: "/bin/sh"
+      option :ohai, type: :boolean, default: false, desc: "This option is DEPRECATED and will be unavailable."
+      option :profile, type: :string, desc: "[EXPERIMENTAL] Save profiling data", banner: "PATH"
+    end
+
+    desc "local RECIPE [RECIPE...]", "Run Itamae locally"
+    define_exec_options
+    def local(*recipe_files)
+      if recipe_files.empty?
+        raise "Please specify recipe files."
+      end
+
+      Runner.run(recipe_files, :local, options)
+    end
+
+    desc "ssh RECIPE [RECIPE...]", "Run Itamae via ssh"
+    define_exec_options
+    option :host, type: :string, aliases: ['-h']
+    option :user, type: :string, aliases: ['-u']
+    option :key, type: :string, aliases: ['-i']
+    option :port, type: :numeric, aliases: ['-p']
+    option :vagrant, type: :boolean, default: false
+    option :ask_password, type: :boolean, default: false
+    option :sudo, type: :boolean, default: true
+    def ssh(*recipe_files)
+      if recipe_files.empty?
+        raise "Please specify recipe files."
+      end
+
+      unless options[:host] || options[:vagrant]
+        raise "Please set '-h <hostname>' or '--vagrant'"
+      end
+
+      Runner.run(recipe_files, :ssh, options)
+    end
+
+    desc "docker RECIPE [RECIPE...]", "Create Docker image"
+    define_exec_options
+    option :image, type: :string, desc: "This option or 'container' option is required."
+    option :container, type: :string, desc: "This option or 'image' option is required."
+    option :tls_verify_peer, type: :boolean, default: true
+    def docker(*recipe_files)
+      if recipe_files.empty?
+        raise "Please specify recipe files."
+      end
+
+      Runner.run(recipe_files, :docker, options)
+    end
+
+    desc "version", "Print version"
+    def version
+      puts "Itamae v#{Itamae::Mitsurin::VERSION}"
+    end
+
+    private
+    def options
+      @itamae_options ||= super.dup.tap do |options|
+        if config = options[:config]
+          options.merge!(YAML.load_file(config))
+        end
+      end
+    end
+  end
+end

data/lib/itamae/definition.rb

@@ -0,0 +1,40 @@
+require 'itamae'
+
+module Itamae
+  class Definition < Resource::Base
+    class << self
+      attr_accessor :definition_block
+      attr_accessor :defined_in_recipe
+
+      def create_class(name, params, defined_in_recipe, &block)
+        Class.new(self).tap do |klass|
+          klass.definition_block = block
+          klass.defined_in_recipe = defined_in_recipe
+
+          klass.define_attribute :action, default: :run
+          params.each_pair do |key, value|
+            klass.define_attribute key.to_sym, type: Object, default: value
+          end
+        end
+      end
+    end
+
+    def initialize(*args)
+      super
+
+      r = Recipe::RecipeFromDefinition.new(
+        runner,
+        self.class.defined_in_recipe.path,
+      )
+      recipe.children << r
+
+      r.definition = self
+      r.load(params: @attributes.merge(name: resource_name))
+    end
+
+    def run(*args)
+      # nothing
+    end
+  end
+end
+

data/lib/itamae/ext.rb

@@ -0,0 +1 @@
+require 'itamae/ext/specinfra'

data/lib/itamae/ext/specinfra.rb

@@ -0,0 +1,39 @@
+# TODO: Send patches to Specinfra
+
+module Specinfra
+  module Backend
+    class Base
+      def receive_file(from, to = nil)
+        raise NotImplementedError
+      end
+    end
+
+    class Exec < Base
+      def receive_file(from, to = nil)
+        if to
+          FileUtils.cp(from, to)
+        else
+          ::File.read(from)
+        end
+      end
+    end
+
+    class Ssh < Exec
+      def receive_file(from, to = nil)
+        scp_download!(from, to)
+      end
+
+      private
+
+      def scp_download!(from, to, opt={})
+        if get_config(:scp).nil?
+          set_config(:scp, create_scp)
+        end
+
+        scp = get_config(:scp)
+        scp.download!(from, to, opt)
+      end
+    end
+  end
+end
+

data/lib/itamae/generators.rb

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

data/lib/itamae/generators/cookbook.rb

@@ -0,0 +1,22 @@
+require 'thor'
+require 'thor/group'
+
+module Itamae
+  module Generators
+    class Cookbook < Thor::Group
+      include Thor::Actions
+
+      def self.source_root
+        File.expand_path('../templates/cookbook', __FILE__)
+      end
+
+      def copy_files
+        directory '.'
+      end
+
+      def remove_files
+        remove_file '.'
+      end
+    end
+  end
+end

data/lib/itamae/generators/project.rb

@@ -0,0 +1,22 @@
+require 'thor'
+require 'thor/group'
+
+module Itamae
+  module Generators
+    class Project < Thor::Group
+      include Thor::Actions
+
+      def self.source_root
+        File.dirname(__FILE__) + '/templates/project'
+      end
+
+      def copy_files
+        directory '.'
+      end
+
+      def bundle
+        run 'bundle install'
+      end
+    end
+  end
+end

data/lib/itamae/generators/role.rb

@@ -0,0 +1,22 @@
+require 'thor'
+require 'thor/group'
+
+module Itamae
+  module Generators
+    class Role < Thor::Group
+      include Thor::Actions
+
+      def self.source_root
+        File.expand_path('../templates/role', __FILE__)
+      end
+
+      def copy_files
+        directory '.'
+      end
+
+      def remove_files
+        remove_file '.'
+      end
+    end
+  end
+end

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

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+gem 'itamae'
+# gem 'serverspec'

data/lib/itamae/handler.rb

@@ -0,0 +1,21 @@
+require 'itamae/handler/base'
+
+module Itamae
+  module Handler
+    def self.from_type(type)
+      first_time = true
+
+      class_name = type.split('_').map(&:capitalize).join
+      self.const_get(class_name)
+    rescue NameError
+      require "itamae/handler/#{type}"
+
+      if first_time
+        first_time = false
+        retry
+      else
+        raise
+      end
+    end
+  end
+end

data/lib/itamae/handler/base.rb

@@ -0,0 +1,40 @@
+require 'socket'
+
+module Itamae
+  module Handler
+    class Base
+      attr_reader :recipes, :resources, :actions
+
+      def initialize(options)
+        @options = options
+
+        @recipes = []
+        @resources = []
+        @actions = []
+      end
+
+      def event(type, payload = {})
+        case type
+        when :recipe_started
+          @recipes << payload
+        when :recipe_completed, :recipe_failed
+          @recipes.pop
+        when :resource_started
+          @resources << payload
+        when :resource_completed, :resource_failed
+          @resources.pop
+        when :action_started
+          @actions << payload
+        when :action_completed, :action_failed
+          @actions.pop
+        end
+      end
+
+      private
+
+      def hostname
+        @hostname ||= @options['hostname'] || Socket.gethostname
+      end
+    end
+  end
+end

data/lib/itamae/handler/debug.rb

@@ -0,0 +1,10 @@
+module Itamae
+  module Handler
+    class Debug < Base
+      def event(type, payload = {})
+        super
+        Itamae.logger.info("EVENT:#{type} #{payload}")
+      end
+    end
+  end
+end

data/lib/itamae/handler/fluentd.rb

@@ -0,0 +1,44 @@
+module Itamae
+  module Handler
+    class Fluentd < Base
+      attr_accessor :fluent_logger # for test
+
+      def initialize(*)
+        super
+        load_fluent_logger
+      end
+
+      def event(type, payload = {})
+        super
+
+        unless @fluent_logger.post(type, payload.merge(hostname: hostname))
+          Itamae.logger.warn "Sending logs to Fluentd failed: #{@fluent_logger.last_error}"
+        end
+      end
+
+      private
+
+      def load_fluent_logger
+        begin
+          require 'fluent-logger'
+        rescue LoadError
+          raise "Loading fluent-logger gem failed. Please install 'fluent-logger' gem to use fluentd handler."
+        end
+
+        @fluent_logger = Fluent::Logger::FluentLogger.new(tag_prefix, host: fluentd_host, port: fluentd_port)
+      end
+
+      def tag_prefix
+        @options['tag_prefix'] || 'itamae_server'
+      end
+
+      def fluentd_host
+        @options['host'] || 'localhost'
+      end
+
+      def fluentd_port
+        (@options['port'] || 24224).to_i
+      end
+    end
+  end
+end

data/lib/itamae/handler/json.rb

@@ -0,0 +1,22 @@
+module Itamae
+  module Handler
+    class Json < Base
+      def initialize(*)
+        super
+        require 'time'
+        open_file
+      end
+
+      def event(type, payload = {})
+        super
+        @f.puts({'time' => Time.now.iso8601, 'event' => type, 'payload' => payload}.to_json)
+      end
+
+      private
+
+      def open_file
+        @f = open(@options.fetch('path'), 'a')
+      end
+    end
+  end
+end

data/lib/itamae/handler_proxy.rb

@@ -0,0 +1,38 @@
+module Itamae
+  class HandlerProxy
+    def initialize
+      @instances = []
+    end
+
+    def register_instance(instance)
+      @instances << instance
+    end
+
+    def event(*args, &block)
+      if block_given?
+        _event_with_block(*args, &block)
+      else
+        _event(*args)
+      end
+    end
+
+    private
+
+    def _event(*args)
+      @instances.each do |i|
+        i.event(*args)
+      end
+    end
+
+    def _event_with_block(event_name, *args, &block)
+      event("#{event_name}_started".to_sym, *args)
+      block.call
+    rescue
+      event("#{event_name}_failed".to_sym, *args)
+      raise
+    else
+      event("#{event_name}_completed".to_sym, *args)
+    end
+  end
+end
+