1 files changed, 143 insertions, 0 deletions
diff --git a/db/migrate/20230129154819_add_widgets_for_work_item_types.rb b/db/migrate/20230129154819_add_widgets_for_work_item_types.rb
new file mode 100644
index 00000000000..b936ea2e409
--- /dev/null
+++ b/db/migrate/20230129154819_add_widgets_for_work_item_types.rb
@@ -0,0 +1,143 @@
+# frozen_string_literal: true
+
+class AddWidgetsForWorkItemTypes < Gitlab::Database::Migration[2.1]
+  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!
+
+  def up
+    widget_names = {
+      assignees: 'Assignees',
+      labels: 'Labels',
+      description: 'Description',
+      hierarchy: 'Hierarchy',
+      start_and_due_date: 'Start and due date',
+      milestone: 'Milestone',
+      notes: 'Notes',
+      iteration: 'Iteration',
+      weight: 'Weight',
+      health_status: 'Health status',
+      progress: 'Progress',
+      status: 'Status',
+      requirement_legacy: 'Requirement legacy',
+      test_reports: 'Test reports'
+    }
+
+    widgets_for_type = {
+      'Issue' => [
+        :assignees,
+        :labels,
+        :description,
+        :hierarchy,
+        :start_and_due_date,
+        :milestone,
+        :notes,
+        # EE widgets
+        :iteration,
+        :weight,
+        :health_status
+      ],
+      'Incident' => [
+        :description,
+        :hierarchy,
+        :notes
+      ],
+      'Test Case' => [
+        :description,
+        :notes
+      ],
+      'Requirement' => [
+        :description,
+        :notes,
+        :status,
+        :requirement_legacy,
+        :test_reports
+      ],
+      'Task' => [
+        :assignees,
+        :labels,
+        :description,
+        :hierarchy,
+        :start_and_due_date,
+        :milestone,
+        :notes,
+        :iteration,
+        :weight
+      ],
+      'Objective' => [
+        :assignees,
+        :labels,
+        :description,
+        :hierarchy,
+        :milestone,
+        :notes,
+        :health_status,
+        :progress
+      ],
+      'Key Result' => [
+        :assignees,
+        :labels,
+        :description,
+        :hierarchy,
+        :start_and_due_date,
+        :notes,
+        :health_status,
+        :progress
+      ]
+    }
+
+    widgets_enum = {
+      assignees: 0,
+      description: 1,
+      hierarchy: 2,
+      labels: 3,
+      milestone: 4,
+      notes: 5,
+      start_and_due_date: 6,
+      health_status: 7, # EE-only
+      weight: 8, # EE-only
+      iteration: 9, # EE-only
+      progress: 10, # EE-only
+      status: 11, # EE-only
+      requirement_legacy: 12, # EE-only
+      test_reports: 13
+    }
+
+    widgets = []
+    widgets_for_type.each do |type_name, widget_syms|
+      type = WorkItemType.find_by_name_and_namespace_id(type_name, nil)
+
+      unless type
+        Gitlab::AppLogger.warn("type #{type_name} is missing, not adding widgets")
+
+        next
+      end
+
+      widgets += widget_syms.map do |widget_sym|
+        {
+          work_item_type_id: type.id,
+          name: widget_names[widget_sym],
+          widget_type: widgets_enum[widget_sym]
+        }
+      end
+    end
+
+    return if widgets.empty?
+
+    WidgetDefinition.upsert_all(
+      widgets,
+      unique_by: :index_work_item_widget_definitions_on_default_witype_and_name
+    )
+  end
+
+  def down
+    WidgetDefinition.delete_all
+  end
+end