@tevm/ts-plugin/src/decorators/getScriptKind.ts

A typescript plugin for tevm

import { createHostDecorator } from '../factories/index.js'
import { isRelativeSolidity, isSolidity } from '../utils/index.js'

/**
 * Decorate `LangaugeServerHost.getScriptKind` to return TS type for `.sol` files
 * This lets the ts-server expect `.sol` files to resolve to `.d.ts` files in `resolveModuleNameLiterals`
 */
export const getScriptKindDecorator = createHostDecorator((createInfo, ts, _logger, _config) => {
	return {
		getScriptKind: (fileName) => {
			// TODO we should check if it is using ts baseUrl or paths in future
			if (isRelativeSolidity(fileName)) {
				return ts.ScriptKind.TS
			}
			if (isSolidity(fileName)) {
				return ts.ScriptKind.External
			}
			if (!createInfo.languageServiceHost.getScriptKind) {
				return ts.ScriptKind.Unknown
			}
			return createInfo.languageServiceHost.getScriptKind(fileName)
		},
	}
})