Stringifying attributes

1 files changed, 19 insertions, 1 deletions

diff --git a/src/mono_opt.sml b/src/mono_opt.sml
index 2f98a9c7..c9cd5f84 100644
--- a/src/mono_opt.sml
+++ b/src/mono_opt.sml
@@ -52,7 +52,6 @@ fun exp e =
             EPrim (Prim.String (String.implode (rev chs)))
         end
                                        
-        
       | EStrcat ((EPrim (Prim.String s1), loc), (EPrim (Prim.String s2), _)) =>
         let
             val s =
@@ -65,8 +64,27 @@ fun exp e =
         in
             EPrim (Prim.String s)
         end
+
+      | EStrcat ((EPrim (Prim.String s1), loc), (EStrcat ((EPrim (Prim.String s2), _), rest), _)) =>
+        let
+            val s =
+                if size s1 > 0 andalso size s2 > 0
+                   andalso Char.isSpace (String.sub (s1, size s1 - 1))
+                   andalso Char.isSpace (String.sub (s2, 0)) then
+                    s1 ^ String.extract (s2, 1, NONE)
+                else
+                    s1 ^ s2
+        in
+            EStrcat ((EPrim (Prim.String s), loc), rest)
+        end
+
+      | EStrcat ((EStrcat (e1, e2), loc), e3) =>
+        optExp (EStrcat (e1, (EStrcat (e2, e3), loc)), loc)
+
       | _ => e
 
+and optExp e = #1 (U.Exp.map {typ = typ, exp = exp} e)
+
 val optimize = U.File.map {typ = typ, exp = exp, decl = decl}
 
 end