CodeNomad/packages/electron-app/electron/main/permissions.ts

import { session, systemPreferences } from "electron"
import { isAllowedRendererOrigin } from "./renderer-origin"

export { isAllowedRendererOrigin } from "./renderer-origin"

const isMac = process.platform === "darwin"

export function configureMediaPermissionHandlers(getAllowedOrigins: () => string[]) {
  const isAudioMediaRequest = (permission: string, details?: unknown) => {
    if (permission !== "media") {
      return false
    }

    const mediaTypes = (details as { mediaTypes?: string[] } | undefined)?.mediaTypes ?? []
    return mediaTypes.length === 0 || mediaTypes.includes("audio")
  }

  session.defaultSession.setPermissionCheckHandler((_webContents, permission, requestingOrigin, details) => {
    if (!isAudioMediaRequest(permission, details)) {
      return false
    }

    return isAllowedRendererOrigin(requestingOrigin, getAllowedOrigins())
  })

  session.defaultSession.setPermissionRequestHandler((webContents, permission, callback, details) => {
    if (!isAudioMediaRequest(permission, details)) {
      callback(false)
      return
    }

    const requestingOrigin = (details as { requestingOrigin?: string } | undefined)?.requestingOrigin || webContents.getURL()
    callback(isAllowedRendererOrigin(requestingOrigin, getAllowedOrigins()))
  })
}

export async function requestMicrophoneAccess(): Promise<boolean> {
  if (!isMac) {
    return true
  }

  const status = systemPreferences.getMediaAccessStatus("microphone")
  if (status === "granted") {
    return true
  }

  return systemPreferences.askForMediaAccess("microphone")
}