mt-lang 0.3.17 → 0.3.22

data/lib/milk_tea/tooling/cli/commands/parse.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module MilkTea
+  class CLI
+    module CommandParse
+      def parse_command
+        sexpr = false
+        args = @argv.dup
+        @argv = []
+        until args.empty?
+          arg = args.shift
+          if arg == "--sexpr"
+            sexpr = true
+            next
+          end
+          @argv << arg
+        end
+
+        unless @argv.any?
+          @err.puts("missing source file path")
+          print_usage(@err)
+          return 1
+        end
+
+        resolution = extract_resolution_flags!
+        input_paths = @argv.dup
+        return 1 unless ensure_known_source_operands!("parse", input_paths)
+
+        paths = expand_source_paths(input_paths)
+        return 0 if print_no_source_files_if_empty(paths, input_paths)
+
+        ensure_current_lockfiles!(paths) if resolution[:frozen]
+
+        multiple = paths.length > 1
+        paths.each_with_index do |path, index|
+          ast = make_module_loader(path, locked: resolution[:locked], platform: ModuleLoader.default_host_platform).load_file(path)
+          if multiple
+            @out.puts("# --- #{path} ---") unless sexpr
+          end
+          if sexpr
+            @out.puts(SexprDumper.dump_ast(ast))
+          else
+            @out.write(PrettyPrinter.format_ast(ast))
+          end
+          @out.puts if multiple && index < paths.length - 1
+        end
+        0
+      end
+    end
+  end
+end

data/lib/milk_tea/tooling/cli/commands/run.rb

@@ -0,0 +1,99 @@
+# frozen_string_literal: true
+
+module MilkTea
+  class CLI
+    module CommandRun
+      def run_command
+        path, options = extract_path_and_options
+        return 1 unless path
+
+        run_and_print_result(path, options)
+      end
+
+      def app_command
+        options = parse_build_options
+        return 1 unless options
+
+        module_name = @argv.shift
+        unless module_name
+          @err.puts("missing module name")
+          print_usage(@err)
+          return 1
+        end
+
+        path = resolve_app_module(module_name)
+        unless path
+          @err.puts("run-module module not found: #{module_name}")
+          return 1
+        end
+
+        frozen = options.delete(:frozen)
+        ensure_current_lockfile!(path) if frozen
+
+        run_and_print_result(path, options)
+      end
+
+      def extract_path_and_options(allow_clean: false)
+        options = parse_build_options(allow_clean:)
+        return nil unless options
+
+        path = @argv.shift
+        unless path
+          if File.file?(File.join(Dir.pwd, "package.toml"))
+            path = Dir.pwd
+          else
+            @err.puts("missing source file path")
+            print_usage(@err)
+            return nil
+          end
+        end
+
+        [path, options]
+      end
+
+      def run_and_print_result(path, options)
+        frozen = options.delete(:frozen)
+        ensure_current_lockfile!(path) if frozen
+        locked = options.delete(:locked)
+        package_graph = package_graph_for(path, locked:)
+        preview_notice_emitted = false
+        preview_started = lambda do |message|
+          preview_notice_emitted = true
+          @out.write(message)
+          @out.flush if @out.respond_to?(:flush)
+        end
+
+        result = Run.run(
+          path,
+          module_roots: module_roots_for(path, locked:),
+          package_graph:,
+          frontend: @build_frontend,
+          preview_started:,
+          argv: @argv.dup,
+          **options.except(:timings)
+        )
+        unless @out.equal?($stdout) || preview_notice_emitted
+          @out.write(result.stdout)
+        end
+        @out.flush if @out.respond_to?(:flush)
+        @err.write(result.stderr) unless @err.equal?($stderr)
+        info("[cached]") if result.cached
+        result.exit_status
+      end
+
+      def resolve_app_module(name)
+        relative = name.tr(".", "/").sub(%r{^/}, "") + ".mt"
+
+        module_roots_for(Dir.pwd).each do |root|
+          candidate = File.join(root, "std", relative)
+          return File.expand_path(candidate) if File.file?(candidate)
+
+          candidate = File.join(root, relative)
+          return File.expand_path(candidate) if File.file?(candidate)
+        end
+
+        nil
+      end
+    end
+  end
+end

