itamae-mitsurin 0.1

data/lib/itamae/resource/http_request.rb

@@ -0,0 +1,71 @@
+require 'itamae'
+require 'uri'
+require 'net/https'
+
+module Itamae
+  module Resource
+    class HttpRequest < File
+      RedirectLimitExceeded = Class.new(StandardError)
+
+      define_attribute :action, default: :get
+      define_attribute :headers, type: Hash, default: {}
+      define_attribute :message, type: String, default: ""
+      define_attribute :redirect_limit, type: Integer, default: 10
+      define_attribute :url, type: String, required: true
+
+      def pre_action
+        uri = URI.parse(attributes.url)
+        response = nil
+        redirects_followed = 0
+
+        loop do
+          http = Net::HTTP.new(uri.host, uri.port)
+          http.use_ssl = true if uri.scheme == "https"
+
+          case attributes.action
+          when :delete, :get, :options
+            response = http.method(attributes.action).call(uri.request_uri, attributes.headers)
+          when :post, :put
+            response = http.method(attributes.action).call(uri.request_uri, attributes.message, attributes.headers)
+          end
+
+          if response.kind_of?(Net::HTTPRedirection)
+            if redirects_followed < attributes.redirect_limit
+              uri = URI.parse(response["location"])
+              redirects_followed += 1
+              Itamae.logger.debug "Following redirect #{redirects_followed}/#{attributes.redirect_limit}"
+            else
+              raise RedirectLimitExceeded
+            end
+          else
+            break
+          end
+        end
+
+        attributes.content = response.body
+
+        super
+      end
+
+      def action_delete(options)
+        action_create(options)
+      end
+
+      def action_get(options)
+        action_create(options)
+      end
+
+      def action_options(options)
+        action_create(options)
+      end
+
+      def action_post(options)
+        action_create(options)
+      end
+
+      def action_put(options)
+        action_create(options)
+      end
+    end
+  end
+end

data/lib/itamae/resource/link.rb

@@ -0,0 +1,33 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class Link < Base
+      define_attribute :action, default: :create
+      define_attribute :link, type: String, default_name: true
+      define_attribute :to, type: String, required: true
+      define_attribute :force, type: [TrueClass, FalseClass], default: false
+
+      def pre_action
+        case @current_action
+        when :create
+          attributes.exist = true
+        end
+      end
+
+      def set_current_attributes
+        current.exist = run_specinfra(:check_file_is_link, attributes.link)
+
+        if current.exist
+          current.to = run_specinfra(:get_file_link_target, attributes.link).stdout.strip
+        end
+      end
+
+      def action_create(options)
+        unless run_specinfra(:check_file_is_linked_to, attributes.link, attributes.to)
+          run_specinfra(:link_file_to, attributes.link, attributes.to, force: attributes.force)
+        end
+      end
+    end
+  end
+end

data/lib/itamae/resource/local_ruby_block.rb

@@ -0,0 +1,15 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class LocalRubyBlock < Base
+      define_attribute :action, default: :run
+      define_attribute :block, type: Proc
+
+      def action_run(options)
+        attributes.block.call
+      end
+    end
+  end
+end
+

data/lib/itamae/resource/package.rb

@@ -0,0 +1,44 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class Package < Base
+      define_attribute :action, default: :install
+      define_attribute :name, type: String, default_name: true
+      define_attribute :version, type: String
+      define_attribute :options, type: String
+
+      def pre_action
+        case @current_action
+        when :install
+          attributes.installed = true
+        when :remove
+          attributes.installed = false
+        end
+      end
+
+      def set_current_attributes
+        current.installed = run_specinfra(:check_package_is_installed, attributes.name)
+
+        if current.installed
+          current.version = run_specinfra(:get_package_version, attributes.name).stdout.strip
+        end
+      end
+
+      def action_install(action_options)
+        unless run_specinfra(:check_package_is_installed, attributes.name, attributes.version)
+          run_specinfra(:install_package, attributes.name, attributes.version, attributes.options)
+          updated!
+        end
+      end
+
+      def action_remove(action_options)
+        if run_specinfra(:check_package_is_installed, attributes.name, nil)
+          run_specinfra(:remove_package, attributes.name, attributes.options)
+          updated!
+        end
+      end
+    end
+  end
+end
+

data/lib/itamae/resource/remote_directory.rb

