module Rouge::Lexers::TypescriptCommon

def self.extended(base)
  base.prepend :root do
    rule %r/[?][.]/, base::Punctuation
    rule %r/[?]{2}/, base::Operator
    # Positive Examples:
    # const cat = { name: "Garfield" } satisfies Person;
    # import {something as thingy} from 'module'
    # export { foo as default }
    # ...spreadOperator as const;
    # Negative Example:
    # cy.get('kitten').as('friend')
    rule %r{(?<![_$[:alnum:]])(?:(?<=\.\.\.)|(?<!\.))(?:(as)|(satisfies))\s+}, base::Keyword::Declaration
  end
  base.prepend :statement do
    rule %r/(#{Javascript.id_regex})(\??)(\s*)(:)/ do
      groups base::Name::Label, base::Punctuation, base::Text, base::Punctuation
      push :expr_start
    end
  end
end

def builtins
  @builtins ||= super + %w(
    Capitalize ConstructorParameters Exclude Extract InstanceType
    Lowercase NonNullable Omit OmitThisParameter Parameters
    Partial Pick Readonly Record Required
    ReturnType ThisParameterType ThisType Uncapitalize Uppercase
  )
end

def declarations
  @declarations ||= super + Set.new(%w(
    type abstract
  ))
end

def keywords
  @keywords ||= super + Set.new(%w(
    is namespace static private protected public
    implements readonly
  ))
end

def reserved
  @reserved ||= super + Set.new(%w(
    string any void number namespace module
    declare default interface keyof
  ))
end