Merge pull request #3 from CatalystCode/feat_file_association

feat(lib/file_association) - file associations!

Author: Steven Edouard

lib/file_association.js

@@ -0,0 +1,33 @@
+'use strict';
+var registry = require('./registry'),
+    windef = require('./windef');
+module.exports = {
+    associateExeForFile: function (handlerName, handlerDescription, iconPath, exePath, extensionName) {
+        var key = registry.openKeyFromPredefined(windef.HKEY.HKEY_CLASSES_ROOT, '', windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        registry.createKey(key, extensionName, windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        var appKey = registry.openKeyFromKeyObject(key, extensionName, windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        registry.setValueForKeyObject(appKey, '', windef.REG_VALUE_TYPE.REG_SZ, handlerName);
+        appKey.close();
+
+        registry.createKey(key, handlerName, windef.KEY_ACCESS.KEY_ALL_ACCESS);
+        var handlerKey = registry.openKeyFromKeyObject(key, handlerName, windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        registry.setValueForKeyObject(handlerKey, '', windef.REG_VALUE_TYPE.REG_SZ, handlerDescription);
+        registry.createKey(handlerKey, 'DefaultIcon', windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        var defaultIconKey = registry.openKeyFromKeyObject(handlerKey, 'DefaultIcon', windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        registry.setValueForKeyObject(defaultIconKey, '', windef.REG_VALUE_TYPE.REG_SZ, iconPath);
+
+        registry.createKey(handlerKey, 'shell\\Open\\Command', windef.KEY_ACCESS.KEY_ALL_ACCESS);
+
+        var commandKey = registry.openKeyFromKeyObject(handlerKey, 'shell\\Open\\Command', windef.KEY_ACCESS.KEY_ALL_ACCESS);
+        registry.setValueForKeyObject(commandKey, '', windef.REG_VALUE_TYPE.REG_SZ, exePath);
+
+        commandKey.close();
+        handlerKey.close();
+    }
+};

lib/key.js

@@ -1,10 +1,13 @@
 'use strict';
+
 class Key {
     constructor (pHKey, path) {
         this.handle = pHKey;
         this.path = path;
     }
     close () {
+        var registry = require('./registry');
+        registry.closeKey(this);
     }
 
     toString () {

lib/registry.js

@@ -32,7 +32,13 @@ var advApi = ffi.Library('Advapi32', {
       _In_opt_ LPCTSTR lpSubKey
       );
     */
-    RegDeleteTreeA: ['uint64', [types.HKEY, 'string']]
+    RegDeleteTreeA: ['uint64', [types.HKEY, 'string']],
+    /*
+    LONG WINAPI RegCloseKey(
+    _In_ HKEY hKey
+    );
+    */
+    RegCloseKey: ['uint64', [types.HKEY]]
 });
 
 var api = {
@@ -148,6 +154,14 @@ var api = {
     deleteKey: function (key, subKeyName) {
         var result = advApi.RegDeleteTreeA(key.handle.deref(), subKeyName);
 
+        if (result !== 0) {
+            throw 'Failed to open key error ' + result + ': + error[result]';
+        }
+    },
+    closeKey: function (key) {
+        console.log('KEY:' + key);
+        var result = advApi.RegCloseKey(key.handle.deref());
+
         if (result !== 0) {
             throw 'Failed to open key error ' + result + ': + error[result]';
         }

test/file_association.js

@@ -0,0 +1,11 @@
+/* global describe, it */
+'use strict';
+var fileAssociation = require('../lib/file_association'),
+    assert = require('assert');
+
+describe('File Association Test', () => {
+    it('Should create a file association for *.zzz files', () => {
+        fileAssociation.associateExeForFile('myTestHandler', 'A test handler for unit tests', '', 'C:\\Program Files\\nodejs\\node.exe %1', '.zzz');
+        assert(true);
+    });
+});