OpenHarmony-v6.0-Release/s

/**
 * Copyright (c) 2025 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "completions.h"
#include "internal_api.h"
#include "compiler/lowering/util.h"
#include "ir/astNode.h"
#include "public/public.h"
#include <algorithm>
#include <regex>
#include <cstddef>
#include <optional>
#include <string>
#include "generated/tokenType.h"
#include <cstdio>
#include "public/es2panda_lib.h"
#include "lexer/token/letters.h"
namespace ark::es2panda::lsp {

std::string ToLowerCase(const std::string &str)
{
    std::string lowerStr = str;
    std::transform(lowerStr.begin(), lowerStr.end(), lowerStr.begin(), [](unsigned char c) { return std::tolower(c); });
    return lowerStr;
}

std::vector<CompletionEntry> AllKeywordsCompletions()
{
    std::vector<CompletionEntry> keywords;
    for (int i = static_cast<int>(lexer::TokenType::FIRST_KEYW); i <= static_cast<int>(lexer::TokenType::KEYW_YIELD);
         i++) {
        keywords.emplace_back(lsp::CompletionEntry(TokenToString(static_cast<lexer::TokenType>(i)),
                                                   CompletionEntryKind::KEYWORD,
                                                   std::string(sort_text::GLOBALS_OR_KEYWORDS)));
    }
    return keywords;
}

std::vector<CompletionEntry> GetKeywordCompletions(const std::string &input)
{
    std::vector<CompletionEntry> allKeywords = AllKeywordsCompletions();
    std::vector<CompletionEntry> completions;

    for (const auto &entry : allKeywords) {
        if (ToLowerCase(entry.GetName()).find(ToLowerCase(input)) == 0) {
            completions.push_back(entry);
        }
    }
    return completions;
}

CompletionEntry GetDeclarationEntry(ir::AstNode *node)
{
    if (node == nullptr) {
        return CompletionEntry();
    }
    // GetClassPropertyName function could get name of ClassDeclaration
    if (node->IsClassDeclaration()) {
        return CompletionEntry(GetClassPropertyName(node), CompletionEntryKind::CLASS,
                               std::string(sort_text::GLOBALS_OR_KEYWORDS));
    }
    if (node->IsTSInterfaceDeclaration()) {
        if (node->AsTSInterfaceDeclaration()->Id() == nullptr) {
            return CompletionEntry();
        }
        return CompletionEntry(std::string(node->AsTSInterfaceDeclaration()->Id()->Name()),
                               CompletionEntryKind::INTERFACE, std::string(sort_text::GLOBALS_OR_KEYWORDS));
    }
    if (node->IsMethodDefinition()) {
        if (node->AsMethodDefinition()->Key() == nullptr && !node->AsMethodDefinition()->Key()->IsIdentifier()) {
            return CompletionEntry();
        }
        return CompletionEntry(std::string(node->AsMethodDefinition()->Key()->AsIdentifier()->Name()),
                               CompletionEntryKind::METHOD, std::string(sort_text::GLOBALS_OR_KEYWORDS));
    }
    if (node->IsClassProperty()) {
        return CompletionEntry(GetClassPropertyName(node), CompletionEntryKind::PROPERTY,
                               std::string(sort_text::GLOBALS_OR_KEYWORDS));
    }
    if (node->IsETSStructDeclaration()) {
        if (node->AsETSStructDeclaration()->Definition() == nullptr) {
            return CompletionEntry();
        }
        if (node->AsETSStructDeclaration()->Definition()->Ident() == nullptr) {
            return CompletionEntry();
        }
        return CompletionEntry(std::string(node->AsETSStructDeclaration()->Definition()->Ident()->Name()),
                               CompletionEntryKind::STRUCT, std::string(sort_text::GLOBALS_OR_KEYWORDS));
    }
    return CompletionEntry();
}

std::vector<CompletionEntry> GetExportsFromProgram(parser::Program *program)
{
    std::vector<CompletionEntry> exportEntries;
    auto ast = program->Ast();
    auto collectExportNames = [&exportEntries](ir::AstNode *node) {
        if (node->IsExported()) {
            auto entry = GetDeclarationEntry(node);
            if (!entry.GetName().empty()) {
                exportEntries.emplace_back(entry);
            }
        }
    };
    ast->IterateRecursivelyPreorder(collectExportNames);

    return exportEntries;
}

std::vector<CompletionEntry> GetSystemInterfaceCompletions(const std::string &input, parser::Program *program)
{
    std::vector<CompletionEntry> allExternalSourceExports;
    std::vector<CompletionEntry> completions;
    for (auto [_, programList] : program->ExternalSources()) {
        for (auto prog : programList) {
            auto exports = GetExportsFromProgram(prog);
            if (!exports.empty()) {
                allExternalSourceExports.insert(allExternalSourceExports.end(), exports.begin(), exports.end());
            }
        }
    }

    for (const auto &entry : allExternalSourceExports) {
        if (ToLowerCase(entry.GetName()).find(ToLowerCase(input)) == 0) {
            completions.emplace_back(entry);
        }
    }
    return completions;
}

bool IsPointValid(const std::string &str)
{
    std::regex pattern(R"(^[a-zA-Z_$][a-zA-Z0-9_$\-]*(\?)?\.$)");
    return std::regex_match(str, pattern);
}

bool IsEndWithValidPoint(std::string str)
{
    return !str.empty() && str.back() == '.' && IsPointValid(str);
}

bool IsEndWithToken(ir::AstNode *preNode, std::string str)
{
    return str.back() != '.' && preNode->IsIdentifier() && preNode->AsIdentifier()->Name() != "*ERROR_LITERAL*";
}

bool IsEndWithWildcard(ir::AstNode *preNode, const std::string &str)
{
    const std::string wildcardStr = "_WILDCARD";
    if (str == wildcardStr) {
        return preNode->IsIdentifier() && preNode->AsIdentifier()->Name() != "*ERROR_LITERAL*";
    }
    return false;
}

size_t GetPrecedingTokenPosition(std::string sourceCode, size_t pos)
{
    while (pos > 0) {
        char c = sourceCode[pos];
        if (std::isalnum(c) != 0 || c == '_') {
            return pos;
        }
        pos--;
    }
    return pos;
}

bool IsIgnoredName(const std::string &name)
{
    static const std::unordered_set<std::string> IGNORED_NAMES = {"constructor", "_$init$_",
                                                                  "_$initializerBlockInit$_"};
    return IGNORED_NAMES.find(name) != IGNORED_NAMES.end();
}

bool IsWordPartOfIdentifierName(ir::AstNode *node, std::string triggerWord)
{
    if (node == nullptr || !node->IsIdentifier()) {
        return false;
    }
    std::string name(node->AsIdentifier()->Name());
    std::string lowerTrigger = std::move(triggerWord);
    std::string lowerName = name;
    std::transform(lowerTrigger.begin(), lowerTrigger.end(), lowerTrigger.begin(), ::tolower);
    std::transform(lowerName.begin(), lowerName.end(), lowerName.begin(), ::tolower);

    return lowerName.find(lowerTrigger) != std::string::npos && !IsIgnoredName(name);
}

std::vector<ir::AstNode *> FilterFromBody(const ark::ArenaVector<ir::AstNode *> &bodyNodes,
                                          const std::string &triggerWord)
{
    std::vector<ir::AstNode *> res;
    if (bodyNodes.empty()) {
        return res;
    }
    for (auto node : bodyNodes) {
        if (node->IsClassProperty() && IsWordPartOfIdentifierName(node->AsClassProperty()->Key(), triggerWord)) {
            res.emplace_back(node);
        }
        if (node->IsClassDeclaration()) {
            auto def = node->AsClassDeclaration()->Definition()->AsClassDefinition();
            if (def != nullptr && def->Ident() != nullptr && IsWordPartOfIdentifierName(def->Ident(), triggerWord)) {
                res.emplace_back(node);
            }
        }
        if (node->IsMethodDefinition() && IsWordPartOfIdentifierName(node->AsMethodDefinition()->Key(), triggerWord)) {
            res.emplace_back(node);
        }
    }
    return res;
}

std::vector<ir::AstNode *> FilterFromEnumMember(const ark::ArenaVector<ir::AstNode *> &members,
                                                const std::string &triggerWord)
{
    std::vector<ir::AstNode *> res;
    if (members.empty()) {
        return res;
    }
    for (auto member : members) {
        if (member->IsTSEnumMember() && IsWordPartOfIdentifierName(member->AsTSEnumMember()->Key(), triggerWord)) {
            res.emplace_back(member);
        }
    }
    return res;
}

std::vector<ir::AstNode *> FilterFromInterfaceBody(const ark::ArenaVector<ir::AstNode *> &members,
                                                   const std::string &triggerWord)
{
    std::vector<ir::AstNode *> res;
    if (members.empty()) {
        return res;
    }
    for (auto member : members) {
        if (member->IsMethodDefinition() &&
            IsWordPartOfIdentifierName(member->AsMethodDefinition()->Key(), triggerWord)) {
            res.emplace_back(member);
        }
    }
    return res;
}

std::string GetClassPropertyName(ir::AstNode *node)
{
    // property in class
    if (node->IsClassProperty()) {
        auto key = node->AsClassProperty()->Key();
        if (key != nullptr && key->IsIdentifier()) {
            return std::string(key->AsIdentifier()->Name());
        }
    }
    // class in namespace
    if (node->IsClassDeclaration()) {
        auto def = node->AsClassDeclaration()->Definition();
        if (def != nullptr && def->IsClassDefinition() && def->AsClassDefinition()->Ident() != nullptr &&
            def->AsClassDefinition()->Ident()->IsIdentifier()) {
            return std::string(def->AsClassDefinition()->Ident()->AsIdentifier()->Name());
        }
    }
    return "";
}

std::string GetEnumMemberName(ir::AstNode *node)
{
    if (!node->IsTSEnumMember()) {
        return "";
    }
    auto id = node->AsTSEnumMember()->Key();
    if (id == nullptr || !id->IsIdentifier()) {
        return "";
    }
    return std::string(id->AsIdentifier()->Name());
}

std::string GetMethodDefinitionName(ir::AstNode *node)
{
    if (!node->IsMethodDefinition()) {
        return "";
    }
    auto key = node->AsMethodDefinition()->Key();
    if (key == nullptr || !key->IsIdentifier()) {
        return "";
    }
    return std::string(key->AsIdentifier()->Name());
}

ir::AstNode *GetClassDefinitionFromClassProperty(ir::AstNode *node)
{
    if (!node->IsClassProperty()) {
        return nullptr;
    }
    auto value = node->AsClassProperty()->Value();
    auto ident = GetIdentFromNewClassExprPart(value);
    if (ident != nullptr) {
        return compiler::DeclarationFromIdentifier(ident);
    }
    auto type = node->AsClassProperty()->TypeAnnotation();
    if (type != nullptr && type->IsETSTypeReference()) {
        auto typeRefPart = type->AsETSTypeReference()->Part();
        if (typeRefPart == nullptr) {
            return nullptr;
        }
        auto id = typeRefPart->Name();
        if (id != nullptr && id->IsIdentifier()) {
            return compiler::DeclarationFromIdentifier(id->AsIdentifier());
        }
    }
    return nullptr;
}

std::vector<CompletionEntry> GetEntriesForClassDeclaration(
    const std::vector<ark::es2panda::ir::AstNode *> &propertyNodes)
{
    if (propertyNodes.empty()) {
        return {};
    }
    std::vector<CompletionEntry> completions;
    completions.reserve(propertyNodes.size());
    for (auto node : propertyNodes) {
        if (node->IsClassProperty()) {
            completions.emplace_back(lsp::CompletionEntry(GetClassPropertyName(node), CompletionEntryKind::PROPERTY,
                                                          std::string(sort_text::SUGGESTED_CLASS_MEMBERS)));
        }
        if (node->IsClassDeclaration()) {
            completions.emplace_back(
                lsp::CompletionEntry(GetClassPropertyName(node), CompletionEntryKind::CLASS,
                                     std::string(sort_text::MEMBER_DECLARED_BY_SPREAD_ASSIGNMENT)));
        }
        if (node->IsMethodDefinition()) {
            completions.emplace_back(lsp::CompletionEntry(GetMethodDefinitionName(node), CompletionEntryKind::METHOD,
                                                          std::string(sort_text::CLASS_MEMBER_SNIPPETS)));
        }
    }
    return completions;
}

std::vector<CompletionEntry> GetEntriesForTSInterfaceDeclaration(
    const std::vector<ark::es2panda::ir::AstNode *> &propertyNodes)
{
    if (propertyNodes.empty()) {
        return {};
    }
    std::vector<CompletionEntry> completions;
    completions.reserve(propertyNodes.size());
    for (auto node : propertyNodes) {
        if (node->IsMethodDefinition()) {
            completions.emplace_back(lsp::CompletionEntry(GetMethodDefinitionName(node), CompletionEntryKind::METHOD,
                                                          std::string(sort_text::CLASS_MEMBER_SNIPPETS)));
        }
    }
    return completions;
}

std::vector<CompletionEntry> GetEntriesForEnumDeclaration(
    const std::vector<ark::es2panda::ir::AstNode *> &qualifiedMembers)
{
    if (qualifiedMembers.empty()) {
        return {};
    }
    std::vector<CompletionEntry> completions;
    completions.reserve(qualifiedMembers.size());
    for (auto member : qualifiedMembers) {
        completions.emplace_back(lsp::CompletionEntry(GetEnumMemberName(member), CompletionEntryKind::ENUM_MEMBER,
                                                      std::string(sort_text::MEMBER_DECLARED_BY_SPREAD_ASSIGNMENT)));
    }

    return completions;
}

ir::AstNode *GetIdentifierFromSuper(ir::AstNode *super)
{
    if (super == nullptr || !super->IsETSTypeReference()) {
        return nullptr;
    }
    auto part = super->AsETSTypeReference()->Part();
    if (part == nullptr || !part->IsETSTypeReferencePart()) {
        return nullptr;
    }
    return part->AsETSTypeReferencePart()->Name();
}

std::vector<CompletionEntry> GetCompletionFromClassDefinition(ir::ClassDefinition *decl, const std::string &triggerWord)
{
    // After enum refactoring, enum declaration is transformed to a class declaration
    if (compiler::ClassDefinitionIsEnumTransformed(decl)) {
        if (decl->AsClassDefinition()->OrigEnumDecl() == nullptr) {
            return {};
        }
        auto members = decl->AsClassDefinition()->OrigEnumDecl()->AsTSEnumDeclaration()->Members();
        auto qualifiedMembers = FilterFromEnumMember(members, triggerWord);
        return GetEntriesForEnumDeclaration(qualifiedMembers);
    }
    auto bodyNodes = decl->AsClassDefinition()->Body();
    std::vector<CompletionEntry> extendCompletions;
    auto super = decl->AsClassDefinition()->Super();
    if (super != nullptr) {
        auto ident = GetIdentifierFromSuper(super);
        extendCompletions = GetPropertyCompletions(ident, triggerWord);
    }
    auto propertyNodes = FilterFromBody(bodyNodes, triggerWord);
    auto res = GetEntriesForClassDeclaration(propertyNodes);
    res.insert(res.end(), extendCompletions.begin(), extendCompletions.end());
    return res;
}

ir::AstNode *GetIdentifierFromTSInterfaceHeritage(ir::AstNode *node)
{
    if (node == nullptr) {
        return nullptr;
    }
    ir::AstNode *expr = nullptr;
    if (node->IsTSInterfaceHeritage()) {
        expr = node->AsTSInterfaceHeritage()->Expr();
    } else if (node->IsTSClassImplements()) {
        expr = node->AsTSClassImplements()->Expr();
    } else {
        return nullptr;
    }
    if (expr == nullptr) {
        return nullptr;
    }
    auto part = expr->AsETSTypeReference()->Part();
    if (part == nullptr) {
        return nullptr;
    }
    return part->AsETSTypeReferencePart()->Name();
}

std::vector<CompletionEntry> GetCompletionFromTSInterfaceDeclaration(ir::TSInterfaceDeclaration *decl,
                                                                     const std::string &triggerWord)
{
    std::vector<CompletionEntry> completions;
    auto body = decl->AsTSInterfaceDeclaration()->Body();
    if (body == nullptr) {
        return {};
    }
    auto bodies = body->Body();
    std::vector<CompletionEntry> extendCompletions;
    auto extends = decl->AsTSInterfaceDeclaration()->Extends();
    for (auto extend : extends) {
        auto ident = GetIdentifierFromTSInterfaceHeritage(extend);
        if (ident != nullptr && ident->IsIdentifier()) {
            auto extendInterf = compiler::DeclarationFromIdentifier(ident->AsIdentifier());
            if (extendInterf == nullptr) {
                continue;
            }
            auto extendCom =
                extendInterf->IsTSInterfaceDeclaration()
                    ? GetCompletionFromTSInterfaceDeclaration(extendInterf->AsTSInterfaceDeclaration(), triggerWord)
                    : completions;
            extendCompletions.insert(extendCompletions.end(), extendCom.begin(), extendCom.end());
        }
    }
    auto qualifiedBodies = FilterFromInterfaceBody(bodies, triggerWord);
    auto res = GetEntriesForTSInterfaceDeclaration(qualifiedBodies);
    res.insert(res.end(), extendCompletions.begin(), extendCompletions.end());
    return res;
}

std::vector<CompletionEntry> GetCompletionFromMethodDefinition(ir::MethodDefinition *decl,
                                                               const std::string &triggerWord)
{
    auto value = decl->AsMethodDefinition()->Value();
    if (value == nullptr || !value->IsFunctionExpression()) {
        return {};
    }
    auto func = value->AsFunctionExpression()->Function();
    if (func == nullptr || func->ReturnTypeAnnotation() == nullptr) {
        return {};
    }
    auto returnType = func->ReturnTypeAnnotation();
    if (returnType == nullptr || !returnType->IsETSTypeReference()) {
        return {};
    }
    auto ident = returnType->AsETSTypeReference()->Part()->Name();
    if (ident == nullptr || !ident->IsIdentifier()) {
        return {};
    }
    return GetPropertyCompletions(reinterpret_cast<ir::AstNode *>(ident), triggerWord);
}

ir::AstNode *GetIndentifierFromCallExpression(ir::AstNode *node)
{
    if (!node->IsCallExpression()) {
        return nullptr;
    }
    auto callee = node->AsCallExpression()->Callee();
    if (callee != nullptr && callee->IsMemberExpression()) {
        auto object = callee->AsMemberExpression()->Object();
        if (object->IsCallExpression()) {
            return GetIndentifierFromCallExpression(object);
        }
    }
    return callee;
}

util::StringView GetNameFromDefinition(ir::AstNode *preNode)
{
    if (preNode == nullptr || !preNode->IsClassDefinition()) {
        return "";
    }
    auto ident = preNode->AsClassDefinition()->Ident();
    if (ident == nullptr) {
        return "";
    }
    return ident->Name();
}

bool IsDeclarationNameDefined(util::StringView name)
{
    static const std::unordered_set<util::StringView> IGNORED_NAMES = {"ETSGLOBAL"};
    return IGNORED_NAMES.find(name) == IGNORED_NAMES.end();
}

bool IsDefinedClassOrStruct(ir::AstNode *preNode)
{
    if (preNode == nullptr) {
        return false;
    }
    if (!preNode->IsClassDeclaration() && !preNode->IsETSStructDeclaration()) {
        return false;
    }
    if (preNode->IsClassDeclaration()) {
        return IsDeclarationNameDefined(GetNameFromDefinition(preNode->AsClassDeclaration()->Definition()));
    }
    if (preNode->IsETSStructDeclaration()) {
        return IsDeclarationNameDefined(GetNameFromDefinition(preNode->AsETSStructDeclaration()->Definition()));
    }
    return false;
}

ir::AstNode *GetDefinitionOfThisExpression(ir::AstNode *preNode)
{
    if (preNode == nullptr || !preNode->IsThisExpression()) {
        return nullptr;
    }
    while (preNode->Parent() != nullptr) {
        preNode = preNode->Parent();
        if (preNode->IsClassDeclaration() && IsDefinedClassOrStruct(preNode)) {
            return preNode->AsClassDeclaration()->Definition();
        }
        if (preNode->IsETSStructDeclaration() && IsDefinedClassOrStruct(preNode)) {
            return preNode->AsETSStructDeclaration()->Definition();
        }
    }
    return nullptr;
}

std::vector<CompletionEntry> GetCompletionFromThisExpression(ir::AstNode *preNode, const std::string &triggerWord)
{
    if (preNode == nullptr || !preNode->IsThisExpression()) {
        return {};
    }
    auto def = GetDefinitionOfThisExpression(preNode);
    if (def == nullptr || !def->IsClassDefinition()) {
        return {};
    }
    return GetCompletionFromClassDefinition(def->AsClassDefinition(), triggerWord);
}

std::vector<CompletionEntry> GetPropertyCompletions(ir::AstNode *preNode, const std::string &triggerWord)
{
    std::vector<CompletionEntry> completions;
    if (preNode == nullptr) {
        return completions;
    }
    if (preNode->IsCallExpression()) {
        preNode = GetIndentifierFromCallExpression(preNode);
    }
    if (preNode->IsThisExpression()) {
        return GetCompletionFromThisExpression(preNode, triggerWord);
    }
    if (!preNode->IsIdentifier()) {
        return completions;
    }
    auto decl = compiler::DeclarationFromIdentifier(preNode->AsIdentifier());
    if (decl == nullptr) {
        return completions;
    }
    if (decl->IsClassProperty()) {
        decl = GetClassDefinitionFromClassProperty(decl);
    }
    if (decl->IsClassDeclaration()) {
        decl = decl->AsClassDeclaration()->Definition();
    }
    if (decl == nullptr) {
        return completions;
    }
    if (decl->IsMethodDefinition()) {
        return GetCompletionFromMethodDefinition(decl->AsMethodDefinition(), triggerWord);
    }
    if (decl->IsTSInterfaceDeclaration()) {
        return GetCompletionFromTSInterfaceDeclaration(decl->AsTSInterfaceDeclaration(), triggerWord);
    }
    if (decl->IsClassDefinition()) {
        return GetCompletionFromClassDefinition(decl->AsClassDefinition(), triggerWord);
    }
    return completions;
}

Request KeywordCompletionData(const std::string &input)
{
    return {
        CompletionDataKind::KEYWORDS,
        GetKeywordCompletions(input),
    };
}

std::string GetDeclName(const ir::AstNode *decl)
{
    switch (decl->Type()) {
        case ir::AstNodeType::IDENTIFIER:
            return decl->AsIdentifier()->ToString();
        case ir::AstNodeType::METHOD_DEFINITION:
            return decl->AsMethodDefinition()->Key()->AsIdentifier()->ToString();
        case ir::AstNodeType::CLASS_PROPERTY:
            return decl->AsClassProperty()->Key()->AsIdentifier()->ToString();
        default:
            return "";
    }
}

bool IsGlobalVar(const ir::AstNode *node)
{
    return node->IsClassProperty() && node->Parent()->IsClassDefinition() &&
           node->Parent()->AsClassDefinition()->IsGlobal();
}

bool IsVariableOfKind(const ir::Identifier *node, ir::VariableDeclaration::VariableDeclarationKind kind)
{
    /** A VariableDeclaration statement:
     *  - type: VariableDeclaration
     *      - type: VariableDeclarator
     *      - id
     *          - type: Identifier
     *  - Declaration kind
     */
    return node->Parent() != nullptr &&  // try to get the VariableDeclarator
           node->Parent()->IsVariableDeclarator() &&
           node->Parent()->Parent() != nullptr &&  // try to get the VariableDeclaration
           node->Parent()->Parent()->IsVariableDeclaration() &&
           node->Parent()->Parent()->AsVariableDeclaration()->Kind() == kind;
}

