itamae-mitsurin 0.4 → 0.7

data/test/test_itamae-mitsurin/resource/test_directory.rb

@@ -0,0 +1,63 @@
+require 'itamae-mitsurin'
+
+module ItamaeMitsurin
+  module Resource
+    class Directory < Base
+      define_attribute :action, default: :create
+      define_attribute :path, type: String, default_name: true
+      define_attribute :mode, type: String
+      define_attribute :owner, type: String
+      define_attribute :group, type: String
+
+      def pre_action
+        case @current_action
+        when :create
+          attributes.exist = true
+        when :delete
+          attributes.exist = false
+        end
+      end
+
+      def show_differences
+        current.mode    = current.mode.rjust(4, '0') if current.mode
+        attributes.mode = attributes.mode.rjust(4, '0') if attributes.mode
+
+        super
+      end
+
+      def set_current_attributes
+        exist = run_specinfra(:check_file_is_directory, attributes.path)
+        current.exist = exist
+
+        if 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 action_create(options)
+        if !run_specinfra(:check_file_is_directory, attributes.path)
+          run_specinfra(:create_file_as_directory, attributes.path)
+        end
+        if attributes.mode
+          run_specinfra(:change_file_mode, attributes.path, attributes.mode)
+        end
+        if attributes.owner || attributes.group
+          run_specinfra(:change_file_owner, attributes.path, attributes.owner, attributes.group)
+        end
+      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/test/test_itamae-mitsurin/resource/test_execute.rb

@@ -0,0 +1,26 @@
+require 'itamae-mitsurin'
+
+module ItamaeMitsurin
+  module Resource
+    class Execute < Base
+      define_attribute :action, default: :run
+      define_attribute :command, type: String, default_name: true
+
+      def pre_action
+        case @current_action
+        when :run
+          attributes.executed = true
+        end
+      end
+
+      def set_current_attributes
+        current.executed = false
+      end
+
+      def action_run(options)
+        run_command(attributes.command)
+        updated!
+      end
+    end
+  end
+end

data/{lib/itamae/resource/file.rb → test/test_itamae-mitsurin/resource/test_file.rb}

@@ -1,6 +1,6 @@
-require 'itamae'
+require 'itamae-mitsurin'
 
-module Itamae
+module ItamaeMitsurin
   module Resource
     class File < Base
       define_attribute :action, default: :create
@@ -125,9 +125,9 @@ module Itamae
         diff = run_command(["diff", "-u", compare_to, @temppath], error: false)
         if diff.exit_status == 0
           # no change
-          Itamae.logger.debug "file content will not change"
+          ItamaeMitsurin.logger.debug "file content will not change"
         else
-          Itamae.logger.info "diff:"
+          ItamaeMitsurin.logger.info "diff:"
           diff.stdout.each_line do |line|
             color = if line.start_with?('+')
                       :green
@@ -136,8 +136,8 @@ module Itamae
                     else
                       :clear
                     end
-            Itamae.logger.color(color) do
-              Itamae.logger.info line.chomp
+            ItamaeMitsurin.logger.color(color) do
+              ItamaeMitsurin.logger.info line.chomp
             end
           end
           runner.handler.event(:file_content_changed, diff: diff.stdout)

data/test/test_itamae-mitsurin/resource/test_gem_package.rb

@@ -0,0 +1,81 @@
+require 'itamae-mitsurin'
+
+module ItamaeMitsurin
+  module Resource
+    class GemPackage < Base
+      define_attribute :action, default: :install
+      define_attribute :package_name, type: String, default_name: true
+      define_attribute :gem_binary, type: [String, Array], default: 'gem'
+      define_attribute :options, type: [String, Array], default: []
+      define_attribute :version, type: String
+      define_attribute :source, type: String
+
+      def pre_action
+        case @current_action
+        when :install
+          attributes.installed = true
+        end
+      end
+
+      def set_current_attributes
+        installed = installed_gems.find {|g| g[:name] == attributes.package_name }
+        current.installed = !!installed
+
+        if current.installed
+          versions = installed[:versions]
+          if versions.include?(attributes.version)
+            current.version = attributes.version
+          else
+            current.version = versions.first
+          end
+        end
+      end
+
+      def action_install(action_options)
+        if current.installed
+          if attributes.version && current.version != attributes.version
+            install!
+            updated!
+          end
+        else
+          install!
+          updated!
+        end
+      end
+
+      def action_upgrade(action_options)
+        return if current.installed && attributes.version && current.version == attributes.version
+
+        install!
+        updated!
+      end
+
+      def installed_gems
+        gems = []
+        run_command([*Array(attributes.gem_binary), 'list', '-l']).stdout.each_line do |line|
+          if /\A([^ ]+) \(([^\)]+)\)\z/ =~ line.strip
+            name = $1
+            versions = $2.split(', ')
+            gems << {name: name, versions: versions}
+          end
+        end
+        gems
+      rescue Backend::CommandExecutionError
+        []
+      end
+
+      def install!
+        cmd = [*Array(attributes.gem_binary), 'install', *Array(attributes.options)]
+        if attributes.version
+          cmd << '-v' << attributes.version
+        end
+        if attributes.source
+          cmd << '--source' << attributes.source
+        end
+        cmd << attributes.package_name
+
+        run_command(cmd)
+      end
+    end
+  end
+end