data/lib/milk_tea/tooling/cli/commands/snapshot.rb

@@ -0,0 +1,117 @@
+# frozen_string_literal: true
+
+module MilkTea
+  class CLI
+    module CommandSnapshot
+      def snapshot_command
+        input_path = nil
+        theme_path = nil
+        output_path = nil
+        textmate_only = false
+
+        until @argv.empty?
+          arg = @argv.first
+          case arg
+          when "--theme", "-t"
+            @argv.shift
+            theme_path = @argv.shift
+            unless theme_path
+              @err.puts("snapshot: missing value for --theme")
+              return 1
+            end
+          when "--output", "-o"
+            @argv.shift
+            output_path = @argv.shift
+            unless output_path
+              @err.puts("snapshot: missing value for --output")
+              return 1
+            end
+          when "--textmate-only"
+            @argv.shift
+            textmate_only = true
+          else
+            if arg.start_with?("-")
+              @err.puts("snapshot: unknown option #{arg}")
+              return 1
+            end
+            unless input_path
+              input_path = @argv.shift
+            else
+              @err.puts("snapshot: unexpected argument #{@argv.shift}")
+              return 1
+            end
+          end
+        end
+
+        unless input_path
+          @err.puts("snapshot: missing source file path")
+          print_usage(@err)
+          return 1
+        end
+
+        unless File.file?(input_path)
+          @err.puts("snapshot: source file not found: #{input_path}")
+          return 1
+        end
+
+        input_path = File.expand_path(input_path)
+        theme_path = File.expand_path(theme_path) if theme_path
+        output_path = File.expand_path(output_path) if output_path
+
+        if theme_path && !File.file?(theme_path)
+          @err.puts("snapshot: theme file not found: #{theme_path}")
+          return 1
+        end
+
+        snapshot_script = MilkTea.root.join("bindings/vscode/scripts/snapshot.js").to_s
+        unless File.file?(snapshot_script)
+          @err.puts("snapshot: internal script not found at #{snapshot_script}")
+          return 1
+        end
+
+        args = ["node", snapshot_script, input_path]
+        args.push("-t", theme_path) if theme_path
+        args.push("-o", output_path) if output_path
+
+        semantic_result = nil
+        unless textmate_only
+          begin
+            semantic_result = MilkTea::LSP::Server.semantic_tokens_for_path(input_path)
+          rescue => e
+            @err.puts("snapshot: semantic analysis skipped: #{e.message}")
+          end
+        end
+
+        if semantic_result && semantic_result[:entries] && !semantic_result[:entries].empty?
+          source = File.read(input_path)
+          lines = source.split("\n", -1)
+          semantic_result[:entries].each do |entry|
+            byte_start = entry[:startChar]
+            byte_len = entry[:length]
+            line_text = lines[entry[:line]]
+            unless line_text && byte_start && byte_len
+              entry[:startChar] = 0
+              entry[:length] = 0
+              next
+            end
+            char_start = line_text.byteslice(0, byte_start).length
+            char_length = line_text.byteslice(byte_start, byte_len).length
+            char_length = line_text.length - char_start if char_start + char_length > line_text.length
+            entry[:startChar] = char_start
+            entry[:length] = char_length
+          end
+
+          temp = Tempfile.new(["mt_semantic", ".json"])
+          temp.write(JSON.generate(semantic_result[:entries]))
+          temp.close
+          args.push("-s", temp.path)
+          system(*args)
+          temp.unlink
+        else
+          system(*args)
+        end
+        $?.success? ? 0 : 1
+      end
+    end
+  end
+end