class Tapioca::Commands::Todo

def execute
  say("Finding all unresolved constants, this may take a few seconds... ")
  # Clean all existing unresolved constants before regenerating the list
  # so Sorbet won't grab them as already resolved.
  File.delete(@todo_file) if File.exist?(@todo_file)
  constants = unresolved_constants
  if constants.empty?
    say("Nothing to do", :green)
    return
  end
  say("Done", :green)
  contents = rbi(constants, command: default_command(:todo))
  create_file(@todo_file, contents.string, verbose: false)
  name = set_color(@todo_file, :yellow, :bold)
  say("\nAll unresolved constants have been written to #{name}.", [:green, :bold])
  say("Please review changes and commit them.", [:green, :bold])
end

def initialize(todo_file:, file_header:)
  @todo_file = todo_file
  @file_header = file_header
  super()
end

def rbi(constants, command:)
  file = RBI::File.new
  if @file_header
    file.comments << RBI::Comment.new("DO NOT EDIT MANUALLY")
    file.comments << RBI::Comment.new("This is an autogenerated file for unresolved constants.")
    file.comments << RBI::Comment.new("Please instead update this file by running `#{command}`.")
    file.comments << RBI::BlankLine.new
  end
  file.comments << RBI::Comment.new("typed: false")
  constants.each do |name|
    file << RBI::Module.new(name)
  end
  file
end

def run_with_deprecation
  say(DEPRECATION_MESSAGE, :red)
  say("")
  run
end

def unresolved_constants
  # Taken from https://github.com/sorbet/sorbet/blob/master/gems/sorbet/lib/todo-rbi.rb
  sorbet("--print=missing-constants", "--quiet", "--stdout-hup-hack", "--no-error-count")
    .out
    .strip
    .each_line
    .filter_map do |line|
      next if line.include?("<")
      line.strip
        .gsub(/T\.class_of\(([:\w]+)\)/, '\1') # Turn T.class_of(Foo)::Bar into Foo::Bar
    end
    .sort
end