bool IsConstVar(const ir::AstNode *node)
{
    if (!node->IsIdentifier()) {
        return false;
    }
    return IsVariableOfKind(node->AsIdentifier(), ir::VariableDeclaration::VariableDeclarationKind::CONST);
}

bool IsLetVar(const ir::AstNode *node)
{
    if (!node->IsIdentifier()) {
        return false;
    }
    return IsVariableOfKind(node->AsIdentifier(), ir::VariableDeclaration::VariableDeclarationKind::LET);
}

bool IsValidDecl(const ir::AstNode *decl)
{
    return decl != nullptr && NodeHasTokens(decl) &&
           (decl->IsMethodDefinition() || IsLetVar(decl) || IsConstVar(decl) || IsGlobalVar(decl));
}

CompletionEntry InitEntry(const ir::AstNode *decl)
{
    auto name = GetDeclName(decl);
    auto sortText = sort_text::GLOBALS_OR_KEYWORDS;
    auto kind = CompletionEntryKind::KEYWORD;
    if (IsLetVar(decl)) {
        kind = CompletionEntryKind::VARIABLE;
    } else if (IsConstVar(decl)) {
        kind = CompletionEntryKind::CONSTANT;
    } else if (IsGlobalVar(decl)) {
        auto globalDefiniton = decl->Parent()->AsClassDefinition();
        auto initMethod = globalDefiniton->FindChild([](ir::AstNode *child) {
            return child->IsMethodDefinition() &&
                   child->AsMethodDefinition()->Key()->AsIdentifier()->Name() == compiler::Signatures::INIT_METHOD;
        });
        if (initMethod == nullptr) {
            return CompletionEntry(name, CompletionEntryKind::CONSTANT, std::string(sortText));
        }
        auto found = initMethod->FindChild([&name](ir::AstNode *child) {
            return child->IsAssignmentExpression() && child->AsAssignmentExpression()->Left()->IsIdentifier() &&
                   child->AsAssignmentExpression()->Left()->AsIdentifier()->ToString() == name;
        });
        if (found != nullptr) {
            // let variable in global definition need to be assigned in _$init$_ method
            kind = CompletionEntryKind::VARIABLE;
        } else {
            kind = CompletionEntryKind::CONSTANT;
        }
    } else if (decl->IsMethodDefinition()) {
        kind = CompletionEntryKind::FUNCTION;
    }
    return CompletionEntry(name, kind, std::string(sortText));
}

