mt-lang 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 990a1e0ede945713a31152caccc4a7de798dc6ead6078d46d5a376f19054a52a
-  data.tar.gz: 97d7877587ca50ab673212d1d0d80d6d60ad8f97a74c439965feb32c2f5a4e37
+  metadata.gz: 7e72d5a921ff9b83cef5fb60e89672e913daa367f7e64b7fd000664664c4bb22
+  data.tar.gz: 7876a95d45c0a11a801ea6ec3bef10adc448e69eacaccd5bc65df5c22e01786b
 SHA512:
-  metadata.gz: c04b4d611724c8b4f42ae14929b3d8d98b4aa4da5157f7c3ee765afb6356be643b6ada8c96ed45a8e61103e71792c7855c25d87f65865b9609cdb0cc85544507
-  data.tar.gz: 82416c2665d49928f0d6dd5e9b8f225177f9df0f782e973339545a6375cc673e082bccc3bbb6126cbdc50875dc6fa29880b5e23b8e3ccd458934b32f5d2f6718
+  metadata.gz: 641eb7f5a45dfd4b737342ccb89ef2e58036ff6e3a889032664b0136ba6a8c632d45180f9a18ce56061c9278e2bd7e68ad37374cf40dc6da2d626940f8457d95
+  data.tar.gz: 399e6943e1207afa3a14201e8a3f1984405810da46d245685b9b2a9b5fd74b92486f982377452f416caf5aa9d6878d6094ffb11c7382d35b37d11820c3df6a89

data/lib/milk_tea/base.rb

@@ -3,7 +3,7 @@
 require "pathname"
 
 module MilkTea
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 
   def self.root
     @root ||= Pathname.new(File.expand_path("../..", __dir__))

data/lib/milk_tea/core/compile_time/method_folding.rb

@@ -116,7 +116,7 @@ module MilkTea
         func && func.respond_to?(:ast) && func.ast.respond_to?(:const) && func.ast.const
       end
 
-      def comptime_const_method_body(binding, arguments, scopes:, receiver_value:)
+      def comptime_const_method_body(binding, arguments, scopes:, receiver_value:, arg_evaluator: nil)
         method = binding.ast
         return nil unless method.respond_to?(:body) && method.body
         return nil if binding.type_params.any?
@@ -130,7 +130,11 @@ module MilkTea
         end
 
         method.params.each_with_index do |param, idx|
-          arg_value = comptime_eval(arguments[idx].value, scopes)
+          arg_value = if arg_evaluator
+            arg_evaluator.call(arguments[idx].value)
+          else
+            comptime_eval(arguments[idx].value, scopes)
+          end
           return nil unless arg_value
 
           initial_vars[param.name] = arg_value

data/lib/milk_tea/core/compile_time.rb

@@ -713,20 +713,31 @@ module MilkTea
         return unless @checker.respond_to?(:comptime_method_binding_for_receiver)
         return unless @checker.respond_to?(:comptime_const_method_body)
 
+        arg_evaluator = ->(expr) { evaluate_expression(expr, scopes:) }
+
         receiver = call_expr.callee.receiver
-        return unless receiver.is_a?(AST::Identifier)
-        return unless @variables.key?(receiver.name)
+        if receiver.is_a?(AST::Identifier) && @variables.key?(receiver.name)
+          receiver_value = @variables[receiver.name]
+          return nil if receiver_value.nil? || receiver_value.is_a?(Types::Base)
+
+          receiver_type = @variable_types[receiver.name]
+          return nil unless receiver_type
+
+          binding = @checker.comptime_method_binding_for_receiver(receiver_type, call_expr.callee.member)
+          return nil unless binding
 
-        receiver_value = @variables[receiver.name]
-        return nil if receiver_value.nil? || receiver_value.is_a?(Types::Base)
+          return @checker.comptime_const_method_body(binding, call_expr.arguments, scopes:, receiver_value:, arg_evaluator:)
+        end
+
+        return nil unless @checker.respond_to?(:resolve_type_expression)
 
-        receiver_type = @variable_types[receiver.name]
+        receiver_type = @checker.resolve_type_expression(receiver)
         return nil unless receiver_type
 
         binding = @checker.comptime_method_binding_for_receiver(receiver_type, call_expr.callee.member)
         return nil unless binding
 
-        @checker.comptime_const_method_body(binding, call_expr.arguments, scopes:, receiver_value:)
+        @checker.comptime_const_method_body(binding, call_expr.arguments, scopes:, receiver_value: nil, arg_evaluator:)
       end
 
       def try_const_function_call(call_expr, scopes:)
@@ -774,9 +785,12 @@ module MilkTea
         elsif call_expr.callee.respond_to?(:name)
           call_expr.callee.name
         end
-        return unless callee_name
 
-        type = types[callee_name]
+        type = if callee_name
+          types[callee_name]
+        elsif call_expr.callee.is_a?(AST::MemberAccess) && @checker.respond_to?(:resolve_type_expression)
+          @checker.resolve_type_expression(call_expr.callee)
+        end
         return unless type.is_a?(Types::Struct)
 
         fields = {}

data/lib/milk_tea/core/lowering/resolve.rb

@@ -1875,6 +1875,10 @@ module MilkTea
           @lowerer.resolve_type_ref(type_ref)
         end
 
+        def resolve_type_expression(expression)
+          @lowerer.resolve_type_expression(expression)
+        end
+
         def comptime_struct_field_names(type_name)
           @lowerer.comptime_struct_field_names(type_name)
         end
@@ -1883,8 +1887,8 @@ module MilkTea
           @lowerer.comptime_method_binding_for_receiver(receiver_type, member)
         end
 
-        def comptime_const_method_body(binding, arguments, scopes: nil, receiver_value: nil)
-          @lowerer.comptime_const_method_body(binding, arguments, scopes:, receiver_value:)
+        def comptime_const_method_body(binding, arguments, scopes: nil, receiver_value: nil, arg_evaluator: nil)
+          @lowerer.comptime_const_method_body(binding, arguments, scopes:, receiver_value:, arg_evaluator:)
         end
 
         def comptime_expression_type(expression, scopes: nil)

data/lib/milk_tea/core/semantic_analyzer/top_level.rb

@@ -424,6 +424,17 @@ module MilkTea
             receiver_value = comptime_member_receiver_value(expression.callee.receiver, scopes)
             return comptime_const_method_body(method_binding, expression.arguments, scopes:, receiver_value:)
           end
+
+          if (struct_type = resolve_type_expression(expression.callee)) && struct_type.is_a?(Types::Struct)
+            fields = {}
+            expression.arguments.each do |argument|
+              val = evaluate_compile_time_const_value(argument.value, scopes:)
+              return nil unless val
+
+              fields[argument.name] = val
+            end
+            return fields
+          end
         when AST::Identifier
           if (struct_type = @ctx.types[expression.callee.name]) && struct_type.is_a?(Types::Struct)
             fields = {}

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mt-lang
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - Long (Teefan) Tran
@@ -629,7 +629,7 @@ metadata:
   homepage_uri: https://teefan.github.io/mt-lang/
   source_code_uri: https://github.com/teefan/mt-lang
 post_install_message: |
-  Milk Tea 0.4.0 installed!
+  Milk Tea 0.4.1 installed!
 
   System requirements:
     - A C compiler (gcc or clang) must be available on PATH