Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

enhancement: migrate bulk publish & unpublish to v5 #20184

Merged
merged 16 commits into from May 7, 2024
Merged
Show file tree
Hide file tree
Changes from 12 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
Expand Up @@ -411,7 +411,7 @@ export default {
const { userAbility } = ctx.state;
const { model } = ctx.params;
const { body } = ctx.request;
const { ids } = body;
const { documentIds } = body;

await validateBulkActionInput(body);

Expand All @@ -422,36 +422,34 @@ export default {
return ctx.forbidden();
}

const permissionQuery = await permissionChecker.sanitizedQuery.publish(ctx.query);
const populate = await getService('populate-builder')(model)
.populateFromQuery(permissionQuery)
.populateDeep(Infinity)
.countRelations()
.build();
const { locale } = getDocumentLocaleAndStatus(body);

const entityPromises = ids.map((id: any) => documentManager.findOne(id, model, { populate }));
const entities = await Promise.all(entityPromises);
const documentLocales = await documentManager.findLocales(documentIds[0], model, {
locale,
isPublished: true,
});

for (const entity of entities) {
if (!entity) {
return ctx.notFound();
}
if (documentLocales.length === 0) {
return ctx.notFound();
}

if (permissionChecker.cannot.publish(entity)) {
for (const document of documentLocales) {
if (permissionChecker.cannot.publish(document)) {
return ctx.forbidden();
}
}

Feranchz marked this conversation as resolved.
Show resolved Hide resolved
// @ts-expect-error - publish many should not return null
const { count } = await documentManager.publishMany(entities, model);
const localeDocumentsIds = documentLocales.map((document) => document.documentId);

const { count } = await documentManager.publishMany(localeDocumentsIds, model, { locale });
ctx.body = { count };
},

async bulkUnpublish(ctx: any) {
const { userAbility } = ctx.state;
const { model } = ctx.params;
const { body } = ctx.request;
const { ids } = body;
const { documentIds } = body;

await validateBulkActionInput(body);

Expand All @@ -462,26 +460,27 @@ export default {
return ctx.forbidden();
}

const permissionQuery = await permissionChecker.sanitizedQuery.publish(ctx.query);
const populate = await getService('populate-builder')(model)
.populateFromQuery(permissionQuery)
.build();
const { locale } = getDocumentLocaleAndStatus(body);

const entityPromises = ids.map((id: any) => documentManager.findOne(id, model, { populate }));
const entities = await Promise.all(entityPromises);
const documentLocales = await documentManager.findLocales(documentIds, model, {
locale,
isPublished: true,
});

for (const entity of entities) {
if (!entity) {
return ctx.notFound();
}
if (documentLocales.length === 0) {
return ctx.notFound();
}

if (permissionChecker.cannot.publish(entity)) {
for (const document of documentLocales) {
if (permissionChecker.cannot.unpublish(document)) {
return ctx.forbidden();
}
}

// @ts-expect-error - unpublish many should not return null
const { count } = await documentManager.unpublishMany(entities, model);
const localeDocumentsIds = documentLocales.map((document) => document.documentId);

const { count } = await documentManager.unpublishMany(localeDocumentsIds, model, { locale });

ctx.body = { count };
},

Expand Down Expand Up @@ -587,7 +586,7 @@ export default {
const { userAbility } = ctx.state;
const { model } = ctx.params;
const { query, body } = ctx.request;
const { ids } = body;
const { documentIds } = body;

await validateBulkActionInput(body);

Expand All @@ -598,18 +597,32 @@ export default {
return ctx.forbidden();
}

// TODO: fix
const permissionQuery = await permissionChecker.sanitizedQuery.delete(query);
const populate = await getService('populate-builder')(model)
.populateFromQuery(permissionQuery)
.build();

const idsWhereClause = { id: { $in: ids } };
const params = {
...permissionQuery,
filters: {
$and: [idsWhereClause].concat(permissionQuery.filters || []),
},
};
const { locale } = getDocumentLocaleAndStatus(body);

const documentLocales = await documentManager.findLocales(documentIds, model, {
populate,
locale,
});

if (documentLocales.length === 0) {
return ctx.notFound();
}

for (const document of documentLocales) {
if (permissionChecker.cannot.delete(document)) {
return ctx.forbidden();
}
}

// We filter out documentsIds that maybe doesn't exist in a specific locale
const localeDocumentsIds = documentLocales.map((document) => document.documentId);

const { count } = await documentManager.deleteMany(params, model);
const { count } = await documentManager.deleteMany(localeDocumentsIds, model, { locale });

ctx.body = { count };
},
Expand Down Expand Up @@ -650,8 +663,7 @@ export default {

async countManyEntriesDraftRelations(ctx: any) {
const { userAbility } = ctx.state;
const ids = ctx.request.query.ids as any;
const locale = ctx.request.query.locale;
const { documentIds, locale } = ctx.request.query;
const { model } = ctx.params;

const documentManager = getService('document-manager');
Expand All @@ -661,21 +673,21 @@ export default {
return ctx.forbidden();
}

const entities = await documentManager.findMany(
const documents = await documentManager.findMany(
{
filters: {
id: ids,
documentId: { $in: documentIds },
},
locale,
},
model
);

if (!entities) {
if (!documents) {
return ctx.notFound();
}

const number = await documentManager.countManyEntriesDraftRelations(ids, model, locale);
const number = await documentManager.countManyEntriesDraftRelations(documentIds, model, locale);

return {
data: number,
Expand Down
Expand Up @@ -12,7 +12,7 @@ const kindSchema = yup.string().oneOf(TYPES).nullable();

const bulkActionInputSchema = yup
.object({
ids: yup.array().of(yup.strapiID()).min(1).required(),
documentIds: yup.array().of(yup.strapiID()).min(1).required(),
})
.required();

Expand Down