void GetIdentifiersInScope(const varbinder::Scope *scope, size_t position, ArenaVector<ir::AstNode *> &results)
{
    if (scope->Node() == nullptr) {
        return;
    }
    auto checkFunc = [scope, position](ir::AstNode *child) -> bool {
        return child->End().index < position && NodeHasTokens(child) && compiler::NearestScope(child) == scope &&
               child->IsIdentifier();
    };
    FindAllChild(scope->Node(), checkFunc, results);
}

auto GetDeclByScopePath(ArenaVector<varbinder::Scope *> &scopePath, size_t position, ArenaAllocator *allocator)
{
    auto hashFunc = [](const ir::AstNode *node) {
        static std::hash<std::string> strHasher;
        return strHasher(GetDeclName(node));
    };
    auto equalFunc = [](const ir::AstNode *lhs, const ir::AstNode *rhs) {
        return GetDeclName(lhs) == GetDeclName(rhs);
    };
    auto decls = ArenaUnorderedSet<ir::AstNode *, decltype(hashFunc), decltype(equalFunc)>(0, hashFunc, equalFunc,
                                                                                           allocator->Adapter());
    for (auto scope : scopePath) {
        auto nodes = ArenaVector<ir::AstNode *>(allocator->Adapter());
        GetIdentifiersInScope(scope, position, nodes);
        for (auto node : nodes) {
            auto decl = compiler::DeclarationFromIdentifier(node->AsIdentifier());
            if (IsValidDecl(decl)) {
                decls.insert(decl);
            }
        }
    }
    return decls;
}

