From b190d7e9369d4335cc89ca3b86b33df2e5a1f7a6 Mon Sep 17 00:00:00 2001
From: qiushengchao <shengchao.qiushch@dtzhejiang.com>
Date: Thu, 7 Dec 2023 13:53:10 +0800
Subject: [PATCH] fix(codemod): configPath maybe .umirc.ts or .umirc.js

---
 codemod/src/prepare.ts       |  1 +
 codemod/src/runner/config.ts | 13 +++----------
 2 files changed, 4 insertions(+), 10 deletions(-)

diff --git a/codemod/src/prepare.ts b/codemod/src/prepare.ts
index 912190d49e68..104d123140b5 100644
--- a/codemod/src/prepare.ts
+++ b/codemod/src/prepare.ts
@@ -75,6 +75,7 @@ export async function prepare(opts: { cwd: string; pattern: any; args?: any }) {
 
   return {
     config,
+    configPath: mainConfigFile,
     absAppJSPath,
     pkg,
     pkgPath,
diff --git a/codemod/src/runner/config.ts b/codemod/src/runner/config.ts
index 0a1ae621a76f..123a79de11b2 100644
--- a/codemod/src/runner/config.ts
+++ b/codemod/src/runner/config.ts
@@ -1,8 +1,6 @@
 import * as t from '@umijs/bundler-utils/compiled/babel/types';
 import { lodash } from '@umijs/utils';
-import assert from 'assert';
 import { existsSync, readFileSync, writeFileSync } from 'fs';
-import { join } from 'path';
 import { update as appJSUpdate } from '../appJSUpdater';
 import { update } from '../configUpdater';
 import { info } from '../logger';
@@ -123,17 +121,12 @@ export class Runner {
       setKeys['layout.locale'] = false;
     }
 
-    const configFile = join(this.cwd, 'config/config.ts');
-    assert(
-      existsSync(configFile),
-      `Could not find config file at ${configFile}`,
-    );
     const {
       config: { code },
       routesConfig: { filePath: routeFilePath, code: routeCode } = {} as any,
     } = update({
-      code: readFileSync(configFile, 'utf-8'),
-      filePath: configFile,
+      code: readFileSync(this.context.configPath, 'utf-8'),
+      filePath: this.context.configPath,
       updates: {
         del: deleteKeys,
         set: setKeys,
@@ -164,7 +157,7 @@ export class Runner {
         },
       },
     });
-    writeFileSync(configFile, code);
+    writeFileSync(this.context.configPath, code);
     if (routeCode) {
       writeFileSync(routeFilePath, routeCode);
     }