From ee2f5f07b4aed193d151a1eec68329bf0d48c1f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9mi=20de=20Juvigny?= <remi.dejuvigny@strapi.io>
Date: Tue, 7 Feb 2023 12:11:28 +0100
Subject: [PATCH] Add auto db migration

---
 packages/core/admin/ee/server/register.js | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/packages/core/admin/ee/server/register.js b/packages/core/admin/ee/server/register.js
index d6e743bdc1..6f03dd937a 100644
--- a/packages/core/admin/ee/server/register.js
+++ b/packages/core/admin/ee/server/register.js
@@ -4,8 +4,31 @@ const { features } = require('@strapi/strapi/lib/utils/ee');
 const executeCERegister = require('../../server/register');
 const createAuditLogsService = require('./services/audit-logs');
 
+// Migrate the audit logs table name for users coming from v4.6.0
+const migrateAuditLogsTable = async ({ oldContentTypes, contentTypes }) => {
+  // Check if the table name needs to be migrated
+  const oldName = oldContentTypes?.['admin::audit-log']?.collectionName;
+  const newName = contentTypes['admin::audit-log']?.collectionName;
+  const shouldMigrate = oldName === 'audit_logs' && newName === 'strapi_audit_logs';
+
+  if (shouldMigrate) {
+    // Migrate the name audit logs table
+    if (await strapi.db.getSchemaConnection().hasTable('audit_logs')) {
+      await strapi.db.getSchemaConnection().renameTable('audit_logs', 'strapi_audit_logs');
+    }
+
+    // Migrate the link table
+    if (await strapi.db.getSchemaConnection().hasTable('audit_logs_user_links')) {
+      await strapi.db
+        .getSchemaConnection()
+        .renameTable('audit_logs_user_links', 'strapi_audit_logs_user_links');
+    }
+  }
+};
+
 module.exports = async ({ strapi }) => {
   if (features.isEnabled('audit-logs')) {
+    strapi.hook('strapi::content-types.beforeSync').register(migrateAuditLogsTable);
     const auditLogsService = createAuditLogsService(strapi);
     strapi.container.register('audit-logs', auditLogsService);
     await auditLogsService.register();