#include <catch2/catch_test_macros.hpp>

#include <nlohmann/json.hpp>
#include <exception>
#include <filesystem>
#include <memory>
#include <optional>
#include <sstream>
#include <string>
#include <vector>

#include "common/test_helpers_catch2.h"

#include <yams/cli/yams_cli.h>
#include <yams/daemon/client/global_io_context.h>
#include <yams/metadata/connection_pool.h>
#include <yams/metadata/document_metadata.h>
#include <yams/metadata/knowledge_graph_store.h>
#include <yams/metadata/metadata_repository.h>
#include <yams/metadata/path_utils.h>
#include <yams/topology/topology_baseline.h>
#include <yams/topology/topology_metadata_store.h>

namespace fs = std::filesystem;

namespace {

class CaptureStdout {
public:
    CaptureStdout() : old_(std::cout.rdbuf(buffer_.rdbuf())) {}
    ~CaptureStdout() { std::cout.rdbuf(old_); }

    std::string str() const { return buffer_.str(); }

private:
    std::ostringstream buffer_;
    std::streambuf* old_{nullptr};
};

class ScopedCurrentPath {
public:
    explicit ScopedCurrentPath(const fs::path& target) : original_(fs::current_path()) {
        fs::current_path(target);
    }

    ~ScopedCurrentPath() {
        std::error_code error;
        fs::current_path(original_, error);
    }

