Include original source in the annotation

Author: NduatiK

lib/live_view_native/swiftui/rules_parser/parser/annotations.ex

@@ -3,7 +3,8 @@ defmodule LiveViewNative.SwiftUI.RulesParser.Parser.Annotations do
   # Helpers
 
   def context_to_annotation(%Context{annotations: true} = context, line) do
-    [file: context.file, line: line, module: context.module]
+    source = Enum.at(context.source_lines, line - context.source_line, "")
+    [file: context.file, line: line, module: context.module, source: String.trim(source)]
   end
 
   def context_to_annotation(_, _) do

lib/live_view_native/swiftui/rules_parser/parser/context.ex

@@ -4,6 +4,7 @@ defmodule LiveViewNative.SwiftUI.RulesParser.Parser.Context do
     :annotations,
     :module,
     source: "",
+    source_lines: [],
     errors: [],
     highlight_error: true,
     # Where in the code does the input start?
@@ -18,6 +19,7 @@ defmodule LiveViewNative.SwiftUI.RulesParser.Parser.Context do
      context
      |> Map.put_new(:context, %__MODULE__{
        source: rest,
+       source_lines: String.split(rest, "\n"),
        file: context[:file] || "",
        module: context[:module] || nil,
        source_line: context[:source_line] || 1,

test/live_view_native/swiftui/rules_parser_test.exs

@@ -18,7 +18,7 @@ defmodule LiveViewNative.SwiftUI.RulesParserTest do
       {line, input} = {__ENV__.line, "\nbold(true)"}
 
       # We add 1 because the modifier is on the second line of the input
-      output = {:bold, [file: __ENV__.file, line: line + 1, module: __ENV__.module], [true]}
+      output = {:bold, [file: __ENV__.file, line: line + 1, module: __ENV__.module, source: "bold(true)"], [true]}
 
       assert parse(input,
                file: __ENV__.file,
@@ -27,6 +27,27 @@ defmodule LiveViewNative.SwiftUI.RulesParserTest do
                annotations: true
              ) ==
                output
+      end
+
+      test "parses modifier function definition with annotation (2)" do
+      {line, input} = {__ENV__.line,"""
+      font(.largeTitle)
+      bold(true)
+      italic(true)
+      """}
+      
+      output = [
+        {:font, [file: __ENV__.file, line: line, module: __ENV__.module, source: "font(.largeTitle)"], [{:., [file: __ENV__.file, line: line, module: __ENV__.module, source: "font(.largeTitle)"], [nil, :largeTitle]}]},
+        {:bold, [file: __ENV__.file, line: line + 1, module: __ENV__.module, source: "bold(true)"], [true]},
+        {:italic, [file: __ENV__.file, line: line + 2, module: __ENV__.module, source: "italic(true)"], [true]}
+      ]
+
+      assert parse(input,
+          file: __ENV__.file,
+          module: __ENV__.module,
+          line: line,
+          annotations: true
+        ) == output
     end
 
     test "parses modifier function definition" do