kalugin66/webtorrent-desktop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

set windows protocol/file icons

Browse Source
This commit is contained in:
Feross Aboukhadijeh
2016-03-20 00:52:54 -07:00
parent 45b36fb7a1
commit fb37b87297
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
main/register-protocol-handler.js
View File

@@ -1,7 +1,9 @@
var config = require('../config')
module.exports = function () { module.exports = function () {
if (process.platform === 'win32') { if (process.platform === 'win32') {
registerProtocolHandler('magnet', 'BitTorrent Magnet URL', process.execPath) registerProtocolHandler('magnet', 'BitTorrent Magnet URL', config.APP_FILE_ICON + '.ico', process.execPath)
registerFileHandler('.torrent', 'io.webtorrent.torrent', 'BitTorrent Document', process.execPath) registerFileHandler('.torrent', 'io.webtorrent.torrent', 'BitTorrent Document', config.APP_FILE_ICON + '.ico', process.execPath)
} }
} }
@@ -25,7 +27,7 @@ module.exports = function () {
* "HKEY_CLASSES_ROOT" anyway, and can be written by unprivileged users. * "HKEY_CLASSES_ROOT" anyway, and can be written by unprivileged users.
*/ */
function registerProtocolHandler (protocol, name, command) { function registerProtocolHandler (protocol, name, icon, command) {
var Registry = require('winreg') var Registry = require('winreg')
var protocolKey = new Registry({ var protocolKey = new Registry({
@@ -35,6 +37,12 @@ function registerProtocolHandler (protocol, name, command) {
protocolKey.set('', Registry.REG_SZ, 'URL:' + name, callback) protocolKey.set('', Registry.REG_SZ, 'URL:' + name, callback)
protocolKey.set('URL Protocol', Registry.REG_SZ, '', callback) protocolKey.set('URL Protocol', Registry.REG_SZ, '', callback)
var iconKey = new Registry({
hive: Registry.HKCU,
key: '\\Software\\Classes\\' + protocol + '\\DefaultIcon'
})
iconKey.set('', Registry.REG_SZ, icon, callback)
var commandKey = new Registry({ var commandKey = new Registry({
hive: Registry.HKCU, hive: Registry.HKCU,
key: '\\Software\\Classes\\' + protocol + '\\shell\\open\\command' key: '\\Software\\Classes\\' + protocol + '\\shell\\open\\command'
@@ -46,7 +54,7 @@ function registerProtocolHandler (protocol, name, command) {
} }
} }
function registerFileHandler (ext, id, name, command) { function registerFileHandler (ext, id, name, icon, command) {
var Registry = require('winreg') var Registry = require('winreg')
var extKey = new Registry({ var extKey = new Registry({
@@ -61,6 +69,12 @@ function registerFileHandler (ext, id, name, command) {
}) })
idKey.set('', Registry.REG_SZ, name, callback) idKey.set('', Registry.REG_SZ, name, callback)
var iconKey = new Registry({
hive: Registry.HKCU,
key: '\\Software\\Classes\\' + id + '\\DefaultIcon'
})
iconKey.set('', Registry.REG_SZ, icon, callback)
var commandKey = new Registry({ var commandKey = new Registry({
hive: Registry.HKCU, hive: Registry.HKCU,
key: '\\Software\\Classes\\' + id + '\\shell\\open\\command' key: '\\Software\\Classes\\' + id + '\\shell\\open\\command'