Merge pull request #19 from sandbox-science/commentCode

File Extension Detection and Commenting Feature

Author: chrisdedman

include/CodeEditor.h

@@ -18,6 +18,8 @@ class CodeEditor : public QPlainTextEdit
     Mode mode = NORMAL;
     void lineNumberAreaPaintEvent(QPaintEvent *event);
     int lineNumberAreaWidth();
+    QString getCurrentFileName() const { return m_currentFileName; }
+    void setCurrentFileName(const QString &fileName) { m_currentFileName = fileName; }
 
 signals:
     void statusMessageChanged(const QString &message);
@@ -33,4 +35,11 @@ private slots:
 
 private:
     QWidget *m_lineNumberArea;
+    QString m_currentFileName;
+
+    QString getFileExtension();
+    void addLanguageSymbol(QTextCursor &cursor, const QString &commentSymbol);
+    void commentSelection(QTextCursor &cursor, const QString &commentSymbol);
+    void commentLine(QTextCursor &cursor, const QString &commentSymbol);
+    void addComment();
 };

src/CodeEditor.cpp

@@ -5,6 +5,7 @@
 #include <QPainter>
 #include <QTextBlock>
 #include <QStatusBar>
+#include <QFileInfo>
 
 CodeEditor::CodeEditor(QWidget *parent)
     : QPlainTextEdit(parent),
@@ -27,6 +28,11 @@ void CodeEditor::keyPressEvent(QKeyEvent *event)
         moveCursor(QTextCursor::WordLeft, QTextCursor::KeepAnchor);
         return;
     }
+    if (event->modifiers() == Qt::ControlModifier && event->key() == Qt::Key_Slash)
+    {
+        addComment();
+        return;
+    }
 
     if (mode == NORMAL)
     {
@@ -67,6 +73,109 @@ void CodeEditor::keyPressEvent(QKeyEvent *event)
     }
 }
 
+QString CodeEditor::getFileExtension()
+{
+    QString filePath = getCurrentFileName();
+    if (!QFile::exists(filePath))
+    {
+        return QString();
+    }
+
+    // Extract the file extension from the file path
+    return QFileInfo(filePath).suffix().toLower();
+}
+
+void CodeEditor::addLanguageSymbol(QTextCursor &cursor, const QString &commentSymbol)
+{
+    if (cursor.hasSelection())
+    {
+        commentSelection(cursor, commentSymbol);
+    }
+    else
+    {
+        commentLine(cursor, commentSymbol);
+    }
+}
+
+// Comment/uncomment the selected text or the current line
+void CodeEditor::commentSelection(QTextCursor &cursor, const QString &commentSymbol)
+{
+    int start = cursor.selectionStart();
+    int end   = cursor.selectionEnd();
+
+    cursor.setPosition(start);
+    int startBlockNumber = cursor.blockNumber();
+    cursor.setPosition(end);
+    int endBlockNumber = cursor.blockNumber();
+
+    cursor.setPosition(start);
+    for (int i = startBlockNumber; i <= endBlockNumber; ++i)
+    {
+        cursor.movePosition(QTextCursor::StartOfLine);
+        QString lineText = cursor.block().text();
+
+        if (lineText.startsWith(commentSymbol))
+        {
+            cursor.movePosition(QTextCursor::Right, QTextCursor::KeepAnchor, 3);
+            cursor.removeSelectedText();
+        }
+        else
+        {
+            cursor.insertText(commentSymbol + " ");
+        }
+
+        cursor.movePosition(QTextCursor::NextBlock);
+    }
+}
+
+// Comment/uncomment the single current line
+void CodeEditor::commentLine(QTextCursor &cursor, const QString &commentSymbol)
+{
+    cursor.select(QTextCursor::LineUnderCursor);
+    QString lineText = cursor.selectedText();
+
+    if (lineText.startsWith(commentSymbol))
+    {
+        lineText.remove(0, 3);
+    }
+    else
+    {
+        lineText.prepend(commentSymbol + " ");
+    }
+
+    cursor.insertText(lineText);
+}
+
+void CodeEditor::addComment()
+{
+    QTextCursor cursor    = textCursor();
+    QString fileExtension = getFileExtension();
+    qDebug() << "File Extension:" << fileExtension;
+
+    if (fileExtension == "cpp" || fileExtension == "h" ||
+        fileExtension == "hpp" || fileExtension == "c" ||
+        fileExtension == "java" || fileExtension == "go" ||
+        fileExtension == "json")
+    {
+        addLanguageSymbol(cursor, "//");
+    }
+    else if (fileExtension == "py" || fileExtension == "yaml" ||
+             fileExtension == "yml" || fileExtension == "sh" ||
+             fileExtension == "bash")
+    {
+        addLanguageSymbol(cursor, "#");
+    }
+    else if (fileExtension == "sql")
+    {
+        addLanguageSymbol(cursor, "--");
+    }
+    else
+    {
+        qDebug() << "Unsupported file extension for commenting.";
+        return;
+    }
+}
+
 int CodeEditor::lineNumberAreaWidth()
 {
     int digits = 1;

src/MainWindow.cpp

@@ -170,13 +170,13 @@ void MainWindow::openFile()
 
 void MainWindow::saveFile()
 {
-    if (m_currentFileName.isEmpty())
+    if (m_editor->getCurrentFileName().isEmpty())
     {
         saveFileAs();
         return;
     }
 
-    QFile file(m_currentFileName);
+    QFile file(m_editor->getCurrentFileName());
     if (!file.open(QFile::WriteOnly | QFile::Text))
     {
         QMessageBox::warning(this, "Error", "Cannot save file: " + file.errorString());
@@ -205,7 +205,7 @@ void MainWindow::saveFileAs()
 
     if (!fileName.isEmpty())
     {
-        m_currentFileName = fileName;
+        m_editor->setCurrentFileName(fileName);
         saveFile();
     }
 }
@@ -231,6 +231,7 @@ void MainWindow::loadFileInEditor(const QString &filePath)
     }
     file.close();
 
-    m_currentFileName = filePath;
+    m_editor->setCurrentFileName(filePath);
     setWindowTitle("CodeAstra ~ " + QFileInfo(filePath).fileName());
+    emit m_editor->statusMessageChanged("File loaded successfully: " + QFileInfo(filePath).fileName());
 }