git.squeep.com Git - akkoma/blob - priv/repo/migrations/20190324222404_add_thread_visibility_function.exs

   1 defmodule Pleroma.Repo.Migrations.AddThreadVisibilityFunction do
   2   use Ecto.Migration
   3   @disable_ddl_transaction true
   4
   5   def up do
   6     statement = """
   7     CREATE OR REPLACE FUNCTION thread_visibility(actor varchar, activity_id varchar) RETURNS boolean AS $$
   8     DECLARE
   9       public varchar := 'https://www.w3.org/ns/activitystreams#Public';
  10       child objects%ROWTYPE;
  11       activity activities%ROWTYPE;
  12       actor_user users%ROWTYPE;
  13       author users%ROWTYPE;
  14       author_fa varchar;
  15     BEGIN
  16       --- Fetch our actor.
  17       SELECT * INTO actor_user FROM users WHERE users.ap_id = actor;
  18
  19       --- Fetch our initial activity.
  20       SELECT * INTO activity FROM activities WHERE activities.data->>'id' = activity_id;
  21
  22       LOOP
  23         --- Ensure that we have an activity before continuing.
  24         IF activity IS NULL THEN
  25           RETURN true;
  26         END IF;
  27
  28         --- Normalize the child object into child.
  29         SELECT * INTO child FROM objects
  30         INNER JOIN activities ON COALESCE(activities.data->'object'->>'id', activities.data->>'object') = objects.data->>'id'
  31         WHERE COALESCE(activity.data->'object'->>'id', activity.data->>'object') = objects.data->>'id';
  32
  33         --- Fetch the author.
  34         SELECT * INTO author FROM users WHERE users.ap_id = activity.actor;
  35
  36         --- Prepare author's AS2 followers collection.
  37         SELECT COALESCE(author.follower_address, '') INTO author_fa;
  38
  39         --- Check visibility.
  40         IF activity.actor = actor THEN
  41           --- activity visible
  42           NULL;
  43         ELSIF ARRAY[public] && activity.recipients THEN
  44           --- activity visible
  45           NULL;
  46         ELSIF ARRAY[author_fa] && activity.recipients AND ARRAY[author_fa] && actor_user.following THEN
  47           --- activity visible
  48           NULL;
  49         ELSIF ARRAY[actor] && activity.recipients THEN
  50           --- activity visible
  51           NULL;
  52         ELSE
  53           --- activity not visible, break out of the loop
  54           RETURN false;
  55         END IF;
  56
  57         --- If there's a parent, load it and do this all over again.
  58         IF (child.data->'inReplyTo' IS NOT NULL) AND (child.data->'inReplyTo' != 'null'::jsonb) THEN
  59           SELECT * INTO activity FROM activities
  60           INNER JOIN objects ON COALESCE(activities.data->'object'->>'id', activities.data->>'object') = objects.data->>'id'
  61           WHERE child.data->>'inReplyTo' = objects.data->>'id';
  62         ELSE
  63           RETURN true;
  64         END IF;
  65       END LOOP;
  66     END;
  67     $$ LANGUAGE plpgsql IMMUTABLE;
  68     """
  69
  70     execute(statement)
  71   end
  72
  73   def down do
  74     execute("drop function thread_visibility(actor varchar, activity_id varchar)")
  75   end
  76 end