From 9c8e3083d687a0aa96999bde3ea0100f8963180b Mon Sep 17 00:00:00 2001
From: Haoqun Jiang <haoqunjiang@gmail.com>
Date: Fri, 28 Jul 2023 16:31:41 +0800
Subject: [PATCH] fix: port the latest `defineConfig` type from vite (#3804)

---
 packages/vitest/src/config.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/vitest/src/config.ts b/packages/vitest/src/config.ts
index 0afb83cccd86..1fdd469cd2bf 100644
--- a/packages/vitest/src/config.ts
+++ b/packages/vitest/src/config.ts
@@ -10,13 +10,19 @@ export { configDefaults, defaultInclude, defaultExclude, coverageConfigDefaults
 export { mergeConfig } from 'vite'
 
 export type { ConfigEnv, ViteUserConfig as UserConfig }
+export type UserConfigFnObject = (env: ConfigEnv) => ViteUserConfig
+export type UserConfigFnPromise = (env: ConfigEnv) => Promise<ViteUserConfig>
 export type UserConfigFn = (env: ConfigEnv) => ViteUserConfig | Promise<ViteUserConfig>
-export type UserConfigExport = ViteUserConfig | Promise<ViteUserConfig> | UserConfigFn
+export type UserConfigExport = ViteUserConfig | Promise<ViteUserConfig> | UserConfigFnObject | UserConfigFnPromise | UserConfigFn
 
 export type UserProjectConfigFn = (env: ConfigEnv) => UserWorkspaceConfig | Promise<UserWorkspaceConfig>
 export type UserProjectConfigExport = UserWorkspaceConfig | Promise<UserWorkspaceConfig> | UserProjectConfigFn
 
-export function defineConfig(config: UserConfigExport) {
+export function defineConfig(config: ViteUserConfig): ViteUserConfig
+export function defineConfig(config: Promise<ViteUserConfig>): Promise<ViteUserConfig>
+export function defineConfig(config: UserConfigFnObject): UserConfigFnObject
+export function defineConfig(config: UserConfigExport): UserConfigExport
+export function defineConfig(config: UserConfigExport): UserConfigExport {
   return config
 }