|
| 1 | +import { |
| 2 | + arrayFrom, |
| 3 | + compareStringsCaseSensitive, |
| 4 | + Debug, |
| 5 | + noop, |
| 6 | +} from "./_namespaces/ts"; |
| 7 | +import { |
| 8 | + AutoImportProviderProject, |
| 9 | + AuxiliaryProject, |
| 10 | + LogLevel, |
| 11 | + Project, |
| 12 | + ProjectKind, |
| 13 | + ProjectService, |
| 14 | + ScriptInfo, |
| 15 | +} from "./_namespaces/ts.server"; |
| 16 | +import { |
| 17 | + LoggerWithInMemoryLogs, |
| 18 | +} from "./tsserverLogger"; |
| 19 | + |
| 20 | +interface ProjectData { |
| 21 | + projectStateVersion: Project["projectStateVersion"]; |
| 22 | + projectProgramVersion: Project["projectProgramVersion"]; |
| 23 | + autoImportProviderHost: Project["autoImportProviderHost"]; |
| 24 | + noDtsResolutionProject: Project["noDtsResolutionProject"]; |
| 25 | +} |
| 26 | + |
| 27 | +interface ScriptInfoData { |
| 28 | + open: boolean; |
| 29 | + version: string; |
| 30 | + pendingReloadFromDisk: boolean; |
| 31 | + containingProjects: Set<Project>; |
| 32 | +} |
| 33 | + |
| 34 | +enum Diff { |
| 35 | + None = "", |
| 36 | + New = " *new*", |
| 37 | + Change = " *changed*", |
| 38 | + Deleted = " *deleted*", |
| 39 | +} |
| 40 | + |
| 41 | +type StateItemLog = [string, string[], string[]?]; |
| 42 | + |
| 43 | +export function patchServiceForStateBaseline(service: ProjectService) { |
| 44 | + if (!service.logger.isTestLogger || !service.logger.hasLevel(LogLevel.verbose)) return; |
| 45 | + if (service.baseline !== noop) return; // Already patched |
| 46 | + |
| 47 | + const projects = new Map<Project, ProjectData>(); |
| 48 | + const scriptInfos = new Map<ScriptInfo, ScriptInfoData>(); |
| 49 | + const allPropertiesOfProjectData: (keyof ProjectData & keyof Project)[] = [ |
| 50 | + "projectStateVersion", |
| 51 | + "projectProgramVersion", |
| 52 | + "autoImportProviderHost", |
| 53 | + "noDtsResolutionProject", |
| 54 | + ]; |
| 55 | + const logger = service.logger as LoggerWithInMemoryLogs; |
| 56 | + service.baseline = title => { |
| 57 | + const projectLogs = baselineProjects(); |
| 58 | + const infoLogs = baselineScriptInfos(); |
| 59 | + if (projectLogs?.length || infoLogs?.length) { |
| 60 | + if (title) logger.log(title); |
| 61 | + sendLogsToLogger("Projects::", projectLogs); |
| 62 | + sendLogsToLogger("ScriptInfos::", infoLogs); |
| 63 | + } |
| 64 | + }; |
| 65 | + |
| 66 | + function sendLogsToLogger(title: string, logs: StateItemLog[] | undefined) { |
| 67 | + if (!logs) return; |
| 68 | + logger.log(title); |
| 69 | + logs.sort((a, b) => compareStringsCaseSensitive(a[0], b[0])) |
| 70 | + .forEach(([title, propertyLogs, additionalPropertyLogs]) => { |
| 71 | + logger.log(title); |
| 72 | + propertyLogs.forEach(s => logger.log(s)); |
| 73 | + additionalPropertyLogs?.forEach(s => logger.log(s)); |
| 74 | + }); |
| 75 | + logger.log(""); |
| 76 | + } |
| 77 | + |
| 78 | + function baselineProjects() { |
| 79 | + const autoImportProviderProjects = [] as AutoImportProviderProject[]; |
| 80 | + const auxiliaryProjects = [] as AuxiliaryProject[]; |
| 81 | + return baselineState( |
| 82 | + [service.externalProjects, service.configuredProjects, service.inferredProjects, autoImportProviderProjects, auxiliaryProjects], |
| 83 | + projects, |
| 84 | + (logs, project, data) => { |
| 85 | + if (project.autoImportProviderHost) autoImportProviderProjects.push(project.autoImportProviderHost); |
| 86 | + if (project.noDtsResolutionProject) auxiliaryProjects.push(project.noDtsResolutionProject); |
| 87 | + let projectDiff = newOrDeleted(project, projects, data); |
| 88 | + const projectPropertyLogs = [] as string[]; |
| 89 | + allPropertiesOfProjectData.forEach(key => { |
| 90 | + let value: Project[typeof key] | string = project[key]; |
| 91 | + if (key === "autoImportProviderHost" || key === "noDtsResolutionProject") { |
| 92 | + if (project[key]) value = (project[key] as Project).projectName; |
| 93 | + else if (project[key] === undefined && data?.[key] === undefined) return; |
| 94 | + } |
| 95 | + projectDiff = printProperty(PrintPropertyWhen.Always, data, key, project[key], projectDiff, projectPropertyLogs, value); |
| 96 | + }); |
| 97 | + logs.push([`${project.projectName} (${ProjectKind[project.projectKind]})${projectDiff}`, projectPropertyLogs]); |
| 98 | + return projectDiff; |
| 99 | + }, |
| 100 | + project => ({ |
| 101 | + projectStateVersion: project.projectStateVersion, |
| 102 | + projectProgramVersion: project.projectProgramVersion, |
| 103 | + autoImportProviderHost: project.autoImportProviderHost, |
| 104 | + noDtsResolutionProject: project.noDtsResolutionProject, |
| 105 | + }), |
| 106 | + ); |
| 107 | + } |
| 108 | + |
| 109 | + function baselineScriptInfos() { |
| 110 | + return baselineState( |
| 111 | + [service.filenameToScriptInfo], |
| 112 | + scriptInfos, |
| 113 | + (logs, info, data) => { |
| 114 | + let infoDiff = newOrDeleted(info, scriptInfos, data); |
| 115 | + const infoPropertyLogs = [] as string[]; |
| 116 | + const isOpen = info.isScriptOpen(); |
| 117 | + infoDiff = printProperty(PrintPropertyWhen.Changed, data, "open", isOpen, infoDiff, infoPropertyLogs); |
| 118 | + infoDiff = printProperty(PrintPropertyWhen.Always, data, "version", info.textStorage.getVersion(), infoDiff, infoPropertyLogs); |
| 119 | + infoDiff = printProperty(PrintPropertyWhen.TruthyOrChangedOrNew, data, "pendingReloadFromDisk", info.textStorage.pendingReloadFromDisk, infoDiff, infoPropertyLogs); |
| 120 | + const containingProjectsLogs = [] as string[]; |
| 121 | + let containingProjectsDiff = Diff.None; |
| 122 | + const defaultProject = isOpen && info.containingProjects.length ? info.getDefaultProject() : undefined; |
| 123 | + info.containingProjects.forEach(project => |
| 124 | + containingProjectsDiff = printPropertyWorker( |
| 125 | + PrintPropertyWhen.Always, |
| 126 | + ` ${project.projectName}${defaultProject === project ? " *default*" : ""}`, |
| 127 | + containingProjectsLogs, |
| 128 | + containingProjectsDiff, |
| 129 | + !!data && !data.containingProjects.has(project), |
| 130 | + Diff.New, |
| 131 | + project, |
| 132 | + ) |
| 133 | + ); |
| 134 | + const infoProjects = new Set(info.containingProjects); |
| 135 | + data?.containingProjects.forEach(project => { |
| 136 | + if (infoProjects.has(project)) return; |
| 137 | + containingProjectsDiff = Diff.Change; |
| 138 | + containingProjectsLogs.push(` ${project.projectName}${Diff.Deleted}`); |
| 139 | + }); |
| 140 | + infoDiff = printPropertyWorker( |
| 141 | + PrintPropertyWhen.Always, |
| 142 | + `containingProjects: ${info.containingProjects.length}`, |
| 143 | + infoPropertyLogs, |
| 144 | + infoDiff, |
| 145 | + !!containingProjectsDiff, |
| 146 | + Diff.Change, |
| 147 | + info.containingProjects, |
| 148 | + ); |
| 149 | + logs.push([ |
| 150 | + `${info.fileName}${isOpen ? " (Open)" : ""}${infoDiff}`, |
| 151 | + infoPropertyLogs, |
| 152 | + containingProjectsLogs, |
| 153 | + ]); |
| 154 | + return infoDiff; |
| 155 | + }, |
| 156 | + info => ({ |
| 157 | + open: info.isScriptOpen(), |
| 158 | + version: info.textStorage.getVersion(), |
| 159 | + pendingReloadFromDisk: info.textStorage.pendingReloadFromDisk, |
| 160 | + containingProjects: new Set(info.containingProjects), |
| 161 | + }), |
| 162 | + ); |
| 163 | + } |
| 164 | + |
| 165 | + function baselineState<T, Data>( |
| 166 | + currentCaches: (T[] | Map<any, T>)[], |
| 167 | + dataMap: Map<T, Data>, |
| 168 | + printWorker: (logs: StateItemLog[], current: T, data: Data | undefined) => Diff, |
| 169 | + toData: (current: T) => Data, |
| 170 | + ) { |
| 171 | + const logs = [] as StateItemLog[]; |
| 172 | + let hasChange = false; |
| 173 | + const currentSet = new Set<T>(); |
| 174 | + currentCaches.forEach(currentCache => currentCache.forEach(printCurrent)); |
| 175 | + if (currentSet.size !== dataMap.size) { |
| 176 | + for (const [key, data] of arrayFrom(dataMap.entries())) { |
| 177 | + if (!currentSet.has(key)) { |
| 178 | + dataMap.delete(key); |
| 179 | + printWorker(logs, key, data); |
| 180 | + } |
| 181 | + } |
| 182 | + hasChange = true; |
| 183 | + } |
| 184 | + return hasChange ? logs : undefined; |
| 185 | + |
| 186 | + function printCurrent(current: T) { |
| 187 | + if (!current) return; |
| 188 | + currentSet.add(current); |
| 189 | + if (printWorker(logs, current, dataMap.get(current))) { |
| 190 | + hasChange = true; |
| 191 | + dataMap.set(current, toData(current)); |
| 192 | + } |
| 193 | + } |
| 194 | + } |
| 195 | + |
| 196 | + function newOrDeleted<T, Data>(current: T, dataMap: Map<T, Data>, data: Data | undefined) { |
| 197 | + return !data ? Diff.New : !dataMap.has(current) ? Diff.Deleted : Diff.None; |
| 198 | + } |
| 199 | + |
| 200 | + enum PrintPropertyWhen { |
| 201 | + Always, |
| 202 | + TruthyOrChangedOrNew, |
| 203 | + Changed, |
| 204 | + } |
| 205 | + function printProperty<Data, Key extends keyof Data & string>( |
| 206 | + printWhen: PrintPropertyWhen, |
| 207 | + data: Data | undefined, |
| 208 | + key: Key, |
| 209 | + value: Data[Key], |
| 210 | + dataDiff: Diff, |
| 211 | + propertyLogs: string[], |
| 212 | + stringValue?: Data[Key] | string, |
| 213 | + ) { |
| 214 | + return printPropertyWorker( |
| 215 | + printWhen, |
| 216 | + `${key}: ${stringValue === undefined ? value : stringValue}`, |
| 217 | + propertyLogs, |
| 218 | + dataDiff, |
| 219 | + !!data && data[key] !== value, |
| 220 | + Diff.Change, |
| 221 | + value, |
| 222 | + ); |
| 223 | + } |
| 224 | + |
| 225 | + function printPropertyWorker( |
| 226 | + printWhen: PrintPropertyWhen, |
| 227 | + stringValue: string, |
| 228 | + propertyLogs: string[], |
| 229 | + dataDiff: Diff, |
| 230 | + propertyChanged: boolean, |
| 231 | + propertyChangeDiff: Diff.Change | Diff.New, |
| 232 | + value: any, |
| 233 | + ) { |
| 234 | + const propertyDiff = propertyChanged ? propertyChangeDiff : Diff.None; |
| 235 | + const result = !dataDiff && propertyDiff ? propertyChangeDiff : dataDiff; |
| 236 | + switch (printWhen) { |
| 237 | + case PrintPropertyWhen.Changed: |
| 238 | + if (!propertyDiff) return result; |
| 239 | + break; |
| 240 | + case PrintPropertyWhen.TruthyOrChangedOrNew: |
| 241 | + if (!value && !propertyDiff) return result; |
| 242 | + break; |
| 243 | + case PrintPropertyWhen.Always: |
| 244 | + break; |
| 245 | + default: |
| 246 | + Debug.assertNever(printWhen); |
| 247 | + } |
| 248 | + propertyLogs.push(` ${stringValue}${propertyDiff}`); |
| 249 | + return result; |
| 250 | + } |
| 251 | +} |
0 commit comments