Enable users to define their env var validations

waspc/data/Generator/templates/sdk/wasp/client/env.ts

@@ -1,14 +1,24 @@
 {{={= =}=}}
 import * as z from 'zod'
 
-import { ensureEnvSchema } from '../env/index.js'
+import { ensureEnvSchema } from '../env/validation.js'
 
-const clientEnvSchema = z.object({
+{=# envValidationFn.isDefined =}
+{=& envValidationFn.importStatement =}
+const userClientEnvSchema = {= envValidationFn.importIdentifier =}()
+{=/ envValidationFn.isDefined =}
+{=^ envValidationFn.isDefined =}
+const userClientEnvSchema = z.object({})
+{=/ envValidationFn.isDefined =}
+
+const waspClientEnvSchema = z.object({
   REACT_APP_API_URL: z
     .string({
       required_error: 'REACT_APP_API_URL is required',
     })
     .default('{= defaultServerUrl =}')
 })
 
+const clientEnvSchema = waspClientEnvSchema.merge(userClientEnvSchema)
+
 export const env = ensureEnvSchema(import.meta.env, clientEnvSchema)

waspc/data/Generator/templates/sdk/wasp/env/index.ts

@@ -1,29 +1,3 @@
-import * as z from 'zod'
+import type { ZodObject } from 'zod'
 
-const redColor = '\x1b[31m'
-
-export function ensureEnvSchema<Schema extends z.ZodTypeAny>(
-  data: unknown,
-  schema: Schema
-): z.infer<Schema> {
-  try {
-    return schema.parse(data)
-  } catch (e) {
-    // TODO: figure out how to output the error message in a better way
-    if (e instanceof z.ZodError) {
-      console.error()
-      console.error(redColor, '╔═════════════════════════════╗');
-      console.error(redColor, '║ Env vars validation failed  ║');
-      console.error(redColor, '╚═════════════════════════════╝');
-      console.error()
-      for (const error of e.errors) {
-        console.error(redColor, `- ${error.message}`)
-      }
-      console.error()
-      console.error(redColor, '═══════════════════════════════');
-      throw new Error('Error parsing environment variables')
-    } else {
-      throw e
-    }
-  }
-}
+export type EnvValidationFn = () => ZodObject<any>

waspc/data/Generator/templates/sdk/wasp/env/validation.ts

@@ -0,0 +1,28 @@
+import * as z from 'zod'
+
+const redColor = '\x1b[31m'
+
+export function ensureEnvSchema<Schema extends z.ZodTypeAny>(
+  data: unknown,
+  schema: Schema
+): z.infer<Schema> {
+  try {
+    return schema.parse(data)
+  } catch (e) {
+    if (e instanceof z.ZodError) {
+      console.error()
+      console.error(redColor, '╔═════════════════════════════╗');
+      console.error(redColor, '║ Env vars validation failed  ║');
+      console.error(redColor, '╚═════════════════════════════╝');
+      console.error()
+      for (const error of e.errors) {
+        console.error(redColor, `- ${error.message}`)
+      }
+      console.error()
+      console.error(redColor, '═══════════════════════════════');
+      throw new Error('Error parsing environment variables')
+    } else {
+      throw e
+    }
+  }
+}

waspc/data/Generator/templates/sdk/wasp/package.json

@@ -109,6 +109,7 @@
       "./client/test": "./dist/client/test/index.js",
       "./client": "./dist/client/index.js",
       "./dev": "./dist/dev/index.js",
+      "./env": "./dist/env/index.js",
 
       {=! todo(filip): Fixes below are for type errors in 0.13.1, remove ASAP =}
       {=! Used by our code (SDK for full-stack type safety), uncodumented (but accessible) for users. =}

waspc/data/Generator/templates/sdk/wasp/server/env.ts

@@ -1,9 +1,17 @@
 {{={= =}=}}
 import * as z from 'zod'
 
-import { ensureEnvSchema } from '../env/index.js'
+import { ensureEnvSchema } from '../env/validation.js'
 
-const serverCommonSchema = z.object({
+{=# envValidationFn.isDefined =}
+{=& envValidationFn.importStatement =}
+const userServerEnvSchema = {= envValidationFn.importIdentifier =}()
+{=/ envValidationFn.isDefined =}
+{=^ envValidationFn.isDefined =}
+const userServerEnvSchema = z.object({})
+{=/ envValidationFn.isDefined =}
+
+const waspServerCommonSchema = z.object({
   PORT: z.coerce.number().default({= defaultServerPort =}),
   {= databaseUrlEnvVarName =}: z.string({
     required_error: '{= databaseUrlEnvVarName =} is required',
@@ -133,6 +141,7 @@ const serverProdSchema = z.object({
   {=/ isAuthEnabled =}
 })
 
+const serverCommonSchema = waspServerCommonSchema.merge(userServerEnvSchema)
 const serverEnvSchema = z.discriminatedUnion('NODE_ENV', [
   serverDevSchema.merge(serverCommonSchema),
   serverProdSchema.merge(serverCommonSchema)

waspc/data/Generator/templates/sdk/wasp/server/jobs/core/pgBoss/pgBoss.ts

@@ -1,6 +1,5 @@
 import PgBoss from 'pg-boss'
-import { env } from '../../../env.js'
-import { config } from '../../../index.js'
+import { config, env } from '../../../index.js'
 
 const boss = createPgBoss()