@@ -0,0 +1,84 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class RemoteDirectory < Base
+      define_attribute :action, default: :create
+      define_attribute :path, type: String, default_name: true
+      define_attribute :source, type: String, required: true
+      define_attribute :mode, type: String
+      define_attribute :owner, type: String
+      define_attribute :group, type: String
+
+      def pre_action
+        directory = ::File.expand_path(attributes.source, ::File.dirname(@recipe.path))
+        src = ::File.expand_path(directory, ::File.dirname(@recipe.path))
+
+        @temppath = ::File.join(runner.tmpdir, Time.now.to_f.to_s)
+        backend.send_directory(src, @temppath)
+
+        case @current_action
+        when :create
+          attributes.exist = true
+        end
+      end
+
+      def set_current_attributes
+        current.exist = run_specinfra(:check_file_is_directory, attributes.path)
+
+        if current.exist
+          current.mode  = run_specinfra(:get_file_mode, attributes.path).stdout.chomp
+          current.owner = run_specinfra(:get_file_owner_user, attributes.path).stdout.chomp
+          current.group = run_specinfra(:get_file_owner_group, attributes.path).stdout.chomp
+        else
+          current.mode  = nil
+          current.owner = nil
+          current.group = nil
+        end
+      end
+
+      def show_differences
+        super
+
+        if current.exist
+          diff = run_command(["diff", "-u", attributes.path, @temppath], error: false)
+          if diff.exit_status == 0
+            # no change
+            Itamae.logger.debug "directory content will not change"
+          else
+            Itamae.logger.info "diff:"
+            diff.stdout.each_line do |line|
+              Itamae.logger.info "#{line.strip}"
+            end
+          end
+        end
+      end
+
+      def action_create(options)
+        if attributes.mode
+          run_specinfra(:change_file_mode, @temppath, attributes.mode)
+        end
+        if attributes.owner || attributes.group
+          run_specinfra(:change_file_owner, @temppath, attributes.owner, attributes.group)
+        end
+
+        if run_specinfra(:check_file_is_file, attributes.path)
+          unless check_command(["diff", "-q", @temppath, attributes.path])
+            updated!
+          end
+        else
+          updated!
+        end
+
+        run_specinfra(:remove_file, attributes.path)
+        run_specinfra(:move_file, @temppath, attributes.path)
+      end
+
+      def action_delete(options)
+        if run_specinfra(:check_file_is_directory, attributes.path)
+          run_specinfra(:remove_file, attributes.path)
+        end
+      end
+    end
+  end
+end

data/lib/itamae/resource/remote_file.rb

@@ -0,0 +1,54 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class RemoteFile < File
+      SourceNotFoundError = Class.new(StandardError)
+
+      define_attribute :source, type: [String, Symbol], default: :auto
+
+      private
+
+      def content_file
+        source_file
+      end
+
+      def source_file
+        @source_file ||= find_source_file
+      end
+
+      def find_source_file
+        if attributes.source == :auto
+          dirs = attributes.path.split(::File::SEPARATOR)
+          dirs.shift if dirs.first == ""
+
+          searched_paths = []
+          dirs.size.times do |i|
+            source_file_exts.each do |ext|
+              path = ::File.join(@recipe.dir, source_file_dir, "#{dirs[i..-1].join("/")}#{ext}")
+              if ::File.exist?(path)
+                Itamae.logger.debug "#{path} is used as a source file."
+                return path
+              else
+                searched_paths << path
+              end
+            end
+          end
+
+          raise SourceNotFoundError, "source file is not found (searched paths: #{searched_paths.join(', ')})"
+        else
+          ::File.expand_path(attributes.source, @recipe.dir)
+        end
+      end
+
+      def source_file_dir
+        "files"
+      end
+
+      def source_file_exts
+        [""]
+      end
+    end
+  end
+end
+

data/lib/itamae/resource/service.rb

@@ -0,0 +1,69 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class Service < Base
+      define_attribute :action, default: :nothing
+      define_attribute :name, type: String, default_name: true
+      define_attribute :provider, type: Symbol, default: nil
+
+      def initialize(*args)
+        super
+        @under = attributes.provider ? "_under_#{attributes.provider}" : ""
+      end
+
+      def pre_action
+        case @current_action
+        when :start, :restart
+          attributes.running = true
+        when :stop
+          attributes.running = false
+        when :enable
+          attributes.enabled = true
+        when :disable
+          attributes.enabled = false
+        end
+      end
+
+      def set_current_attributes
+        current.running = run_specinfra(:"check_service_is_running#{@under}", attributes.name)
+        current.enabled = run_specinfra(:"check_service_is_enabled#{@under}", attributes.name)
+      end
+
+      def action_start(options)
+        unless current.running
+          run_specinfra(:"start_service#{@under}", attributes.name)
+        end
+      end
+
+      def action_stop(options)
+        if current.running
+          run_specinfra(:"stop_service#{@under}", attributes.name)
+        end
+      end
+
+      def action_restart(options)
+        run_specinfra(:"restart_service#{@under}", attributes.name)
+      end
+
+      def action_reload(options)
+        if current.running
+          run_specinfra(:"reload_service#{@under}", attributes.name)
+        end
+      end
+
+      def action_enable(options)
+        unless current.enabled
+          run_specinfra(:"enable_service#{@under}", attributes.name)
+        end
+      end
+
+      def action_disable(options)
+        if current.enabled
+          run_specinfra(:"disable_service#{@under}", attributes.name)
+        end
+      end
+    end
+  end
+end
+

