[compile] Add a context field to ast_pragma

PlasmaLang · Aug 22, 2024 · da4ad72 · da4ad72
1 parent 3cbd140
commit da4ad72
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/ast.m b/src/ast.m
@@ -310,7 +310,8 @@
 :- type ast_pragma
     --->    ast_pragma(
                 astp_name           :: string,
-                astp_args           :: list(ast_pragma_arg)
+                astp_args           :: list(ast_pragma_arg),
+                astp_context        :: context
             ).
 
 :- type ast_pragma_arg

diff --git a/src/foreign.m b/src/foreign.m
@@ -81,7 +81,7 @@
 :- pred find_foreign_include_pragma(ast_pragma::in,
     list(foreign_include)::in, list(foreign_include)::out) is det.
 
-find_foreign_include_pragma(ast_pragma(Name, Args), !Includes) :-
+find_foreign_include_pragma(ast_pragma(Name, Args, _Context), !Includes) :-
     ( if Name = "foreign_include" then
         ( if Args = [ast_pragma_arg(String)] then
             Include = foreign_include(String),

diff --git a/src/parse.m b/src/parse.m
@@ -1612,6 +1612,7 @@
     is det.
 
 parse_pragma(Result, !Tokens) :-
+    get_context(!.Tokens, Context),
     match_token(pragma_, MatchPragma, !Tokens),
     match_token(ident, MatchIdent, !Tokens),
     within(l_paren, zero_or_more_delimited(comma, parse_pragma_arg),
@@ -1621,7 +1622,7 @@
         MatchIdent = ok(Ident),
         MatchArgs = ok(Args)
     then
-        Result = ok(ast_pragma(ast_pragma(Ident, Args)))
+        Result = ok(ast_pragma(ast_pragma(Ident, Args, Context)))
     else
         Result = combine_errors_3(MatchPragma, MatchIdent, MatchArgs)
     ).