#ifndef SORBET_LSP_QUERYRESPONSE
#define SORBET_LSP_QUERYRESPONSE
#include "core/Loc.h"
#include "core/LocalVariable.h"
#include "core/Types.h"
#include <variant>

namespace sorbet::core::lsp {
class TypeConstraint;

class SendResponse final {
public:
    SendResponse(std::shared_ptr<core::DispatchResult> dispatchResult, absl::Span<const core::LocOffsets> argLocOffsets,
                 absl::Span<const core::TypePtr> argTypes, core::NameRef callerSideName, core::NameRef originalName,
                 core::MethodRef enclosingMethod, bool isPrivateOk, uint16_t numPosArgs, core::FileRef file,
                 core::LocOffsets termLocOffsets, core::LocOffsets receiverLocOffsets, core::LocOffsets funLocOffsets,
                 core::LocOffsets locOffsetsWithoutBlock)
        : dispatchResult(std::move(dispatchResult)), argLocOffsets(argLocOffsets.begin(), argLocOffsets.end()),
          argTypes(argTypes.begin(), argTypes.end()), callerSideName(callerSideName), originalName(originalName),
          enclosingMethod(enclosingMethod), isPrivateOk(isPrivateOk), numPosArgs(numPosArgs), file(file),
          termLocOffsets(termLocOffsets), receiverLocOffsets(receiverLocOffsets), funLocOffsets(funLocOffsets),
          locOffsetsWithoutBlock(locOffsetsWithoutBlock){};
    const std::shared_ptr<core::DispatchResult> dispatchResult;
    const InlinedVector<core::LocOffsets, 2> argLocOffsets;
    const InlinedVector<core::TypePtr, 2> argTypes;
    // The actual name we wind up invoking; in the case of `<Magic>` methods
    // like `<call-with-splat>`, this is the name that would be invoked.
    const core::NameRef callerSideName;
    // The method name from the send with none of the filtering involved in
    // `callerSideName`.
    const core::NameRef originalName;
    const core::MethodRef enclosingMethod;
    const bool isPrivateOk;
    const uint16_t numPosArgs;
    const core::FileRef file;
    const core::LocOffsets termLocOffsets;
    const core::LocOffsets receiverLocOffsets;
    const core::LocOffsets funLocOffsets;
    const core::LocOffsets locOffsetsWithoutBlock;

    core::Loc termLoc() const {
        return core::Loc(file, termLocOffsets);
    }
    core::Loc receiverLoc() const {
        return core::Loc(file, receiverLocOffsets);
    }
    core::Loc funLoc() const {
        return core::Loc(file, funLocOffsets);
    }
    core::Loc locWithoutBlock() const {
        return core::Loc(file, locOffsetsWithoutBlock);
    }

    const std::optional<core::Loc> getMethodNameLoc(const core::GlobalState &gs) const;
};
CheckSize(SendResponse, 120, 8);

class IdentResponse final {
public:
    IdentResponse(core::Loc termLoc, core::LocalVariable variable, core::TypeAndOrigins retType,
                  core::MethodRef enclosingMethod, core::Loc enclosingMethodLoc)
        : termLoc(termLoc), variable(variable), enclosingMethod(enclosingMethod),
          enclosingMethodLoc(enclosingMethodLoc), retType(std::move(retType)) {}
    const core::Loc termLoc;
    const core::LocalVariable variable;
    const core::MethodRef enclosingMethod;
    // The loc of the MethodDef this ident was in.
    // (not the declLoc, which can be found by way of the enclosingMethod's entry in the symbol table)
    const core::Loc enclosingMethodLoc;
    const core::TypeAndOrigins retType;
};
CheckSize(IdentResponse, 72, 8);

class LiteralResponse final {
public:
    LiteralResponse(core::Loc termLoc, core::TypeAndOrigins retType) : termLoc(termLoc), retType(std::move(retType)){};
    const core::Loc termLoc;
    const core::TypeAndOrigins retType;
};
CheckSize(LiteralResponse, 48, 8);

class KeywordArgResponse final {
public:
    KeywordArgResponse(Loc termLoc, const MethodRef owner, const ParamInfo &param)
        : termLoc(termLoc), owner(owner), paramLoc(param.loc), paramName(param.name), paramType(param.type) {}
    const Loc termLoc;
    MethodRef owner;
    Loc paramLoc;
    NameRef paramName;
    TypePtr paramType;
};
CheckSize(KeywordArgResponse, 40, 8);

class ConstantResponse final {
public:
    using Scopes = InlinedVector<core::SymbolRef, 1>;
    ConstantResponse(core::SymbolRef symbolBeforeDealias, core::Loc termLoc, Scopes scopes, core::NameRef name,
                     core::TypeAndOrigins retType, core::MethodRef enclosingMethod)
        : symbolBeforeDealias(symbolBeforeDealias), termLoc(termLoc), scopes(scopes), name(name),
          enclosingMethod(enclosingMethod), retType(std::move(retType)) {}
    const core::SymbolRef symbolBeforeDealias;
    const core::Loc termLoc;
    const Scopes scopes;
    const core::NameRef name;
    const core::MethodRef enclosingMethod;
    const core::TypeAndOrigins retType;
};
CheckSize(ConstantResponse, 80, 8);

class FieldResponse final {
public:
    FieldResponse(core::FieldRef symbol, core::Loc termLoc, core::NameRef name, core::TypeAndOrigins retType)
        : symbol(symbol), termLoc(termLoc), name(name), retType(std::move(retType)){};
    const core::FieldRef symbol;
    const core::Loc termLoc;
    const core::NameRef name;
    const core::TypeAndOrigins retType;
};
CheckSize(FieldResponse, 56, 8);

class MethodDefResponse final {
public:
    MethodDefResponse(core::MethodRef symbol, core::FileRef file, core::LocOffsets termLocOffsets,
                      core::LocOffsets declLocOffsets, core::NameRef name, bool isAttrBestEffortUIOnly,
                      core::TypeAndOrigins retType)
        : symbol(symbol), file(file), termLocOffsets(termLocOffsets), declLocOffsets(declLocOffsets), name(name),
          isAttrBestEffortUIOnly(isAttrBestEffortUIOnly), retType(std::move(retType)){};
    const core::MethodRef symbol;
    const core::FileRef file;
    const core::LocOffsets termLocOffsets;
    const core::LocOffsets declLocOffsets;
    const core::NameRef name;
    const bool isAttrBestEffortUIOnly;
    const core::TypeAndOrigins retType;