// Support: global variables, local variables, functions, keywords
std::vector<CompletionEntry> GetGlobalCompletions(es2panda_Context *context, size_t position)
{
    auto ctx = reinterpret_cast<public_lib::Context *>(context);
    auto allocator = ctx->allocator;
    if (allocator == nullptr) {
        return {};
    }
    auto precedingToken = FindPrecedingToken(position, ctx->parserProgram->Ast(), allocator);
    if (precedingToken == nullptr) {
        return {};
    }
    auto scopePath = BuildScopePath(compiler::NearestScope(precedingToken), allocator);
    auto prefix = GetCurrentTokenValueImpl(context, position);
    auto decls = GetDeclByScopePath(scopePath, position, allocator);
    std::vector<CompletionEntry> completions;

    for (auto decl : decls) {
        auto entry = InitEntry(decl);
        if (entry.GetName().find(prefix) != 0) {
            continue;
        }
        entry = ProcessAutoImportForEntry(entry);
        completions.push_back(entry);
    }

    auto keywordCompletions = GetKeywordCompletions(prefix);
    completions.insert(completions.end(), keywordCompletions.begin(), keywordCompletions.end());
    auto systemInterfaceCompletions = GetSystemInterfaceCompletions(prefix, ctx->parserProgram);
    completions.insert(completions.end(), systemInterfaceCompletions.begin(), systemInterfaceCompletions.end());

    return completions;
}

