feat: Add additional language for handling to CodeCompletion settings

Author: Palm1r

CodeHandler.cpp

@@ -21,6 +21,7 @@
 #include "CodeHandler.hpp"
 #include <QFileInfo>
 #include <QHash>
+#include <settings/CodeCompletionSettings.hpp>
 
 namespace QodeAssist {
 
@@ -32,6 +33,45 @@ struct LanguageProperties
     QVector<QString> fileExtensions;
 };
 
+const QVector<LanguageProperties> customLanguagesFromSettings()
+{
+    QVector<LanguageProperties> customLanguages;
+
+    const QStringList customLanguagesList = Settings::codeCompletionSettings().customLanguages();
+    for (const QString &entry : customLanguagesList) {
+        if (entry.trimmed().isEmpty()) {
+            continue;
+        }
+
+        QStringList parts = entry.split(',');
+        if (parts.size() < 4) {
+            continue;
+        }
+
+        QString name = parts[0].trimmed();
+        QString commentStyle = parts[1].trimmed();
+        QStringList modelNamesList = parts[2].trimmed().split(' ', Qt::SkipEmptyParts);
+        QStringList extensionsList = parts[3].trimmed().split(' ', Qt::SkipEmptyParts);
+
+        if (!name.isEmpty() && !commentStyle.isEmpty() && !modelNamesList.isEmpty()
+            && !extensionsList.isEmpty()) {
+            QVector<QString> modelNames;
+            for (const auto &modelName : modelNamesList) {
+                modelNames.append(modelName);
+            }
+
+            QVector<QString> extensions;
+            for (const auto &ext : extensionsList) {
+                extensions.append(ext);
+            }
+
+            customLanguages.append({name, commentStyle, modelNames, extensions});
+        }
+    }
+
+    return customLanguages;
+}
+
 const QVector<LanguageProperties> &getKnownLanguages()
 {
     static QVector<LanguageProperties> knownLanguages = {
@@ -55,6 +95,8 @@ const QVector<LanguageProperties> &getKnownLanguages()
         {"qml", "//", {"qml"}, {"qml"}},
     };
 
+    knownLanguages.append(customLanguagesFromSettings());
+
     return knownLanguages;
 }
 

settings/CodeCompletionSettings.cpp

@@ -192,6 +192,15 @@ CodeCompletionSettings::CodeCompletionSettings()
         "Here is the code context with insertion points:\n"
         "<code_context>\n${prefix}<cursor>${suffix}\n</code_context>\n\n");
 
+    customLanguages.setSettingsKey(Constants::CC_CUSTOM_LANGUAGES);
+    customLanguages.setLabelText(Tr::tr("Additional Programming Languages for handling:"));
+    customLanguages.setToolTip(Tr::tr("Specify additional programming languages in format: "
+                                      "name,comment_style,model_names,extensions\n"
+                                      "Example: rust,//,rust rs,rs"
+                                      "Fields: language name, comment prefix, names from LLM "
+                                      "(space-separated), file extensions (space-separated)"));
+    customLanguages.setDefaultValue({{"cmake,#,cmake,CMakeLists.txt"}, {"qmake,#,qmake,pro pri"}});
+
     useProjectChangesCache.setSettingsKey(Constants::CC_USE_PROJECT_CHANGES_CACHE);
     useProjectChangesCache.setDefaultValue(true);
     useProjectChangesCache.setLabelText(Tr::tr("Max Changes Cache Size:"));
@@ -249,18 +258,18 @@ CodeCompletionSettings::CodeCompletionSettings()
         contextGrid.addRow({Row{readFullFile}});
         contextGrid.addRow({Row{readFileParts, readStringsBeforeCursor, readStringsAfterCursor}});
 
-        auto contextItem = Column{
-            Row{contextGrid, Stretch{1}},
-            Row{useSystemPrompt, Stretch{1}},
-            Group{title(Tr::tr("Prompts for FIM models")), Column{systemPrompt}},
-            Group{
-                title(Tr::tr("Prompts for Non FIM models")),
-                Column{
-                    Row{useUserMessageTemplateForCC, Stretch{1}},
-                    systemPromptForNonFimModels,
-                    userMessageTemplateForCC,
-                }},
-            Row{useProjectChangesCache, maxChangesCacheSize, Stretch{1}}};
+        auto contextItem = Column{Row{contextGrid, Stretch{1}},
+                                  Row{useSystemPrompt, Stretch{1}},
+                                  Group{title(Tr::tr("Prompts for FIM models")),
+                                        Column{systemPrompt}},
+                                  Group{title(Tr::tr("Prompts for Non FIM models")),
+                                        Column{
+                                            Row{useUserMessageTemplateForCC, Stretch{1}},
+                                            systemPromptForNonFimModels,
+                                            userMessageTemplateForCC,
+                                            customLanguages,
+                                        }},
+                                  Row{useProjectChangesCache, maxChangesCacheSize, Stretch{1}}};
 
         return Column{
             Row{Stretch{1}, resetToDefaults},
@@ -355,6 +364,7 @@ void CodeCompletionSettings::resetSettingsToDefaults()
         resetAspect(useUserMessageTemplateForCC);
         resetAspect(userMessageTemplateForCC);
         resetAspect(systemPromptForNonFimModels);
+        resetAspect(customLanguages);
     }
 }
 

settings/CodeCompletionSettings.hpp

@@ -42,6 +42,8 @@ class CodeCompletionSettings : public Utils::AspectContainer
     Utils::IntegerAspect autoCompletionCharThreshold{this};
     Utils::IntegerAspect autoCompletionTypingInterval{this};
 
+    Utils::StringListAspect customLanguages{this};
+
     // General Parameters Settings
     Utils::DoubleAspect temperature{this};
     Utils::IntegerAspect maxTokens{this};

settings/SettingsConstants.hpp

@@ -72,6 +72,7 @@ const char CA_TOKENS_THRESHOLD[] = "QodeAssist.caTokensThreshold";
 const char CA_LINK_OPEN_FILES[] = "QodeAssist.caLinkOpenFiles";
 const char CA_STREAM[] = "QodeAssist.caStream";
 const char CA_AUTOSAVE[] = "QodeAssist.caAutosave";
+const char CC_CUSTOM_LANGUAGES[] = "QodeAssist.ccCustomLanguages";
 
 const char QODE_ASSIST_GENERAL_OPTIONS_ID[] = "QodeAssist.GeneralOptions";
 const char QODE_ASSIST_GENERAL_SETTINGS_PAGE_ID[] = "QodeAssist.1GeneralSettingsPageId";