    ScopedCurrentPath(const ScopedCurrentPath&) = delete;
    ScopedCurrentPath& operator=(const ScopedCurrentPath&) = delete;

private:
    fs::path original_;
};

int run_cli(const std::vector<std::string>& args, std::string* output = nullptr,
            std::optional<std::string> stdinData = std::nullopt) {
    std::vector<std::string> effectiveArgs = args;
    const bool hasDataDirFlag =
        std::find(effectiveArgs.begin(), effectiveArgs.end(), "--data-dir") !=
            effectiveArgs.end() ||
        std::find(effectiveArgs.begin(), effectiveArgs.end(), "--storage") != effectiveArgs.end();
    if (!hasDataDirFlag) {
        if (const char* dataDir = std::getenv("YAMS_DATA_DIR"); dataDir && *dataDir) {
            effectiveArgs.insert(effectiveArgs.begin() + 1, std::string(dataDir));
            effectiveArgs.insert(effectiveArgs.begin() + 1, "--data-dir");
        }
    }
    int rc = 0;
    std::string captured;
    try {
        yams::cli::YamsCLI cli;
        std::vector<char*> argv;
        argv.reserve(effectiveArgs.size());
        for (const auto& arg : effectiveArgs) {
            argv.push_back(const_cast<char*>(arg.c_str()));
        }

        CaptureStdout capture;

        std::istringstream in;
        std::streambuf* oldIn = nullptr;
        if (stdinData.has_value()) {
            in.str(*stdinData);
            oldIn = std::cin.rdbuf(in.rdbuf());
        }

        {
            rc = cli.run(static_cast<int>(argv.size()), argv.data());
        }

        if (oldIn) {
            std::cin.rdbuf(oldIn);
        }
        if (captured.empty()) {
            captured = capture.str();
        }
    } catch (const std::exception& e) {
        rc = -1;
        captured = std::string("EXCEPTION: ") + e.what();
    } catch (...) {
        rc = -1;
        captured = "EXCEPTION: unknown";
    }
    if (output) {
        *output = std::move(captured);
    }
    return rc;
}

yams::metadata::DocumentInfo makeDocumentWithPath(const std::string& path,
                                                  const std::string& hash) {
    yams::metadata::DocumentInfo info;
    info.filePath = path;
    info.fileName = fs::path(path).filename().string();
    info.fileExtension = fs::path(path).extension().string();
    info.fileSize = 123;
    info.sha256Hash = hash;
    info.mimeType = "text/plain";
    info.createdTime = std::chrono::floor<std::chrono::seconds>(std::chrono::system_clock::now());
    info.modifiedTime = info.createdTime;
    info.indexedTime = info.createdTime;
    info.contentExtracted = true;
    info.extractionStatus = yams::metadata::ExtractionStatus::Success;
    auto derived = yams::metadata::computePathDerivedValues(path);
    info.filePath = derived.normalizedPath;
    info.pathPrefix = derived.pathPrefix;
    info.reversePath = derived.reversePath;
    info.pathHash = derived.pathHash;
    info.parentHash = derived.parentHash;
    info.pathDepth = derived.pathDepth;
    return info;
}

struct StoredTopologyFixture {
    fs::path dataDir;
    std::string snapshotId;
    std::string firstClusterId;
};

std::string symbolNodeKey(const yams::metadata::SymbolMetadata& sym) {
    return sym.kind + ":" + sym.qualifiedName + "@" + sym.filePath;
}

yams::metadata::SymbolMetadata makeSymbol(const fs::path& path, const std::string& hash,
                                          const std::string& name, const std::string& qualifiedName,
                                          std::int32_t startLine, std::int32_t endLine) {
    yams::metadata::SymbolMetadata sym;
    sym.documentHash = hash;
    sym.filePath = path.string();
    sym.symbolName = name;
    sym.qualifiedName = qualifiedName;
    sym.kind = "function";
    sym.startLine = startLine;
    sym.endLine = endLine;
    return sym;
}

StoredTopologyFixture createStoredTopologyFixture(const fs::path& root) {
    using namespace yams::metadata;
    using namespace yams::topology;

    StoredTopologyFixture fixture;
    fixture.dataDir = root / "data";
    fs::create_directories(fixture.dataDir);
    const fs::path dbPath = fixture.dataDir / "yams.db";

    ConnectionPoolConfig poolConfig;
    poolConfig.minConnections = 1;
    poolConfig.maxConnections = 2;

    auto pool = std::make_unique<ConnectionPool>(dbPath.string(), poolConfig);
    REQUIRE(pool->initialize().has_value());

    auto repository = std::make_shared<MetadataRepository>(*pool);
    auto kgResult = makeSqliteKnowledgeGraphStore(*pool, KnowledgeGraphStoreConfig{});
    REQUIRE(kgResult.has_value());
    auto kgStore = std::shared_ptr<KnowledgeGraphStore>(kgResult.value().release());

    REQUIRE(repository->insertDocument(makeDocumentWithPath((root / "src/a.cpp").string(), "aaa"))
                .has_value());
    REQUIRE(repository->insertDocument(makeDocumentWithPath((root / "src/b.cpp").string(), "bbb"))
                .has_value());
    REQUIRE(
        repository->insertDocument(makeDocumentWithPath((root / "include/c.hpp").string(), "ccc"))
            .has_value());

    ConnectedComponentTopologyEngine engine;
    std::vector<TopologyDocumentInput> docs{
        TopologyDocumentInput{
            .documentHash = "aaa",
            .filePath = (root / "src/a.cpp").string(),
            .neighbors = {{.documentHash = "bbb", .score = 0.9F, .reciprocal = true}}},
        TopologyDocumentInput{
            .documentHash = "bbb",
            .filePath = (root / "src/b.cpp").string(),
            .neighbors = {{.documentHash = "aaa", .score = 0.9F, .reciprocal = true}}},
        TopologyDocumentInput{
            .documentHash = "ccc", .filePath = (root / "include/c.hpp").string(), .neighbors = {}},
    };
    auto batchResult = engine.buildArtifacts(docs, TopologyBuildConfig{});
    REQUIRE(batchResult.has_value());

    MetadataKgTopologyArtifactStore store(repository, kgStore);
    REQUIRE(store.storeBatch(batchResult.value()).has_value());

    fixture.snapshotId = batchResult.value().snapshotId;
    REQUIRE_FALSE(batchResult.value().clusters.empty());
    fixture.firstClusterId = batchResult.value().clusters.front().clusterId;

    kgStore.reset();
    repository.reset();
    pool->shutdown();
    pool.reset();
    return fixture;
}

void createGraphExploreFixture(const fs::path& root) {
    using namespace yams::metadata;

    const fs::path dataDir = root / "data";
    const fs::path sourceDir = root / "repo" / "src";
    const fs::path foreignSourceDir = root / "foreign" / "src";
    fs::create_directories(dataDir);
    fs::create_directories(sourceDir);
    fs::create_directories(foreignSourceDir);
    const fs::path sourcePath = sourceDir / "explore.cpp";
    const fs::path foreignSourcePath = foreignSourceDir / "explore.cpp";
    yams::test::write_file(sourcePath, "int exploreTarget() {\n"
                                       "    return 7;\n"
                                       "}\n"
                                       "int exploreEntry() {\n"
                                       "    return exploreTarget();\n"
                                       "}\n");
    yams::test::write_file(foreignSourcePath, "int exploreTarget() {\n"
                                              "    return 99;\n"
                                              "}\n"
                                              "int exploreEntry() {\n"
                                              "    return exploreTarget();\n"
                                              "}\n");

    const fs::path dbPath = dataDir / "yams.db";
    ConnectionPoolConfig poolConfig;
    poolConfig.minConnections = 1;
    poolConfig.maxConnections = 2;
    auto pool = std::make_unique<ConnectionPool>(dbPath.string(), poolConfig);
    REQUIRE(pool->initialize().has_value());

    auto repository = std::make_shared<MetadataRepository>(*pool);
    auto kgResult = makeSqliteKnowledgeGraphStore(*pool, KnowledgeGraphStoreConfig{});
    REQUIRE(kgResult.has_value());
    auto kgStore = std::shared_ptr<KnowledgeGraphStore>(kgResult.value().release());
    repository->setKnowledgeGraphStore(kgStore);

    REQUIRE(repository->insertDocument(makeDocumentWithPath(sourcePath.string(), "explore-hash"))
                .has_value());
    REQUIRE(repository
                ->insertDocument(
                    makeDocumentWithPath(foreignSourcePath.string(), "foreign-explore-hash"))
                .has_value());

    auto entry = makeSymbol(sourcePath, "explore-hash", "exploreEntry", "demo::exploreEntry", 4, 6);
    auto target =
        makeSymbol(sourcePath, "explore-hash", "exploreTarget", "demo::exploreTarget", 1, 3);
    auto foreignTarget = makeSymbol(foreignSourcePath, "foreign-explore-hash", "exploreTarget",
                                    "foreign::exploreTarget", 1, 3);
    auto foreignEntry = makeSymbol(foreignSourcePath, "foreign-explore-hash", "exploreEntry",
                                   "foreign::exploreEntry", 4, 6);
    REQUIRE(
        kgStore->upsertSymbolMetadata({entry, target, foreignEntry, foreignTarget}).has_value());

    KGNode entryNode;
    entryNode.nodeKey = symbolNodeKey(entry);
    entryNode.label = entry.symbolName;
    entryNode.type = entry.kind;
    const auto entryId = kgStore->upsertNode(entryNode);
    REQUIRE(entryId.has_value());

    KGNode targetNode;
    targetNode.nodeKey = symbolNodeKey(target);
    targetNode.label = target.symbolName;
    targetNode.type = target.kind;
    const auto targetId = kgStore->upsertNode(targetNode);
    REQUIRE(targetId.has_value());

    KGEdge edge;
    edge.srcNodeId = entryId.value();
    edge.dstNodeId = targetId.value();
    edge.relation = "call";
    edge.weight = 1.0F;
    REQUIRE(kgStore->addEdge(edge).has_value());

    KGNode foreignEntryNode;
    foreignEntryNode.nodeKey = symbolNodeKey(foreignEntry);
    foreignEntryNode.label = foreignEntry.symbolName;
    foreignEntryNode.type = foreignEntry.kind;
    const auto foreignEntryId = kgStore->upsertNode(foreignEntryNode);
    REQUIRE(foreignEntryId.has_value());

    KGNode foreignTargetNode;
    foreignTargetNode.nodeKey = symbolNodeKey(foreignTarget);
    foreignTargetNode.label = foreignTarget.symbolName;
    foreignTargetNode.type = foreignTarget.kind;
    const auto foreignTargetId = kgStore->upsertNode(foreignTargetNode);
    REQUIRE(foreignTargetId.has_value());

    KGEdge foreignEdge;
    foreignEdge.srcNodeId = foreignEntryId.value();
    foreignEdge.dstNodeId = foreignTargetId.value();
    foreignEdge.relation = "call";
    foreignEdge.weight = 1.0F;
    REQUIRE(kgStore->addEdge(foreignEdge).has_value());

    kgStore.reset();
    repository.reset();
    pool->shutdown();
    pool.reset();
}

} // namespace

