Save permissions grant / deny per domain

resources/js/index.js

@@ -384,24 +384,56 @@ function loadService(serviceId, service) {
             setContextMenu(webContents);
 
             // Set permission request handler
-            session.fromPartition(service.view.partition)
+            function getUrlDomain(url) {
-                .setPermissionRequestHandler(((webContents, permission, callback, details) => {
+                let domain = url.match(/^https?:\/\/((.+?)\/|(.+))/i)[1];
-                    dialog.showMessageBox(remote.getCurrentWindow(), {
+                if (domain.endsWith('/')) domain = domain.substr(0, domain.length - 1);
-                        type: 'question',
+                return domain;
-                        title: 'Grant ' + permission + ' permission',
+            }
-                        message: 'Do you wish to grant the ' + permission + ' permission to ' + details.requestingUrl + '?',
+
-                        buttons: ['Deny', 'Authorize'],
+            function getDomainPermissions(domain) {
-                        cancelId: 0,
+                let domainPermissions = service.permissions[domain];
-                    }).then(result => {
+                if (!domainPermissions) domainPermissions = service.permissions[domain] = [];
-                        if (result.response === 1) {
+                return domainPermissions;
-                            console.log('Granted', permission, 'for service', details.requestingUrl);
+            }
-                            callback(true);
+
-                        } else {
+            let serviceSession = session.fromPartition(service.view.partition);
-                            console.log('Denied', permission, 'for service', details.requestingUrl);
+            serviceSession.setPermissionRequestHandler(((webContents, permissionName, callback, details) => {
-                            callback(false);
+                let domain = getUrlDomain(details.requestingUrl);
-                        }
+                let domainPermissions = getDomainPermissions(domain);
-                    }).catch(console.error);
+
-                }));
+                let existingPermissions = domainPermissions.filter(p => p.name === permissionName);
+                if (existingPermissions.length > 0) {
+                    callback(existingPermissions[0].authorized);
+                    return;
+                }
+
+                dialog.showMessageBox(remote.getCurrentWindow(), {
+                    type: 'question',
+                    title: 'Grant ' + permissionName + ' permission',
+                    message: 'Do you wish to grant the ' + permissionName + ' permission to ' + domain + '?',
+                    buttons: ['Deny', 'Authorize'],
+                    cancelId: 0,
+                }).then(result => {
+                    const authorized = result.response === 1;
+
+                    domainPermissions.push({
+                        name: permissionName,
+                        authorized: authorized,
+                    });
+                    updateServicePermissions(serviceId);
+
+                    console.log(authorized ? 'Granted' : 'Denied', permissionName, 'for domain', domain);
+                    callback(authorized);
+                }).catch(console.error);
+            }));
+            serviceSession.setPermissionCheckHandler((webContents1, permissionName, requestingOrigin, details) => {
+                console.log('Permission check', permissionName, requestingOrigin, details);
+                let domain = getUrlDomain(details.requestingUrl);
+                let domainPermissions = getDomainPermissions(domain);
+
+                let existingPermissions = domainPermissions.filter(p => p.name === permissionName);
+                return existingPermissions.length > 0 && existingPermissions[0].authorized;
+            });
 
             service.view.setAttribute('src', service.url);
         });
@@ -469,6 +501,11 @@ function reloadService(serviceId) {
     }
 }
 
+function updateServicePermissions(serviceId) {
+    const service = services[serviceId];
+    ipcRenderer.send('updateServicePermissions', serviceId, service.permissions);
+}
+
 function updateNavigation() {
     console.debug('Updating navigation');
     // Update active list element

src/Service.js

@@ -36,6 +36,7 @@ Service.requiredProperties = {
     'autoLoad': false,
     'customCSS': null,
     'customUserAgent': null,
+    'permissions': {},
 };
 
 export default Service;

src/main.js

@@ -183,6 +183,11 @@ function createWindow() {
         config.save();
     });
 
+    ipcMain.on('updateServicePermissions', (e, serviceId, permissions) => {
+        config.services[serviceId].permissions = permissions;
+        config.save();
+    });
+
     ipcMain.on('updateWindowTitle', (event, serviceId, viewTitle) => {
         if (serviceId === null) {
             window.setTitle(Meta.title);