// Copyright (c) Microsoft Corporation. All rights reserved.

// Licensed under the MIT License.

import * as fs from 'fs-extra';

import { applyEdits, ModificationOptions, modify } from 'jsonc-parser';

import * as path from 'path';

import { IS_CI_SERVER } from '../ciConstants';

import { EXTENSION_ROOT_DIR_FOR_TESTS } from '../constants';

const settingsFile = path.join(EXTENSION_ROOT_DIR_FOR_TESTS, 'src/test/datascience/.vscode/settings.json');

function updateTestsForNativeNotebooks() {

/**

* Modify package.json to ensure VSC Notebooks have been setup so tests can run.

* This is required because we modify package.json during runtime, hence we need to do the same thing for tests.

*/

const packageJsonFile = path.join(EXTENSION_ROOT_DIR_FOR_TESTS, 'package.json');

const content = JSON.parse(fs.readFileSync(packageJsonFile).toString());

// This code is temporary.

if (

!content.enableProposedApi ||

!Array.isArray(content.contributes.notebookOutputRenderer) ||

!Array.isArray(content.contributes.notebookProvider)

) {

content.enableProposedApi = true;

content.contributes.notebookProvider = [

{

viewType: 'jupyter-notebook',

displayName: 'Jupyter Notebook',

selector: [

{

filenamePattern: '*.ipynb'

}

]

}

];

}

// Update package.json to pick experiments from our custom settings.json file.

content.contributes.configuration.properties['python.experiments.optInto'].scope = 'resource';

content.contributes.configuration.properties['python.logging.level'].scope = 'resource';

fs.writeFileSync(packageJsonFile, JSON.stringify(content, undefined, 4));

updateSettings(true);

}

function updateSettings(useNativeNotebooks: boolean) {

const modificationOptions: ModificationOptions = {

formattingOptions: {

tabSize: 4,

insertSpaces: true

}

};

let settingsJson = fs.readFileSync(settingsFile).toString();

const experiments = useNativeNotebooks ? ['NativeNotebook - experiment'] : [];

const autoSave = useNativeNotebooks ? 'off' : 'afterDelay';

settingsJson = applyEdits(

settingsJson,

modify(settingsJson, ['python.experiments.optInto'], experiments, modificationOptions)

);

settingsJson = applyEdits(settingsJson, modify(settingsJson, ['files.autoSave'], autoSave, modificationOptions));

fs.writeFileSync(settingsFile, settingsJson);

}

function updateTestsForOldNotebooks() {

updateSettings(false);

}

if (!IS_CI_SERVER) {

// Noop.

} else if (

process.env.VSC_PYTHON_CI_TEST_VSC_CHANNEL === 'insiders' &&

process.env.TEST_FILES_SUFFIX === 'native.vscode.test'

) {

updateTestsForNativeNotebooks();

} else {

updateTestsForOldNotebooks();

}