sitespeed.io/lib/api/send.js

sitespeed.io is an open-source tool for comprehensive web performance analysis, enabling you to test, monitor, and optimize your website’s speed using real browsers in various environments.

import http from 'node:http';
import https from 'node:https';

const delay = ms => new Promise(res => setTimeout(res, ms));

export async function addTest(hostname, options) {
  const port = options.api.port ?? 3000;
  const postData = options;
  const postOptions = {
    hostname: hostname,
    port: port,
    path: '/api/add',
    method: 'POST',
    headers: {
      'Content-Type': 'application/json',
      'Content-Length': Buffer.byteLength(JSON.stringify(postData))
    }
  };

  return new Promise((resolve, reject) => {
    // not perfect but maybe work for us
    const library = options.api.port === 443 ? https : http;
    const request = library.request(postOptions, res => {
      let data = [];
      res.on('data', chunk => data.push(chunk));
      res.on('end', () => {
        if (res.statusCode === 200) {
          return resolve(Buffer.concat(data).toString('utf8'));
        } else {
          try {
            return reject(JSON.parse(Buffer.concat(data).toString('utf8')));
          } catch {
            return reject({
              message:
                'Could parse result:' + Buffer.concat(data).toString('utf8')
            });
          }
        }
      });
    });
    request.on('error', error => {
      reject({ message: error.toString() });
    });
    request.write(JSON.stringify(postData));
    request.end();
  });
}

export async function waitAndGetResult(testId, hostname, options, spinner) {
  do {
    await delay(2000);
    const response = await get(testId, hostname, options);
    spinner.text = response.message;
    if (response.status === 'completed') {
      return response;
    } else if (response.status === 'failed') {
      return response;
    }
    // eslint-disable-next-line no-constant-condition
  } while (true);
}

export async function get(testId, hostname, options) {
  return new Promise((resolve, reject) => {
    const port = options.api.port ?? 3000;
    const library = port === 443 ? https : http;

    const url = `${port === 443 ? 'https' : 'http'}://${hostname}${
      port != 80 && port != 443 ? `:${port}` : ''
    }/api/status/${testId}`;

    library
      .get(url, res => {
        let data = [];
        res.on('data', chunk => {
          data.push(chunk);
        });

        res.on('end', () => {
          if (res.statusCode === 200) {
            return resolve(JSON.parse(Buffer.concat(data).toString('utf8')));
          } else {
            try {
              return reject(JSON.parse(Buffer.concat(data).toString('utf8')));
            } catch {
              return reject({
                message:
                  'Could parse result:' + Buffer.concat(data).toString('utf8')
              });
            }
          }
        });
      })
      .on('error', () => {
        return reject();
      });
  });
}