Initial commit

qodeassist.cpp

@@ -0,0 +1,138 @@
+/* 
+ * Copyright (C) 2024 Petr Mironychev
+ *
+ * This file is part of QodeAssist.
+ *
+ * QodeAssist is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * QodeAssist is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with QodeAssist. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#include "QodeAssistConstants.hpp"
+#include "QodeAssisttr.h"
+
+#include <coreplugin/actionmanager/actioncontainer.h>
+#include <coreplugin/actionmanager/actionmanager.h>
+#include <coreplugin/actionmanager/command.h>
+#include <coreplugin/coreconstants.h>
+#include <coreplugin/icontext.h>
+#include <coreplugin/icore.h>
+#include <coreplugin/statusbarmanager.h>
+
+#include <extensionsystem/iplugin.h>
+#include <languageclient/languageclientmanager.h>
+
+#include <QAction>
+#include <QMainWindow>
+#include <QMenu>
+#include <QMessageBox>
+#include <texteditor/texteditor.h>
+#include <utils/icon.h>
+
+#include "LLMProvidersManager.hpp"
+#include "PromptTemplateManager.hpp"
+#include "QodeAssistClient.hpp"
+#include "providers/LMStudioProvider.hpp"
+#include "providers/OllamaProvider.hpp"
+#include "templates/CodeLLamaTemplate.hpp"
+#include "templates/StarCoder2Template.hpp"
+
+using namespace Utils;
+using namespace Core;
+using namespace ProjectExplorer;
+
+namespace QodeAssist::Internal {
+
+class QodeAssistPlugin final : public ExtensionSystem::IPlugin
+{
+    Q_OBJECT
+    Q_PLUGIN_METADATA(IID "org.qt-project.Qt.QtCreatorPlugin" FILE "QodeAssist.json")
+
+public:
+    QodeAssistPlugin()
+    {
+    }
+
+    ~QodeAssistPlugin() final
+    {
+
+    }
+
+    void initialize() final
+    {
+        auto &providerManager = LLMProvidersManager::instance();
+        providerManager.registerProvider<Providers::OllamaProvider>();
+        providerManager.registerProvider<Providers::LMStudioProvider>();
+
+        auto &templateManager = PromptTemplateManager::instance();
+        templateManager.registerTemplate<Templates::CodeLLamaTemplate>();
+        templateManager.registerTemplate<Templates::StarCoder2Template>();
+
+        Utils::Icon QCODEASSIST_ICON(
+            {{":/resources/images/qoderassist-icon.png", Utils::Theme::IconsBaseColor}});
+
+        ActionBuilder requestAction(this, Constants::QODE_ASSIST_REQUEST_SUGGESTION);
+        requestAction.setToolTip(
+            Tr::tr("Request Ollama suggestion at the current editor's cursor position."));
+        requestAction.setText(Tr::tr("Request Ollama Suggestion"));
+        requestAction.setIcon(QCODEASSIST_ICON.icon());
+        requestAction.addOnTriggered(this, [this] {
+            if (auto editor = TextEditor::TextEditorWidget::currentTextEditorWidget()) {
+                if (m_qodeAssistClient && m_qodeAssistClient->reachable()) {
+                    m_qodeAssistClient->requestCompletions(editor);
+                } else
+                    qWarning() << "The Qode Assist is not ready. Please check your connection and "
+                                  "settings.";
+            }
+        });
+
+        auto toggleButton = new QToolButton;
+        toggleButton->setDefaultAction(requestAction.contextAction());
+        StatusBarManager::addStatusBarWidget(toggleButton, StatusBarManager::RightCorner);
+    }
+
+    void extensionsInitialized() final
+    {
+    }
+
+    void restartClient()
+    {
+        LanguageClient::LanguageClientManager::shutdownClient(m_qodeAssistClient);
+
+        m_qodeAssistClient = new QodeAssistClient();
+    }
+
+    bool delayedInitialize() final
+    {
+        restartClient();
+
+        return true;
+    }
+
+    ShutdownFlag aboutToShutdown() final
+    {
+        if (!m_qodeAssistClient)
+            return SynchronousShutdown;
+        connect(m_qodeAssistClient,
+                &QObject::destroyed,
+                this,
+                &IPlugin::asynchronousShutdownFinished);
+        return AsynchronousShutdown;
+    }
+
+private:
+    QPointer<QodeAssistClient> m_qodeAssistClient;
+};
+
+} // namespace QodeAssist::Internal
+
+#include <qodeassist.moc>