@stacksjs/env 0.58.47

package/src/index.ts

@@ -0,0 +1,76 @@
+import p from 'node:process'
+import fs from 'fs-extra'
+import { projectPath } from '@stacksjs/path'
+import { ValidationBoolean, ValidationEnum, ValidationNumber } from '@stacksjs/validation'
+import type { EnvKey } from '../../../env'
+import type { Env } from './types'
+
+interface EnumObject {
+  [key: string]: string[]
+}
+
+export const enums: EnumObject = {
+  APP_ENV: ['local', 'dev', 'development', 'staging', 'prod', 'production'],
+  DB_CONNECTION: ['mysql', 'sqlite', 'postgres', 'planetscale'],
+  MAIL_MAILER: ['smtp', 'mailgun', 'ses', 'postmark', 'sendmail', 'log'],
+  SEARCH_ENGINE_DRIVER: ['meilisearch', 'algolia', 'typesense'],
+  FRONTEND_APP_ENV: ['development', 'staging', 'production'],
+}
+
+const handler = {
+  get: (target: Env, key: EnvKey) => {
+    const value = target[key] as any
+
+    // if value is a string but only contains numbers, and the key is not AWS_ACCOUNT_ID, return it as a number
+    if (typeof value === 'string' && /^\d+$/.test(value) && key !== 'AWS_ACCOUNT_ID')
+      return Number(value)
+
+    // if value is a string but only contains boolean values, return it as a boolean
+    if (typeof value === 'string' && /^(true|false)$/.test(value))
+      return value === 'true'
+
+    // at some point, let's see if we can remove the need for below
+    if (value instanceof ValidationEnum)
+      return target[key] as string
+
+    if (value instanceof ValidationBoolean)
+      return !!target[key]
+
+    if (value instanceof ValidationNumber)
+      return Number(target[key])
+
+    return value as string
+  },
+}
+
+export function process() {
+  return typeof Bun !== 'undefined'
+    ? Bun.env
+    : p.env as unknown as Env
+}
+
+export const env: Env = new Proxy(process(), handler)
+
+export function writeEnv(key: EnvKey, value: string, options?: { path: string }) {
+  const envPath = options?.path || projectPath('.env')
+  const env = fs.readFileSync(envPath, 'utf-8')
+
+  // Split the file into lines
+  const lines = env.split('\n')
+
+  // Find the line with the variable we want to update
+  const index = lines.findIndex(line => line.startsWith(`${key}=`))
+
+  // If the variable exists, update it
+  if (index !== -1)
+    lines[index] = `${key}=${value}`
+
+  // Otherwise, add a new line
+  else
+    lines.push(`${key}=${value}`)
+
+  // Join the lines back into a string and write it to the .env file
+  fs.writeFileSync(envPath, lines.join('\n'))
+}
+
+export * from './types'

package/src/types.ts

@@ -0,0 +1,54 @@
+import type { Infer, VineBoolean, VineEnum, VineNumber, VineString } from '@stacksjs/validation'
+import { validator } from '@stacksjs/validation'
+import env from '../../../../../config/env'
+import type { EnvKey } from '../../../env'
+
+// import type { Validate } from '@stacksjs/validation'
+
+// we need to get this just into right format so we can infer the type
+type EnvValue = string | boolean | number | readonly string[]
+type EnvType = typeof env
+type EnvKeys = keyof EnvType
+type EnvMap = { [K in EnvKeys]: EnvType[K] extends string ? VineString :
+  EnvType[K] extends number ? VineNumber :
+    EnvType[K] extends boolean ? VineBoolean :
+      EnvType[K] extends readonly string[] ? VineEnum<string[]> : unknown }
+
+type ValidatorType = VineString | VineNumber | VineBoolean | VineEnum<string[]>
+
+const envStructure = Object.entries(env).reduce((acc, [key, value]) => {
+  let validatorType: ValidatorType
+  switch (typeof value) {
+    case 'string':
+      validatorType = validator.string()
+      break
+    case 'number':
+      validatorType = validator.number()
+      break
+    case 'boolean':
+      validatorType = validator.boolean()
+      break
+    default:
+      if (Array.isArray(value)) {
+        validatorType = validator.enum(value as string[])
+        break
+      }
+      throw new Error(`Invalid env value for ${key}`)
+  }
+  const envKey = key as EnvKeys
+
+  acc[envKey] = validatorType as any
+  return acc
+}, {} as EnvMap)
+
+export const envSchema = validator.object(envStructure)
+export type Env = Infer<typeof envSchema>
+
+export type EnvOptions = Env
+export type EnvConfig = Partial<Record<EnvKey, EnvValue>>
+
+export interface FrontendEnv {
+  FRONTEND_APP_ENV: 'local' | 'development' | 'staging' | 'production'
+  FRONTEND_APP_URL: string
+}
+export type FrontendEnvKeys = keyof FrontendEnv