Add server plugin helpers

[github/Chocobozzz/PeerTube.git] / server / tests / fixtures / peertube-plugin-test-four / main.js

diff --git a/server/tests/fixtures/peertube-plugin-test-four/main.js b/server/tests/fixtures/peertube-plugin-test-four/main.js
index 9abb73646b048ac158884f22d37c2e3b9065336c..ea0599997bf10ea33375e3e34403b31235f2dd2d 100644 (file)
--- a/server/tests/fixtures/peertube-plugin-test-four/main.js
+++ b/server/tests/fixtures/peertube-plugin-test-four/main.js
@@ -1,18 +1,88 @@
 async function register ({
-  peertubeHelpers
+  peertubeHelpers,
+  registerHook,
+  getRouter
 }) {
-  peertubeHelpers.logger.info('Hello world from plugin four')
-
-  const username = 'root'
-  const results = await peertubeHelpers.database.query(
-    'SELECT "email" from "user" WHERE "username" = $username',
-    {
-      type: 'SELECT',
-      bind: { username }
+  const logger = peertubeHelpers.logger
+
+  logger.info('Hello world from plugin four')
+
+  {
+    const username = 'root'
+    const results = await peertubeHelpers.database.query(
+      'SELECT "email" from "user" WHERE "username" = $username',
+      {
+        type: 'SELECT',
+        bind: { username }
+      }
+    )
+
+    logger.info('root email is ' + results[0]['email'])
+  }
+
+  {
+    registerHook({
+      target: 'action:api.video.viewed',
+      handler: async ({ video }) => {
+        const videoFromDB1 = await peertubeHelpers.videos.loadByUrl(video.url)
+        const videoFromDB2 = await peertubeHelpers.videos.loadByIdOrUUID(video.id)
+        const videoFromDB3 = await peertubeHelpers.videos.loadByIdOrUUID(video.uuid)
+
+        if (videoFromDB1.uuid !== videoFromDB2.uuid || videoFromDB2.uuid !== videoFromDB3.uuid) return
+
+        logger.info('video from DB uuid is %s.', videoFromDB1.uuid)
+
+        await peertubeHelpers.videos.removeVideo(video.id)
+
+        logger.info('Video deleted by plugin four.')
+      }
+    })
+  }
+
+  {
+    const serverActor = await peertubeHelpers.server.getServerActor()
+    logger.info('server actor name is %s', serverActor.preferredUsername)
+  }
+
+  {
+    logger.info('server url is %s', peertubeHelpers.config.getWebserverUrl())
+  }
+
+  {
+    const actions = {
+      blockServer,
+      unblockServer,
+      blockAccount,
+      unblockAccount,
+      blacklist,
+      unblacklist
     }
-  )
 
-  peertubeHelpers.logger.info('root email is ' + results[0]['email'])
+    const router = getRouter()
+    router.post('/commander', async (req, res) => {
+      try {
+        await actions[req.body.command](peertubeHelpers, req.body)
+
+        res.sendStatus(204)
+      } catch (err) {
+        logger.error('Error in commander.', { err })
+        res.sendStatus(500)
+      }
+    })
+
+    router.get('/server-config', async (req, res) => {
+      const serverConfig = await peertubeHelpers.config.getServerConfig()
+
+      return res.json({ serverConfig })
+    })
+
+    router.get('/static-route', async (req, res) => {
+      const staticRoute = await peertubeHelpers.plugin.getBaseStaticRoute()
+
+      return res.json({ staticRoute })
+    })
+  }
+
 }
 
 async function unregister () {
@@ -25,3 +95,38 @@ module.exports = {
 }
 
 // ###########################################################################
+
+async function blockServer (peertubeHelpers, body) {
+  const serverActor = await peertubeHelpers.server.getServerActor()
+
+  await peertubeHelpers.moderation.blockServer({ byAccountId: serverActor.Account.id, hostToBlock: body.hostToBlock })
+}
+
+async function unblockServer (peertubeHelpers, body) {
+  const serverActor = await peertubeHelpers.server.getServerActor()
+
+  await peertubeHelpers.moderation.unblockServer({ byAccountId: serverActor.Account.id, hostToUnblock: body.hostToUnblock })
+}
+
+async function blockAccount (peertubeHelpers, body) {
+  const serverActor = await peertubeHelpers.server.getServerActor()
+
+  await peertubeHelpers.moderation.blockAccount({ byAccountId: serverActor.Account.id, handleToBlock: body.handleToBlock })
+}
+
+async function unblockAccount (peertubeHelpers, body) {
+  const serverActor = await peertubeHelpers.server.getServerActor()
+
+  await peertubeHelpers.moderation.unblockAccount({ byAccountId: serverActor.Account.id, handleToUnblock: body.handleToUnblock })
+}
+
+async function blacklist (peertubeHelpers, body) {
+  await peertubeHelpers.moderation.blacklistVideo({
+    videoIdOrUUID: body.videoUUID,
+    createOptions: body
+  })
+}
+
+async function unblacklist (peertubeHelpers, body) {
+  await peertubeHelpers.moderation.unblacklistVideo({ videoIdOrUUID: body.videoUUID })
+}