83 lines
2.3 KiB
TypeScript
83 lines
2.3 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { useState, useCallback, useEffect } from 'react';
|
|
import { Settings, LoadedSettings } from '../../config/settings.js';
|
|
import { FolderTrustChoice } from '../components/FolderTrustDialog.js';
|
|
import {
|
|
loadTrustedFolders,
|
|
TrustLevel,
|
|
isWorkspaceTrusted,
|
|
} from '../../config/trustedFolders.js';
|
|
import * as process from 'process';
|
|
|
|
export const useFolderTrust = (
|
|
settings: LoadedSettings,
|
|
onTrustChange: (isTrusted: boolean | undefined) => void,
|
|
) => {
|
|
const [isTrusted, setIsTrusted] = useState<boolean | undefined>(undefined);
|
|
const [isFolderTrustDialogOpen, setIsFolderTrustDialogOpen] = useState(false);
|
|
const [isRestarting, setIsRestarting] = useState(false);
|
|
|
|
const { folderTrust, folderTrustFeature } = settings.merged;
|
|
useEffect(() => {
|
|
const trusted = isWorkspaceTrusted({
|
|
folderTrust,
|
|
folderTrustFeature,
|
|
} as Settings);
|
|
setIsTrusted(trusted);
|
|
setIsFolderTrustDialogOpen(trusted === undefined);
|
|
onTrustChange(trusted);
|
|
}, [onTrustChange, folderTrust, folderTrustFeature]);
|
|
|
|
const handleFolderTrustSelect = useCallback(
|
|
(choice: FolderTrustChoice) => {
|
|
const trustedFolders = loadTrustedFolders();
|
|
const cwd = process.cwd();
|
|
let trustLevel: TrustLevel;
|
|
|
|
const wasTrusted = isTrusted ?? true;
|
|
|
|
switch (choice) {
|
|
case FolderTrustChoice.TRUST_FOLDER:
|
|
trustLevel = TrustLevel.TRUST_FOLDER;
|
|
break;
|
|
case FolderTrustChoice.TRUST_PARENT:
|
|
trustLevel = TrustLevel.TRUST_PARENT;
|
|
break;
|
|
case FolderTrustChoice.DO_NOT_TRUST:
|
|
trustLevel = TrustLevel.DO_NOT_TRUST;
|
|
break;
|
|
default:
|
|
return;
|
|
}
|
|
|
|
trustedFolders.setValue(cwd, trustLevel);
|
|
const newIsTrusted =
|
|
trustLevel === TrustLevel.TRUST_FOLDER ||
|
|
trustLevel === TrustLevel.TRUST_PARENT;
|
|
setIsTrusted(newIsTrusted);
|
|
onTrustChange(newIsTrusted);
|
|
|
|
const needsRestart = wasTrusted !== newIsTrusted;
|
|
if (needsRestart) {
|
|
setIsRestarting(true);
|
|
setIsFolderTrustDialogOpen(true);
|
|
} else {
|
|
setIsFolderTrustDialogOpen(false);
|
|
}
|
|
},
|
|
[onTrustChange, isTrusted],
|
|
);
|
|
|
|
return {
|
|
isTrusted,
|
|
isFolderTrustDialogOpen,
|
|
handleFolderTrustSelect,
|
|
isRestarting,
|
|
};
|
|
};
|