add invite link and registration validation

lib/cannery_web/controllers/user_registration_controller.ex

@@ -1,18 +1,67 @@
 defmodule CanneryWeb.UserRegistrationController do
   use CanneryWeb, :controller
 
-  alias Cannery.Accounts
+  alias Cannery.{Accounts, Invites}
   alias Cannery.Accounts.User
   alias CanneryWeb.UserAuth
 
-  def new(conn, _params) do
-    changeset = Accounts.change_user_registration(%User{})
-    render(conn, "new.html", changeset: changeset)
+  def new(conn, %{"invite" => invite_token}) do
+    invite = Invites.get_invite_by_token(invite_token)
+
+    if invite do
+      conn |> render_new(invite)
+    else
+      conn
+      |> put_flash(:error, "Sorry, this invite was not found or expired")
+      |> redirect(to: Routes.home_path(CanneryWeb.Endpoint, :index))
+    end
   end
 
-  def create(conn, %{"user" => user_params}) do
+  def new(conn, _params) do
+    if Accounts.allow_registration?() do
+      conn |> render_new()
+    else
+      conn
+      |> put_flash(:error, "Sorry, public registration is disabled")
+      |> redirect(to: Routes.home_path(CanneryWeb.Endpoint, :index))
+    end
+  end
+
+  # renders new user registration page
+  defp render_new(conn, invite \\ nil) do
+    changeset = Accounts.change_user_registration(%User{})
+    conn |> render("new.html", changeset: changeset, invite: invite)
+  end
+
+  def create(conn, %{"user" => %{"invite_token" => invite_token}} = attrs) do
+    invite = Invites.get_invite_by_token(invite_token)
+
+    if invite do
+      conn |> create_user(attrs, invite)
+    else
+      conn
+      |> put_flash(:error, "Sorry, this invite was not found or expired")
+      |> redirect(to: Routes.home_path(CanneryWeb.Endpoint, :index))
+    end
+  end
+
+  def create(conn, attrs) do
+    if Accounts.allow_registration?() do
+      conn |> create_user(attrs)
+    else
+      conn
+      |> put_flash(:error, "Sorry, public registration is disabled")
+      |> redirect(to: Routes.home_path(CanneryWeb.Endpoint, :index))
+    end
+  end
+
+  defp create_user(conn, %{"user" => user_params}, invite \\ nil) do
     case Accounts.register_user(user_params) do
       {:ok, user} ->
+        unless invite |> is_nil() do
+          invite |> Invites.use_invite!()
+        end
+
         {:ok, _} =
           Accounts.deliver_user_confirmation_instructions(
             user,
@@ -24,7 +73,7 @@ defmodule CanneryWeb.UserRegistrationController do
         |> UserAuth.log_in_user(user)
 
       {:error, %Ecto.Changeset{} = changeset} ->
-        render(conn, "new.html", changeset: changeset)
+        render(conn, "new.html", changeset: changeset, invite: invite)
     end
   end
 end