module Berkshelf
  # @author Jamie Winsor <reset@riotgames.com>
  class CookbookGenerator < BaseGenerator
    argument :name,
      type: :string,
      required: true

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

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

    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.instance.cookbook.license

    class_option :maintainer,
      type: :string,
      default: Berkshelf::Config.instance.cookbook.copyright

    class_option :maintainer_email,
      type: :string,
      default: Berkshelf::Config.instance.cookbook.email

    def generate
      empty_directory target.join("files/default")
      empty_directory target.join("templates/default")
      empty_directory target.join("attributes")
      empty_directory target.join("definitions")
      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")
      template "metadata.rb.erb", target.join("metadata.rb")
      template license_file, target.join("LICENSE")
      template "README.md.erb", target.join("README.md")

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

    private

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

      def license_name
        case options[:license]
        when "apachev2"; "Apache 2.0"
        when "gplv2"; "GNU Public License 2.0"
        when "gplv3"; "GNU Public License 3.0"
        when "mit"; "MIT"
        when "reserved"; "All rights reserved"
        else
          raise Berkshelf::InternalError, "Unknown license: '#{options[: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"; "licenses/apachev2.erb"
        when "gplv2"; "licenses/gplv2.erb"
        when "gplv3"; "licenses/gplv3.erb"
        when "mit"; "licenses/mit.erb"
        when "reserved"; "licenses/reserved.erb"
        else
          raise Berkshelf::InternalError, "Unknown license: '#{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