From af8a1309bde2a85bd94d06f3f152ae6355677095 Mon Sep 17 00:00:00 2001
From: Fire Demon <firedemon@creature.cafe>
Date: Fri, 17 Jul 2020 19:56:35 -0500
Subject: [Feature] Introduce nest_level attribute to posts and API to help
 apps visualize nesting in threads

---
 app/models/status.rb | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'app/models')

diff --git a/app/models/status.rb b/app/models/status.rb
index 731bb6eee..bee7d1e67 100644
--- a/app/models/status.rb
+++ b/app/models/status.rb
@@ -26,6 +26,7 @@
 #  content_type           :string
 #  deleted_at             :datetime
 #  edited                 :integer          default(0), not null
+#  nest_level             :integer          default(0), not null
 #
 
 class Status < ApplicationRecord
@@ -272,6 +273,7 @@ class Status < ApplicationRecord
   around_create Mastodon::Snowflake::Callbacks
 
   before_create :set_locality
+  before_create :set_nest_level
 
   before_validation :prepare_contents, if: :local?
   before_validation :set_reblog
@@ -546,6 +548,16 @@ class Status < ApplicationRecord
     self.local = account.local?
   end
 
+  def set_nest_level
+    return if attribute_changed?(:nest_level)
+
+    if reply?
+      self.nest_level = [thread&.account_id == account_id ? thread&.nest_level.to_i : thread&.nest_level.to_i + 1, 127].min
+    else
+      self.nest_level = 0
+    end
+  end
+
   def update_statistics
     return unless distributable?
 
-- 
cgit