#!/usr/bin/env ruby
# -*- coding: utf-8 -*-

require "chef"
require "logger"
require "optparse"
require "chef/fork/commands"
require "chef/fork/version"

class Chef
  class Fork
    class Application
      def initialize()
        @logger = Logger.new(STDERR)
        @optparse = OptionParser.new
        @optparse.version = "#{Chef::Fork::VERSION} (chef #{Chef::VERSION})"
        @options = {}
        define_options
      end
      attr_reader :options
      attr_reader :optparse

      def main(args=[])
        configure_chef(locate_config_file("fork") || locate_config_file("knife"))
        rest = @optparse.order(args)
        begin
          command = get_command(rest.shift || "help").new(self)
          command.run(rest)
        rescue Errno::EPIPE
          # noop
        end
      end

      private
      def define_options()
        options.merge!({
          editor: ENV["EDITOR"],
          verbose: false,
        })
        optparse.on("-c CONFIG", "--config CONFIG", "The configuration file to use") do |value|
          options[:config_file] = value
        end

        optparse.on("-V", "--[no-]verbose", "Run verbosely") do |value|
          options[:verbose] = value
        end

        optparse.on("-E ENVIRONMENT", "--environment ENVIRONMENT", "Set the Chef environment") do |value|
          options[:environment] = value
        end

        optparse.on("-e EDITOR", "--editor EDITOR", "Set the editor to use for interactive commands") do |value|
          options[:editor] = value
        end

        optparse.on("-u USER", "--user USER", "API Client Username") do |value|
          options[:node_name] = value
        end

        optparse.on("-k KEY", "--key KEY", "API Client Key") do |value|
          options[:client_key] = value
        end

        optparse.on("-s URL", "--server-url URL", "Chef Server URL") do |value|
          options[:chef_server_url] = value
        end
      end

      def locate_config_file(name)
        if options.key?(:config_file)
          options[:config_file]
        else
          candidate_configs = []
          if ENV.key?("#{name.upcase}_HOME")
            candidate_config << File.join(File.expand_path(ENV["#{name.upcase}_CONFIG"]), "#{name.downcase}.rb")
          end
          candidate_configs << File.expand_path("#{name.downcase}.rb")
          if config_dir = chef_config_dir(Dir.pwd)
            candidate_configs << File.join(config_dir, "#{name.downcase}.rb")
          end
          if ENV.key?("HOME")
            candidate_configs << File.join(File.expand_path(ENV["HOME"]), "#{name.downcase}.rb")
          end
          candidate_configs.find { |candidate_config| File.exist?(candidate_config) }
        end
      end

      def chef_config_dir(path)
        config_dir = File.join(path, ".chef")
        if File.exist?(config_dir)
          return config_dir
        else
          if path == "/"
            return nil
          else
            return chef_config_dir(File.dirname(path))
          end
        end
      end

      def configure_chef(config_file)
        if config_file
          @logger.info("Using configuration from #{config_file}")
          Chef::Config.from_file(config_file)
        else
          @logger.warn("No fork/knife configuration file found")
        end
        # override with command-line options
        Chef::Config.merge!(options)
      end

      def get_command(name)
        begin
          require "chef/fork/commands/#{name}"
        rescue LoadError
          # ignore LoadError for built-in commands
        end
        class_name = name.to_s.split(/[^\w]+/).map { |s| s.capitalize }.join
        begin
          Chef::Fork::Commands.const_get(class_name)
        rescue NameError
          require "chef/fork/commands/help"
          Chef::Fork::Commands::Help
        end
      end
    end
  end
end