fix: handle literals for getTranslations

.changeset/many-sloths-fry.md

@@ -0,0 +1,5 @@
+---
+"@labdigital/intl-extractor": patch
+---
+
+Handle literals for `getTranslations()` call

src/extract.test.ts

@@ -44,6 +44,30 @@ describe("Test parseSource", () => {
 		expect(result).toEqual(expected);
 	});
 
+	test("should parse translator object functions", () => {
+		const source = `
+		export const MyComponent = () => {
+			const t = useTranslations("MyComponent");
+
+			const foobar = t.html("foobar");
+
+			return (
+				<div>
+					<h1>{t.rich("title")}</h1>
+				</div>
+			)
+		}
+		`;
+
+		const result = extractLabels("MyComponent.tsx", source);
+		const expected = {
+			MyComponent: new Set(["foobar", "title"]),
+		};
+		expect(result).toEqual(expected);
+	});
+});
+
+describe("getTranslator usage", () => {
 	test("should parse source from server component using getTranslations", () => {
 		const source = `
 		export const MyComponent = async () => {
@@ -71,10 +95,32 @@ describe("Test parseSource", () => {
 		expect(result).toEqual(expected);
 	});
 
-	test("should parse translator object functions", () => {
+	test("should parse translator object functions (literal)", () => {
 		const source = `
 		export const MyComponent = () => {
-			const t = useTranslations("MyComponent");
+			const t = await getTranslations("MyComponent");
+
+			const foobar = t.html("foobar");
+
+			return (
+				<div>
+					<h1>{t.rich("title")}</h1>
+				</div>
+			)
+		}
+		`;
+
+		const result = extractLabels("MyComponent.tsx", source);
+		const expected = {
+			MyComponent: new Set(["foobar", "title"]),
+		};
+		expect(result).toEqual(expected);
+	});
+
+	test("should parse translator object functions (object) ", () => {
+		const source = `
+		export const MyComponent = () => {
+			const t = await getTranslations({ namespace: "MyComponent", locale });
 
 			const foobar = t.html("foobar");
 

src/extract.ts

@@ -65,6 +65,8 @@ export function extractLabels(filename: string, source: string) {
 			) {
 				if (node.name && ts.isIdentifier(node.name)) {
 					const arg = callExpr.arguments[0];
+
+					// If the argument is an object, parse the object
 					if (ts.isObjectLiteralExpression(arg)) {
 						// Iterate over the object properties
 						for (const prop of arg.properties) {
@@ -82,6 +84,12 @@ export function extractLabels(filename: string, source: string) {
 								}
 							}
 						}
+					} else {
+						currentScope.variables.set(
+							node.name.text,
+							// Remove the surrounding quotes
+							callExpr.arguments[0].getText().slice(1, -1),
+						);
 					}
 				}
 			}