TEST_CASE("IntegrationSmoke.GraphCommandFallsBackToInProcessWhenDaemonUnavailable",
          "[smoke][integrationsmoke]") {
    const fs::path root = yams::test::make_temp_dir("yams_graph_fallback_");
    const fs::path dataDir = root / "data";
    const fs::path blockedSocketDir = root / "blocked-socket";
    fs::create_directories(dataDir);
    fs::create_directories(blockedSocketDir);

    yams::test::ScopedEnvVar embedded("YAMS_EMBEDDED", std::nullopt);
    yams::test::ScopedEnvVar inDaemon("YAMS_IN_DAEMON", std::nullopt);
    yams::test::ScopedEnvVar dataEnv("YAMS_DATA_DIR", dataDir.string());
    yams::test::ScopedEnvVar storageEnv("YAMS_STORAGE", dataDir.string());
    yams::test::ScopedEnvVar disableVectors("YAMS_DISABLE_VECTORS", std::string("1"));
    yams::test::ScopedEnvVar skipModelLoading("YAMS_SKIP_MODEL_LOADING", std::string("1"));
    yams::test::ScopedEnvVar disableWatcher("YAMS_DISABLE_SESSION_WATCHER", std::string("1"));
    yams::test::ScopedEnvVar daemonSocket("YAMS_DAEMON_SOCKET",
                                          (blockedSocketDir / "daemon.sock").string());

    std::error_code ec;
    fs::permissions(blockedSocketDir, fs::perms::none, fs::perm_options::replace, ec);

    std::string out;
    const int rc = run_cli({"yams", "graph", "--list-types", "--json"}, &out);

    fs::permissions(blockedSocketDir, fs::perms::owner_all, fs::perm_options::replace, ec);

    INFO(out);
    CHECK((rc == 0));
    INFO(out);
    CHECK((out.find("Connection failed") == std::string::npos));
    INFO(out);
    CHECK((out.find("Operation not permitted") == std::string::npos));
}

