script.ts

import { request, gql } from 'graphql-request';
import { customAlphabet } from 'nanoid';
import * as retry from 'async-await-retry';
import * as puppeteer from 'puppeteer';
import * as fs from 'fs';

import config from './config';

const invite: string = process.argv[2] || config.invite;

const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz0123456789', 12);

const regex: RegExp = /[\d]{6}/gm;
const query = gql`
  query getEmails($username: String!) {
    emails(username: $username) {
      token
      subject
      sender
      timestamp
      stripped_text
    }
  }
`;

const makeStepScreenshot = () => {
  var i = 0;
  async function inc(page: any) {
    return await page.screenshot({ path: config.screenshotsDir + '/' + i++ + '.png' });
  }
  return inc;
};

const run = async () => {
  if (!fs.existsSync(config.screenshotsDir)) {
    fs.mkdirSync(config.screenshotsDir);
  }

  console.log('invite:', invite, '\n');

  for (let i = 1; i <= config.runs; i++) {
    const username: string = nanoid();
    const screenshot = makeStepScreenshot();

    console.log(`[${i}] Starting run`);

    console.log(`[${i}] Email:`, username + '@emaildrop.io');

    const browser = await puppeteer.launch();
    const page = await browser.newPage();
    await page.goto('https://treasure.cloud', { waitUntil: 'load', timeout: 0 });
    await page.goto('https://app.treasure.cloud/auth/signup?code=' + invite, { waitUntil: 'load', timeout: 0 });
    await page.goto('https://app.treasure.cloud/auth/signup?code=' + invite, { waitUntil: 'load', timeout: 0 });

    await screenshot(page);

    await page.type('#mat-input-0', `${username}@emaildrop.io`);
    await page.type('#mat-input-1', username);
    await page.type('#mat-input-2', config.password);
    await page.type('#mat-input-3', config.password);

    await screenshot(page);

    await page.keyboard.press('Enter');

    console.log(`[${i}] Submited form`);

    await page.waitForNavigation({ waitUntil: 'load', timeout: 0 });

    await page.waitForTimeout(10000);

    await screenshot(page);

    const getOTP = async () => {
      const data: any = await request('https://api.emaildrop.io/graphql', query, { username });

      if (data.emails.length == 0) {
        return await getOTP();
      } else {
        return regex.exec(data.emails[0].stripped_text.replaceAll(/\s/g, ''))[0];
      }
    };

    const otp = await retry(getOTP);
    console.log(`[${i}] Obtained OTP:`, otp);

    const nums = otp.split('');
    await page.type(`#verify-code-form > div:nth-child(1) > div:nth-child(1) > input`, nums[0]);
    await page.type(`#verify-code-form > div:nth-child(1) > div:nth-child(2) > input`, nums[1]);
    await page.type(`#verify-code-form > div:nth-child(1) > div:nth-child(3) > input`, nums[2]);
    await page.type(`#verify-code-form > div:nth-child(2) > div:nth-child(1) > input`, nums[3]);
    await page.type(`#verify-code-form > div:nth-child(2) > div:nth-child(2) > input`, nums[4]);
    await page.type(`#verify-code-form > div:nth-child(2) > div:nth-child(3) > input`, nums[5]);

    await page.waitForTimeout(2000);

    await screenshot(page);

    await page.waitForNavigation({ waitUntil: 'load', timeout: 0 });

    await screenshot(page);

    await page.waitForTimeout(5000);

    await browser.close();

    console.log(`[${i}] Run completed successfully!\n`);
  }
};

run();