From 1a40132c67e50c94a7bd3f6b63c075f471b3d6cc Mon Sep 17 00:00:00 2001
From: Chocobozzz <me@florianbigard.com>
Date: Fri, 12 Jun 2020 16:24:58 +0200
Subject: [PATCH] Use default nsfw instance policy for search index

---
 server/controllers/api/search.ts        |  9 ++++++-
 server/tests/api/search/search-index.ts | 31 +++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/server/controllers/api/search.ts b/server/controllers/api/search.ts
index 1dea77d29..e35a7346e 100644
--- a/server/controllers/api/search.ts
+++ b/server/controllers/api/search.ts
@@ -169,7 +169,14 @@ async function searchVideosIndex (query: VideosSearchQuery, res: express.Respons
 
   const result = await buildMutedForSearchIndex(res)
 
-  const body = Object.assign(query, result)
+  const body: VideosSearchQuery = Object.assign(query, result)
+
+  // Use the default instance NSFW policy if not specified
+  if (!body.nsfw) {
+    body.nsfw = CONFIG.INSTANCE.DEFAULT_NSFW_POLICY === 'do_not_list'
+      ? 'false'
+      : 'both'
+  }
 
   const url = sanitizeUrl(CONFIG.SEARCH.SEARCH_INDEX.URL) + '/api/v1/search/videos'
 
diff --git a/server/tests/api/search/search-index.ts b/server/tests/api/search/search-index.ts
index 2354aaa8b..40065d162 100644
--- a/server/tests/api/search/search-index.ts
+++ b/server/tests/api/search/search-index.ts
@@ -209,6 +209,37 @@ describe('Test videos search', function () {
       expect(res.body.total).to.be.greaterThan(5)
       expect(res.body.data).to.have.lengthOf(5)
     })
+
+    it('Should use the nsfw instance policy as default', async function () {
+      let nsfwUUID: string
+
+      {
+        await updateCustomSubConfig(server.url, server.accessToken, { instance: { defaultNSFWPolicy: 'display' } })
+
+        const res = await searchVideo(server.url, 'NSFW search index')
+        const video = res.body.data[0] as Video
+
+        expect(res.body.data).to.have.length.greaterThan(0)
+        expect(video.nsfw).to.be.true
+
+        nsfwUUID = video.uuid
+      }
+
+      {
+        await updateCustomSubConfig(server.url, server.accessToken, { instance: { defaultNSFWPolicy: 'do_not_list' } })
+
+        const res = await searchVideo(server.url, 'NSFW search index')
+
+        try {
+          expect(res.body.data).to.have.lengthOf(0)
+        } catch (err) {
+          //
+          const video = res.body.data[0] as Video
+
+          expect(video.uuid).not.equal(nsfwUUID)
+        }
+      }
+    })
   })
 
   describe('Channels search', async function () {
-- 
2.25.1