TEST_CASE("IntegrationSmoke.GraphCommandRespectsForcedSocketMode", "[smoke][integrationsmoke]") {
    const fs::path root = yams::test::make_temp_dir("yams_graph_socket_forced_");
    const fs::path dataDir = root / "data";
    const fs::path pinnedSocket = root / "missing-socket" / "daemon.sock";
    fs::create_directories(dataDir);

    yams::test::ScopedEnvVar embedded("YAMS_EMBEDDED", std::string("0"));
    yams::test::ScopedEnvVar inDaemon("YAMS_IN_DAEMON", std::nullopt);
    yams::test::ScopedEnvVar dataEnv("YAMS_DATA_DIR", dataDir.string());
    yams::test::ScopedEnvVar storageEnv("YAMS_STORAGE", dataDir.string());
    yams::test::ScopedEnvVar disableVectors("YAMS_DISABLE_VECTORS", std::string("1"));
    yams::test::ScopedEnvVar skipModelLoading("YAMS_SKIP_MODEL_LOADING", std::string("1"));
    yams::test::ScopedEnvVar disableWatcher("YAMS_DISABLE_SESSION_WATCHER", std::string("1"));
    yams::test::ScopedEnvVar daemonSocket("YAMS_DAEMON_SOCKET", pinnedSocket.string());
    yams::test::ScopedEnvVar daemonSocketPath("YAMS_DAEMON_SOCKET_PATH", pinnedSocket.string());
    yams::test::ScopedEnvVar disableAutoStart("YAMS_CLI_DISABLE_DAEMON_AUTOSTART",
                                              std::string("1"));

    std::string out;
    const int rc = run_cli({"yams", "graph", "--list-types", "--json"}, &out);

    INFO(out);
    CHECK((rc != 0));
}

