class Generators::Avo::EjectGenerator

def handle
  if @filename.starts_with?(":")
    template_id = path_to_sym @filename
    template_path = TEMPLATES[template_id]
    if path_exists? template_path
      eject template_path
    else
      say("Failed to find the `#{template_id.to_sym}` template.", :yellow)
    end
  elsif path_exists? @filename
    eject @filename
  else
    say("Failed to find the `#{@filename}` template.", :yellow)
  end
end