data/lib/itamae/resource/template.rb

@@ -0,0 +1,53 @@
+require 'itamae'
+require 'erb'
+require 'tempfile'
+
+module Itamae
+  module Resource
+    class Template < RemoteFile
+      define_attribute :variables, type: Hash, default: {}
+
+      def pre_action
+        attributes.content = RenderContext.new(self).render_file(source_file)
+
+        super
+      end
+
+      private
+
+      def content_file
+        nil
+      end
+
+      def source_file_dir
+        "templates"
+      end
+
+      def source_file_exts
+        [".erb", ""]
+      end
+
+      class RenderContext
+        def initialize(resource)
+          @resource = resource
+
+          @resource.attributes.variables.each_pair do |key, value|
+            instance_variable_set("@#{key}".to_sym, value)
+          end
+        end
+
+        def render_file(src)
+          template = ::File.read(src)
+          ERB.new(template, nil, '-').tap do |erb|
+            erb.filename = src
+          end.result(binding)
+        end
+
+        def node
+          @resource.recipe.runner.node
+        end
+      end
+    end
+  end
+end
+

data/lib/itamae/resource/user.rb

@@ -0,0 +1,93 @@
+require 'itamae'
+
+module Itamae
+  module Resource
+    class User < Base
+      define_attribute :action, default: :create
+      define_attribute :username, type: String, default_name: true
+      define_attribute :gid, type: [Integer, String]
+      define_attribute :home, type: String
+      define_attribute :password, type: String
+      define_attribute :system_user, type: [TrueClass, FalseClass]
+      define_attribute :uid, type: Integer
+      define_attribute :shell, type: String
+      define_attribute :create_home, type: [TrueClass, FalseClass], default: false
+
+      def pre_action
+        case @current_action
+        when :create
+          attributes.exist = true
+        end
+      end
+
+      def set_current_attributes
+        current.exist = exist?
+
+        if current.exist
+          current.uid = run_specinfra(:get_user_uid, attributes.username).stdout.strip.to_i
+          current.gid = run_specinfra(:get_user_gid, attributes.username).stdout.strip.to_i
+          current.home = run_specinfra(:get_user_home_directory, attributes.username).stdout.strip
+          current.shell = run_specinfra(:get_user_login_shell, attributes.username).stdout.strip
+          current.password = current_password
+        end
+      end
+
+      def action_create(options)
+        if run_specinfra(:check_user_exists, attributes.username)
+          if attributes.uid && attributes.uid != current.uid
+            run_specinfra(:update_user_uid, attributes.username, attributes.uid)
+            updated!
+          end
+
+          if attributes.gid && attributes.gid != current.gid
+            run_specinfra(:update_user_gid, attributes.username, attributes.gid)
+            updated!
+          end
+
+          if attributes.password && attributes.password != current.password
+            run_specinfra(:update_user_encrypted_password, attributes.username, attributes.password)
+            updated!
+          end
+
+          if attributes.home && attributes.home != current.home
+            run_specinfra(:update_user_home_directory, attributes.username, attributes.home)
+            updated!
+          end
+
+          if attributes.shell && attributes.shell != current.shell
+            run_specinfra(:update_user_login_shell, attributes.username, attributes.shell)
+            updated!
+          end
+        else
+          options = {
+            gid:            attributes.gid,
+            home_directory: attributes.home,
+            password:       attributes.password,
+            system_user:    attributes.system_user,
+            uid:            attributes.uid,
+            shell:          attributes.shell,
+            create_home:    attributes.create_home,
+          }
+
+          run_specinfra(:add_user, attributes.username, options)
+
+          updated!
+        end
+      end
+
+      private
+      def exist?
+        run_specinfra(:check_user_exists, attributes.username)
+      end
+
+      def current_password
+        result = run_specinfra(:get_user_encrypted_password, attributes.username)
+        if result.success?
+          result.stdout.strip
+        else
+          nil
+        end
+      end
+    end
+  end
+end