# typed: strict
# frozen_string_literal: true

module RubyLsp
  module Requests
    module Support
      class Sorbet
        class << self
          extend T::Sig

          ANNOTATIONS = T.let(
            {
              abstract!: Annotation.new(arity: 0),
              absurd: Annotation.new(arity: 1, receiver: true),
              all: Annotation.new(arity: (2..), receiver: true),
              any: Annotation.new(arity: (2..), receiver: true),
              assert_type!: Annotation.new(arity: 2, receiver: true),
              attached_class: Annotation.new(arity: 0, receiver: true),
              bind: Annotation.new(arity: 2, receiver: true),
              cast: Annotation.new(arity: 2, receiver: true),
              class_of: Annotation.new(arity: 1, receiver: true),
              enums: Annotation.new(arity: 0),
              interface!: Annotation.new(arity: 0),
              let: Annotation.new(arity: 2, receiver: true),
              mixes_in_class_methods: Annotation.new(arity: 1),
              must: Annotation.new(arity: 1, receiver: true),
              must_because: Annotation.new(arity: 1, receiver: true),
              nilable: Annotation.new(arity: 1, receiver: true),
              noreturn: Annotation.new(arity: 0, receiver: true),
              requires_ancestor: Annotation.new(arity: 0),
              reveal_type: Annotation.new(arity: 1, receiver: true),
              sealed!: Annotation.new(arity: 0),
              self_type: Annotation.new(arity: 0, receiver: true),
              sig: Annotation.new(arity: 0),
              type_member: Annotation.new(arity: (0..1)),
              type_template: Annotation.new(arity: 0),
              unsafe: Annotation.new(arity: 1),
              untyped: Annotation.new(arity: 0, receiver: true),
            },
            T::Hash[Symbol, Annotation],
          )

          sig do
            params(
              node: Prism::CallNode,
            ).returns(T::Boolean)
          end
          def annotation?(node)
            !!ANNOTATIONS[node.name]&.match?(node)
          end
        end
      end
    end
  end
end