A11yWatch/a11ywatch-core

import { issuesCollection } from "../../../../database";
import { getHostName, websiteSearchParams } from "../../../utils";
import type { Issue } from "../../../../types/schema";

export const getIssue = async (
  { url, pageUrl, userId, noRetries }: any,
  chain?: boolean
) => {
  const queryUrl = decodeURIComponent(String(url || pageUrl));

  const searchProps = websiteSearchParams({
    pageUrl: queryUrl,
    userId,
  });

  // todo: set default type
  let issue = null;

  // TODO: remove props and allow all
  if (Object.keys(searchProps).length) {
    issue = await issuesCollection.findOne(searchProps);

    // get issues from general bucket [marketing]
    if (!issue && !noRetries) {
      issue = await issuesCollection.findOne({ pageUrl: queryUrl });
      if (!issue) {
        issue = await issuesCollection.findOne({
          domain: getHostName(queryUrl),
        });
      }
    }
  }

  return chain ? [issue, issuesCollection] : issue;
};

// query issue collection by limit
export const getIssues = async (
  {
    userId,
    domain,
    pageUrl,
  }: { userId: number; domain?: string; pageUrl?: string },
  limit: number = 2000
) => {
  const searchProps = websiteSearchParams({
    domain: domain || getHostName(pageUrl),
    userId,
  });

  return await issuesCollection
    .find(searchProps)
    .sort({ pageUrl: 1 })
    .limit(limit)
    .toArray();
};

// get issues for a user with pagination offsets.
export const getIssuesPaging = async (params) => {
  const { userId, domain, pageUrl, limit = 10, offset = 0, all } = params ?? {};

  const searchParams = websiteSearchParams({
    domain: domain || getHostName(pageUrl),
    userId,
    all,
  });

  return (await issuesCollection
    .find(searchParams)
    .skip(offset)
    .limit(limit)
    .toArray()) as Issue[];
};