@flakiness/sdk 0.95.0

package/lib/serverapi.js

@@ -0,0 +1,331 @@
+// ../server/lib/common/typedHttp.js
+var TypedHTTP;
+((TypedHTTP2) => {
+  TypedHTTP2.StatusCodes = {
+    Informational: {
+      CONTINUE: 100,
+      SWITCHING_PROTOCOLS: 101,
+      PROCESSING: 102,
+      EARLY_HINTS: 103
+    },
+    Success: {
+      OK: 200,
+      CREATED: 201,
+      ACCEPTED: 202,
+      NON_AUTHORITATIVE_INFORMATION: 203,
+      NO_CONTENT: 204,
+      RESET_CONTENT: 205,
+      PARTIAL_CONTENT: 206,
+      MULTI_STATUS: 207
+    },
+    Redirection: {
+      MULTIPLE_CHOICES: 300,
+      MOVED_PERMANENTLY: 301,
+      MOVED_TEMPORARILY: 302,
+      SEE_OTHER: 303,
+      NOT_MODIFIED: 304,
+      USE_PROXY: 305,
+      TEMPORARY_REDIRECT: 307,
+      PERMANENT_REDIRECT: 308
+    },
+    ClientErrors: {
+      BAD_REQUEST: 400,
+      UNAUTHORIZED: 401,
+      PAYMENT_REQUIRED: 402,
+      FORBIDDEN: 403,
+      NOT_FOUND: 404,
+      METHOD_NOT_ALLOWED: 405,
+      NOT_ACCEPTABLE: 406,
+      PROXY_AUTHENTICATION_REQUIRED: 407,
+      REQUEST_TIMEOUT: 408,
+      CONFLICT: 409,
+      GONE: 410,
+      LENGTH_REQUIRED: 411,
+      PRECONDITION_FAILED: 412,
+      REQUEST_TOO_LONG: 413,
+      REQUEST_URI_TOO_LONG: 414,
+      UNSUPPORTED_MEDIA_TYPE: 415,
+      REQUESTED_RANGE_NOT_SATISFIABLE: 416,
+      EXPECTATION_FAILED: 417,
+      IM_A_TEAPOT: 418,
+      INSUFFICIENT_SPACE_ON_RESOURCE: 419,
+      METHOD_FAILURE: 420,
+      MISDIRECTED_REQUEST: 421,
+      UNPROCESSABLE_ENTITY: 422,
+      LOCKED: 423,
+      FAILED_DEPENDENCY: 424,
+      UPGRADE_REQUIRED: 426,
+      PRECONDITION_REQUIRED: 428,
+      TOO_MANY_REQUESTS: 429,
+      REQUEST_HEADER_FIELDS_TOO_LARGE: 431,
+      UNAVAILABLE_FOR_LEGAL_REASONS: 451
+    },
+    ServerErrors: {
+      INTERNAL_SERVER_ERROR: 500,
+      NOT_IMPLEMENTED: 501,
+      BAD_GATEWAY: 502,
+      SERVICE_UNAVAILABLE: 503,
+      GATEWAY_TIMEOUT: 504,
+      HTTP_VERSION_NOT_SUPPORTED: 505,
+      INSUFFICIENT_STORAGE: 507,
+      NETWORK_AUTHENTICATION_REQUIRED: 511
+    }
+  };
+  const AllErrorCodes = {
+    ...TypedHTTP2.StatusCodes.ClientErrors,
+    ...TypedHTTP2.StatusCodes.ServerErrors
+  };
+  class HttpError extends Error {
+    constructor(status, message) {
+      super(message);
+      this.status = status;
+    }
+    static withCode(code, message) {
+      const statusCode = AllErrorCodes[code];
+      const defaultMessage = code.split("_").map((word) => word.charAt(0) + word.slice(1).toLowerCase()).join(" ");
+      return new HttpError(statusCode, message ?? defaultMessage);
+    }
+  }
+  TypedHTTP2.HttpError = HttpError;
+  function isInformationalResponse(response) {
+    return response.status >= 100 && response.status < 200;
+  }
+  TypedHTTP2.isInformationalResponse = isInformationalResponse;
+  function isSuccessResponse(response) {
+    return response.status >= 200 && response.status < 300;
+  }
+  TypedHTTP2.isSuccessResponse = isSuccessResponse;
+  function isRedirectResponse(response) {
+    return response.status >= 300 && response.status < 400;
+  }
+  TypedHTTP2.isRedirectResponse = isRedirectResponse;
+  function isErrorResponse(response) {
+    return response.status >= 400 && response.status < 600;
+  }
+  TypedHTTP2.isErrorResponse = isErrorResponse;
+  function info(status) {
+    return { status };
+  }
+  TypedHTTP2.info = info;
+  function ok(data, status) {
+    return {
+      status: status ?? TypedHTTP2.StatusCodes.Success.OK,
+      data
+    };
+  }
+  TypedHTTP2.ok = ok;
+  function redirect(url, status = 302) {
+    return { status, url };
+  }
+  TypedHTTP2.redirect = redirect;
+  function error(message, status = TypedHTTP2.StatusCodes.ServerErrors.INTERNAL_SERVER_ERROR) {
+    return { status, message };
+  }
+  TypedHTTP2.error = error;
+  class Router {
+    constructor(_resolveContext) {
+      this._resolveContext = _resolveContext;
+    }
+    static create() {
+      return new Router(async (e) => e.ctx);
+    }
+    rawMethod(method, route) {
+      return {
+        [method]: {
+          method,
+          input: route.input,
+          etag: route.etag,
+          resolveContext: this._resolveContext,
+          handler: route.handler
+        }
+      };
+    }
+    get(route) {
+      return this.rawMethod("GET", {
+        ...route,
+        handler: (...args) => Promise.resolve(route.handler(...args)).then((result) => TypedHTTP2.ok(result))
+      });
+    }
+    post(route) {
+      return this.rawMethod("POST", {
+        ...route,
+        handler: (...args) => Promise.resolve(route.handler(...args)).then((result) => TypedHTTP2.ok(result))
+      });
+    }
+    use(resolveContext) {
+      return new Router(async (options) => {
+        const m = await this._resolveContext(options);
+        return resolveContext({ ...options, ctx: m });
+      });
+    }
+  }
+  TypedHTTP2.Router = Router;
+  function createClient(base, fetchCallback) {
+    function buildUrl(path, input, options) {
+      const method = path.at(-1);
+      const url = new URL(path.slice(0, path.length - 1).join("/"), base);
+      const signal = options?.signal;
+      let body = void 0;
+      if (method === "GET" && input)
+        url.searchParams.set("input", JSON.stringify(input));
+      else if (method !== "GET" && input)
+        body = JSON.stringify(input);
+      return {
+        url,
+        method,
+        headers: body ? { "Content-Type": "application/json" } : void 0,
+        body,
+        signal
+      };
+    }
+    function createProxy(path = []) {
+      return new Proxy(() => {
+      }, {
+        get(target, prop) {
+          if (typeof prop === "symbol")
+            return void 0;
+          if (prop === "prepare")
+            return (input, options) => buildUrl(path, input, options);
+          const newPath = [...path, prop];
+          return createProxy(newPath);
+        },
+        apply(target, thisArg, args) {
+          const options = buildUrl(path, args[0], args[1]);
+          return fetchCallback(options.url, {
+            method: options.method,
+            body: options.body,
+            headers: options.headers,
+            signal: options.signal
+          }).then(async (response) => {
+            if (response.status >= 200 && response.status < 300) {
+              if (response.headers.get("content-type")?.includes("application/json")) {
+                const text = await response.text();
+                return text.length ? JSON.parse(text) : void 0;
+              }
+              return await response.blob();
+            }
+            if (response.status >= 400 && response.status < 600) {
+              const text = await response.text();
+              if (text)
+                throw new Error(`HTTP request failed with status ${response.status}: ${text}`);
+              else
+                throw new Error(`HTTP request failed with status ${response.status}`);
+            }
+          });
+        }
+      });
+    }
+    return createProxy();
+  }
+  TypedHTTP2.createClient = createClient;
+})(TypedHTTP || (TypedHTTP = {}));
+
+// src/utils.ts
+import http from "http";
+import https from "https";
+import util from "util";
+import zlib from "zlib";
+var gzipAsync = util.promisify(zlib.gzip);
+var gunzipAsync = util.promisify(zlib.gunzip);
+var gunzipSync = zlib.gunzipSync;
+var brotliCompressAsync = util.promisify(zlib.brotliCompress);
+var brotliCompressSync = zlib.brotliCompressSync;
+async function retryWithBackoff(job, backoff = []) {
+  for (const timeout of backoff) {
+    try {
+      return await job();
+    } catch (e) {
+      if (e instanceof AggregateError)
+        console.error(`[flakiness.io err]`, e.errors[0].message);
+      else if (e instanceof Error)
+        console.error(`[flakiness.io err]`, e.message);
+      else
+        console.error(`[flakiness.io err]`, e);
+      await new Promise((x) => setTimeout(x, timeout));
+    }
+  }
+  return await job();
+}
+var httpUtils;
+((httpUtils2) => {
+  function createRequest({ url, method = "get", headers = {} }) {
+    let resolve;
+    let reject;
+    const responseDataPromise = new Promise((a, b) => {
+      resolve = a;
+      reject = b;
+    });
+    const protocol = url.startsWith("https") ? https : http;
+    const request = protocol.request(url, { method, headers }, (res) => {
+      const chunks = [];
+      res.on("data", (chunk) => chunks.push(chunk));
+      res.on("end", () => {
+        if (res.statusCode && res.statusCode >= 200 && res.statusCode < 300)
+          resolve(Buffer.concat(chunks));
+        else
+          reject(new Error(`Request to ${url} failed with ${res.statusCode}`));
+      });
+      res.on("error", (error) => reject(error));
+    });
+    request.on("error", reject);
+    return { request, responseDataPromise };
+  }
+  httpUtils2.createRequest = createRequest;
+  async function getBuffer(url, backoff) {
+    return await retryWithBackoff(async () => {
+      const { request, responseDataPromise } = createRequest({ url });
+      request.end();
+      return await responseDataPromise;
+    }, backoff);
+  }
+  httpUtils2.getBuffer = getBuffer;
+  async function getText(url, backoff) {
+    const buffer = await getBuffer(url, backoff);
+    return buffer.toString("utf-8");
+  }
+  httpUtils2.getText = getText;
+  async function getJSON(url) {
+    return JSON.parse(await getText(url));
+  }
+  httpUtils2.getJSON = getJSON;
+  async function postText(url, text, backoff) {
+    const headers = {
+      "Content-Type": "application/json",
+      "Content-Length": Buffer.byteLength(text) + ""
+    };
+    return await retryWithBackoff(async () => {
+      const { request, responseDataPromise } = createRequest({ url, headers, method: "post" });
+      request.write(text);
+      request.end();
+      return await responseDataPromise;
+    }, backoff);
+  }
+  httpUtils2.postText = postText;
+  async function postJSON(url, json, backoff) {
+    const buffer = await postText(url, JSON.stringify(json), backoff);
+    return JSON.parse(buffer.toString("utf-8"));
+  }
+  httpUtils2.postJSON = postJSON;
+})(httpUtils || (httpUtils = {}));
+var ansiRegex = new RegExp("[\\u001B\\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[-a-zA-Z\\d\\/#&.:=?%@~_]*)*)?\\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PR-TZcf-ntqry=><~]))", "g");
+var IS_WIN32_PATH = new RegExp("^[a-zA-Z]:\\\\", "i");
+var IS_ALMOST_POSIX_PATH = new RegExp("^[a-zA-Z]:/", "i");
+
+// src/serverapi.ts
+function createServerAPI(endpoint, options) {
+  endpoint += "/api/";
+  const fetcher = options?.auth ? (url, init) => fetch(url, {
+    ...init,
+    headers: {
+      ...init.headers,
+      "Authorization": `Bearer ${options.auth}`
+    }
+  }) : fetch;
+  if (options?.retries)
+    return TypedHTTP.createClient(endpoint, (url, init) => retryWithBackoff(() => fetcher(url, init), options.retries));
+  return TypedHTTP.createClient(endpoint, fetcher);
+}
+export {
+  createServerAPI
+};
+//# sourceMappingURL=serverapi.js.map

