Add utility functions for objects and activities.

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/pleroma/activity.ex b/lib/pleroma/activity.ex
index c4efc6283..46568bb13 100644
--- a/lib/pleroma/activity.ex
+++ b/lib/pleroma/activity.ex
@@ -1,9 +1,21 @@
 defmodule Pleroma.Activity do
   use Ecto.Schema
+  alias Pleroma.{Repo, Activity}
+  import Ecto.Query
 
   schema "activities" do
     field :data, :map
 
     timestamps()
   end
+
+  def get_by_ap_id(ap_id) do
+    Repo.one(from activity in Activity,
+      where: fragment("? @> ?", activity.data, ^%{id: ap_id}))
+  end
+
+  def all_by_object_ap_id(ap_id) do
+    Repo.all(from activity in Activity,
+      where: fragment("? @> ?", activity.data, ^%{object: %{id: ap_id}}))
+  end
 end