make custom extension mapping json option work

Author: p0358

CodeStats/CodeStatsPackage.cs

@@ -108,15 +108,14 @@ private static void InitializeAsync()
             {
                 currentPulse = new Pulse();
 
-                Logger.Info("Initialing settings form...");
-
                 // Load config file
                 _CodeStatsConfigFile = new ConfigFile();
                 GetSettings(true);
 
                 Logger.Debug("Loaded config");
 
                 LoadExtensionMapping();
+                LoadCustomExtensionMapping();
 
                 // Check for updates
                 Task.Run(() =>
@@ -242,6 +241,37 @@ private static void LoadExtensionMapping()
             }
         }
 
+        private static void LoadCustomExtensionMapping()
+        {
+            try
+            {
+                // get path of plugin configuration
+                StringBuilder sbConfigFilePath = new StringBuilder(Win32.MAX_PATH);
+                Win32.SendMessage(PluginBase.nppData._nppHandle, (uint)NppMsg.NPPM_GETPLUGINSCONFIGDIR, Win32.MAX_PATH, sbConfigFilePath);
+                string configFilePath = sbConfigFilePath.ToString();
+
+                string customExtensionMappingFilePath = ConfigFile.GetCustomExtensionMappingFilePath();
+
+                StreamReader _textStreamReader;
+                string customExtensionMappingJson;
+                var serializer = new JavaScriptSerializer();
+
+                _textStreamReader = File.OpenText(customExtensionMappingFilePath);
+                customExtensionMappingJson = _textStreamReader.ReadToEnd();
+                customExtensionMapping = serializer.Deserialize<Dictionary<string, string>>(customExtensionMappingJson);
+
+                Logger.Debug("Loaded custom extension mapping");
+            }
+            catch (FileNotFoundException)
+            {
+                Logger.Debug("Custom extension mapping JSON file does not exist");
+            }
+            catch (Exception ex)
+            {
+                Logger.Error("Error loading custom extension mappings!", ex);
+            }
+        }
+
         internal static void SetToolBarIcon()
         {
             // TODO: Maybe check out https://docs.microsoft.com/en-us/windows/desktop/controls/embed-nonbutton-controls-in-toolbars to create counter within toolbar as alternative?
@@ -601,6 +631,22 @@ public static string GetCurrentLanguage() // Code::Stats
             {
                 switch (detectionMethod)
                 {
+                    case Constants.DetectionType.CUSTOM_MAPPING:
+                        if (!String.IsNullOrWhiteSpace(extension))
+                        {
+                            try
+                            {
+                                language = customExtensionMapping[extension];
+                                found = true;
+                            }
+                            catch (KeyNotFoundException)
+                            {
+                                Logger.Debug("Extension \"" + extension + "\" is not found in custom extension mappings file.");
+                                found = false;
+                            }
+                        }
+                        break;
+                    
                     case Constants.DetectionType.EXTENSION_MAPPING:
                         if (!String.IsNullOrWhiteSpace(extension))
                         {
@@ -754,6 +800,7 @@ private static void SettingsPopup()
         private static void SettingsFormOnConfigSaved(object sender, EventArgs eventArgs)
         {
             GetSettings();
+            LoadCustomExtensionMapping();
             _settingsForm.OnConfigSaved -= SettingsFormOnConfigSaved;
             _settingsForm.Dispose();
             _settingsForm = null;

CodeStats/ConfigFile.cs

@@ -172,5 +172,14 @@ static string GetConfigFilePath()
 
             return Path.Combine(iniFilePath, "CodeStats.ini");
         }
+        
+        public static string GetCustomExtensionMappingFilePath()
+        {
+            StringBuilder sbConfigFilePath = new StringBuilder(Win32.MAX_PATH);
+            Win32.SendMessage(PluginBase.nppData._nppHandle, (uint)NppMsg.NPPM_GETPLUGINSCONFIGDIR, Win32.MAX_PATH, sbConfigFilePath);
+            string configFilePath = sbConfigFilePath.ToString();
+
+            return Path.Combine(configFilePath, "CodeStatsCustomExtensionMapping.json");
+        }
     }
 }

CodeStats/Forms/SettingsForm.Designer.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Windows.Forms;
 using static CodeStats.Constants;
 
@@ -37,7 +38,7 @@ private void SettingsForm_Load(object sender, EventArgs e)
                 }
 
                 chkUseExtensionMapping.Checked = _CodeStatsConfigFile.UseExtensionMapping;
-                chkUseExtensionMapping.Checked = _CodeStatsConfigFile.UseLexerLanguage;
+                chkUseLexerLanguage.Checked = _CodeStatsConfigFile.UseLexerLanguage;
                 radioDetectionPriority_extensionMapping.Checked = _CodeStatsConfigFile.DetectionPriority == Constants.DetectionType.EXTENSION_MAPPING;
                 radioDetectionPriority_lexerLanguage.Checked = _CodeStatsConfigFile.DetectionPriority == Constants.DetectionType.LEXER_LANGUAGE;
                 chkUseCustomMapping.Checked = _CodeStatsConfigFile.UseCustomMapping;
@@ -102,7 +103,7 @@ private void btnOk_Click(object sender, EventArgs e)
 
                     LanguageDetectionUIRefresh();
                     _CodeStatsConfigFile.UseExtensionMapping = chkUseExtensionMapping.Checked;
-                    _CodeStatsConfigFile.UseLexerLanguage = chkUseExtensionMapping.Checked;
+                    _CodeStatsConfigFile.UseLexerLanguage = chkUseLexerLanguage.Checked;
                     _CodeStatsConfigFile.DetectionPriority = !radioDetectionPriority_extensionMapping.Checked ? Constants.DetectionType.LEXER_LANGUAGE : Constants.DetectionType.EXTENSION_MAPPING;
                     _CodeStatsConfigFile.UseCustomMapping = chkUseCustomMapping.Checked;
                     _CodeStatsConfigFile.RefreshDetectionOrder();
@@ -219,5 +220,11 @@ private void LanguageDetectionUIRefresh(object sender, EventArgs e)
         {
             LanguageDetectionUIRefresh();
         }
+
+        private void btnOpenCustomMappingFile_Click(object sender, EventArgs e)
+        {
+            string path = ConfigFile.GetCustomExtensionMappingFilePath();
+            Process.Start("notepad.exe", path);
+        }
     }
 }