ArenaVector<varbinder::Scope *> BuildScopePath(varbinder::Scope *startScope, ArenaAllocator *allocator)
{
    ArenaVector<varbinder::Scope *> scopePath(allocator->Adapter());
    for (auto scope = startScope; scope != nullptr; scope = scope->Parent()) {
        scopePath.push_back(scope);
    }
    return scopePath;
}

CompletionEntry ProcessAutoImportForEntry(CompletionEntry &entry)
{
    auto dataOpt = entry.GetCompletionEntryData();
    if (!dataOpt.has_value()) {
        return entry;
    }

    auto config = GetArkTsConfigFromFile(dataOpt->GetFileName());
    if (config == nullptr) {
        return entry;
    }

    auto autoImportData = GetAutoImportCompletionEntry(&dataOpt.value(), config, entry.GetName());
    if (!autoImportData.has_value()) {
        return entry;
    }

    return CompletionEntry(entry.GetName(), entry.GetCompletionKind(), entry.GetSortText(), entry.GetInsertText(),
                           autoImportData);
}

std::vector<CompletionEntry> GetCompletionsAtPositionImpl(es2panda_Context *context, size_t pos)
{
    if (context == nullptr) {
        return {};
    }
    auto ctx = reinterpret_cast<public_lib::Context *>(context);
    if (ctx->parserProgram == nullptr || ctx->parserProgram->Ast() == nullptr) {
        return {};
    }
    auto allocator = ctx->allocator;
    std::string sourceCode(ctx->parserProgram->SourceCode());
    // Current GetPrecedingPosition cannot get token of "obj." with position.
    auto precedingToken = FindPrecedingToken(pos, ctx->parserProgram->Ast(), allocator);
    if (precedingToken == nullptr) {
        return {};
    }
    auto triggerValue = GetCurrentTokenValueImpl(context, pos);
    // Unsupported yet because of ast parsing problem
    if (IsEndWithValidPoint(triggerValue)) {
        return GetPropertyCompletions(precedingToken, "");
    }
    auto memberExpr = precedingToken->Parent();
    // This is a temporary solution to support "obj." with wildcard for better solution in internal issue.
    if (IsEndWithWildcard(precedingToken, triggerValue) && memberExpr != nullptr) {
        if (memberExpr->IsMemberExpression()) {
            precedingToken = memberExpr->AsMemberExpression()->Object();
            return GetPropertyCompletions(precedingToken, "");
        }
        if (memberExpr->IsTSQualifiedName()) {
            precedingToken = memberExpr->AsTSQualifiedName()->Left();
            return GetPropertyCompletions(precedingToken, "");
        }
    }
    if (IsEndWithToken(precedingToken, triggerValue) && memberExpr != nullptr) {
        if (memberExpr->IsMemberExpression()) {
            precedingToken = memberExpr->AsMemberExpression()->Object();
            return GetPropertyCompletions(precedingToken, triggerValue);
        }
        if (memberExpr->IsTSQualifiedName()) {
            precedingToken = memberExpr->AsTSQualifiedName()->Left();
            return GetPropertyCompletions(precedingToken, triggerValue);
        }
    }
    return GetGlobalCompletions(context, pos);
}

