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         --- We only care about Create activities.
  29         IF activity.data->>'type' != 'Create' THEN
  30           RETURN true;
  31         END IF;
  32
  33         --- Normalize the child object into child.
  34         SELECT * INTO child FROM objects
  35         INNER JOIN activities ON COALESCE(activities.data->'object'->>'id', activities.data->>'object') = objects.data->>'id'
  36         WHERE COALESCE(activity.data->'object'->>'id', activity.data->>'object') = objects.data->>'id';
  37
  38         --- Fetch the author.
  39         SELECT * INTO author FROM users WHERE users.ap_id = activity.actor;
  40
  41         --- Prepare author's AS2 followers collection.
  42         SELECT COALESCE(author.follower_address, '') INTO author_fa;
  43
  44         --- Check visibility.
  45         IF activity.actor = actor THEN
  46           --- activity visible
  47           NULL;
  48         ELSIF ARRAY[public] && activity.recipients THEN
  49           --- activity visible
  50           NULL;
  51         ELSIF ARRAY[author_fa] && activity.recipients AND ARRAY[author_fa] && actor_user.following THEN
  52           --- activity visible
  53           NULL;
  54         ELSIF ARRAY[actor] && activity.recipients THEN
  55           --- activity visible
  56           NULL;
  57         ELSE
  58           --- activity not visible, break out of the loop
  59           RETURN false;
  60         END IF;
  61
  62         --- If there's a parent, load it and do this all over again.
  63         IF (child.data->'inReplyTo' IS NOT NULL) AND (child.data->'inReplyTo' != 'null'::jsonb) THEN
  64           SELECT * INTO activity FROM activities
  65           INNER JOIN objects ON COALESCE(activities.data->'object'->>'id', activities.data->>'object') = objects.data->>'id'
  66           WHERE child.data->>'inReplyTo' = objects.data->>'id';
  67         ELSE
  68           RETURN true;
  69         END IF;
  70       END LOOP;
  71     END;
  72     $$ LANGUAGE plpgsql IMMUTABLE;
  73     """
  74
  75     execute(statement)
  76   end
  77
  78   def down do
  79     execute("drop function thread_visibility(actor varchar, activity_id varchar)")
  80   end
  81 end