Last active
March 12, 2021 02:08
-
-
Save cometkim/3b1f426159920c1ce3c448bb93251a34 to your computer and use it in GitHub Desktop.
coc.nvim + tsserver + pnpify (lbrayner/vim-rzip plugin required)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env node | |
const {existsSync} = require(`fs`); | |
const {createRequire, createRequireFromPath} = require(`module`); | |
const {resolve} = require(`path`); | |
const relPnpApiPath = "../../../../.pnp.js"; | |
const absPnpApiPath = resolve(__dirname, relPnpApiPath); | |
const absRequire = (createRequire || createRequireFromPath)(absPnpApiPath); | |
const moduleWrapper = tsserver => { | |
const {isAbsolute} = require(`path`); | |
const pnpApi = require(`pnpapi`); | |
const dependencyTreeRoots = new Set(pnpApi.getDependencyTreeRoots().map(locator => { | |
return `${locator.name}@${locator.reference}`; | |
})); | |
// VSCode sends the zip paths to TS using the "zip://" prefix, that TS | |
// doesn't understand. This layer makes sure to remove the protocol | |
// before forwarding it to TS, and to add it back on all returned paths. | |
function toEditorPath(str) { | |
// We add the `zip:` prefix to both `.zip/` paths and virtual paths | |
if (isAbsolute(str) && !str.match(/^\^zip:/) && (str.match(/\.zip\//) || str.match(/\$\$virtual\//))) { | |
// We also take the opportunity to turn virtual paths into physical ones; | |
// this makes is much easier to work with workspaces that list peer | |
// dependencies, since otherwise Ctrl+Click would bring us to the virtual | |
// file instances instead of the real ones. | |
// | |
// We only do this to modules owned by the the dependency tree roots. | |
// This avoids breaking the resolution when jumping inside a vendor | |
// with peer dep (otherwise jumping into react-dom would show resolution | |
// errors on react). | |
// | |
const resolved = pnpApi.resolveVirtual(str); | |
if (resolved) { | |
const locator = pnpApi.findPackageLocator(resolved); | |
if (locator && dependencyTreeRoots.has(`${locator.name}@${locator.reference}`)) { | |
str = resolved; | |
} | |
} | |
str = str.replace(/\\/g, `/`) | |
str = str.replace(/^\/?/, `/`); | |
// Absolute VSCode `Uri.fsPath`s need to start with a slash. | |
// VSCode only adds it automatically for supported schemes, | |
// so we have to do it manually for the `zip` scheme. | |
// The path needs to start with a caret otherwise VSCode doesn't handle the protocol | |
// | |
// Ref: https://github.com/microsoft/vscode/issues/105014#issuecomment-686760910 | |
// | |
if (str.match(/\.zip\//)) { | |
if (hostInfo === `vscode`) { | |
str = `^zip:${str}`; | |
} else (hostInfo === `coc-nvim`) { | |
// to make "go to definition" work | |
// replace the virtual path to actual file system path | |
// that can be opened using [vim-rzip](https://github.com/lbrayner/vim-rzip) | |
str = str.replace(/\$\$virtual\/([\-\w\/]+)\/cache/, 'cache'); | |
str = str.replace(/\.zip\//, '.zip::'); | |
// resolve(...) resolves the prepending `/` issue, but don't know why. | |
str = resolve(`zipfile:${str}`); | |
} else { | |
str = `zip:${str}`; | |
} | |
} | |
} | |
return str; | |
} | |
function fromEditorPath(str) { | |
return process.platform === `win32` | |
? str.replace(/^\^?zip:\//, ``) | |
: str.replace(/^\^?zip:/, ``); | |
} | |
// And here is the point where we hijack the VSCode <-> TS communications | |
// by adding ourselves in the middle. We locate everything that looks | |
// like an absolute path of ours and normalize it. | |
const Session = tsserver.server.Session; | |
const {onMessage: originalOnMessage, send: originalSend} = Session.prototype; | |
let hostInfo = `unknown`; | |
return Object.assign(Session.prototype, { | |
onMessage(/** @type {string} */ message) { | |
const parsedMessage = JSON.parse(message) | |
if ( | |
parsedMessage != null && | |
typeof parsedMessage === `object` && | |
parsedMessage.arguments && | |
typeof parsedMessage.arguments.hostInfo === `string` | |
) { | |
hostInfo = parsedMessage.arguments.hostInfo; | |
} | |
return originalOnMessage.call(this, JSON.stringify(parsedMessage, (key, value) => { | |
return typeof value === `string` ? fromEditorPath(value) : value; | |
})); | |
}, | |
send(/** @type {any} */ msg) { | |
return originalSend.call(this, JSON.parse(JSON.stringify(msg, (key, value) => { | |
return typeof value === `string` ? toEditorPath(value) : value; | |
}))); | |
} | |
}); | |
}; | |
if (existsSync(absPnpApiPath)) { | |
if (!process.versions.pnp) { | |
// Setup the environment to be able to require typescript/lib/tsserver.js | |
require(absPnpApiPath).setup(); | |
} | |
} | |
// Defer to the real typescript/lib/tsserver.js your application uses | |
module.exports = moduleWrapper(absRequire(`typescript/lib/tsserver.js`)); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment