context-bridge.ts 1.4 KB

123456789101112131415161718192021222324252627282930313233343536
  1. const binding = process._linkedBinding('electron_renderer_context_bridge');
  2. const checkContextIsolationEnabled = () => {
  3. if (!process.contextIsolated) throw new Error('contextBridge API can only be used when contextIsolation is enabled');
  4. };
  5. const contextBridge: Electron.ContextBridge = {
  6. exposeInMainWorld: (key: string, api: any) => {
  7. checkContextIsolationEnabled();
  8. return binding.exposeAPIInWorld(0, key, api);
  9. },
  10. exposeInIsolatedWorld: (worldId: number, key: string, api: any) => {
  11. checkContextIsolationEnabled();
  12. return binding.exposeAPIInWorld(worldId, key, api);
  13. }
  14. };
  15. export default contextBridge;
  16. export const internalContextBridge = {
  17. contextIsolationEnabled: process.contextIsolated,
  18. overrideGlobalValueFromIsolatedWorld: (keys: string[], value: any) => {
  19. return binding._overrideGlobalValueFromIsolatedWorld(keys, value, false);
  20. },
  21. overrideGlobalValueWithDynamicPropsFromIsolatedWorld: (keys: string[], value: any) => {
  22. return binding._overrideGlobalValueFromIsolatedWorld(keys, value, true);
  23. },
  24. overrideGlobalPropertyFromIsolatedWorld: (keys: string[], getter: Function, setter?: Function) => {
  25. return binding._overrideGlobalPropertyFromIsolatedWorld(keys, getter, setter || null);
  26. },
  27. isInMainWorld: () => binding._isCalledFromMainWorld() as boolean
  28. };
  29. if (binding._isDebug) {
  30. contextBridge.internalContextBridge = internalContextBridge;
  31. }