@@ -6,8 +6,8 @@ import getCustomCodeActions, { REFACTORS_CATEGORY } from './getCodeActions'
6
6
import improveBuiltin from './improveBuiltin'
7
7
8
8
export default ( proxy : ts . LanguageService , languageService : ts . LanguageService , languageServiceHost : ts . LanguageServiceHost , c : GetConfig ) => {
9
- proxy . getApplicableRefactors = ( fileName , positionOrRange , preferences ) => {
10
- let prior = languageService . getApplicableRefactors ( fileName , positionOrRange , preferences )
9
+ proxy . getApplicableRefactors = ( fileName , positionOrRange , preferences , ... args ) => {
10
+ let prior = languageService . getApplicableRefactors ( fileName , positionOrRange , preferences , ... args )
11
11
12
12
previousGetCodeActionsResult . value = compact (
13
13
prior . flatMap ( refactor => {
@@ -34,7 +34,7 @@ export default (proxy: ts.LanguageService, languageService: ts.LanguageService,
34
34
return prior
35
35
}
36
36
37
- proxy . getEditsForRefactor = ( fileName , formatOptions , positionOrRange , refactorName , actionName , preferences ) => {
37
+ proxy . getEditsForRefactor = ( fileName , formatOptions , positionOrRange , refactorName , actionName , preferences , ... args ) => {
38
38
const category = refactorName
39
39
if ( category === REFACTORS_CATEGORY ) {
40
40
const program = languageService . getProgram ( )
@@ -43,10 +43,19 @@ export default (proxy: ts.LanguageService, languageService: ts.LanguageService,
43
43
return edit
44
44
}
45
45
if ( refactorName === 'Extract Symbol' && actionName . startsWith ( 'function_scope' ) ) {
46
- const handledResult = handleFunctionRefactorEdits ( actionName , languageService , fileName , formatOptions , positionOrRange , refactorName , preferences )
46
+ const handledResult = handleFunctionRefactorEdits (
47
+ actionName ,
48
+ languageService ,
49
+ fileName ,
50
+ formatOptions ,
51
+ positionOrRange ,
52
+ refactorName ,
53
+ preferences ,
54
+ ...args ,
55
+ )
47
56
if ( handledResult ) return handledResult
48
57
}
49
- const prior = languageService . getEditsForRefactor ( fileName , formatOptions , positionOrRange , refactorName , actionName , preferences )
58
+ const prior = languageService . getEditsForRefactor ( fileName , formatOptions , positionOrRange , refactorName , actionName , preferences , ... args )
50
59
if ( ! prior ) return
51
60
return improveBuiltin ( fileName , refactorName , actionName , languageService , c , prior ) ?? prior
52
61
}
0 commit comments