index.js

#!/usr/bin/env node
import { load } from 'cheerio'
import axios from 'axios'
import { parse } from 'acorn';
import { simple } from 'acorn-walk'
import { performance } from 'node:perf_hooks'
import ms from 'ms'
import chalk from 'chalk'
import clear from 'console-clear';
import { address, mac, dns } from 'address/promises';
import si from 'systeminformation';


async function systemStats() {
  const cpu = await si.cpu();
  const os = await si.osInfo();
  const wifi = await si.wifiConnections()
  const gateway = await si.networkGatewayDefault()

  let system = {
    manufacturer: cpu.manufacturer,
    brand: cpu.brand,
    vendor: cpu.vendor,
    model: cpu.model,
    speed: cpu.speed,
    platform: os.platform,
    distro: os.distro,
    release: os.release,
    osName: os.codename,
    hostname: os.hostname,
    arch: os.arch,
  }
  let response = {
    device: system,
    wifi: wifi,
    gateway,
  }

  return response
}

async function parseToken(scriptPath) {
  const { data: script } = await axios.get(scriptPath, { responseType: 'text' });
  const result = parse(script, {
    allowHashBang: true,
    ecmaVersion: 2022
  });

  return new Promise((resolve) => {
    let token = null;
    simple(result, {
      Property(node) {
        if (token != null) return;
        if (node.key.name === "token") {
          token = node.value.value;
        }
      },
    });
    resolve(token)
  })
}


clear(true)
console.log('')
console.log(chalk.red.bold("ETHGlobal Speed Test"))
console.log('')
console.log(chalk.green('Starting..'));
console.log('')
const { data } = await axios.get("https://fast.com/", { responseType: 'text' });
const $ = load(data);
const token = await parseToken(new URL($("script[src]").first().attr('src'), "https://fast.com/"));

clear(true)
console.log('')
console.log(chalk.red.bold("ETHGlobal Speed Test"))
console.log('')
console.log(chalk.green('Initializing..'));
console.log('')
const { data: { client: _client, targets } } = await axios.get(`https://api.fast.com/netflix/speedtest/v2?https=true&token=${token}&urlCount=5`);

let bytes = 0;
const avg = [];

const controller = new AbortController();

let startedAt = Date.now();
let elapsed = Date.now();

const responseTimeSamples = [];

const payload = {}
const history = []

let tick = 0;
let lastReport = {};
const spinner = '◐◓◑◒'.split('')

await Promise.all(
  targets.map(async ({ url }) => {
    try {
      const { data: stream } = await axios.get(url, { responseType: 'stream', signal: controller.signal });
      let responseTime = performance.now();
  
      stream.on('data', buffer => {
  
        bytes += buffer.length;
  
        responseTime = performance.now() - responseTime;
        responseTimeSamples.push(responseTime)
        responseTime = performance.now()
  
        if (Date.now() - startedAt > 100) {
          tick++;
          avg.push(bytes * 8 * 10);
          const averageBits = avg.reduce((a, b) => a + b, 0) / avg.length;
          const latency = ms(Math.round(responseTimeSamples.reduce((a, b) => a + b) / responseTimeSamples.length), { long: true })
          const spin = offset => chalk.dim(`${spinner[(tick + offset) % spinner.length]}`);
          clear(true)
  
          const timeElapsed = ms(Date.now() - elapsed);
  
          console.log('')
          console.log(chalk.red.bold("ETHGlobal Speed Test"))
          console.log('')
          console.log(spin(4) + chalk.yellow(' Speed  \t') + Math.round(averageBits / 1000000) + " Mbps ");
          console.log(spin(3) + chalk.yellow(' Latency\t') + latency);
          console.log(spin(2) + chalk.yellow(' Elapsed\t') + timeElapsed);
          
          lastReport = {
            averageBits,
            latency,
            timeElapsed,
            ts: Math.round((new Date()).getTime()/1000),
          }

          if (Math.random() > 0.9) {
            history.push(lastReport)
          }

          bytes = 0;
          startedAt += 100;
        }
      });
      stream.on('end', async () => {
        controller.abort();
        const {timeElapsed, latency, averageBits}  = lastReport;
        const _addr = await address();
        const _mac = await mac();
        const _dns = await dns();
        const stats = await systemStats();
        payload["summary"] = {
          speed: Math.round(averageBits / 1000)/1000,
          isp: _client.isp,
          latency,
          ip: _addr.ip,
          mac: _mac,
          duration: timeElapsed,
          city: _client.location.city,
          country: _client.location.country,
          publicIP: _client.ip,
          ...stats,
          ipv6: _addr.ipv6,
          bits: averageBits,
          asn: _client.asn,
          dns: _dns,
          time: (new Date()).toISOString(),
          timestamp: Math.round((new Date()).getTime()/1000),
        }
        payload["raw"] = history
        const spin = () => chalk.green("✔");
        clear(true)
        console.log('')
        console.log(chalk.red.bold("ETHGlobal Speed Test"))
        console.log('')
        console.log(spin(4) + chalk.green(' Speed  \t') + Math.round(averageBits / 1000000) + " Mbps ");
        console.log(spin(3) + chalk.green(' Latency\t') + latency);
        console.log(spin(2) + chalk.green(' Elapsed\t') + timeElapsed);
        console.log('')
        console.log(chalk.magenta(' IP\t') + _addr.ip);
        console.log(chalk.magenta(' Mac\t') + _mac);
        console.log('')
        console.log(chalk.green.bold(' Your internet speed is ' + Math.round(averageBits / 10000)/100 + " Mbps on " + (new Date()).toUTCString() + ' '));
        console.log('')

        const postUpdate = await fetch('https://speed.ethglobal.com/api/create', {
          method: 'POST',
          headers: {'Content-Type': 'application/json'},
          body: JSON.stringify(payload)
        })
        const postUpdateJson = await postUpdate.json()

        console.log(chalk.redBright(' Link: ') + 'https://speed.ethglobal.com/run/' + postUpdateJson.key);
        console.log('')
        // console.log([history.length, payload])
      })
    } catch(e) {
      if (!axios.isAxiosError(e)) {
        throw e;
      }
    }
  })
)