std::optional<CompletionEntryData> GetAutoImportCompletionEntry(ark::es2panda::lsp::CompletionEntryData *data,
                                                                const std::shared_ptr<ArkTsConfig> &config,
                                                                const std::string &name)
{
    const char *fileName = data->GetFileName();
    if (fileName == nullptr || std::strlen(fileName) == 0) {
        return std::nullopt;
    }
    if (config == nullptr) {
        return std::nullopt;
    }
    return CompletionEntryDataToOriginInfo(data, config, name);
}

std::optional<CompletionEntryData> CompletionEntryDataToOriginInfo(ark::es2panda::lsp::CompletionEntryData *data,
                                                                   const std::shared_ptr<ArkTsConfig> &config,
                                                                   const std::string &name)
{
    if (IsCompletionEntryDataResolved(data, config) == true) {
        return CompletionEntryData(data->GetFileName(), data->GetNamedExport(), data->GetImportDeclaration(), name,
                                   ResolutionStatus::RESOLVED);
    }
    if (IsCompletionEntryDataResolved(data, config) == false) {
        return CompletionEntryData(data->GetFileName(), data->GetNamedExport(), data->GetImportDeclaration(), name,
                                   ResolutionStatus::UNRESOLVED);
    }
    return std::nullopt;
}

bool StartsWith(const std::string &str, const std::string &prefix)
{
    return str.size() >= prefix.size() && str.compare(0, prefix.size(), prefix) == 0;
}

std::optional<bool> IsCompletionEntryDataResolved(ark::es2panda::lsp::CompletionEntryData *data,
                                                  const std::shared_ptr<ArkTsConfig> &config)
{
    auto importDecl = data->GetImportDeclaration();
    if (importDecl.length() == 0) {
        return std::nullopt;
    }
    if (StartsWith(importDecl, "./") || StartsWith(importDecl, "../")) {
        return true;
    }

    const char slash = static_cast<char>(lexer::LEX_CHAR_SLASH);
    const int pos = importDecl.find(slash);
    const std::string importSub = importDecl.substr(0, pos);
    auto configPaths = config->Paths();
    if (configPaths.count(importSub) != 0) {
        return false;
    }

    return std::nullopt;
}

std::shared_ptr<ArkTsConfig> GetArkTsConfigFromFile(const char *fileName)
{
    Initializer initializer = Initializer();

    auto ctx = initializer.CreateContext(fileName, ES2PANDA_STATE_CHECKED);
    auto config = reinterpret_cast<ark::es2panda::public_lib::Context *>(ctx)->config->options->ArkTSConfig();
    initializer.DestroyContext(ctx);

    return config;
}
}  // namespace ark::es2panda::lsp