A11yWatch/a11ywatch-core

import {
  pageSpeedCollection,
  analyticsCollection,
  pagesCollection,
  issuesCollection,
  actionsCollection,
  websitesCollection,
} from "../../../../database";
import {
  WEBSITE_NOT_FOUND,
  SUCCESS,
  SUCCESS_DELETED_ALL,
} from "../../../strings";
import { getWebsite } from "../find";
import { HistoryController } from "../../history";
import { domainNameFind } from "../../../utils";
import { validateUID } from "../../../../web/params/extracter";

// remove a website or all website and related data.
export const removeWebsite = async ({
  userId,
  url = "",
  domain = "",
  deleteMany = false,
}) => {
  if (deleteMany) {
    // todo: get all websites and send request to cdn server for assets removal
    await websitesCollection.deleteMany({ userId });
    await analyticsCollection.deleteMany({ userId });
    await pagesCollection.deleteMany({ userId });
    await issuesCollection.deleteMany({ userId });
    await actionsCollection.deleteMany({ userId });
    await pageSpeedCollection.deleteMany({ userId });

    return { code: 200, success: true, message: SUCCESS_DELETED_ALL };
  }

  const [siteExist, collection] = await getWebsite({ userId, url, domain });

  if (!siteExist) {
    throw new Error(WEBSITE_NOT_FOUND);
  }

  const removeRelative = siteExist.subdomains || siteExist.tld;

  const baseRemoveQuery = validateUID(userId)
    ? { domain: siteExist.domain, userId }
    : { domain: siteExist.domain };

  const [history, historyCollection] = await HistoryController().getHistoryItem(
    baseRemoveQuery
  );

  let deleteQuery: Record<string, unknown> = baseRemoveQuery;

  if (removeRelative) {
    deleteQuery = domainNameFind({ userId }, siteExist.domain);
  }

  await analyticsCollection.deleteMany(deleteQuery);
  await pagesCollection.deleteMany(deleteQuery);
  await issuesCollection.deleteMany(deleteQuery);
  await actionsCollection.deleteMany(deleteQuery);
  await pageSpeedCollection.deleteMany(deleteQuery);
  await collection.findOneAndDelete(deleteQuery);

  // PREVENT DUPLICATE ITEMS IN HISTORY
  if (!history) {
    if ((await historyCollection.countDocuments({ userId })) >= 20) {
      await historyCollection.deleteOne({ userId });
    }
    await historyCollection.insertOne({
      ...siteExist,
      deletedDate: new Date(),
    });
  }

  return { website: siteExist, code: 200, success: true, message: SUCCESS };
};