fix: handle custom canary environment

Author: danielroe

config/env.ts

@@ -44,13 +44,15 @@ export const gitBranch = process.env.BRANCH || process.env.VERCEL_GIT_COMMIT_REF
  * Whether this is the canary environment (main.npmx.dev).
  *
  * Detected as any non-PR Vercel deploy from the `main` branch
- * (which may receive `VERCEL_ENV === 'production'` or `'preview'`
- * depending on the project's production branch configuration).
+ * (which may receive `VERCEL_ENV === 'production'`, `'preview'`, or a
+ * custom `'canary'` environment depending on the project configuration).
  *
  * @see {@link https://vercel.com/docs/environment-variables/system-environment-variables#VERCEL_ENV}
  */
 export const isCanary =
-  (process.env.VERCEL_ENV === 'production' || process.env.VERCEL_ENV === 'preview') &&
+  (process.env.VERCEL_ENV === 'production' ||
+    process.env.VERCEL_ENV === 'preview' ||
+    process.env.VERCEL_ENV === 'canary') &&
   gitBranch === 'main' &&
   !isPR
 

test/unit/config/env.spec.ts

@@ -43,6 +43,14 @@ describe('isCanary', () => {
     expect(isCanary).toBe(true)
   })
 
+  it('returns true when VERCEL_ENV is custom "canary" and branch is "main"', async () => {
+    vi.stubEnv('VERCEL_ENV', 'canary')
+    vi.stubEnv('VERCEL_GIT_COMMIT_REF', 'main')
+    const { isCanary } = await import('../../../config/env')
+
+    expect(isCanary).toBe(true)
+  })
+
   it('returns false when VERCEL_ENV is "preview", branch is "main", but is a PR', async () => {
     vi.stubEnv('VERCEL_ENV', 'preview')
     vi.stubEnv('VERCEL_GIT_COMMIT_REF', 'main')
@@ -97,6 +105,15 @@ describe('getEnv', () => {
     expect(result.env).toBe('canary')
   })
 
+  it('returns "canary" for custom Vercel "canary" environment on main branch', async () => {
+    vi.stubEnv('VERCEL_ENV', 'canary')
+    vi.stubEnv('VERCEL_GIT_COMMIT_REF', 'main')
+    const { getEnv } = await import('../../../config/env')
+    const result = await getEnv(false)
+
+    expect(result.env).toBe('canary')
+  })
+
   it('returns "preview" for Vercel preview PR deploys', async () => {
     vi.stubEnv('VERCEL_ENV', 'preview')
     vi.stubEnv('VERCEL_GIT_PULL_REQUEST_ID', '123')