(PUP-11981) WIP

Author: AriaXLi

lib/puppet/info_service.rb

@@ -6,6 +6,7 @@ module Puppet::InfoService
   require_relative 'info_service/plan_information_service'
 
   def self.classes_per_environment(env_file_hash)
+    #require 'pry-byebug'; binding.pry
     Puppet::InfoService::ClassInformationService.new.classes_per_environment(env_file_hash)
   end
 

lib/puppet/info_service/class_information_service.rb

@@ -26,6 +26,7 @@ def classes_per_environment(env_file_hash)
     #     if file already processed, use last result or error
     #
     env_file_hash.each do |env, files|
+      #require 'pry-byebug'; binding.pry
       env_result = result[env] = {}
       files.each do |f|
         env_result[f] = result_of(f)
@@ -56,6 +57,7 @@ def parse_file(f)
     return {:error => _("The file %{f} does not exist") % { f: f }} unless Puppet::FileSystem.exist?(f)
 
     begin
+      #require 'pry-byebug'; binding.pry
       parse_result = @parser.parse_file(f)
       {:classes =>
         parse_result.definitions.select {|d| d.is_a?(Puppet::Pops::Model::HostClassDefinition)}.map do |d|
@@ -77,9 +79,11 @@ def extract_type(structure, p)
     return structure if p.type_expr.nil?
     structure[:type] = typeexpr_to_string(p.type_expr)
     structure
+    # require 'pry-byebug'; binding.pry
   end
 
   def extract_default(structure, p)
+    # require 'pry-byebug'; binding.pry
     value_expr = p.value
     return structure if value_expr.nil?
     default_value = value_as_literal(value_expr)

lib/puppet/pops/evaluator/literal_evaluator.rb

@@ -40,6 +40,7 @@ def literal_Program(o)
   end
 
   def literal_LiteralString(o)
+    #require 'pry-byebug'; binding.pry
     o.value
   end
 
@@ -67,6 +68,11 @@ def literal_LiteralRegularExpression(o)
     o.value
   end
 
+  def literal_QualifiedReference(o)
+    #require 'pry-byebug'; binding.pry
+    o.value
+  end
+
   def literal_ConcatenatedString(o)
     # use double quoted string value if there is no interpolation
     throw :not_literal unless o.segments.size == 1 && o.segments[0].is_a?(Model::LiteralString)

lib/puppet/pops/issues.rb

@@ -207,6 +207,10 @@ def self.hard_issue(issue_code, *args, &block)
     _("The numeric parameter name '$%{name}' cannot be used (clashes with numeric match result variables)") % { name: name }
   end
 
+  ILLEGAL_NONLITERAL_PARAMETER_TYPE = issue :ILLEGAL_NONLITERAL_PARAMETER_TYPE, :name, :type_class do
+    _("The parameter '$%{name}' must be a literal type, not %{type_class}") % { name: name, type_class: label.a_an(type_class) }
+  end
+
   # In certain versions of Puppet it may be allowed to assign to a not already assigned key
   # in an array or a hash. This is an optional validation that may be turned on to prevent accidental
   # mutation.

lib/puppet/pops/validation/checker4_0.rb

@@ -471,6 +471,7 @@ def check_HostClassDefinition(o)
     internal_check_parameter_name_uniqueness(o)
     internal_check_reserved_params(o)
     internal_check_no_idem_last(o)
+    internal_check_parameter_type_literal(o)
   end
 
   def check_ResourceTypeDefinition(o)
@@ -481,6 +482,18 @@ def check_ResourceTypeDefinition(o)
     internal_check_no_idem_last(o)
   end
 
+  def internal_check_parameter_type_literal(o)
+    o.parameters.each do |p|
+      next unless p.type_expr
+      catch :not_literal do
+        #require 'pry-byebug'; binding.pry
+        literal(p.type_expr)
+        next
+      end
+      acceptor.accept(Issues::ILLEGAL_NONLITERAL_PARAMETER_TYPE, p, {name: p.name, type_class: p.type_expr.class})
+    end
+  end
+
   def internal_check_return_type(o)
     r = o.return_type
     internal_check_type_ref(o, r) unless r.nil?

spec/unit/info_service_spec.rb

@@ -313,7 +313,7 @@ class Borked($Herp+$Derp) {}
         CODE
         'json_unsafe.pp' => <<-CODE,
              class json_unsafe($arg1 = /.*/, $arg2 = default, $arg3 = {1 => 1}) {}
-          CODE
+        CODE
        })
     end
 
@@ -326,6 +326,7 @@ class json_unsafe($arg1 = /.*/, $arg2 = default, $arg3 = {1 => 1}) {}
     end
 
     it "produces classes and parameters from a given file" do
+      #require 'pry-byebug'; binding.pry
       files = ['foo.pp'].map {|f| File.join(code_dir, f) }
       result = Puppet::InfoService.classes_per_environment({'production' => files })
       expect(result).to eq({