TEST_CASE("IntegrationSmoke.GraphExploreRendersAgentContext", "[smoke][integrationsmoke]") {
    const fs::path root = yams::test::make_temp_dir("yams_graph_explore_");
    createGraphExploreFixture(root);
    ScopedCurrentPath cwdGuard(root / "repo");

    yams::test::ScopedEnvVar embedded("YAMS_EMBEDDED", std::string("1"));
    yams::test::ScopedEnvVar inDaemon("YAMS_IN_DAEMON", std::nullopt);
    yams::test::ScopedEnvVar dataEnv("YAMS_DATA_DIR", (root / "data").string());
    yams::test::ScopedEnvVar storageEnv("YAMS_STORAGE", (root / "data").string());
    yams::test::ScopedEnvVar disableVectors("YAMS_DISABLE_VECTORS", std::string("1"));
    yams::test::ScopedEnvVar skipModelLoading("YAMS_SKIP_MODEL_LOADING", std::string("1"));
    yams::test::ScopedEnvVar disableWatcher("YAMS_DISABLE_SESSION_WATCHER", std::string("1"));

    std::string jsonOut;
    const int jsonRc = run_cli(
        {"yams", "graph", "--explore", "exploreEntry", "--max-files", "1", "--json"}, &jsonOut);
    INFO(jsonOut);
    REQUIRE((jsonRc == 0));
    auto parsed = nlohmann::json::parse(jsonOut);
    CHECK((parsed["query"] == "exploreEntry"));
    REQUIRE_FALSE(parsed["entrySymbols"].empty());
    CHECK((parsed["entrySymbols"][0]["label"] == "exploreEntry"));
    CHECK((parsed["entrySymbols"].size() == 1));
    CHECK((parsed["entrySymbols"][0]["qualifiedName"] == "demo::exploreEntry"));
    REQUIRE_FALSE(parsed["files"].empty());
    CHECK((parsed["files"][0]["content"].get<std::string>().find("4\tint exploreEntry()") !=
           std::string::npos));
    REQUIRE_FALSE(parsed["relationships"].empty());
    CHECK((parsed["relationships"][0]["relation"] == "calls"));

    std::string humanOut;
    const int humanRc =
        run_cli({"yams", "graph", "--explore", "exploreEntry", "--max-files", "1"}, &humanOut);
    INFO(humanOut);
    REQUIRE((humanRc == 0));
    CHECK((humanOut.find("Graph Explore") != std::string::npos));
    CHECK((humanOut.find("exploreEntry --calls--> exploreTarget") != std::string::npos));
    CHECK((humanOut.find("4\tint exploreEntry()") != std::string::npos));

    std::string globalOut;
    const int globalRc =
        run_cli({"yams", "graph", "--explore", "exploreEntry", "--global", "--json"}, &globalOut);
    INFO(globalOut);
    REQUIRE((globalRc == 0));
    const auto global = nlohmann::json::parse(globalOut);
    CHECK((global["entrySymbols"].size() == 2));

    std::string impactOut;
    const int impactRc =
        run_cli({"yams", "graph", "--impact", "exploreTarget", "--json"}, &impactOut);
    INFO(impactOut);
    REQUIRE((impactRc == 0));
    const auto impact = nlohmann::json::parse(impactOut);
    REQUIRE((impact["affectedSymbols"].size() == 1));
    CHECK((impact["affectedSymbols"][0]["filePath"] ==
           (root / "repo" / "src" / "explore.cpp").string()));

    std::string globalImpactOut;
    const int globalImpactRc = run_cli(
        {"yams", "graph", "--impact", "exploreTarget", "--global", "--json"}, &globalImpactOut);
    INFO(globalImpactOut);
    REQUIRE((globalImpactRc == 0));
    const auto globalImpact = nlohmann::json::parse(globalImpactOut);
    CHECK((globalImpact["affectedSymbols"].size() == 2));
}

