#include "format.h"

#include "grammar.h"

#include <yql/essentials/sql/v1/ide/completion/antlr4/vocabulary.h>

#include <util/generic/hash_set.h>
#include <util/charset/utf8.h>

namespace NSQLComplete {

namespace {

const THashSet<std::string> Keywords = [] {
    const auto& grammar = GetSqlGrammar();
    const auto& vocabulary = grammar.GetVocabulary();

    THashSet<std::string> keywords;
    for (auto& token : grammar.GetKeywordTokens()) {
        keywords.emplace(Display(vocabulary, token));
    }
    return keywords;
}();

} // namespace

TString FormatKeywords(const TVector<TString>& seq) {
    if (seq.empty()) {
        return "";
    }

    TString text = seq[0];
    for (size_t i = 1; i < seq.size(); ++i) {
        const auto& token = seq[i];
        if (Keywords.contains(token)) {
            text += " ";
        }
        text += token;
    }
    return text;
}

bool IsPlain(TStringBuf content) {
    return GetSqlGrammar().IsPlainIdentifier(content);
}

bool IsQuoted(TStringBuf content) {
    return 2 <= content.size() && content.front() == '`' && content.back() == '`';
}

TString Quoted(TString content) {
    content.prepend('`');
    content.append('`');
    return content;
}

TStringBuf Unquoted(TStringBuf content) {
    Y_ENSURE(IsQuoted(content));
    return content.SubStr(1, content.size() - 2);
}

bool IsBinding(TStringBuf content) {
    return !content.empty() && content.front() == '$';
}

TStringBuf Unbinded(TStringBuf content) {
    Y_ENSURE(IsBinding(content));
    return content.SubStr(1);
}

} // namespace NSQLComplete