Add plugin override for watchFactory

Author: sheetalkamat

src/compiler/sys.ts

@@ -963,6 +963,11 @@ namespace ts {
                 ),
         };
 
+        function getWatchFactoryPlugin(options: WatchOptions) {
+            const plugin = isString(options.watchFactory) ? { name: options.watchFactory } : options.watchFactory!;
+            return options.watchFactoryWithConfigOverride?.(plugin) || plugin;
+        }
+
         function createWatcherConsideringFactory(
             options: WatchOptions | undefined,
             createWatcher: (factory?: UserWatchModule) => FileWatcher,
@@ -1001,7 +1006,7 @@ namespace ts {
                         setImportPluginResult(
                             options,
                             resolveModule<UserWatchFactory>(
-                                isString(options.watchFactory) ? { name: options.watchFactory } : options.watchFactory,
+                                getWatchFactoryPlugin(options),
                                 searchPaths,
                                 getSystem(),
                                 sysLog
@@ -1022,7 +1027,7 @@ namespace ts {
         function importFactory(options: WatchOptions, searchPaths: readonly string[], system: System) {
             return setGetResolvedWatchFactory(options, {
                 pending: importPlugin<UserWatchFactory>(
-                    isString(options.watchFactory) ? { name: options.watchFactory } : options.watchFactory!,
+                    getWatchFactoryPlugin(options),
                     searchPaths,
                     system,
                     sysLog
@@ -1037,7 +1042,13 @@ namespace ts {
             { resolvedModule, errorLogs, pluginConfigEntry }: ImportPluginResult<UserWatchFactory>,
         ) {
             if (typeof resolvedModule === "function") {
-                return setGetResolvedWatchFactory(options, { resolved: resolvedModule({ typescript: ts }) });
+                return setGetResolvedWatchFactory(options, {
+                    resolved: resolvedModule({
+                        typescript: ts,
+                        options,
+                        config: pluginConfigEntry
+                    })
+                });
             }
             else if (!resolvedModule) {
                 forEach(errorLogs, sysLog);

src/compiler/types.ts

@@ -6673,7 +6673,7 @@ namespace ts {
         [option: string]: CompilerOptionsValue | TsConfigSourceFile | undefined;
     }
 
-    export type UserWatchFactory = (mod: { typescript: typeof ts }) => UserWatchModule;
+    export type UserWatchFactory = (mod: { typescript: typeof ts, options: WatchOptions, config: any }) => UserWatchModule;
     export interface UserWatchModule {
         watchFile?(fileName: string, callback: FileWatcherCallback, pollingInterval: number, options: WatchOptions | undefined): FileWatcher;
         watchDirectory?(fileName: string, callback: DirectoryWatcherCallback, recursive: boolean, options: WatchOptions | undefined): FileWatcher;
@@ -6696,6 +6696,7 @@ namespace ts {
         /* @internal */ readonly configFile?: TsConfigSourceFile;
         /* @internal */ getResolvedWatchFactory?(): ResolvedWatchFactory | undefined;
         /* @internal */ watchFactorySearchPaths?(): readonly string[];
+        /* @internal */ watchFactoryWithConfigOverride?(plugin: PluginImport): PluginImport;
 
         [option: string]: CompilerOptionsValue | TsConfigSourceFile | undefined;
     }

src/server/editorServices.ts

@@ -784,7 +784,7 @@ namespace ts.server {
         public readonly pluginProbeLocations: readonly string[];
         public readonly allowLocalPluginLoads: boolean;
         /*@internal*/
-        currentPluginConfigOverrides: ESMap<string, any> | undefined;
+        private currentPluginConfigOverrides: ESMap<string, any> | undefined;
 
         public readonly typesMapLocation: string | undefined;
 
@@ -2237,7 +2237,10 @@ namespace ts.server {
                 configFileExistenceInfo.config.watchedDirectoriesStale = true;
                 configFileExistenceInfo.config.reloadLevel = undefined;
             }
-            if (parsedCommandLine.watchOptions) parsedCommandLine.watchOptions.watchFactorySearchPaths = () => this.getProjectPluginSearchPaths(canonicalConfigFilePath);
+            if (parsedCommandLine.watchOptions) {
+                parsedCommandLine.watchOptions.watchFactorySearchPaths = () => this.getProjectPluginSearchPaths(canonicalConfigFilePath);
+                parsedCommandLine.watchOptions.watchFactoryWithConfigOverride = plugin => this.getPluginWithConfigOverride(plugin);
+            }
 
             // If watch options different than older options when setting for the first time, update the config file watcher
             if (!oldCommandLine && !isJsonEqual(
@@ -3053,6 +3056,7 @@ namespace ts.server {
                     if (this.hostConfiguration.watchOptions?.watchFactory) {
                         // Set the watchFactory searchPaths same as global paths
                         this.hostConfiguration.watchOptions.watchFactorySearchPaths = () => this.getGlobalPluginSearchPaths();
+                        this.hostConfiguration.watchOptions.watchFactoryWithConfigOverride = plugin => this.getPluginWithConfigOverride(plugin);
                     }
                     this.projectWatchOptions.clear();
                     this.logger.info(`Host watch options changed to ${JSON.stringify(this.hostConfiguration.watchOptions)}, it will be take effect for next watches.`);
@@ -4132,6 +4136,18 @@ namespace ts.server {
             return searchPaths;
         }
 
+        /*@internal*/
+        getPluginWithConfigOverride(pluginConfigEntry: PluginImport) {
+            const configurationOverride = this.currentPluginConfigOverrides?.get(pluginConfigEntry.name);
+            if (configurationOverride) {
+                // Preserve the name property since it's immutable
+                const pluginName = pluginConfigEntry.name;
+                pluginConfigEntry = configurationOverride;
+                pluginConfigEntry.name = pluginName;
+            }
+            return pluginConfigEntry;
+        }
+
         /*@internal*/
         requestEnablePlugin(project: Project, pluginConfigEntry: PluginImport, searchPaths: string[]) {
             if (!this.host.importPlugin && !this.host.require) {

src/server/project.ts

@@ -1442,6 +1442,7 @@ namespace ts.server {
             this.watchOptions = watchOptions;
             if (this.watchOptions?.watchFactory) {
                 this.watchOptions.watchFactorySearchPaths = () => this.getProjectPluginSearchPaths();
+                this.watchOptions.watchFactoryWithConfigOverride = plugin => this.projectService.getPluginWithConfigOverride(plugin);
             }
         }
 
@@ -1612,15 +1613,7 @@ namespace ts.server {
         /*@internal*/
         endEnablePlugin({ pluginConfigEntry, resolvedModule, errorLogs }: BeginEnablePluginResult) {
             if (resolvedModule) {
-                const configurationOverride = this.projectService.currentPluginConfigOverrides?.get(pluginConfigEntry.name);
-                if (configurationOverride) {
-                    // Preserve the name property since it's immutable
-                    const pluginName = pluginConfigEntry.name;
-                    pluginConfigEntry = configurationOverride;
-                    pluginConfigEntry.name = pluginName;
-                }
-
-                this.enableProxy(resolvedModule, pluginConfigEntry);
+                this.enableProxy(resolvedModule, this.projectService.getPluginWithConfigOverride(pluginConfigEntry));
             }
             else {
                 forEach(errorLogs, message => this.projectService.logger.info(message));

src/testRunner/unittests/tscWatch/watchEnvironment.ts

@@ -900,10 +900,13 @@ namespace ts.tscWatch {
                 system.require = (initialPath, moduleName) => {
                     system.write(`Require:: Resolving ${moduleName} from ${initialPath}\n`);
                     return {
-                        module: () => ({
-                            watchFile: !excludeWatchFile ? system.factoryData.watchFile : undefined,
-                            watchDirectory: system.factoryData.watchDirectory,
-                        }),
+                        module: (({ options, config }) => {
+                            system.write(`Require:: Module ${moduleName} created with config: ${JSON.stringify(config)} and options: ${JSON.stringify(options)}\n`);
+                            return {
+                                watchFile: !excludeWatchFile ? system.factoryData.watchFile : undefined,
+                                watchDirectory: system.factoryData.watchDirectory,
+                            };
+                        }) as UserWatchFactory,
                         error: undefined
                     };
                 };

src/testRunner/unittests/tsserver/watchEnvironment.ts

@@ -831,9 +831,12 @@ namespace ts.projectSystem {
             return (initialPath, moduleName) => {
                 session.logger.logs.push(`CustomRequire:: Resolving ${moduleName} from ${initialPath}`);
                 return {
-                    module: (): UserWatchModule => !plugin2 || moduleName === "myplugin" ?
-                        plugin :
-                        plugin2,
+                    module: (({ options, config }) => {
+                        session.logger.logs.push(`Require:: Module ${moduleName} created with config: ${JSON.stringify(config)} and options: ${JSON.stringify(options)}`);
+                        return !plugin2 || moduleName === "myplugin" ?
+                            plugin :
+                            plugin2;
+                    }) as UserWatchFactory,
                     error: undefined
                 };
             };

tests/baselines/reference/api/tsserverlibrary.d.ts

@@ -3092,6 +3092,8 @@ declare namespace ts {
     }
     export type UserWatchFactory = (mod: {
         typescript: typeof ts;
+        options: WatchOptions;
+        config: any;
     }) => UserWatchModule;
     export interface UserWatchModule {
         watchFile?(fileName: string, callback: FileWatcherCallback, pollingInterval: number, options: WatchOptions | undefined): FileWatcher;

tests/baselines/reference/api/typescript.d.ts

@@ -3092,6 +3092,8 @@ declare namespace ts {
     }
     export type UserWatchFactory = (mod: {
         typescript: typeof ts;
+        options: WatchOptions;
+        config: any;
     }) => UserWatchModule;
     export interface UserWatchModule {
         watchFile?(fileName: string, callback: FileWatcherCallback, pollingInterval: number, options: WatchOptions | undefined): FileWatcher;

tests/baselines/reference/tscWatch/watchEnvironment/watchFactory/in-config-file-object.js

@@ -31,6 +31,7 @@ FileWatcher:: Added:: WatchInfo: /user/username/projects/myproject/tsconfig.json
 Enabling watchFactory myplugin from candidate paths: /user/username/projects/myproject,/a/lib/tsc.js/../../..
 Loading myplugin from /user/username/projects/myproject (resolved to /user/username/projects/myproject/node_modules)
 Require:: Resolving myplugin from /user/username/projects/myproject/node_modules
+Require:: Module myplugin created with config: {"name":"myplugin","myconfig":"somethingelse"} and options: {"watchFactory":{"name":"myplugin","myconfig":"somethingelse"}}
 Custom watchFile: /user/username/projects/myproject/tsconfig.json 2000 {"watchFactory":{"name":"myplugin","myconfig":"somethingelse"}}
 Synchronizing program
 CreatingProgramWith::

tests/baselines/reference/tscWatch/watchEnvironment/watchFactory/in-config-file.js

@@ -31,6 +31,7 @@ FileWatcher:: Added:: WatchInfo: /user/username/projects/myproject/tsconfig.json
 Enabling watchFactory myplugin from candidate paths: /user/username/projects/myproject,/a/lib/tsc.js/../../..
 Loading myplugin from /user/username/projects/myproject (resolved to /user/username/projects/myproject/node_modules)
 Require:: Resolving myplugin from /user/username/projects/myproject/node_modules
+Require:: Module myplugin created with config: {"name":"myplugin"} and options: {"watchFactory":"myplugin"}
 Custom watchFile: /user/username/projects/myproject/tsconfig.json 2000 {"watchFactory":"myplugin"}
 Synchronizing program
 CreatingProgramWith::