huridocs/uwazi

import { ObjectIdSchema, PropertySchema } from 'shared/types/commonTypes';
import { generateID } from 'shared/IDGenerator';
import { propertyTypes } from 'shared/propertyTypes';
//@ts-ignore
import PromisePool from '@supercharge/promise-pool';
import model from './entitiesModel';

const updateRecursively = async (
  templateId: ObjectIdSchema,
  generatedIdProperties: PropertySchema[],
  searchQuery: {}
): Promise<void> => {
  const batchSize = 1000;
  const sharedIds = (
    await model.db.aggregate([
      { $match: { $and: [{ template: templateId }, { ...searchQuery }] } },
      { $group: { _id: '$sharedId' } },
      { $limit: batchSize },
    ])
  ).map(g => g._id);

  if (sharedIds.length === 0) {
    return Promise.resolve();
  }

  await Promise.all(
    sharedIds.map(async (sharedId: string) =>
      model.updateMany(
        { sharedId },
        {
          ...generatedIdProperties.reduce(
            (values, property: PropertySchema) => ({
              ...{ ...values, [`metadata.${property.name}`]: [{ value: generateID(3, 4, 4) }] },
            }),
            {}
          ),
        }
      )
    )
  );

  if (sharedIds.length === batchSize) {
    return updateRecursively(templateId, generatedIdProperties, searchQuery);
  }
  return Promise.resolve();
};

const populateGeneratedIdByTemplate = async (
  templateId: ObjectIdSchema,
  properties: PropertySchema[]
) => {
  const generatedIdProperties = properties.filter(prop => prop.type === propertyTypes.generatedid);
  const searchQuery = generatedIdProperties.reduce(
    (values, property: PropertySchema) => ({
      ...{ ...values, [`metadata.${property.name}`]: { $exists: false } },
    }),
    {}
  );

  const stepSize = 5000;
  const entitiesQty = await model.count({ template: templateId });
  const steps = Array(Math.floor(entitiesQty / stepSize) || 1);
  await new PromisePool()
    .for(steps)
    .withConcurrency(Math.min(5, steps.length))
    .process(async () => {
      await updateRecursively(templateId, generatedIdProperties, searchQuery);
    });
};

export { populateGeneratedIdByTemplate };