module Berkshelf
  class CookbookGenerator < BaseGenerator
    require_relative "config"

    LICENSE_MAP = {
      "apachev2" => "Apache 2.0",
      "gplv2"    => "GNU Public License 2.0",
      "gplv3"    => "GNU Public License 3.0",
      "mit"      => "MIT",
      "reserved" => "All rights reserved",
    }.freeze

    (LICENSES = LICENSE_MAP.keys).freeze

    argument :name,
      type: :string,
      required: true

    class_option :skip_vagrant,
      type: :boolean,
      default: false

    class_option :skip_git,
      type: :boolean,
      default: false

    class_option :skip_test_kitchen,
      type: :boolean,
      default: false,
      desc: "Skip adding a testing environment to your cookbook"

    class_option :foodcritic,
      type: :boolean,
      default: false

    class_option :chef_minitest,
      type: :boolean,
      default: false

    class_option :scmversion,
      type: :boolean,
      default: false

    class_option :no_bundler,
      type: :boolean,
      default: false

    class_option :license,
      type: :string,
      default: Berkshelf.config.cookbook.license

    class_option :maintainer,
      type: :string,
      default: Berkshelf.config.cookbook.copyright

    class_option :maintainer_email,
      type: :string,
      default: Berkshelf.config.cookbook.email

    def generate
      case options[:pattern]
      when "library"
        empty_directory target.join("libraries")
        empty_directory target.join("providers")
        empty_directory target.join("resources")
      when "wrapper"
        empty_directory target.join("attributes")
        empty_directory target.join("recipes")
        template "default_recipe.erb", target.join("recipes/default.rb")
      when "environment", "application"
        empty_directory target.join("files/default")
        empty_directory target.join("templates/default")
        empty_directory target.join("attributes")
        empty_directory target.join("libraries")
        empty_directory target.join("providers")
        empty_directory target.join("recipes")
        empty_directory target.join("resources")
        template "default_recipe.erb", target.join("recipes/default.rb")
      end

      template "metadata.rb.erb", target.join("metadata.rb")
      template license_file, target.join("LICENSE")
      template "README.md.erb", target.join("README.md")
      template "CHANGELOG.md.erb", target.join("CHANGELOG.md")

      Berkshelf::InitGenerator.new([target], options.merge(default_options)).invoke_all
    end

    private

    def commented(content)
      content.split("\n").collect { |s| s == "" ? "#" : "# #{s}" }.join("\n")
    end

    def license_name
      LICENSE_MAP.fetch(options[:license]) do |license|
        raise Berkshelf::LicenseNotFound.new(license)
      end
    end

    def license
      ERB.new(File.read(File.join(self.class.source_root, license_file))).result(binding)
    end

    def license_file
      case options[:license]
      when "apachev2" then "licenses/apachev2.erb"
      when "gplv2" then "licenses/gplv2.erb"
      when "gplv3" then "licenses/gplv3.erb"
      when "mit" then "licenses/mit.erb"
      when "reserved" then "licenses/reserved.erb"
      else
        raise Berkshelf::LicenseNotFound.new(options[:license])
      end
    end

    def copyright_year
      Time.now.year
    end

    def maintainer
      options[:maintainer]
    end

    def maintainer_email
      options[:maintainer_email]
    end

    def default_options
      { metadata_entry: true, chefignore: true, cookbook_name: name }
    end
  end
end