Enhance !reference tag support to handle scalar values and update tests

Author: noxify

.changeset/extend-reference-tag-scalar-support.md

@@ -0,0 +1,5 @@
+---
+"@noxify/gitlab-ci-builder": patch
+---
+
+Extended `!reference` tag support to handle scalar values (e.g., `image`, `extends`) in addition to arrays, ensuring inline format without quotes for all use cases.

src/export.ts

@@ -126,7 +126,7 @@ export function toYaml(config: GitLabCi) {
   while (i < lines.length) {
     const line = lines[i]
 
-    // Check if this line contains a multiline !reference tag
+    // Case 1: Check if this line contains a multiline !reference tag in an array
     if (line && line.trim() === "- !reference") {
       // Next two lines should contain the array elements
       const nextLine1 = lines[i + 1]
@@ -154,6 +154,37 @@ export function toYaml(config: GitLabCi) {
       }
     }
 
+    // Case 2: Check if this line contains a scalar !reference (e.g., "image: !reference")
+    if (line?.includes(": !reference")) {
+      // Next two lines should contain the array elements
+      const nextLine1 = lines[i + 1]
+      const nextLine2 = lines[i + 2]
+
+      if (
+        nextLine1 &&
+        nextLine2 &&
+        nextLine1.trim().startsWith("- ") &&
+        nextLine2.trim().startsWith("- ")
+      ) {
+        const elem1 = nextLine1.trim().slice(2)
+        const elem2 = nextLine2.trim().slice(2)
+
+        // Get the key part (e.g., "image:")
+        const keyMatch = /^(\s*)(.+):\s*!reference\s*$/.exec(line)
+        if (keyMatch) {
+          const indent = keyMatch[1] ?? ""
+          const key = keyMatch[2]
+
+          // Create inline format
+          resultLines.push(`${indent}${key}: !reference [${elem1}, ${elem2}]`)
+
+          // Skip the next two lines
+          i += 3
+          continue
+        }
+      }
+    }
+
     if (line !== undefined) {
       resultLines.push(line)
     }

tests/export.test.ts

@@ -299,5 +299,24 @@ describe("export", () => {
       expect(yaml).not.toContain('"!reference')
       expect(yaml).toContain("- pnpm run test")
     })
+
+    it("should handle !reference in scalar values like image", () => {
+      const config = new Config()
+      config.template(".database_template", {
+        image: "postgres:15",
+      })
+      config.job("test", {
+        image: "!reference [.database_template, image]",
+        script: ["npm test"],
+      })
+
+      const yaml = toYaml(config.getPlainObject())
+
+      expect(yaml).toContain("test:")
+      expect(yaml).toContain("image: !reference [.database_template, image]")
+      expect(yaml).not.toContain('"!reference')
+      expect(yaml).toContain("script:")
+      expect(yaml).toContain("- npm test")
+    })
   })
 })