location is unknown

Author: andrew-johnson-4

PLATFORM/C/LIB/common-macros.lsts

@@ -66,3 +66,7 @@ typed macro macro::for(item: lazy, iter: Vector<?>, loop: lazy): lazy = (
 typed macro macro::while(cond: lazy, body: lazy): lazy = (
    primitive::while( body as Nil, into-branch-conditional(cond) )
 );
+
+typed macro macro::assert(cond: lazy): lazy = (
+   if not(cond) then fail("Assertion Failed", macro::location(here))
+);

PLATFORM/C/LIB/io.lm

@@ -19,6 +19,11 @@ fail := λ(: msg SmartString). (: (
    (as (exit 1_u32) Never)
  ) Never);
 
+fail := λ(: msg SmartString)(: loc SmartString). (: (
+   (print msg)(print '\sat\s_s)(print loc)
+   (as (exit 1_u32) Never)
+) Never);
+
 fail := λ(: msg String). (: (
    (print msg)
    (as (exit 1_u32) Never)

PLUGINS/FRONTEND/LSTS/lsts-parse.lsts

@@ -1286,9 +1286,7 @@ let lsts-make-lit(t: Token): AST = (
       while non-zero(s) && s != c"_ss" {
          if s.has-prefix(c"\\{") {
             if non-zero(buffer) {
-               buffer = buffer + SAtom{c"_ss"};
-               let bs = clone-rope(buffer);
-               let be = Lit { bs, with-location(mk-token(bs),loc) };
+               let be = mk-lit(with-location(mk-token(clone-rope(buffer)),loc)).ascript(t1(c"SmartString"));
                if non-zero(base) {
                   base = mk-app(
                      Var{ c"+", with-location(mk-token("+"),loc) },
@@ -1330,9 +1328,7 @@ let lsts-make-lit(t: Token): AST = (
          }
       };
       if non-zero(buffer) {
-         buffer = buffer + SAtom{ c"_ss" };
-         let bs = clone-rope(buffer);
-         let be = Lit{ bs, with-location(mk-token(bs),loc) };
+         let be = mk-lit(with-location(mk-token(clone-rope(buffer)),loc)).ascript(t1(c"SmartString"));
          if non-zero(base) {
             base = mk-app(
                Var{ c"+", with-location(mk-token("+"),loc) },
@@ -1342,6 +1338,8 @@ let lsts-make-lit(t: Token): AST = (
             base = be;
          };
       };
+   } else if t.key.has-suffix(c"_ss") {
+      base = with-location(mk-lit(t.key.remove-suffix(c"_ss")),loc).ascript(t1(c"SmartString"));
    };
    base
 );
@@ -1493,6 +1491,8 @@ let lsts-parse-atom-without-tail(tokens: List<Token>): Tuple<AST,List<Token>> =
       tokens = term-rest.second;
       lsts-parse-expect(c")", tokens); tokens = tail(tokens);
       term = AType{ term-rest.first };
+   } else if lsts-parse-head(tokens).has-suffix(c"_ss") {
+      (term, tokens) = lsts-parse-lit(tokens);
    } else if lsts-parse-head(tokens)==c"sizeof" {
       let loc = head(tokens).location;
       tokens = tail(tokens);

SRC/macro-table.lsts

@@ -1,6 +1,7 @@
 
 let index-macro-table = {} :: HashtableEq<CString,List<(Type,Type,AST)>>;
 index-macro-table = index-macro-table.bind(c"macro::concat", [] :: List<(Type,Type,AST)>);
+index-macro-table = index-macro-table.bind(c"macro::location", [] :: List<(Type,Type,AST)>);
 
 let bind-new-macro(mname: CString, mterm: AST): Nil = (
    let row = index-macro-table.lookup(mname, [] :: List<(Type,Type,AST)>);

SRC/std-apply-macro.lsts

@@ -17,9 +17,14 @@ let std-apply-macro-concat(tctx: Maybe<TContext>, mname: CString, margs: AST): (
    }
 );
 
+let std-apply-macro-location(tctx: Maybe<TContext>, mname: CString, margs: AST): (TContext?, AST) = (
+   (tctx, mk-lit(to-smart-string(margs.location)).ascript(t1(c"SmartString")))
+);
+
 let std-apply-macro(tctx: Maybe<TContext>, mname: CString, margs: AST, strong: Bool): (TContext?, AST) = (
    let result = ASTEOF;
    if mname==c"macro::concat" then (tctx, result) = std-apply-macro-concat(tctx, mname, margs)
+   else if mname==c"macro::location" then (tctx, result) = std-apply-macro-location(tctx, mname, margs)
    else {
       let row = index-macro-table.lookup(mname, [] :: List<(Type,Type,AST)>);
       let peep-holes = TAny;

tests/unit/ast-macros.lsts

@@ -30,6 +30,9 @@ type A = A{a:U64};
    macro::while( wi < 4, ( print(wi); wi = wi + 1; ) )
 );
 
+macro::assert(true);
+macro::assert(false);
+
 # assert
 
 # match