@@ -4,12 +4,12 @@ import {
44 addV3ClientModules ,
55 getClientMetadata ,
66 getV2ClientNames ,
7- getV2ClientNamesFromDefault ,
8- getV2DefaultModuleName ,
7+ getV2ClientNamesFromGlobal ,
8+ getV2GlobalName ,
99 isTypeScriptFile ,
10- removeDefaultModuleIfNotUsed ,
1110 removePromiseCalls ,
1211 removeV2ClientModule ,
12+ removeV2GlobalModule ,
1313 replaceClientCreation ,
1414 replaceTSTypeReference ,
1515} from "./utils" ;
@@ -18,36 +18,34 @@ export default function transformer(file: FileInfo, api: API) {
1818 const j = isTypeScriptFile ( file . path ) ? api . jscodeshift . withParser ( "ts" ) : api . jscodeshift ;
1919 const source = j ( file . source ) ;
2020
21- // ToDo: Make v2DefaultModuleName optional downstream as it can be undefined.
22- // ToDo: Rename v2DefaultModuleName to v2GlobalName to align with v2ClientName.
23- const v2DefaultModuleName = getV2DefaultModuleName ( j , source ) as string ;
21+ // ToDo: Make v2GlobalName optional downstream as it can be undefined.
22+ const v2GlobalName = getV2GlobalName ( j , source ) as string ;
2423 const v2ClientNames = getV2ClientNames ( j , source ) ;
2524
26- if ( ! v2DefaultModuleName && v2ClientNames . length === 0 ) {
25+ if ( ! v2GlobalName && v2ClientNames . length === 0 ) {
2726 return source . toSource ( ) ;
2827 }
2928
30- if ( v2DefaultModuleName ) {
31- v2ClientNames . push ( ...getV2ClientNamesFromDefault ( j , source , v2DefaultModuleName ) ) ;
29+ if ( v2GlobalName ) {
30+ v2ClientNames . push ( ...getV2ClientNamesFromGlobal ( j , source , v2GlobalName ) ) ;
3231 }
3332
3433 const clientMetadata = getClientMetadata ( v2ClientNames ) ;
3534
3635 for ( const [ v2ClientName , v3ClientMetadata ] of Object . entries ( clientMetadata ) . reverse ( ) ) {
3736 const { v3ClientName, v3ClientPackageName } = v3ClientMetadata ;
38- addV3ClientModules ( j , source , {
39- v2ClientName,
40- v3ClientName,
41- v3ClientPackageName,
42- v2DefaultModuleName,
43- } ) ;
44- replaceTSTypeReference ( j , source , { v2ClientName, v2DefaultModuleName, v3ClientName } ) ;
45- removeV2ClientModule ( j , source , { v2ClientName, v2DefaultModuleName } ) ;
46- removePromiseCalls ( j , source , { v2ClientName, v2DefaultModuleName } ) ;
47- replaceClientCreation ( j , source , { v2ClientName, v2DefaultModuleName, v3ClientName } ) ;
37+
38+ const v2Options = { v2ClientName, v2GlobalName } ;
39+ const v3Options = { v3ClientName, v3ClientPackageName } ;
40+
41+ addV3ClientModules ( j , source , { ...v2Options , ...v3Options } ) ;
42+ replaceTSTypeReference ( j , source , { ...v2Options , v3ClientName } ) ;
43+ removeV2ClientModule ( j , source , v2Options ) ;
44+ removePromiseCalls ( j , source , v2Options ) ;
45+ replaceClientCreation ( j , source , { ...v2Options , v3ClientName } ) ;
4846 }
4947
50- removeDefaultModuleIfNotUsed ( j , source , v2DefaultModuleName ) ;
48+ removeV2GlobalModule ( j , source , v2GlobalName ) ;
5149
5250 return source . toSource ( ) ;
5351}
0 commit comments