shortdudey123/yamllint

# frozen_string_literal: true

require 'rake'
require 'rake/tasklib'

require 'yamllint'

module YamlLint
  ###
  # RakeTast execution
  #
  class RakeTask < Rake::TaskLib
    attr_accessor :name, :debug, :disable_ext_check, :exclude_paths, :extensions, :fail_on_error, :paths

    def initialize(name = :yamllint)
      super()

      @debug = false
      @disable_ext_check = false
      @exclude_paths = []
      @extensions = nil
      @fail_on_error = true
      @name = name

      yield self if block_given?

      define_task
    end

    private

    # Rake task
    def define_task
      desc 'Run yamllint' unless ::Rake.application.last_description

      task(name) do
        puts 'Running YamlLint...'

        YamlLint.logger.level = Logger::DEBUG if debug

        files_to_check_raw = Rake::FileList.new(paths)
        files_to_exclude = Rake::FileList.new(exclude_paths)
        files_to_check = files_to_check_raw - files_to_exclude

        puts "Checking #{files_to_check.flatten.length} files"
        puts "Excluding #{files_to_exclude.flatten.length} files"

        linter = ::YamlLint::Linter.new(disable_ext_check: disable_ext_check,
                                        extensions: extensions)
        linter.check_all(files_to_check)

        if linter.errors?
          linter.display_errors
          puts "YAML lint found #{linter.errors_count} errors"
          abort('YamlLint failed!') if fail_on_error
        else
          puts 'YamlLint found no errors'
        end
      end
    end
  end
end