1 files changed, 21 insertions, 0 deletions
diff --git a/lib/rook_web/controllers/app_controller.ex b/lib/rook_web/controllers/app_controller.ex
new file mode 100644
index 0000000..4c36fda
--- /dev/null
+++ b/lib/rook_web/controllers/app_controller.ex
@@ -0,0 +1,21 @@
+defmodule RookWeb.AppController do
+  use RookWeb, :controller
+
+  plug :add_token
+
+  def share(conn, _params) do
+    render(conn, "share.html")
+  end
+
+  def request(conn, _params) do
+    render(conn, "request.html")
+  end
+
+  defp add_token(conn, _params) do
+    if conn.assigns[:token] do
+      conn
+    else
+      assign(conn, :token, Rook.Utils.Token.token())
+    end
+  end
+end