add get-apos-doc-id and set-apos-doc-id tasks

CHANGELOG.md

@@ -5,6 +5,7 @@
 ### Adds
 
 * Translation strings added for the layout- and layout-column-widgets.
+* Adds `@apostrophecms/doc:get-apos-doc-id` and `@apostrophecms/doc:set-apos-doc-id` tasks.
 * When switching locale from the doc editor, ask if the user wants to localize the current document in the target locale or want to start a blank document.
 
 ### Changes

modules/@apostrophecms/doc/index.js

@@ -2,6 +2,8 @@ const _ = require('lodash');
 const { createId } = require('@paralleldrive/cuid2');
 const { SemanticAttributes } = require('@opentelemetry/semantic-conventions');
 const { klona } = require('klona');
+const legacyMigrations = require('./lib/legacy-migrations.js');
+const migrations = require('./lib/migrations.js');
 
 // This module is responsible for managing all of the documents (apostrophe
 // "docs") in the `aposDocs` mongodb collection.
@@ -1843,8 +1845,130 @@ module.exports = {
         return existing?.aposDocId || self.apos.util.generateId();
       },
 
-      ...require('./lib/legacy-migrations')(self),
-      ...require('./lib/migrations')(self)
+      async getAposDocId({
+        _id, slug, locale
+      }) {
+        if (!_id && !slug) {
+          throw self.apos.error('invalid', 'Either _id or slug must be provided');
+        }
+        if (!locale) {
+          throw self.apos.error('invalid', 'Missing locale');
+        }
+
+        // const req = self.apos.task.getReq({ mode: 'draft', locale });
+        const aposLocale = new RegExp(`^${self.apos.util.regExpQuote(locale)}:`);
+        const criteria = _id
+          ? {
+            _id,
+            aposLocale
+          }
+          : {
+            slug,
+            aposLocale
+          };
+
+        const doc = await self.apos.doc.db
+          .findOne(criteria, { projection: { aposDocId: 1 } });
+        if (!doc || !doc.aposDocId) {
+          throw self.apos.error('notfound');
+        }
+
+        return doc.aposDocId;
+      },
+      async setAposDocId({
+        newId, oldId, slug, locale
+      }) {
+        if (!newId) {
+          throw self.apos.error('invalid', 'Missing newId');
+        }
+        if (!oldId && !slug) {
+          throw self.apos.error('invalid', 'Either oldId or slug must be provided');
+        }
+        if (!locale) {
+          throw self.apos.error('invalid', 'Missing locale');
+        }
+
+        const originalId = (slug && !oldId)
+          ? await self.getAposDocId({
+            slug,
+            locale
+          })
+          : oldId;
+
+        const modes = [ 'previous', 'draft', 'published' ];
+        const pairs = modes.map(mode =>
+          [
+            `${originalId}:${locale}:${mode}`,
+            `${newId}:${locale}:${mode}`
+          ]
+        );
+
+        // Filter non existing from _id from the list
+        const existing = (
+          await self.apos.doc.db
+            .find(
+              { _id: { $in: pairs.map(([ from ]) => from) } },
+              { projection: { _id: 1 } }
+            )
+            .toArray()
+        )
+          .map(({ _id }) => _id);
+        const { renamed } = await self.changeDocIds(
+          pairs.filter(([ from ]) => existing.includes(from)),
+          { keep: false }
+        );
+
+        return {
+          oldId: originalId,
+          newId,
+          locale,
+          renamed
+        };
+      },
+
+      ...legacyMigrations(self),
+      ...migrations(self)
+    };
+  },
+  tasks(self) {
+    return {
+      'get-apos-doc-id': {
+        usage: 'Retrieve the aposDocId from the _id or slug, and the locale. Usage: "node app.js @apostrophecms/doc:get-apos-doc-id --_id=pfh0haxfpzowht3oi213cqos:fr:draft --locale=fr" or "node app.js @apostrophecms/doc:get-apos-doc-id --slug=test --locale=fr"',
+        task: async (argv) => {
+          const {
+            _id, slug, locale
+          } = argv;
+
+          const aposDocId = await self.getAposDocId({
+            _id,
+            slug,
+            locale
+          });
+
+          self.apos.util.info(aposDocId);
+
+          return aposDocId;
+        }
+      },
+      'set-apos-doc-id': {
+        usage: 'Change the aposDocId of an existing document. You need the new aposDocId (new-id), the old aposDocId (old-id) or the slug, and the locale. Usage: "node app.js @apostrophecms/doc:set-apos-doc-id --new-id=tz4a98xxat96iws9zmbrgj3a --old-id=pfh0haxfpzowht3oi213cqos --locale=fr" or "node app.js @apostrophecms/doc:set-apos-doc-id --new-id=tz4a98xxat96iws9zmbrgj3a --slug=test --locale=fr"',
+        task: async (argv) => {
+          const {
+            'new-id': newId, 'old-id': oldId, slug, locale
+          } = argv;
+
+          const result = await self.setAposDocId({
+            newId,
+            oldId,
+            slug,
+            locale
+          });
+
+          self.apos.util.info(`"${result.oldId}" has been changed to "${result.newId}" for locale "${result.locale}", ${result.renamed} documents changed.`);
+
+          return result;
+        }
+      }
     };
   }
 };