draffensperger/type_tracer

# frozen_string_literal: true
require 'type_tracer'

desc 'Checks for undefined instance methods'
namespace :type_tracer do
  task check_method_calls: :environment do |_, _args|
    root_dir = defined?(Rails) ? Rails.root : '.'
    ruby_files = Dir.glob(root_dir.join('app/**/*.rb'))
    ruby_files.each { |file| load(file) }

    # Call the configured attribute methods definer to define
    # methods on classes, e.g. initialize ActiveRecord models
    TypeTracer.config.attribute_methods_definer.try(:call)

    exit(1) unless undefined_method_messages(ruby_files).empty?
  end

  def undefined_method_messages(ruby_files)
    ruby_files.flat_map do |file|
      ast = TypeTracer.parse_file(file)
      messages = TypeTracer::InstanceMethodChecker.new(ast).undefined_method_messages
      messages.each(&method(:puts))
      messages
    end
  end

  task check_arg_sends: :environment do |_, _args|
    root_dir = TypeTracer.config.type_check_root_path.to_s + '/'
    files = Dir.glob(File.join(root_dir, '**/*.rb'))
    TypeTracer.config.type_check_path_regex = %r{\A(app|lib)/}
    type_check_files = files.select do |file|
      project_path = file[root_dir.size..-1]
      project_path =~ TypeTracer.config.type_check_path_regex
    end

    type_check_files.each { |file| load(file) }

    runner = TypeTracer::ArgSendTypeCheck::Runner.new(type_check_files)
    bad_arg_sends = runner.bad_arg_sends
    bad_arg_sends.each { |message| puts message }
    exit(1) if bad_arg_sends.present?
  end
end