From e8a311ecffe1fb19a3d194b1c5628853263909a7 Mon Sep 17 00:00:00 2001
From: Roger Braun <roger@rogerbraun.net>
Date: Thu, 27 Apr 2017 09:44:20 +0200
Subject: [PATCH] Add user and hub to websub client subscriptions.

---
 lib/pleroma/web/websub/websub_client_subscription.ex   |  3 +++
 .../migrations/20170427054757_add_user_and_hub.exs     | 10 ++++++++++
 2 files changed, 13 insertions(+)
 create mode 100644 priv/repo/migrations/20170427054757_add_user_and_hub.exs

diff --git a/lib/pleroma/web/websub/websub_client_subscription.ex b/lib/pleroma/web/websub/websub_client_subscription.ex
index 341e27c51..c7a25ea22 100644
--- a/lib/pleroma/web/websub/websub_client_subscription.ex
+++ b/lib/pleroma/web/websub/websub_client_subscription.ex
@@ -1,5 +1,6 @@
 defmodule Pleroma.Web.Websub.WebsubClientSubscription do
   use Ecto.Schema
+  alias Pleroma.User
 
   schema "websub_client_subscriptions" do
     field :topic, :string
@@ -7,6 +8,8 @@ defmodule Pleroma.Web.Websub.WebsubClientSubscription do
     field :valid_until, :naive_datetime
     field :state, :string
     field :subscribers, {:array, :string}, default: []
+    field :hub, :string
+    belongs_to :user, User
 
     timestamps()
   end
diff --git a/priv/repo/migrations/20170427054757_add_user_and_hub.exs b/priv/repo/migrations/20170427054757_add_user_and_hub.exs
new file mode 100644
index 000000000..4f9a520bd
--- /dev/null
+++ b/priv/repo/migrations/20170427054757_add_user_and_hub.exs
@@ -0,0 +1,10 @@
+defmodule Pleroma.Repo.Migrations.AddUserAndHub do
+  use Ecto.Migration
+
+  def change do
+    alter table(:websub_client_subscriptions) do
+      add :hub, :string
+      add :user_id, references(:users)
+    end
+  end
+end