khozlov/kitchen-scribe

#
# Author:: Pawel Kozlowski (<pawel.kozlowski@u2i.com>)
# Copyright:: Copyright (c) 2013 Pawel Kozlowski
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#


require 'chef/mixin/shell_out'

class Chef
  class Knife
    class ScribeHire < Chef::Knife

      include Chef::Mixin::ShellOut

      DEFAULT_CHRONICLE_PATH = ".chronicle" unless const_defined?(:DEFAULT_CHRONICLE_PATH)
      DEFAULT_REMOTE_NAME = "origin" unless const_defined?(:DEFAULT_REMOTE_NAME)

      banner "knife scribe hire"

      option :chronicle_path,
      :short => "-p PATH",
      :long => "--chronicle-path PATH",
      :description => "Path to the directory where the chronicle should be located",
      :default => nil

      option :remote_name,
      :long => "--remote-name REMOTE_NAME",
      :description => "Name of the remote chronicle repository",
      :default => nil

      option :remote_url,
      :short => "-r REMOTE_URL",
      :long => "--remote-url REMOTE_URL",
      :description => "Url of the remote chronicle repository",
      :default => nil

      def run
        configure
        Dir.mkdir(config[:chronicle_path]) unless File.directory?(config[:chronicle_path])
        init_chronicle
        setup_remote if config[:remote_url]
        ["environments", "nodes", "roles"].each do |dir|
          path = File.join(config[:chronicle_path], dir)
          Dir.mkdir(path) unless File.directory?(path)
        end
      end

      def configure
        conf = { :chronicle_path => DEFAULT_CHRONICLE_PATH,
          :remote_name => DEFAULT_REMOTE_NAME }
        conf.merge!(Chef::Config[:knife][:scribe]) if Chef::Config[:knife][:scribe].kind_of? Hash
        conf.each do |key, value|
          config[key] ||= value
        end
      end

      def init_chronicle
        shell_out!("git init", { :cwd => config[:chronicle_path] })
      end

      def setup_remote
        check_remote_command = "git config --get remote.#{config[:remote_name]}.url"
        remote_status = shell_out!(check_remote_command, { :cwd => config[:chronicle_path], :returns => [0,1,2] })
        case remote_status.exitstatus
        when 0, 2
          # In theory 2 should not happen unless somebody messed with
          # the checkout manually, but using --replace-all option will fix it
          unless remote_status.exitstatus != 2 && remote_status.stdout.strip.eql?(config[:remote_url])
            update_remote_url_command = "git config --replace-all remote.#{config[:remote_name]}.url #{config[:remote_url]}"
            shell_out!(update_remote_url_command, { :cwd => config[:chronicle_path] })
          end
        when 1
          add_remote_command = "git remote add #{config[:remote_name]} #{config[:remote_url]}"
          shell_out!(add_remote_command, { :cwd => config[:chronicle_path] })
        end
      end
    end
  end
end