Add secrets

Author: owjs3901

README.md

@@ -11,6 +11,7 @@ This GitHub Action allows you to collect environment variables with a specific p
     prefix: 'API_'
     output: '.env.api'
     remove-prefix: true
+    secrets: ${{ toJson(secrets) }}
 ```
 
 #### Inputs
@@ -43,6 +44,7 @@ jobs:
           prefix: 'API_'
           output: '.env.api'
           remove-prefix: true
+          secrets: ${{ toJson(secrets) }}
 ```
 
 #### Example

action.yml

@@ -13,6 +13,11 @@ inputs:
   output:
     description: The file to write to
     required: true
+  secrets:
+    description: The secrets to collect. This is a JSON object stringified.
+    required: false
+    type: string
+    default: "{}"
   remove-prefix:
     description: The prefix to remove
     required: false

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "collect-env",
-	"version": "1.0.0",
+	"version": "1.0.1",
 	"description": "Collect env values and write to a file",
 	"author": "JeongMin Oh",
 	"license": "Apache-2.0",

src/__tests__/run.test.ts

@@ -28,11 +28,13 @@ test('should write filtered environment variables to file', async () => {
   process.env.OTHER_VAR = 'other'
 
   mockGetInput.mockImplementation((name: string) => {
-    switch (name as 'prefix' | 'output') {
+    switch (name as 'prefix' | 'output' | 'secrets') {
       case 'prefix':
         return testPrefix
       case 'output':
         return testOutput
+      case 'secrets':
+        return '{"TEST_SECRET": "secret123"}'
     }
   })
 
@@ -46,7 +48,7 @@ test('should write filtered environment variables to file', async () => {
   expect(mockGetBooleanInput).toHaveBeenCalledWith('remove-prefix')
   expect(mockWriteFile).toHaveBeenCalledWith(
     testOutput,
-    'TEST_VAR1=value1\nTEST_VAR2=value2',
+    'TEST_SECRET=secret123\nTEST_VAR1=value1\nTEST_VAR2=value2',
   )
 })
 
@@ -58,11 +60,13 @@ test('should handle writeFile error', async () => {
   process.env.TEST_VAR = 'value'
 
   mockGetInput.mockImplementation((name: string) => {
-    switch (name as 'prefix' | 'output') {
+    switch (name as 'prefix' | 'output' | 'secrets') {
       case 'prefix':
         return testPrefix
       case 'output':
         return testOutput
+      case 'secrets':
+        return '{"TEST_SECRET": "secret123"}'
     }
   })
 
@@ -80,11 +84,13 @@ test('should handle empty environment variables', async () => {
   const testOutput = 'empty.env'
 
   mockGetInput.mockImplementation((name: string) => {
-    switch (name as 'prefix' | 'output') {
+    switch (name as 'prefix' | 'output' | 'secrets') {
       case 'prefix':
         return testPrefix
       case 'output':
         return testOutput
+      case 'secrets':
+        return '{"TEST_SECRET": "secret123"}'
     }
   })
 
@@ -94,6 +100,7 @@ test('should handle empty environment variables', async () => {
   await run()
 
   expect(mockWriteFile).toHaveBeenCalledWith(testOutput, '')
+  expect(mockGetInput).toHaveBeenCalledWith('secrets')
 })
 
 test('should filter environment variables by prefix', async () => {
@@ -106,11 +113,13 @@ test('should filter environment variables by prefix', async () => {
   process.env.API_TIMEOUT = '5000'
 
   mockGetInput.mockImplementation((name: string) => {
-    switch (name as 'prefix' | 'output') {
+    switch (name as 'prefix' | 'output' | 'secrets') {
       case 'prefix':
         return testPrefix
       case 'output':
         return testOutput
+      case 'secrets':
+        return '{"TEST_SECRET": "secret123"}'
     }
   })
 
@@ -136,11 +145,13 @@ test('should remove prefix from environment variables', async () => {
   process.env.API_TIMEOUT = '5000'
 
   mockGetInput.mockImplementation((name: string) => {
-    switch (name as 'prefix' | 'output') {
+    switch (name as 'prefix' | 'output' | 'secrets') {
       case 'prefix':
         return testPrefix
       case 'output':
         return testOutput
+      case 'secrets':
+        return '{"TEST_SECRET": "secret123"}'
     }
   })
 

src/run.ts

@@ -1,24 +1,20 @@
 import { writeFile } from 'node:fs/promises'
-import { debug, getBooleanInput, getInput, setFailed } from '@actions/core'
+import { getBooleanInput, getInput, setFailed } from '@actions/core'
 
 export async function run() {
   const prefix = getInput('prefix')
   const output = getInput('output')
+  const secrets = JSON.parse(getInput('secrets'))
   const removePrefix = getBooleanInput('remove-prefix')
-  debug(`prefix: ${prefix}`)
-  debug(`output: ${output}`)
-  debug(`removePrefix: ${removePrefix}`)
-  debug(`process.env: ${JSON.stringify(Object.keys(process.env), null, 2)}`)
-  debug(`process.env: ${JSON.stringify(Object.entries(process.env), null, 2)}`)
 
   try {
     await writeFile(
       output,
-      Object.keys(process.env)
-        .filter((key) => key.startsWith(prefix))
+      Object.entries({ ...secrets, ...process.env })
+        .filter(([key]) => key.startsWith(prefix))
         .map(
-          (key) =>
-            `${removePrefix ? key.substring(prefix.length) : key}=${process.env[key]}`,
+          ([key, value]) =>
+            `${removePrefix ? key.substring(prefix.length) : key}=${value}`,
         )
         .join('\n'),
     )