// SPDX-FileCopyrightText: 2024 - 2025 UnionTech Software Technology Co., Ltd.
//
// SPDX-License-Identifier: GPL-3.0-or-later

#include "scilexerpython.h"

#include <DGuiApplicationHelper>

#include <QColor>
#include <QFont>
#include <QFileInfo>

DGUI_USE_NAMESPACE

SciLexerPython::SciLexerPython(QObject *parent)
    : AbstractLexerProxy(parent)
{
}

const char *SciLexerPython::language() const
{
    return "Python";
}

const char *SciLexerPython::lexer() const
{
    return "python";
}

QString SciLexerPython::description(int style) const
{
    switch (style) {
    case Default:
        return tr("Default");

    case Comment:
        return tr("Comment");

    case Number:
        return tr("Number");

    case DoubleQuotedString:
        return tr("Double-quoted string");

    case SingleQuotedString:
        return tr("Single-quoted string");

    case Keyword:
        return tr("Keyword");

    case TripleSingleQuotedString:
        return tr("Triple single-quoted string");

    case TripleDoubleQuotedString:
        return tr("Triple double-quoted string");

    case ClassName:
        return tr("Class name");

    case FunctionMethodName:
        return tr("Function or method name");

    case Operator:
        return tr("Operator");

    case Identifier:
        return tr("Identifier");

    case CommentBlock:
        return tr("Comment block");

    case UnclosedString:
        return tr("Unclosed string");

    case HighlightedIdentifier:
        return tr("Highlighted identifier");

    case Decorator:
        return tr("Decorator");

    case DoubleQuotedFString:
        return tr("Double-quoted f-string");

    case SingleQuotedFString:
        return tr("Single-quoted f-string");

    case TripleSingleQuotedFString:
        return tr("Triple single-quoted f-string");

    case TripleDoubleQuotedFString:
        return tr("Triple double-quoted f-string");
    }

    return QString();
}

bool SciLexerPython::isSupport(const QString &file) const
{
    QFileInfo info(file);
    return info.suffix().compare("py", Qt::CaseInsensitive) == 0;
}

int SciLexerPython::blockLookback() const
{
    return 0;
}

const char *SciLexerPython::blockStart(int *style) const
{
    if (style)
        *style = Operator;

    return ":";
}

int SciLexerPython::braceStyle() const
{
    return Operator;
}

QColor SciLexerPython::defaultColor(int style) const
{
    bool isDarkTheme = DGuiApplicationHelper::instance()->themeType() == DGuiApplicationHelper::DarkType;
    switch (style) {
    case Default:
        return isDarkTheme ? QColor("#d6cf9a") : QColor("#000000");

    case Comment:
    case CommentBlock:
        return isDarkTheme ? QColor("#a8abb0") : QColor("#008000");

    case Number:
        return isDarkTheme ? QColor("#8a602c") : QColor("#000080");

    case UnclosedString:
    case DoubleQuotedString:
    case SingleQuotedString:
    case DoubleQuotedFString:
    case SingleQuotedFString:
    case TripleSingleQuotedString:
    case TripleDoubleQuotedString:
    case TripleSingleQuotedFString:
    case TripleDoubleQuotedFString:
        return isDarkTheme ? QColor("#d69545") : QColor("#008000");

    case Keyword:
        return isDarkTheme ? QColor("#ff8080") : QColor("#800080");

    case ClassName:
        return isDarkTheme ? QColor("#bcf0ff") : QColor("#000000");

    case FunctionMethodName:
        return isDarkTheme ? QColor("#56a8f5") : QColor("#00627a");

    case Operator:
    case Identifier:
        break;

    case HighlightedIdentifier:
        return isDarkTheme ? QColor("#aa4926") : QColor("#660099");

    case Decorator:
        return isDarkTheme ? QColor("#b3ae60") : QColor("#9e880d");
    }

    return AbstractLexerProxy::defaultColor(style);
}

bool SciLexerPython::defaultEolFill(int style) const
{
    if (style == UnclosedString)
        return true;

    return AbstractLexerProxy::defaultEolFill(style);
}

QColor SciLexerPython::defaultPaper(int style) const
{
    if (style == UnclosedString) {
        bool isDarkTheme = DGuiApplicationHelper::instance()->themeType() == DGuiApplicationHelper::DarkType;
        return isDarkTheme ? QColor("#fa6675") : QColor("#ffcccc");
    }

    return AbstractLexerProxy::defaultPaper(style);
}

const char *SciLexerPython::keywords(int set) const
{
    if (set != 1)
        return nullptr;

    return "and as assert break class continue def del elif else except exec "
           "finally for from global if import in is lambda None not or pass "
           "print raise return try while with yield";
}

void SciLexerPython::refreshProperties()
{
    emit propertyChanged("fold.comment.python", "0");
    emit propertyChanged("fold.compact", "1");
    emit propertyChanged("fold.quotes.python", "0");
    emit propertyChanged("tab.timmy.whinge.level", "0");
    emit propertyChanged("lexer.python.strings.over.newline", "0");
    emit propertyChanged("lexer.python.strings.u", "1");
    emit propertyChanged("lexer.python.literals.binary", "1");
    emit propertyChanged("lexer.python.strings.b", "1");
    emit propertyChanged("lexer.python.keywords2.no.sub.identifiers", "0");
}
