feat: support declarations comment

saurabhdaware · saurabhdaware · commit 4ccc09c16aec · 2022-08-07T20:11:39.000+05:30
diff --git a/packages/abell/package.json b/packages/abell/package.json
@@ -1,6 +1,6 @@
 {
   "name": "abell",
-  "version": "1.0.0-alpha.65",
+  "version": "1.0.0-alpha.66",
   "description": "Abell is a static-site-generator for JavaScript developers. Powered by Vite, It tries to stay close to fundamentals while providing a great DX",
   "bin": "./dist/bin.js",
   "main": "./dist/index.js",
diff --git a/packages/abell/src/vite-plugin-abell/compiler/compiler.spec.ts b/packages/abell/src/vite-plugin-abell/compiler/compiler.spec.ts
@@ -15,6 +15,27 @@ describe('compile()', () => {
     expect(out.declarationsBlock.text).toMatchInlineSnapshot('""');
   });
 
+  test('should create first block as declaration block with declarations comment', () => {
+    const abellCode = `
+    {{
+      // declarations
+      const a = 3;
+    }}
+    <body>{{ a }}</body>
+    `;
+    const out = compile(abellCode, {
+      filepath: __dirname,
+      outputType: 'syntax-blocks'
+    });
+    expect(out.out.text.trim()).toMatchInlineSnapshot(
+      '"<body data-abell-heumYD>${e( a )}</body>"'
+    );
+    expect(out.declarationsBlock.text.trim()).toMatchInlineSnapshot(`
+      "// declarations
+            const a = 3;"
+    `);
+  });
+
   test('should successfully compile with declarations', () => {
     const abellCode = `
     {{
diff --git a/packages/abell/src/vite-plugin-abell/compiler/utils.ts b/packages/abell/src/vite-plugin-abell/compiler/utils.ts
@@ -37,7 +37,11 @@ export const isDeclarationBlock = (
   blockCount: number,
   blockContent: string
 ): boolean => {
-  if (blockCount < 2 && blockContent.includes('import ')) {
+  if (
+    blockCount < 2 &&
+    (blockContent.includes('import ') ||
+      blockContent.includes('// declaration'))
+  ) {
     return true;
   }
   return false;

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "abell",`
`3`		`- "version": "1.0.0-alpha.65",`
	`3`	`+ "version": "1.0.0-alpha.66",`
`4`	`4`	`"description": "Abell is a static-site-generator for JavaScript developers. Powered by Vite, It tries to stay close to fundamentals while providing a great DX",`
`5`	`5`	`"bin": "./dist/bin.js",`
`6`	`6`	`"main": "./dist/index.js",`