require 'thor'
require 'berkshelf'

module Berkshelf
  # @author Jamie Winsor <jamie@vialstudios.com>
  class Cli < Thor
    def initialize(*)
      super
      # JW TODO: Replace Chef::Knife::UI with our own UI class
      ::Berkshelf.ui = Chef::Knife::UI.new(STDOUT, STDERR, STDIN, {})
      load_config
      @options = options.dup # unfreeze frozen options Hash from Thor
    rescue BerkshelfError => e
      Berkshelf.ui.fatal e
      exit e.status_code
    end

    namespace "berkshelf"

    map 'in'        => :install
    map 'up'        => :upload
    map 'ud'        => :update
    map 'ver'       => :version

    class_option :config,
      type: :string,
      default: File.expand_path("~/.chef/knife.rb"),
      desc: "Path to Knife or Chef configuration to use.",
      aliases: "-c",
      banner: "PATH"

    method_option :shims,
      type: :string,
      default: nil,
      lazy_default: File.join(Dir.pwd, "cookbooks"),
      desc: "Create a directory of shims pointing to Cookbook Versions.",
      banner: "PATH"
    method_option :without,
      type: :array,
      default: Array.new,
      desc: "Exclude cookbooks that are in these groups.",
      aliases: "-w"
    method_option :berksfile,
      type: :string,
      default: File.join(Dir.pwd, Berkshelf::DEFAULT_FILENAME),
      desc: "Path to a Berksfile to operate off of.",
      aliases: "-b",
      banner: "PATH"
    desc "install", "Install the Cookbooks specified by a Berksfile or a Berskfile.lock."
    def install
      berksfile = ::Berkshelf::Berksfile.from_file(options[:berksfile])
      berksfile.install(options)
    rescue BerkshelfError => e
      Berkshelf.ui.fatal e
      exit e.status_code
    end

    method_option :berksfile,
      type: :string,
      default: File.join(Dir.pwd, Berkshelf::DEFAULT_FILENAME),
      desc: "Path to a Berksfile to operate off of.",
      aliases: "-b",
      banner: "PATH"
    method_option :without,
      type: :array,
      default: Array.new,
      desc: "Exclude cookbooks that are in these groups.",
      aliases: "-w"
    desc "update", "Update all Cookbooks and their dependencies specified by a Berksfile to their latest versions."
    def update
      Lockfile.remove!
      invoke :install
    rescue BerkshelfError => e
      Berkshelf.ui.fatal e
      exit e.status_code
    end

    method_option :berksfile,
      type: :string,
      default: File.join(Dir.pwd, Berkshelf::DEFAULT_FILENAME),
      desc: "Path to a Berksfile to operate off of.",
      aliases: "-b",
      banner: "PATH"
    method_option :without,
      type: :array,
      default: Array.new,
      desc: "Exclude cookbooks that are in these groups.",
      aliases: "-w"
    method_option :freeze,
      type: :boolean,
      default: false,
      desc: "Freeze the uploaded cookbooks so that they cannot be overwritten"
    option :force,
      type: :boolean,
      default: false,
      desc: "Upload all cookbooks even if a frozen one exists on the target Chef Server"
    desc "upload", "Upload the Cookbooks specified by a Berksfile or a Berksfile.lock to a Chef Server."
    def upload
      berksfile = ::Berkshelf::Berksfile.from_file(options[:berksfile])
      berksfile.upload(Chef::Config[:chef_server_url], options)
    rescue BerkshelfError => e
      Berkshelf.ui.fatal e
      exit e.status_code
    end

    desc "init [PATH]", "Prepare a local path to have it's Cookbook dependencies managed by Berkshelf."
    def init(path = Dir.pwd)
      if File.chef_cookbook?(path)
        options[:chefignore] = true
        options[:metadata_entry] = true
      end

      generator = ::Berkshelf::InitGenerator.new([path], options)
      generator.invoke_all

      ::Berkshelf.ui.info "Successfully initialized"
    rescue BerkshelfError => e
      Berkshelf.ui.fatal e
      exit e.status_code
    end

    desc "version", "Display version and copyright information"
    def version
      Berkshelf.ui.info version_header
      Berkshelf.ui.info "\n"
      Berkshelf.ui.info license
    end

    private

      def version_header
        "Berkshelf (#{Berkshelf::VERSION})"
      end

      def license
        File.read(Berkshelf.root.join('LICENSE'))
      end

      def load_config
        Chef::Config.from_file(File.expand_path(options[:config]))
      rescue Errno::ENOENT
        raise KnifeConfigNotFound, "Unable to find a Knife config at #{options[:config]}. Specify a different path with --config."
      end
  end
end