    core::Loc termLoc() const {
        return core::Loc(file, termLocOffsets);
    }
    core::Loc declLoc() const {
        return core::Loc(file, declLocOffsets);
    }
};
CheckSize(MethodDefResponse, 64, 8);

class ClassDefResponse final {
public:
    ClassDefResponse(core::ClassOrModuleRef symbol, core::Loc termLoc, core::Loc declLoc)
        : symbol(symbol), termLoc(termLoc), declLoc(declLoc){};
    const core::ClassOrModuleRef symbol;
    const core::Loc termLoc;
    const core::Loc declLoc;
};
CheckSize(ClassDefResponse, 28, 4);

class EditResponse final {
public:
    EditResponse(core::Loc loc, std::string replacement) : loc(loc), replacement(std::move(replacement)){};
    const core::Loc loc;
    const std::string replacement;
};
CheckSize(EditResponse, 40, 8);

using QueryResponseVariant = std::variant<SendResponse, IdentResponse, LiteralResponse, ConstantResponse, FieldResponse,
                                          MethodDefResponse, ClassDefResponse, EditResponse, KeywordArgResponse>;

/**
 * Represents a response to a LSP query. Wraps a variant that contains one of several response types.
 */
class QueryResponse final {
private:
    QueryResponseVariant response;

public:
    /**
     * Pushes the given query response on to the error queue.
     */
    static void pushQueryResponse(core::Context ctx, QueryResponseVariant rawResponse);
    static void pushQueryResponse(const GlobalState &gs, FileRef file, QueryResponseVariant rawResponse);

    QueryResponse(QueryResponseVariant response);

    const QueryResponseVariant &asResponse() const {
        return this->response;
    };

    /**
     * Returns nullptr unless this is a Send.
     */
    const SendResponse *isSend() const;

    /**
     * Returns nullptr unless this is an Ident.
     */
    const IdentResponse *isIdent() const;

    /**
     * Returns nullptr unless this is a Literal.
     */
    const LiteralResponse *isLiteral() const;

    /**
     * Returns nullptr unless this is a KeywordArg.
     */
    const KeywordArgResponse *isKeywordArg() const;

    /**
     * Returns nullptr unless this is a Constant.
     */
    const ConstantResponse *isConstant() const;

    /**
     * Returns nullptr unless this is a Field.
     */
    const FieldResponse *isField() const;

    /**
     * Returns nullptr unless this is a MethodDef.
     */
    const MethodDefResponse *isMethodDef() const;

    /**
     * Returns nullptr unless this is a ClassDef.
     */
    const ClassDefResponse *isClassDef() const;

    /**
     * Returns nullptr unless this is an Edit.
     */
    const EditResponse *isEdit() const;

    /**
     * Returns the source code location for the specific expression that this
     * response points to.
     */
    core::Loc getLoc() const;

    /**
     * Returns the type of this expression's rval.
     */
    core::TypePtr getRetType() const;
};

} // namespace sorbet::core::lsp
#endif
