#pragma once

#include <yql/essentials/sql/v1/ide/completion/core/environment.h>
#include <yql/essentials/sql/v1/ide/completion/core/input.h>
#include <yql/essentials/sql/v1/ide/completion/core/name.h>

#include <util/generic/ptr.h>
#include <util/generic/maybe.h>
#include <util/generic/string.h>
#include <util/generic/vector.h>
#include <util/generic/hash.h>
#include <util/generic/hash_set.h>

namespace NSQLComplete {

struct TClusterContext {
    TString Provider;
    TString Name;

    friend bool operator==(const TClusterContext& lhs, const TClusterContext& rhs) = default;
};

struct TFunctionContext {
    TString Name;
    size_t ArgumentNumber = 0;
    TMaybe<TString> Arg0 = Nothing();
    TMaybe<TString> Arg1 = Nothing();
    TMaybe<TClusterContext> Cluster = Nothing();

    friend bool operator==(const TFunctionContext& lhs, const TFunctionContext& rhs) = default;
};

// TODO(YQL-19747): Try to refactor to use Map/Set data structures
struct TColumnContext {
    TVector<TAliased<TTableId>> Tables;
    TVector<TColumnId> Columns;
    THashMap<TString, THashSet<TString>> WithoutByTableAlias;

    [[nodiscard]] bool IsAsterisk() const;
    TColumnContext ExtractAliased(TMaybe<TStringBuf> alias);
    TColumnContext Renamed(TStringBuf alias) &&;

    friend bool operator==(const TColumnContext& lhs, const TColumnContext& rhs) = default;
    friend TColumnContext operator|(TColumnContext lhs, TColumnContext rhs);

    static TColumnContext Asterisk();
};

struct TGlobalContext {
    TMaybe<TClusterContext> Use;
    TVector<TString> Names;
    TMaybe<TFunctionContext> EnclosingFunction;
    TMaybe<TColumnContext> Column;
};

// TODO(YQL-19747): Make it thread-safe to make ISqlCompletionEngine thread-safe.
class IGlobalAnalysis {
public:
    using TPtr = THolder<IGlobalAnalysis>;

    virtual ~IGlobalAnalysis() = default;
    virtual TGlobalContext Analyze(TCompletionInput input, TEnvironment env) = 0;
};

IGlobalAnalysis::TPtr MakeGlobalAnalysis();

} // namespace NSQLComplete