web-check/api/legacy-rank.js

import axios from 'axios';
import unzipper from 'unzipper';
import csv from 'csv-parser';
import fs from 'fs';
import middleware from './_common/middleware.js';

// Should also work with the following sources:
// https://www.domcop.com/files/top/top10milliondomains.csv.zip
// https://tranco-list.eu/top-1m.csv.zip
// https://www.domcop.com/files/top/top10milliondomains.csv.zip
// https://radar.cloudflare.com/charts/LargerTopDomainsTable/attachment?id=525&top=1000000
// https://statvoo.com/dl/top-1million-sites.csv.zip

const FILE_URL = 'https://s3-us-west-1.amazonaws.com/umbrella-static/top-1m.csv.zip';
const TEMP_FILE_PATH = '/tmp/top-1m.csv';

const rankHandler = async (url) => {
  let domain = null;

  try {
    domain = new URL(url).hostname;
  } catch (e) {
    throw new Error('Invalid URL');
  }

// Download and unzip the file if not in cache
if (!fs.existsSync(TEMP_FILE_PATH)) {
  const response = await axios({
    method: 'GET',
    url: FILE_URL,
    responseType: 'stream'
  });

  await new Promise((resolve, reject) => {
    response.data
      .pipe(unzipper.Extract({ path: '/tmp' }))
      .on('close', resolve)
      .on('error', reject);
  });
}

// Parse the CSV and find the rank
return new Promise((resolve, reject) => {
  const csvStream = fs.createReadStream(TEMP_FILE_PATH)
    .pipe(csv({
      headers: ['rank', 'domain'],
    }))
    .on('data', (row) => {
      if (row.domain === domain) {
        csvStream.destroy();
        resolve({
          domain: domain,
          rank: row.rank,
          isFound: true,
        });
      }
    })
    .on('end', () => {
      resolve({
        skipped: `Skipping, as ${domain} is not present in the Umbrella top 1M list.`,
        domain: domain,
        isFound: false,
      });
    })
    .on('error', reject);
});
};

export const handler = middleware(rankHandler);
export default handler;