Add latest changes from gitlab-org/gitlab@16-8-stable-eev16.8.0-rc42

1 files changed, 44 insertions, 0 deletions

diff --git a/db/post_migrate/20231220115201_add_color_work_item_widget.rb b/db/post_migrate/20231220115201_add_color_work_item_widget.rb
new file mode 100644
index 00000000000..eb410cb8b0b
--- /dev/null
+++ b/db/post_migrate/20231220115201_add_color_work_item_widget.rb
@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+class AddColorWorkItemWidget < Gitlab::Database::Migration[2.2]
+  milestone '16.8'
+
+  class WorkItemType < MigrationRecord
+    self.table_name = 'work_item_types'
+  end
+
+  class WidgetDefinition < MigrationRecord
+    self.table_name = 'work_item_widget_definitions'
+  end
+
+  restrict_gitlab_migration gitlab_schema: :gitlab_main
+  disable_ddl_transaction!
+
+  WIDGET_NAME = 'Color'
+  WIDGET_ENUM_VALUE = 18
+  WORK_ITEM_TYPE = 'Epic'
+
+  def up
+    type = WorkItemType.find_by_name_and_namespace_id(WORK_ITEM_TYPE, nil)
+
+    unless type
+      Gitlab::AppLogger.warn("type #{WORK_ITEM_TYPE} is missing, not adding widget")
+      return
+    end
+
+    widget = {
+      work_item_type_id: type.id,
+      name: WIDGET_NAME,
+      widget_type: WIDGET_ENUM_VALUE
+    }
+
+    WidgetDefinition.upsert_all(
+      [widget],
+      unique_by: :index_work_item_widget_definitions_on_default_witype_and_name
+    )
+  end
+
+  def down
+    WidgetDefinition.where(name: WIDGET_NAME).delete_all
+  end
+end