Wrap existing defineConfig with other function

[aheissenberger]

I want to wrap an existing exported defineConfig with a function which loads environment values from files.

gives is this:
vite.config.ts:

import { defineConfig } from 'vite';
export default defineConfig({
   server: {
      port: process.env.PORT
   }
})

.env:

PORT=80

The result after transformation should look like this:

vite.config.ts:

import { defineConfig, loadEnv } from 'vite';
export default ({ mode }) => {
    process.env = {...process.env, ...loadEnv(mode, process.cwd())};
    return defineConfig({
       server: {
         port: process.env.PORT
       }
    })
})

I solved adding loadEnv but have no idea how to wrap the existing default export:

// add loadEnv to import `import { defineConfig, loadEnv } from 'vite';`
mod.imports.$add({
    from: "vite",
    imported: "loadEnv",
});