Function initializeCommonNodeApi

src/test/common.node.ts:205–269

()

let remoteUrisCleared = false;
export function initializeCommonNodeApi() {
    const { commands, Uri } = vscode;
    const { initialize } = initializeModule;

    initializeCommonApi({
        async createTemporaryFile(options: {
            contents?: string;
            extension: string;
        }): Promise<{ file: Uri } & IDisposable> {
            const extension = options.extension || '.py';
            const tempFile = tmp.tmpNameSync({ postfix: extension });
            if (options.contents) {
                await fs.writeFile(tempFile, options.contents);
            }
            return { file: Uri.file(tempFile), dispose: () => swallowExceptions(() => fs.unlinkSync(tempFile)) };
        },
        async startJupyterServer(
            options: {
                token?: string;
                port?: number;
                useCert?: boolean;
                jupyterLab?: boolean;
                password?: string;
                detached?: boolean;
                standalone?: boolean;
            } = {}
        ): Promise<{ url: string } & IDisposable> {
            if (IS_REMOTE_NATIVE_TEST()) {
                if (options.standalone) {
                    return JupyterServer.instance.startJupyter(options);
                }
                if (!remoteUrisCleared) {
                    await commands.executeCommand('deepnote.clearSavedJupyterUris');
                    remoteUrisCleared = true;
                }
                const url = options.useCert
                    ? await JupyterServer.instance.startJupyterWithCert()
                    : await JupyterServer.instance.startJupyterWithToken();
                console.info(`Jupyter started and listening at ${url}`);
                try {
                    await commands.executeCommand('deepnote.selectjupyteruri', Uri.parse(url));
                } catch (ex) {
                    console.error('Failed to select jupyter server, retry in 1s', ex);
                }
                // Todo: Fix in debt week, we need to retry, some changes have caused the first connection attempt to fail on CI.
                // Possible we're trying to connect before the server is ready.
                await sleep(5_000);
                await commands.executeCommand('deepnote.selectjupyteruri', Uri.parse(url));
                return { url, dispose: noop };
            } else {
                return { url: '', dispose: noop };
            }
        },
        async stopJupyterServer() {
            if (IS_REMOTE_NATIVE_TEST()) {
                return;
            }
            await JupyterServer.instance.dispose().catch(noop);