Class: RBS::Types::Function

Inherits:
Object
  • Object
show all
Defined in:
lib/rbs/types.rb

Defined Under Namespace

Classes: Param

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(required_positionals:, optional_positionals:, rest_positionals:, trailing_positionals:, required_keywords:, optional_keywords:, rest_keywords:, return_type:) ⇒ Function

Returns a new instance of Function.



631
632
633
634
635
636
637
638
639
640
 
# File 'lib/rbs/types.rb', line 631

def initialize(required_positionals:, optional_positionals:, rest_positionals:, trailing_positionals:, required_keywords:, optional_keywords:, rest_keywords:, return_type:)
  @return_type = return_type
  @required_positionals = required_positionals
  @optional_positionals = optional_positionals
  @rest_positionals = rest_positionals
  @trailing_positionals = trailing_positionals
  @required_keywords = required_keywords
  @optional_keywords = optional_keywords
  @rest_keywords = rest_keywords
end

Instance Attribute Details

#optional_keywords ⇒ Object (readonly)

Returns the value of attribute optional_keywords.



627
628
629
 
# File 'lib/rbs/types.rb', line 627

def optional_keywords
  @optional_keywords
end

#optional_positionals ⇒ Object (readonly)

Returns the value of attribute optional_positionals.



623
624
625
 
# File 'lib/rbs/types.rb', line 623

def optional_positionals
  @optional_positionals
end

#required_keywords ⇒ Object (readonly)

Returns the value of attribute required_keywords.



626
627
628
 
# File 'lib/rbs/types.rb', line 626

def required_keywords
  @required_keywords
end

#required_positionals ⇒ Object (readonly)

Returns the value of attribute required_positionals.



622
623
624
 
# File 'lib/rbs/types.rb', line 622

def required_positionals
  @required_positionals
end

#rest_keywords ⇒ Object (readonly)

Returns the value of attribute rest_keywords.



628
629
630
 
# File 'lib/rbs/types.rb', line 628

def rest_keywords
  @rest_keywords
end

#rest_positionals ⇒ Object (readonly)

Returns the value of attribute rest_positionals.



624
625
626
 
# File 'lib/rbs/types.rb', line 624

def rest_positionals
  @rest_positionals
end

#return_type ⇒ Object (readonly)

Returns the value of attribute return_type.



629
630
631
 
# File 'lib/rbs/types.rb', line 629

def return_type
  @return_type
end

#trailing_positionals ⇒ Object (readonly)

Returns the value of attribute trailing_positionals.



625
626
627
 
# File 'lib/rbs/types.rb', line 625

def trailing_positionals
  @trailing_positionals
end

Class Method Details

.empty(return_type) ⇒ Object 



759
760
761
762
763
764
765
766
767
768
769
770
 
# File 'lib/rbs/types.rb', line 759

def self.empty(return_type)
  Function.new(
    required_positionals: [],
    optional_positionals: [],
    rest_positionals: nil,
    trailing_positionals: [],
    required_keywords: {},
    optional_keywords: {},
    rest_keywords: nil,
    return_type: return_type
  )
end

Instance Method Details

#==(other) ⇒ Object Also known as: eql? 



642
643
644
645
646
647
648
649
650
651
652
 
# File 'lib/rbs/types.rb', line 642

def ==(other)
  other.is_a?(Function) &&
    other.required_positionals == required_positionals &&
    other.optional_positionals == optional_positionals &&
    other.rest_positionals == rest_positionals &&
    other.trailing_positionals == trailing_positionals &&
    other.required_keywords == required_keywords &&
    other.optional_keywords == optional_keywords &&
    other.rest_keywords == rest_keywords &&
    return_type == return_type
end

#drop_head ⇒ Object 



826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
 
# File 'lib/rbs/types.rb', line 826

def drop_head
  case
  when !required_positionals.empty?
    [
      required_positionals[0],
      update(required_positionals: required_positionals.drop(1))
    ]
  when !optional_positionals.empty?
    [
      optional_positionals[0],
      update(optional_positionals: optional_positionals.drop(1))
    ]
  else
    raise "Cannot #drop_head"
  end
end

#drop_tail ⇒ Object 



843
844
845
846
847
848
849
850
851
852
853
 
# File 'lib/rbs/types.rb', line 843

def drop_tail
  case
  when !trailing_positionals.empty?
    [
      trailing_positionals.last,
      update(trailing_positionals: trailing_positionals.take(trailing_positionals.size - 1))
    ]
  else
    raise "Cannot #drop_tail"
  end
end

#each_param(&block) ⇒ Object 



728
729
730
731
732
733
734
735
736
737
738
739
740
 
# File 'lib/rbs/types.rb', line 728

def each_param(&block)
  if block_given?
    required_positionals.each(&block)
    optional_positionals.each(&block)
    rest_positionals&.yield_self(&block)
    trailing_positionals.each(&block)
    required_keywords.each_value(&block)
    optional_keywords.each_value(&block)
    rest_keywords&.yield_self(&block)
  else
    enum_for :each_param
  end
end

#each_type ⇒ Object 



713
714
715
716
717
718
719
720
721
722
723
724
725
726
 
# File 'lib/rbs/types.rb', line 713

def each_type
  if block_given?
    required_positionals.each {|param| yield param.type }
    optional_positionals.each {|param| yield param.type }
    rest_positionals&.yield_self {|param| yield param.type }
    trailing_positionals.each {|param| yield param.type }
    required_keywords.each_value {|param| yield param.type }
    optional_keywords.each_value {|param| yield param.type }
    rest_keywords&.yield_self {|param| yield param.type }
    yield(return_type)
  else
    enum_for :each_type
  end
end

#empty? ⇒ Boolean

Returns:

  • (Boolean) 


799
800
801
802
803
804
805
806
807
 
# File 'lib/rbs/types.rb', line 799

def empty?
  required_positionals.empty? &&
    optional_positionals.empty? &&
    !rest_positionals &&
    trailing_positionals.empty? &&
    required_keywords.empty? &&
    optional_keywords.empty? &&
    !rest_keywords
end

#free_variables(set = Set.new) ⇒ Object 



668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
 
# File 'lib/rbs/types.rb', line 668

def free_variables(set = Set.new)
  set.tap do
    required_positionals.each do |param|
      param.type.free_variables(set)
    end
    optional_positionals.each do |param|
      param.type.free_variables(set)
    end
    rest_positionals&.yield_self do |param|
      param.type.free_variables(set)
    end
    trailing_positionals.each do |param|
      param.type.free_variables(set)
    end
    required_keywords.each_value do |param|
      param.type.free_variables(set)
    end
    optional_keywords.each_value do |param|
      param.type.free_variables(set)
    end
    rest_keywords&.yield_self do |param|
      param.type.free_variables(set)
    end

    return_type.free_variables(set)
  end
end

#has_keyword? ⇒ Boolean

Returns:

  • (Boolean) 


855
856
857
 
# File 'lib/rbs/types.rb', line 855

def has_keyword?
  !required_keywords.empty? || !optional_keywords.empty? || rest_keywords
end

#hash ⇒ Object 



656
657
658
659
660
661
662
663
664
665
666
 
# File 'lib/rbs/types.rb', line 656

def hash
  self.class.hash ^
    required_positionals.hash ^
    optional_positionals.hash ^
    rest_positionals.hash ^
    trailing_positionals.hash ^
    required_keywords.hash ^
    optional_keywords.hash ^
    rest_keywords.hash ^
    return_type.hash
end

#map_type(&block) ⇒ Object 



696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
 
# File 'lib/rbs/types.rb', line 696

def map_type(&block)
  if block_given?
    Function.new(
      required_positionals: required_positionals.map {|param| param.map_type(&block) },
      optional_positionals: optional_positionals.map {|param| param.map_type(&block) },
      rest_positionals: rest_positionals&.yield_self {|param| param.map_type(&block) },
      trailing_positionals: trailing_positionals.map {|param| param.map_type(&block) },
      required_keywords: required_keywords.transform_values {|param| param.map_type(&block) },
      optional_keywords: optional_keywords.transform_values {|param| param.map_type(&block) },
      rest_keywords: rest_keywords&.yield_self {|param| param.map_type(&block) },
      return_type: yield(return_type)
    )
  else
    enum_for :map_type
  end
end

#param_to_s ⇒ Object 



809
810
811
812
813
814
815
816
817
818
819
820
 
# File 'lib/rbs/types.rb', line 809

def param_to_s
  params = []
  params.push(*required_positionals.map(&:to_s))
  params.push(*optional_positionals.map {|p| "?#{p}"})
  params.push("*#{rest_positionals}") if rest_positionals
  params.push(*trailing_positionals.map(&:to_s))
  params.push(*required_keywords.map {|name, param| "#{name}: #{param}" })
  params.push(*optional_keywords.map {|name, param| "?#{name}: #{param}" })
  params.push("**#{rest_keywords}") if rest_keywords

  params.join(", ")
end

#return_to_s ⇒ Object 



822
823
824
 
# File 'lib/rbs/types.rb', line 822

def return_to_s
  return_type.to_s(1)
end

#sub(s) ⇒ Object 



755
756
757
 
# File 'lib/rbs/types.rb', line 755

def sub(s)
  map_type {|ty| ty.sub(s) }
end

#to_json(*a) ⇒ Object 



742
743
744
745
746
747
748
749
750
751
752
753
 
# File 'lib/rbs/types.rb', line 742

def to_json(*a)
  {
    required_positionals: required_positionals,
    optional_positionals: optional_positionals,
    rest_positionals: rest_positionals,
    trailing_positionals: trailing_positionals,
    required_keywords: required_keywords,
    optional_keywords: optional_keywords,
    rest_keywords: rest_keywords,
    return_type: return_type
  }.to_json(*a)
end

#update(required_positionals: self.required_positionals, optional_positionals: self.optional_positionals, rest_positionals: self.rest_positionals, trailing_positionals: self.trailing_positionals, required_keywords: self.required_keywords, optional_keywords: self.optional_keywords, rest_keywords: self.rest_keywords, return_type: self.return_type) ⇒ Object 



785
786
787
788
789
790
791
792
793
794
795
796
797
 
# File 'lib/rbs/types.rb', line 785

def update(required_positionals: self.required_positionals, optional_positionals: self.optional_positionals, rest_positionals: self.rest_positionals, trailing_positionals: self.trailing_positionals,
           required_keywords: self.required_keywords, optional_keywords: self.optional_keywords, rest_keywords: self.rest_keywords, return_type: self.return_type)
  Function.new(
    required_positionals: required_positionals,
    optional_positionals: optional_positionals,
    rest_positionals: rest_positionals,
    trailing_positionals: trailing_positionals,
    required_keywords: required_keywords,
    optional_keywords: optional_keywords,
    rest_keywords: rest_keywords,
    return_type: return_type
  )
end

#with_return_type(type) ⇒ Object 



772
773
774
775
776
777
778
779
780
781
782
783
 
# File 'lib/rbs/types.rb', line 772

def with_return_type(type)
  Function.new(
    required_positionals: required_positionals,
    optional_positionals: optional_positionals,
    rest_positionals: rest_positionals,
    trailing_positionals: trailing_positionals,
    required_keywords: required_keywords,
    optional_keywords: optional_keywords,
    rest_keywords: rest_keywords,
    return_type: type
  )
end