1 files changed, 19 insertions, 0 deletions
diff --git a/app/models/work_item.rb b/app/models/work_item.rb
index 557694da35a..bdd9aae90a4 100644
--- a/app/models/work_item.rb
+++ b/app/models/work_item.rb
@@ -4,10 +4,29 @@ class WorkItem < Issue
   self.table_name = 'issues'
   self.inheritance_column = :_type_disabled
 
+  has_one :parent_link, class_name: '::WorkItems::ParentLink', foreign_key: :work_item_id
+  has_one :work_item_parent, through: :parent_link, class_name: 'WorkItem'
+
+  has_many :child_links, class_name: '::WorkItems::ParentLink', foreign_key: :work_item_parent_id
+  has_many :work_item_children, through: :child_links, class_name: 'WorkItem',
+            foreign_key: :work_item_id, source: :work_item
+
+  scope :inc_relations_for_permission_check, -> { includes(:author, project: :project_feature) }
+
+  def self.assignee_association_name
+    'issue'
+  end
+
   def noteable_target_type_name
     'issue'
   end
 
+  def widgets
+    work_item_type.widgets.map do |widget_class|
+      widget_class.new(self)
+    end
+  end
+
   private
 
   def record_create_action