Correct webtorrent download cleanup

[github/Chocobozzz/PeerTube.git] / server / helpers / activitypub.ts

diff --git a/server/helpers/activitypub.ts b/server/helpers/activitypub.ts
index ecb509b66c81dd9edea35d7466717493ff77b422..a9de11fb0984ff8b76f95bb84259b9c6b5a03b71 100644 (file)
--- a/server/helpers/activitypub.ts
+++ b/server/helpers/activitypub.ts
@@ -1,123 +1,117 @@
-import * as url from 'url'
-
-import { database as db } from '../initializers'
-import { logger } from './logger'
-import { doRequest } from './requests'
-import { isRemoteAccountValid } from './custom-validators'
-import { ActivityPubActor } from '../../shared/models/activitypub/activitypub-actor'
-import { ResultList } from '../../shared/models/result-list.model'
-
-async function fetchRemoteAccountAndCreatePod (accountUrl: string) {
-  const options = {
-    uri: accountUrl,
-    method: 'GET'
-  }
-
-  let requestResult
-  try {
-    requestResult = await doRequest(options)
-  } catch (err) {
-    logger.warning('Cannot fetch remote account %s.', accountUrl, err)
-    return undefined
-  }
-
-  const accountJSON: ActivityPubActor = requestResult.body
-  if (isRemoteAccountValid(accountJSON) === false) return undefined
-
-  const followersCount = await fetchAccountCount(accountJSON.followers)
-  const followingCount = await fetchAccountCount(accountJSON.following)
-
-  const account = db.Account.build({
-    uuid: accountJSON.uuid,
-    name: accountJSON.preferredUsername,
-    url: accountJSON.url,
-    publicKey: accountJSON.publicKey.publicKeyPem,
-    privateKey: null,
-    followersCount: followersCount,
-    followingCount: followingCount,
-    inboxUrl: accountJSON.inbox,
-    outboxUrl: accountJSON.outbox,
-    sharedInboxUrl: accountJSON.endpoints.sharedInbox,
-    followersUrl: accountJSON.followers,
-    followingUrl: accountJSON.following
-  })
-
-  const accountHost = url.parse(account.url).host
-  const podOptions = {
-    where: {
-      host: accountHost
-    },
-    defaults: {
-      host: accountHost
-    }
-  }
-  const pod = await db.Pod.findOrCreate(podOptions)
-
-  return { account, pod }
-}
-
-function activityPubContextify (data: object) {
-  return Object.assign(data,{
+import * as Bluebird from 'bluebird'
+import * as validator from 'validator'
+import { ResultList } from '../../shared/models'
+import { Activity, ActivityPubActor } from '../../shared/models/activitypub'
+import { ACTIVITY_PUB } from '../initializers'
+import { ActorModel } from '../models/activitypub/actor'
+import { signObject } from './peertube-crypto'
+import { pageToStartAndCount } from './core-utils'
+
+function activityPubContextify <T> (data: T) {
+  return Object.assign(data, {
     '@context': [
       'https://www.w3.org/ns/activitystreams',
       'https://w3id.org/security/v1',
       {
-        'Hashtag': 'as:Hashtag',
-        'uuid': 'http://schema.org/identifier',
-        'category': 'http://schema.org/category',
-        'licence': 'http://schema.org/license',
-        'nsfw': 'as:sensitive',
-        'language': 'http://schema.org/inLanguage',
-        'views': 'http://schema.org/Number',
-        'size': 'http://schema.org/Number'
+        RsaSignature2017: 'https://w3id.org/security#RsaSignature2017',
+        Hashtag: 'as:Hashtag',
+        uuid: 'http://schema.org/identifier',
+        category: 'http://schema.org/category',
+        licence: 'http://schema.org/license',
+        subtitleLanguage: 'http://schema.org/subtitleLanguage',
+        sensitive: 'as:sensitive',
+        language: 'http://schema.org/inLanguage',
+        views: 'http://schema.org/Number',
+        stats: 'http://schema.org/Number',
+        size: 'http://schema.org/Number',
+        fps: 'http://schema.org/Number',
+        commentsEnabled: 'http://schema.org/Boolean',
+        waitTranscoding: 'http://schema.org/Boolean',
+        support: 'http://schema.org/Text'
+      },
+      {
+        likes: {
+          '@id': 'as:likes',
+          '@type': '@id'
+        },
+        dislikes: {
+          '@id': 'as:dislikes',
+          '@type': '@id'
+        },
+        shares: {
+          '@id': 'as:shares',
+          '@type': '@id'
+        },
+        comments: {
+          '@id': 'as:comments',
+          '@type': '@id'
+        }
       }
     ]
   })
 }
 
-function activityPubCollectionPagination (url: string, page: number, result: ResultList<any>) {
-  const baseUrl = url.split('?').shift
+type ActivityPubCollectionPaginationHandler = (start: number, count: number) => Bluebird<ResultList<any>> | Promise<ResultList<any>>
+async function activityPubCollectionPagination (url: string, handler: ActivityPubCollectionPaginationHandler, page?: any) {
+  if (!page || !validator.isInt(page)) {
+    // We just display the first page URL, we only need the total items
+    const result = await handler(0, 1)
 
-  const obj = {
-    id: baseUrl,
-    type: 'Collection',
-    totalItems: result.total,
-    first: {
-      id: baseUrl + '?page=' + page,
-      type: 'CollectionPage',
+    return {
+      id: url,
+      type: 'OrderedCollection',
       totalItems: result.total,
-      next: baseUrl + '?page=' + (page + 1),
-      partOf: baseUrl,
-      items: result.data
+      first: url + '?page=1'
     }
   }
 
-  return activityPubContextify(obj)
-}
+  const { start, count } = pageToStartAndCount(page, ACTIVITY_PUB.COLLECTION_ITEMS_PER_PAGE)
+  const result = await handler(start, count)
 
-// ---------------------------------------------------------------------------
+  let next: string | undefined
+  let prev: string | undefined
 
-export {
-  fetchRemoteAccountAndCreatePod,
-  activityPubContextify,
-  activityPubCollectionPagination
-}
+  // Assert page is a number
+  page = parseInt(page, 10)
 
-// ---------------------------------------------------------------------------
+  // There are more results
+  if (result.total > page * ACTIVITY_PUB.COLLECTION_ITEMS_PER_PAGE) {
+    next = url + '?page=' + (page + 1)
+  }
 
-async function fetchAccountCount (url: string) {
-  const options = {
-    uri: url,
-    method: 'GET'
+  if (page > 1) {
+    prev = url + '?page=' + (page - 1)
   }
 
-  let requestResult
-  try {
-    requestResult = await doRequest(options)
-  } catch (err) {
-    logger.warning('Cannot fetch remote account count %s.', url, err)
-    return undefined
+  return {
+    id: url + '?page=' + page,
+    type: 'OrderedCollectionPage',
+    prev,
+    next,
+    partOf: url,
+    orderedItems: result.data,
+    totalItems: result.total
   }
 
-  return requestResult.totalItems ? requestResult.totalItems : 0
+}
+
+function buildSignedActivity (byActor: ActorModel, data: Object) {
+  const activity = activityPubContextify(data)
+
+  return signObject(byActor, activity) as Promise<Activity>
+}
+
+function getActorUrl (activityActor: string | ActivityPubActor) {
+  if (typeof activityActor === 'string') return activityActor
+
+  return activityActor.id
+}
+
+// ---------------------------------------------------------------------------
+
+export {
+  getActorUrl,
+  activityPubContextify,
+  activityPubCollectionPagination,
+  buildSignedActivity
 }