should preserve source

nguyenyou · Dec 18, 2024 · 3d59135 · 3d59135
1 parent 5f72dfe
commit 3d59135
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/packages/scalawind/src/transform.js b/packages/scalawind/src/transform.js
@@ -147,5 +147,11 @@ export function transformSource(source) {
 }
 
 export const scalaSourceTransform = {
-  scala: (content) => transformSource(content),
+  scala: (content) => {
+    const transformed = transformSource(content)
+    return `${content}
+    
+    ${transformed}
+    `
+  },
 };
diff --git a/packages/scalawind/tests/transform.test.js b/packages/scalawind/tests/transform.test.js
@@ -1,5 +1,5 @@
 import { expect, test, describe } from "bun:test";
-import { transformSource } from "../src/transform";
+import { transformSource, scalaSourceTransform } from "../src/transform";
 
 describe("transform scala source", () => {
   test("case 1", () => {
@@ -305,4 +305,19 @@ describe("transform scala source", () => {
     const expected = "dark:md:hover:focus:active:group-hover:disabled:bg-red-500";
     expect(actual).toBe(expected);
   });
+
+  test("should preserve original source", () => {
+    const source = `
+      div(
+        cls := "bg-red-500",
+        tw.flex.items_center
+      )
+    `;
+    const actual = scalaSourceTransform.scala(source)
+    const expected = `${source}
+    
+    flex items-center
+    `
+    expect(actual).toBe(expected);
+  })
 });