data/test/test_itamae-mitsurin/resource/test_git.rb

@@ -0,0 +1,94 @@
+require 'itamae-mitsurin'
+
+module ItamaeMitsurin
+  module Resource
+    class Git < Base
+      DEPLOY_BRANCH = "deploy"
+
+      define_attribute :action, default: :sync
+      define_attribute :destination, type: String, default_name: true
+      define_attribute :repository, type: String, required: true
+      define_attribute :revision, type: String
+      define_attribute :recursive, type: [TrueClass, FalseClass], default: false
+
+      def pre_action
+        case @current_action
+        when :sync
+          attributes.exist = true
+        end
+      end
+
+      def set_current_attributes
+        current.exist = run_specinfra(:check_file_is_directory, attributes.destination)
+      end
+
+      def action_sync(options)
+        ensure_git_available
+
+        new_repository = false
+
+        if check_empty_dir
+          cmd = ['git', 'clone']
+          cmd << '--recursive' if attributes.recursive
+          cmd << attributes.repository << attributes.destination
+          run_command(cmd)
+          new_repository = true
+        end
+
+        target = if attributes.revision
+                   get_revision(attributes.revision)
+                 else
+                   fetch_origin!
+                   run_command_in_repo("git ls-remote origin HEAD | cut -f1").stdout.strip
+                 end
+
+        if new_repository || target != get_revision('HEAD')
+          updated!
+
+          deploy_old_created = false
+          if current_branch == DEPLOY_BRANCH
+            run_command_in_repo("git branch -m deploy-old")
+            deploy_old_created = true
+          end
+
+          fetch_origin!
+          run_command_in_repo(["git", "checkout", target, "-b", DEPLOY_BRANCH])
+
+          if deploy_old_created
+            run_command_in_repo("git branch -d deploy-old")
+          end
+        end
+      end
+
+      private
+      def ensure_git_available
+        unless run_command("which git", error: false).exit_status == 0
+          raise "`git` command is not available. Please install git."
+        end
+      end
+
+      def check_empty_dir
+        run_command("test -z \"$(ls -A #{shell_escape(attributes.destination)})\"", error: false).success?
+      end
+
+      def run_command_in_repo(*args)
+        run_command(*args, cwd: attributes.destination)
+      end
+
+      def current_branch
+        run_command_in_repo("git rev-parse --abbrev-ref HEAD").stdout.strip
+      end
+
+      def get_revision(branch)
+        run_command_in_repo("git rev-list #{shell_escape(branch)} | head -n1").stdout.strip
+      end
+
+      def fetch_origin!
+        return if @origin_fetched
+        @origin_fetched = true
+        run_command_in_repo(['git', 'fetch', 'origin'])
+      end
+    end
+  end
+end
+

data/test/test_itamae-mitsurin/resource/test_group.rb

@@ -0,0 +1,42 @@
+require 'itamae-mitsurin'
+
+module ItamaeMitsurin
+  module Resource
+    class Group < Base
+      define_attribute :action, default: :create
+      define_attribute :groupname, type: String, default_name: true
+      define_attribute :gid, type: Integer
+
+      def set_current_attributes
+        current.exist = exist?
+
+        if current.exist
+          current.gid = run_specinfra(:get_group_gid, attributes.groupname).stdout.strip.to_i
+        end
+      end
+
+      def action_create(options)
+        if run_specinfra(:check_group_exists, attributes.groupname)
+          if attributes.gid && attributes.gid != current.gid
+            run_specinfra(:update_group_gid, attributes.groupname, attributes.gid)
+            updated!
+          end
+        else
+          options = {
+            gid: attributes.gid,
+          }
+
+          run_specinfra(:add_group, attributes.groupname, options)
+
+          updated!
+        end
+      end
+
+      private
+      def exist?
+        run_specinfra(:check_group_exists, attributes.groupname)
+      end
+    end
+  end
+end
+

data/test/test_itamae-mitsurin/resource/test_http_request.rb

@@ -0,0 +1,71 @@
+require 'itamae-mitsurin'
+require 'uri'
+require 'net/https'
+
+module ItamaeMitsurin
+  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
+              ItamaeMitsurin.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/test/test_itamae-mitsurin/resource/test_link.rb

@@ -0,0 +1,33 @@
+require 'itamae-mitsurin'
+
+module ItamaeMitsurin
+  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