From 2e7cdd0516d063adb748e242a93264548ea4e501 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?KMY=EF=BC=88=E9=9B=AA=E3=81=82=E3=81=99=E3=81=8B=EF=BC=89?=
 <tt@kmycode.net>
Date: Thu, 12 Oct 2023 20:06:21 +0900
Subject: [PATCH] =?UTF-8?q?Fix:=20#89=20NodeInfo=E5=90=91=E3=81=91?=
 =?UTF-8?q?=E3=81=AE=E6=A9=9F=E8=83=BD=E4=B8=80=E8=A6=A7=E8=A1=A8=E3=82=92?=
 =?UTF-8?q?=E4=BD=9C=E6=88=90=E3=81=97=E3=81=A6=E5=8C=BA=E5=88=A5=20(#103)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/helpers/kmyblue_capabilities_helper.rb    | 25 +++++++++++++++++++
 app/serializers/nodeinfo/serializer.rb        |  2 +-
 spec/serializers/nodeinfo/serializer_spec.rb  |  5 ++++
 .../rest/instance_serializer_spec.rb          |  4 +++
 4 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/app/helpers/kmyblue_capabilities_helper.rb b/app/helpers/kmyblue_capabilities_helper.rb
index 8653a3f9da..455561a597 100644
--- a/app/helpers/kmyblue_capabilities_helper.rb
+++ b/app/helpers/kmyblue_capabilities_helper.rb
@@ -30,4 +30,29 @@ module KmyblueCapabilitiesHelper
 
     capabilities
   end
+
+  def capabilities_for_nodeinfo
+    capabilities = %i(
+      wide_emoji
+      status_reference
+      quote
+      kmyblue_quote
+      searchability
+      kmyblue_searchability
+      visibility_mutual
+      visibility_limited
+      kmyblue_antenna
+      kmyblue_bookmark_category
+      kmyblue_searchability_limited
+      kmyblue_circle_history
+    )
+
+    capabilities << :full_text_search if Chewy.enabled?
+    if Setting.enable_emoji_reaction
+      capabilities << :emoji_reaction
+      capabilities << :enable_wide_emoji_reaction
+    end
+
+    capabilities
+  end
 end
diff --git a/app/serializers/nodeinfo/serializer.rb b/app/serializers/nodeinfo/serializer.rb
index 3555f0bd8d..83d63bb397 100644
--- a/app/serializers/nodeinfo/serializer.rb
+++ b/app/serializers/nodeinfo/serializer.rb
@@ -40,7 +40,7 @@ class NodeInfo::Serializer < ActiveModel::Serializer
 
   def metadata
     {
-      features: fedibird_capabilities,
+      features: capabilities_for_nodeinfo,
     }
   end
 
diff --git a/spec/serializers/nodeinfo/serializer_spec.rb b/spec/serializers/nodeinfo/serializer_spec.rb
index c43b0b569f..8352a44143 100644
--- a/spec/serializers/nodeinfo/serializer_spec.rb
+++ b/spec/serializers/nodeinfo/serializer_spec.rb
@@ -28,5 +28,10 @@ describe NodeInfo::Serializer do # rubocop:disable RSpec/FilePath
     it 'returns features' do
       expect(serialization['metadata']['features']).to include 'emoji_reaction'
     end
+
+    it 'returns nodeinfo own features' do
+      expect(serialization['metadata']['features']).to include 'quote'
+      expect(serialization['metadata']['features']).to_not include 'kmyblue_markdown'
+    end
   end
 end
diff --git a/spec/serializers/rest/instance_serializer_spec.rb b/spec/serializers/rest/instance_serializer_spec.rb
index a9a3259aa8..4b33d15daa 100644
--- a/spec/serializers/rest/instance_serializer_spec.rb
+++ b/spec/serializers/rest/instance_serializer_spec.rb
@@ -22,5 +22,9 @@ describe REST::InstanceSerializer do
     it 'returns fedibird_capabilities' do
       expect(serialization['fedibird_capabilities']).to include 'emoji_reaction'
     end
+
+    it 'returns api own fedibird_capabilities' do
+      expect(serialization['fedibird_capabilities']).to include 'kmyblue_markdown'
+    end
   end
 end