feat: support declarations comment

abelljs · Aug 7, 2022 · 4ccc09c · 4ccc09c
1 parent 2b33109
commit 4ccc09c
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 2 deletions.
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;