feat: Add file suggestion edit tool and chat UI (#240)

* feat: Add settings for write to system tool access
2025-11-13 13:32:55 -05:00 · 2025-10-20 11:48:18 +02:00
parent 238ca00227
commit 8a338ecb69
15 changed files with 1400 additions and 22 deletions
--- a/ChatView/FileEditItem.hpp
+++ b/ChatView/FileEditItem.hpp
@ -0,0 +1,122 @@
+/* 
+ * Copyright (C) 2025 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/>.
+ */
+
+#pragma once
+
+#include <QMutex>
+#include <QQuickItem>
+#include <QSet>
+#include <QString>
+#include <QtQmlIntegration>
+
+namespace QodeAssist::Chat {
+
+class FileEditItem : public QQuickItem
+{
+    Q_OBJECT
+    QML_ELEMENT
+
+public:
+    enum class EditStatus {
+        Pending,
+        Applied,
+        Rejected,
+        Reverted
+    };
+    Q_ENUM(EditStatus)
+
+    static constexpr const char *EDIT_MARKER = "QODEASSIST_FILE_EDIT:";
+    static constexpr int RETRY_DELAY_MS = 100;
+    static constexpr int AUTO_APPLY_RETRY_DELAY_MS = 50;
+    static constexpr int MAX_RETRY_COUNT = 10;
+
+    Q_PROPERTY(QString editId READ editId NOTIFY editIdChanged FINAL)
+    Q_PROPERTY(QString filePath READ filePath NOTIFY filePathChanged FINAL)
+    Q_PROPERTY(QString editMode READ editMode NOTIFY editModeChanged FINAL)
+    Q_PROPERTY(QString originalContent READ originalContent NOTIFY originalContentChanged FINAL)
+    Q_PROPERTY(QString newContent READ newContent NOTIFY newContentChanged FINAL)
+    Q_PROPERTY(int addedLines READ addedLines NOTIFY addedLinesChanged FINAL)
+    Q_PROPERTY(int removedLines READ removedLines NOTIFY removedLinesChanged FINAL)
+    Q_PROPERTY(EditStatus status READ status NOTIFY statusChanged FINAL)
+    Q_PROPERTY(QString statusMessage READ statusMessage NOTIFY statusMessageChanged FINAL)
+
+public:
+    explicit FileEditItem(QQuickItem *parent = nullptr);
+
+    QString editId() const { return m_editId; }
+    QString filePath() const { return m_filePath; }
+    QString editMode() const { return m_editMode; }
+    QString originalContent() const { return m_originalContent; }
+    QString newContent() const { return m_newContent; }
+    int addedLines() const { return m_addedLines; }
+    int removedLines() const { return m_removedLines; }
+    EditStatus status() const { return m_status; }
+    QString statusMessage() const { return m_statusMessage; }
+
+    Q_INVOKABLE void parseFromContent(const QString &content);
+    Q_INVOKABLE void applyEdit();
+    Q_INVOKABLE void revertEdit();
+
+signals:
+    void editIdChanged();
+    void filePathChanged();
+    void editModeChanged();
+    void originalContentChanged();
+    void newContentChanged();
+    void addedLinesChanged();
+    void removedLinesChanged();
+    void statusChanged();
+    void statusMessageChanged();
+
+private:
+    void setStatus(EditStatus status);
+    void setStatusMessage(const QString &message);
+    void applyEditInternal(bool isAutomatic, int retryCount = 0);
+    void performApply();
+    void performRevert();
+    void rejectWithError(const QString &errorMessage);
+    void finishWithSuccess(EditStatus status, const QString &message);
+    
+    bool writeFile(const QString &filePath, const QString &content);
+    QString readFile(const QString &filePath);
+    QString applyEditToContent(const QString &content, bool &success);
+    QString applyReverseEdit(const QString &content, bool &success);
+    
+    static bool acquireFileLock(const QString &filePath);
+    static void releaseFileLock(const QString &filePath);
+    static QMutex s_fileLockMutex;
+    static QSet<QString> s_lockedFiles;
+
+    QString m_editId;
+    QString m_filePath;
+    QString m_editMode;
+    QString m_originalContent;
+    QString m_newContent;
+    QString m_contextBefore;
+    QString m_contextAfter;
+    QString m_searchText;
+    int m_lineNumber = -1;
+    int m_addedLines = 0;
+    int m_removedLines = 0;
+    EditStatus m_status = EditStatus::Pending;
+    QString m_statusMessage;
+};
+
+} // namespace QodeAssist::Chat
+