require 'rake/tasklib'

module Rake::Funnel::Tasks
  class SideBySideSpecs < Rake::TaskLib
    include Rake::Funnel::Support

    attr_accessor :name, :projects, :references, :specs, :enabled

    def initialize(*args, &task_block)
      setup_ivars(args)

      define(args, &task_block)
    end

    private
    def setup_ivars(args)
      @name = args.shift || :compile

      @projects = %w(**/*.csproj **/*.vbproj **/*.fsproj)
      @references = []
      @specs = %w(*Specs.cs **/*Specs.cs *Tests.cs **/*Tests.cs)
      @enabled = false
    end

    def define(args, &task_block)
      desc 'Remove tests from projects' unless Rake.application.last_description
      task(name, *args) do |_, task_args|
        task_block.call(*[self, task_args].slice(0, task_block.arity)) if task_block

        next unless enabled
        SpecsRemover.remove({ projects: projects, references: references, specs: specs })
      end

      self
    end
  end
end