TEST_CASE("IntegrationSmoke.GraphTopologyModesReadStoredSnapshot", "[smoke][integrationsmoke]") {
    const fs::path root = yams::test::make_temp_dir("yams_graph_topology_");
    const auto fixture = createStoredTopologyFixture(root);

    yams::test::ScopedEnvVar embedded("YAMS_EMBEDDED", std::string("1"));
    yams::test::ScopedEnvVar inDaemon("YAMS_IN_DAEMON", std::nullopt);
    yams::test::ScopedEnvVar dataEnv("YAMS_DATA_DIR", fixture.dataDir.string());
    yams::test::ScopedEnvVar storageEnv("YAMS_STORAGE", fixture.dataDir.string());
    yams::test::ScopedEnvVar disableVectors("YAMS_DISABLE_VECTORS", std::string("1"));
    yams::test::ScopedEnvVar skipModelLoading("YAMS_SKIP_MODEL_LOADING", std::string("1"));
    yams::test::ScopedEnvVar disableWatcher("YAMS_DISABLE_SESSION_WATCHER", std::string("1"));

    std::string snapshotOut;
    const int snapshotRc =
        run_cli({"yams", "graph", "--topology-snapshots", "--json"}, &snapshotOut);
    INFO(snapshotOut);
    REQUIRE((snapshotRc == 0));
    auto snapshotJson = nlohmann::json::parse(snapshotOut);
    CHECK((snapshotJson["snapshot"]["snapshot_id"] == fixture.snapshotId));
    CHECK((snapshotJson["snapshot"]["cluster_count"].get<std::size_t>() >= 1));

    std::string clustersOut;
    const int clustersRc =
        run_cli({"yams", "graph", "--topology-clusters", "--json"}, &clustersOut);
    INFO(clustersOut);
    REQUIRE((clustersRc == 0));
    auto clustersJson = nlohmann::json::parse(clustersOut);
    CHECK((clustersJson["snapshot_id"] == fixture.snapshotId));
    REQUIRE_FALSE(clustersJson["clusters"].empty());
    CHECK(clustersJson["clusters"][0].contains("role_summary"));
    CHECK(clustersJson["clusters"][0].contains("scoped_member_count"));

    std::string clusterOut;
    const int clusterRc =
        run_cli({"yams", "graph", "--cluster", fixture.firstClusterId, "--json"}, &clusterOut);
    INFO(clusterOut);
    REQUIRE((clusterRc == 0));
    auto clusterJson = nlohmann::json::parse(clusterOut);
    CHECK((clusterJson["snapshot_id"] == fixture.snapshotId));
    CHECK((clusterJson["cluster"]["cluster_id"] == fixture.firstClusterId));
    CHECK(clusterJson["cluster"].contains("role_summary"));
    CHECK(clusterJson["cluster"].contains("role_counts"));
    REQUIRE(clusterJson["members"].is_array());
    REQUIRE_FALSE(clusterJson["members"].empty());
}