Allow identifier with modules in tagged template literals (e.g. Pg.sql``)

Author: tsnobip

CHANGELOG.md

@@ -14,6 +14,7 @@
 
 #### :nail_care: Polish
 - No parens around tagged template literals. https://github.com/rescript-lang/rescript-compiler/pull/6639
+- Allow identifier with modules in tagged template literals (e.g. Pg.sql`select * from ${table} where id = ${id}`). https://github.com/rescript-lang/rescript-compiler/pull/6645
 
 #### :bug: Bug Fix
 

jscomp/syntax/src/res_core.ml

@@ -2071,8 +2071,7 @@ and parsePrimaryExpr ~operand ?(noCall = false) p =
     | Backtick
       when noCall = false && p.prevEndPos.pos_lnum == p.startPos.pos_lnum -> (
       match expr.pexp_desc with
-      | Pexp_ident {txt = Longident.Lident ident} ->
-        parseTemplateExpr ~prefix:ident p
+      | Pexp_ident long_ident -> parseTemplateExpr ~prefix:long_ident p
       | _ ->
         Parser.err ~startPos:expr.pexp_loc.loc_start
           ~endPos:expr.pexp_loc.loc_end p
@@ -2253,13 +2252,15 @@ and parseBinaryExpr ?(context = OrdinaryExpr) ?a p prec =
 (* | _ -> false *)
 (* ) *)
 
-and parseTemplateExpr ?(prefix = "js") p =
+and parseTemplateExpr ?prefix p =
   let partPrefix =
     (* we could stop treating js and j prefix as something special
        for json, we would first need to remove @as(json`true`) feature *)
     match prefix with
-    | "js" | "j" | "json" -> Some prefix
-    | _ -> None
+    | Some {txt = Longident.Lident (("js" | "j" | "json") as prefix); _} ->
+      Some prefix
+    | Some _ -> None
+    | None -> Some "js"
   in
   let startPos = p.Parser.startPos in
 
@@ -2296,8 +2297,7 @@ and parseTemplateExpr ?(prefix = "js") p =
   let values = Ext_list.filter_map parts snd in
   let endPos = p.Parser.endPos in
 
-  let genTaggedTemplateCall () =
-    let lident = Longident.Lident prefix in
+  let genTaggedTemplateCall lident =
     let ident =
       Ast_helper.Exp.ident ~attrs:[] ~loc:Location.none
         (Location.mknoloc lident)
@@ -2348,8 +2348,9 @@ and parseTemplateExpr ?(prefix = "js") p =
   in
 
   match prefix with
-  | "js" | "j" | "json" -> genInterpolatedString ()
-  | _ -> genTaggedTemplateCall ()
+  | Some {txt = Longident.Lident ("js" | "j" | "json"); _} | None ->
+    genInterpolatedString ()
+  | Some {txt = lident} -> genTaggedTemplateCall lident
 
 (* Overparse: let f = a : int => a + 1, is it (a : int) => or (a): int =>
  * Also overparse constraints:

jscomp/syntax/tests/printer/other/expected/string.res.txt

@@ -17,3 +17,5 @@ let heart = "\u2665"
 let smile = "emoji: \u{1F600}"
 
 let taggedTemplate = sql`select * from ${table} where id = ${id}`
+
+let taggedTemplate = Pg.sql`select * from ${table} where id = ${id}`

jscomp/syntax/tests/printer/other/string.res

@@ -16,4 +16,6 @@ let heart = "\u2665"
 
 let smile = "emoji: \u{1F600}"
 
-let taggedTemplate = sql`select * from ${table} where id = ${id}`
+let taggedTemplate = sql`select * from ${table} where id = ${id}`
+
+let taggedTemplate = Pg.sql`select * from ${table} where id = ${id}`

jscomp/test/tagged_template_test.js

@@ -1,9 +1,14 @@
-@module("./tagged_template_lib.js") @taggedTemplate
-external sql: (array<string>, array<string>) => string = "sql"
+module Pg = {
+  @module("./tagged_template_lib.js") @taggedTemplate
+  external sql: (array<string>, array<string>) => string = "sql"
+}
 
 let table = "users"
 let id = "5"
 
+let queryWithModule = Pg.sql`SELECT * FROM ${table} WHERE id = ${id}`
+
+open Pg 
 let query = sql`SELECT * FROM ${table} WHERE id = ${id}`
 
 @module("./tagged_template_lib.js") @taggedTemplate
@@ -29,6 +34,10 @@ Mt.from_pair_suites(
     (
       "with externals, it should return a string with the correct interpolations",
       () => Eq(query, "SELECT * FROM 'users' WHERE id = '5'"),
+    ),
+    (
+      "with module scoped externals, it should also return a string with the correct interpolations",
+      () => Eq(queryWithModule, "SELECT * FROM 'users' WHERE id = '5'"),
     ),
         (
       "with externals, it should return the result of the function",