package/lib/systemUtilizationSampler.js

@@ -0,0 +1,71 @@
+// src/systemUtilizationSampler.ts
+import { spawnSync } from "child_process";
+import os from "os";
+function getAvailableMemMacOS() {
+  const lines = spawnSync("vm_stat", { encoding: "utf8" }).stdout.trim().split("\n");
+  const pageSize = parseInt(lines[0].match(/page size of (\d+) bytes/)[1], 10);
+  if (isNaN(pageSize)) {
+    console.warn("[flakiness.io] Error detecting macos page size");
+    return 0;
+  }
+  let totalFree = 0;
+  for (const line of lines) {
+    if (/Pages (free|inactive|speculative):/.test(line)) {
+      const match = line.match(/\d+/);
+      if (match)
+        totalFree += parseInt(match[0], 10);
+    }
+  }
+  return totalFree * pageSize;
+}
+function getSystemUtilization() {
+  let idleTicks = 0;
+  let totalTicks = 0;
+  for (const cpu of os.cpus()) {
+    totalTicks += cpu.times.user + cpu.times.nice + cpu.times.sys + cpu.times.irq + cpu.times.idle;
+    idleTicks += cpu.times.idle;
+  }
+  return {
+    idleTicks,
+    totalTicks,
+    timestamp: Date.now(),
+    freeBytes: os.platform() === "darwin" ? getAvailableMemMacOS() : os.freemem()
+  };
+}
+function toFKUtilization(sample, previous) {
+  const idleTicks = sample.idleTicks - previous.idleTicks;
+  const totalTicks = sample.totalTicks - previous.totalTicks;
+  const cpuUtilization = Math.floor((1 - idleTicks / totalTicks) * 1e4) / 100;
+  const memoryUtilization = Math.floor((1 - sample.freeBytes / os.totalmem()) * 1e4) / 100;
+  return {
+    cpuUtilization,
+    memoryUtilization,
+    dts: sample.timestamp - previous.timestamp
+  };
+}
+var SystemUtilizationSampler = class {
+  result;
+  _lastSample = getSystemUtilization();
+  _timer;
+  constructor() {
+    this.result = {
+      samples: [],
+      startTimestamp: this._lastSample.timestamp,
+      totalMemoryBytes: os.totalmem()
+    };
+    this._timer = setTimeout(this._addSample.bind(this), 50);
+  }
+  _addSample() {
+    const sample = getSystemUtilization();
+    this.result.samples.push(toFKUtilization(sample, this._lastSample));
+    this._lastSample = sample;
+    this._timer = setTimeout(this._addSample.bind(this), 1e3);
+  }
+  dispose() {
+    clearTimeout(this._timer);
+  }
+};
+export {
+  SystemUtilizationSampler
+};
+//# sourceMappingURL=systemUtilizationSampler.js.map