feat: onboarding, org dashboard, exclusive bounties, provider account mgmt (#97)

Author: zcesur

lib/algora/accounts/accounts.ex

@@ -6,11 +6,15 @@ defmodule Algora.Accounts do
   alias Algora.Accounts.Identity
   alias Algora.Accounts.User
   alias Algora.Bounties.Bounty
+  alias Algora.Contracts.Contract
   alias Algora.Github
   alias Algora.Organizations
   alias Algora.Organizations.Member
   alias Algora.Payments.Transaction
   alias Algora.Repo
+  alias Algora.Workspace.Contributor
+  alias Algora.Workspace.Installation
+  alias Algora.Workspace.Repository
 
   require Algora.SQL
 
@@ -246,7 +250,7 @@ defmodule Algora.Accounts do
   @doc """
   Registers a user from their GitHub information.
   """
-  def register_github_user(primary_email, info, emails, token) do
+  def register_github_user(current_user, primary_email, info, emails, token) do
     query =
       from(u in User,
         left_join: i in Identity,
@@ -257,10 +261,16 @@ defmodule Algora.Accounts do
         select: {u, i}
       )
 
-    case Repo.one(query) do
+    account =
+      case Repo.all(query) do
+        [] -> nil
+        [{user, identity}] -> {user, identity}
+        records -> Enum.find(records, fn {user, _} -> user.id == current_user.id end)
+      end
+
+    case account do
       nil -> create_user(info, primary_email, emails, token)
-      {user, nil} -> update_user(user, info, primary_email, emails, token)
-      {user, _identity} -> update_github_token(user, token)
+      {user, identity} -> update_user(user, identity, info, primary_email, emails, token)
     end
   end
 
@@ -274,15 +284,74 @@ defmodule Algora.Accounts do
     |> Repo.insert()
   end
 
-  def update_user(user, info, primary_email, emails, token) do
-    with {:ok, _} <-
-           user
-           |> Identity.github_registration_changeset(info, primary_email, emails, token)
-           |> Repo.insert() do
-      user
-      |> User.github_registration_changeset(info, primary_email, emails, token)
-      |> Repo.update()
-    end
+  def update_user(user, identity, info, primary_email, emails, token) do
+    old_user = Repo.get_by(User, provider: "github", provider_id: to_string(info["id"]))
+
+    identity_changeset = Identity.github_registration_changeset(user, info, primary_email, emails, token)
+
+    user_changeset = User.github_registration_changeset(user, info, primary_email, emails, token)
+
+    Repo.transact(fn ->
+      delete_result =
+        if identity do
+          Repo.delete(identity)
+        else
+          {:ok, nil}
+        end
+
+      migrate_result =
+        if old_user && old_user.id != user.id do
+          {:ok, old_user} =
+            old_user
+            |> change(provider: nil, provider_id: nil, provider_login: nil, provider_meta: nil)
+            |> Repo.update()
+
+          # TODO: enqueue job
+          migrate_user(old_user, user)
+
+          {:ok, old_user}
+        else
+          {:ok, nil}
+        end
+
+      with {:ok, _} <- delete_result,
+           {:ok, _} <- migrate_result,
+           {:ok, _} <- Repo.insert(identity_changeset) do
+        Repo.update(user_changeset)
+      end
+    end)
+  end
+
+  def migrate_user(old_user, new_user) do
+    Repo.update_all(
+      from(r in Repository, where: r.user_id == ^old_user.id),
+      set: [user_id: new_user.id]
+    )
+
+    Repo.update_all(
+      from(c in Contributor, where: c.user_id == ^old_user.id),
+      set: [user_id: new_user.id]
+    )
+
+    Repo.update_all(
+      from(c in Contract, where: c.contractor_id == ^old_user.id),
+      set: [contractor_id: new_user.id]
+    )
+
+    Repo.update_all(
+      from(i in Installation, where: i.owner_id == ^old_user.id),
+      set: [owner_id: new_user.id]
+    )
+
+    Repo.update_all(
+      from(i in Installation, where: i.provider_user_id == ^old_user.id),
+      set: [provider_user_id: new_user.id]
+    )
+
+    Repo.update_all(
+      from(i in Installation, where: i.connected_user_id == ^old_user.id),
+      set: [connected_user_id: new_user.id]
+    )
   end
 
   # def get_user_by_provider_email(provider, email) when provider in [:github] do
@@ -352,19 +421,6 @@ defmodule Algora.Accounts do
     end
   end
 
-  defp update_github_token(%User{} = user, new_token) do
-    identity =
-      Repo.one!(from(i in Identity, where: i.user_id == ^user.id and i.provider == "github"))
-
-    {:ok, _} =
-      identity
-      |> change()
-      |> put_change(:provider_token, new_token)
-      |> Repo.update()
-
-    {:ok, Repo.preload(user, :identities, force: true)}
-  end
-
   def last_context(nil), do: "nil"
 
   def last_context(%User{last_context: nil} = user) do
@@ -413,8 +469,20 @@ defmodule Algora.Accounts do
 
   def get_last_context_user(%User{} = user) do
     case last_context(user) do
-      "personal" -> user
-      last_context -> get_user_by_handle(last_context)
+      "personal" ->
+        user
+
+      "preview/" <> ctx ->
+        case String.split(ctx, "/") do
+          [id, _repo_owner, _repo_name] -> get_user(id)
+          _ -> nil
+        end
+
+      "repo/" <> _repo_full_name ->
+        user
+
+      last_context ->
+        get_user_by_handle(last_context)
     end
   end
 

lib/algora/accounts/schemas/user.ex

@@ -129,7 +129,7 @@ defmodule Algora.Accounts.User do
       params = %{
         "handle" => info["login"],
         "email" => primary_email,
-        "display_name" => get_change(identity_changeset, :provider_name),
+        "display_name" => info["name"],
         "bio" => info["bio"],
         "location" => info["location"],
         "avatar_url" => info["avatar_url"],
@@ -176,21 +176,33 @@ defmodule Algora.Accounts.User do
       Identity.github_registration_changeset(user, info, primary_email, emails, token)
 
     if identity_changeset.valid? do
-      params = %{
-        "display_name" => user.display_name || get_change(identity_changeset, :provider_name),
-        "bio" => user.bio || info["bio"],
-        "location" => user.location || info["location"],
-        "avatar_url" => user.avatar_url || info["avatar_url"],
-        "website_url" => user.website_url || info["blog"],
-        "github_url" => user.github_url || info["html_url"],
-        "provider" => "github",
-        "provider_id" => to_string(info["id"]),
-        "provider_login" => info["login"],
-        "provider_meta" => info
-      }
+      params =
+        %{
+          "handle" => user.handle || Algora.Organizations.ensure_unique_handle(info["login"]),
+          "email" => user.email || primary_email,
+          "display_name" => user.display_name || info["name"],
+          "bio" => user.bio || info["bio"],
+          "location" => user.location || info["location"],
+          "avatar_url" => user.avatar_url || info["avatar_url"],
+          "website_url" => user.website_url || info["blog"],
+          "github_url" => user.github_url || info["html_url"],
+          "provider" => "github",
+          "provider_id" => to_string(info["id"]),
+          "provider_login" => info["login"],
+          "provider_meta" => info
+        }
+
+      params =
+        if is_nil(user.provider_id) do
+          Map.put(params, "display_name", info["name"])
+        else
+          params
+        end
 
       user
       |> cast(params, [
+        :handle,
+        :email,
         :display_name,
         :bio,
         :location,
@@ -203,8 +215,11 @@ defmodule Algora.Accounts.User do
         :provider_meta
       ])
       |> generate_id()
-      |> validate_required([:display_name])
+      |> validate_required([:email, :display_name, :handle])
+      |> validate_handle()
       |> validate_email()
+      |> unique_constraint(:email)
+      |> unique_constraint(:handle)
     else
       user
       |> change()
@@ -270,7 +285,7 @@ defmodule Algora.Accounts.User do
 
   def validate_handle(changeset) do
     reserved_words =
-      ~w(personal org admin support help security team staff official auth tip home dashboard bounties community user payment claims orgs projects jobs leaderboard onboarding pricing developers companies contracts community blog docs open hiring sdk api)
+      ~w(personal org admin support help security team staff official auth tip home dashboard bounties community user payment claims orgs projects jobs leaderboard onboarding pricing developers companies contracts community blog docs open hiring sdk api repo go preview)
 
     changeset
     |> validate_format(:handle, ~r/^[a-zA-Z0-9_-]{2,32}$/)
@@ -320,8 +335,9 @@ defmodule Algora.Accounts.User do
     validate_inclusion(changeset, :timezone, Tzdata.zone_list())
   end
 
-  defp type_from_provider(:github, "Organization"), do: :organization
-  defp type_from_provider(:github, _), do: :individual
+  def type_from_provider(:github, "Bot"), do: :bot
+  def type_from_provider(:github, "Organization"), do: :organization
+  def type_from_provider(:github, _), do: :individual
 
   def handle(%{handle: handle}) when is_binary(handle), do: handle
   def handle(%{provider_login: handle}), do: handle

lib/algora/admin/admin.ex

@@ -16,6 +16,17 @@ defmodule Algora.Admin do
 
   require Logger
 
+  def alert(message) do
+    %{
+      title: "Alert: #{message}",
+      body: message,
+      name: "Algora Alert",
+      email: "[email protected]"
+    }
+    |> Algora.Activities.SendEmail.changeset()
+    |> Repo.insert()
+  end
+
   def token!, do: System.fetch_env!("ADMIN_GITHUB_TOKEN")
 
   def run(worker) do