Brunobkr/llama.cpp_AlgMor24_github

1"use strict";2var __create = Object.create;3var __defProp = Object.defineProperty;4var __getOwnPropDesc = Object.getOwnPropertyDescriptor;5var __getOwnPropNames = Object.getOwnPropertyNames;6var __getProtoOf = Object.getPrototypeOf;7var __hasOwnProp = Object.prototype.hasOwnProperty;8var __export = (target, all) => {9  for (var name in all)10    __defProp(target, name, { get: all[name], enumerable: true });11};12var __copyProps = (to, from, except, desc) => {13  if (from && typeof from === "object" || typeof from === "function") {14    for (let key of __getOwnPropNames(from))15      if (!__hasOwnProp.call(to, key) && key !== except)16        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });17  }18  return to;19};20var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(21  // If the importer is in node compatibility mode or this is not an ESM22  // file that has been converted to a CommonJS file using a Babel-23  // compatible transform (i.e. "__esModule" has not been set), then set24  // "default" to the CommonJS "module.exports" for node compatibility.25  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,26  mod27));28var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);29var http_exports = {};30__export(http_exports, {31  httpAddressToString: () => httpAddressToString,32  installHttpTransport: () => installHttpTransport,33  startHttpServer: () => startHttpServer34});35module.exports = __toCommonJS(http_exports);36var import_assert = __toESM(require("assert"));37var import_http = __toESM(require("http"));38var import_crypto = __toESM(require("crypto"));39var import_utilsBundle = require("playwright-core/lib/utilsBundle");40var mcpBundle = __toESM(require("./bundle"));41var mcpServer = __toESM(require("./server"));42const testDebug = (0, import_utilsBundle.debug)("pw:mcp:test");43async function startHttpServer(config, abortSignal) {44  const { host, port } = config;45  const httpServer = import_http.default.createServer();46  decorateServer(httpServer);47  await new Promise((resolve, reject) => {48    httpServer.on("error", reject);49    abortSignal?.addEventListener("abort", () => {50      httpServer.close();51      reject(new Error("Aborted"));52    });53    httpServer.listen(port, host, () => {54      resolve();55      httpServer.removeListener("error", reject);56    });57  });58  return httpServer;59}60function httpAddressToString(address) {61  (0, import_assert.default)(address, "Could not bind server socket");62  if (typeof address === "string")63    return address;64  const resolvedPort = address.port;65  let resolvedHost = address.family === "IPv4" ? address.address : `[${address.address}]`;66  if (resolvedHost === "0.0.0.0" || resolvedHost === "[::]")67    resolvedHost = "localhost";68  return `http://${resolvedHost}:${resolvedPort}`;69}70async function installHttpTransport(httpServer, serverBackendFactory, allowedHosts) {71  const url = httpAddressToString(httpServer.address());72  const host = new URL(url).host;73  allowedHosts = (allowedHosts || [host]).map((h) => h.toLowerCase());74  const allowAnyHost = allowedHosts.includes("*");75  const sseSessions = /* @__PURE__ */ new Map();76  const streamableSessions = /* @__PURE__ */ new Map();77  httpServer.on("request", async (req, res) => {78    if (!allowAnyHost) {79      const host2 = req.headers.host?.toLowerCase();80      if (!host2) {81        res.statusCode = 400;82        return res.end("Missing host");83      }84      if (!allowedHosts.includes(host2)) {85        res.statusCode = 403;86        return res.end("Access is only allowed at " + allowedHosts.join(", "));87      }88    }89    const url2 = new URL(`http://localhost${req.url}`);90    if (url2.pathname === "/killkillkill" && req.method === "GET") {91      res.statusCode = 200;92      res.end("Killing process");93      process.emit("SIGINT");94      return;95    }96    if (url2.pathname.startsWith("/sse"))97      await handleSSE(serverBackendFactory, req, res, url2, sseSessions);98    else99      await handleStreamable(serverBackendFactory, req, res, streamableSessions);100  });101}102async function handleSSE(serverBackendFactory, req, res, url, sessions) {103  if (req.method === "POST") {104    const sessionId = url.searchParams.get("sessionId");105    if (!sessionId) {106      res.statusCode = 400;107      return res.end("Missing sessionId");108    }109    const transport = sessions.get(sessionId);110    if (!transport) {111      res.statusCode = 404;112      return res.end("Session not found");113    }114    return await transport.handlePostMessage(req, res);115  } else if (req.method === "GET") {116    const transport = new mcpBundle.SSEServerTransport("/sse", res);117    sessions.set(transport.sessionId, transport);118    testDebug(`create SSE session: ${transport.sessionId}`);119    await mcpServer.connect(serverBackendFactory, transport, false);120    res.on("close", () => {121      testDebug(`delete SSE session: ${transport.sessionId}`);122      sessions.delete(transport.sessionId);123    });124    return;125  }126  res.statusCode = 405;127  res.end("Method not allowed");128}129async function handleStreamable(serverBackendFactory, req, res, sessions) {130  const sessionId = req.headers["mcp-session-id"];131  if (sessionId) {132    const transport = sessions.get(sessionId);133    if (!transport) {134      res.statusCode = 404;135      res.end("Session not found");136      return;137    }138    return await transport.handleRequest(req, res);139  }140  if (req.method === "POST") {141    const transport = new mcpBundle.StreamableHTTPServerTransport({142      sessionIdGenerator: () => import_crypto.default.randomUUID(),143      onsessioninitialized: async (sessionId2) => {144        testDebug(`create http session: ${transport.sessionId}`);145        await mcpServer.connect(serverBackendFactory, transport, true);146        sessions.set(sessionId2, transport);147      }148    });149    transport.onclose = () => {150      if (!transport.sessionId)151        return;152      sessions.delete(transport.sessionId);153      testDebug(`delete http session: ${transport.sessionId}`);154    };155    await transport.handleRequest(req, res);156    return;157  }158  res.statusCode = 400;159  res.end("Invalid request");160}161function decorateServer(server) {162  const sockets = /* @__PURE__ */ new Set();163  server.on("connection", (socket) => {164    sockets.add(socket);165    socket.once("close", () => sockets.delete(socket));166  });167  const close = server.close;168  server.close = (callback) => {169    for (const socket of sockets)170      socket.destroy();171    sockets.clear();172    return close.call(server, callback);173  };174}175// Annotate the CommonJS export names for ESM import in node:1760 && (module.exports = {177  httpAddressToString,178  installHttpTransport,179  startHttpServer180});181