2.4.1 upload user photo

.gitignore

@@ -38,3 +38,4 @@ npm-debug.log
 mix.lock
 
 .elixir_ls/*
+uploads

config/config.exs

@@ -34,6 +34,9 @@ config :rent_cars, RentCars.Mailer, adapter: Swoosh.Adapters.Local
 config :money,
   default_currency: :USD
 
+config :waffle,
+  storage: Waffle.Storage.Local
+
 # Configure esbuild (the version is required)
 config :esbuild,
   version: "0.17.11",

lib/rent_cars/accounts.ex

@@ -14,7 +14,14 @@ defmodule RentCars.Accounts do
     |> Repo.update()
   end
 
-  def get_user(id) do
+  def get_user!(id) do
     Repo.get(User, id)
   end
+
+  def upload_photo(user_id, photo) do
+    user_id
+    |> get_user!()
+    |> User.update_photo(%{avatar: photo})
+    |> Repo.update()
+  end
 end

lib/rent_cars/accounts/avatar.ex

@@ -0,0 +1,19 @@
+defmodule RentCars.Accounts.Avatar do
+  use Waffle.Definition
+  use Waffle.Ecto.Definition
+
+  @extension_whitelist ~w(.jpg .jpeg .png)
+
+  def validate({file, _}) do
+    file_extension = file.file_name |> Path.extname() |> String.downcase()
+
+    case Enum.member?(@extension_whitelist, file_extension) do
+      true -> :ok
+      false -> {:error, "file type is invalid"}
+    end
+  end
+
+  def storage_dir(_, {_file, user}) do
+    "uploads/accounts/users/#{user.id}"
+  end
+end

lib/rent_cars/accounts/user.ex

@@ -1,7 +1,9 @@
 defmodule RentCars.Accounts.User do
   use Ecto.Schema
   import Ecto.Changeset
+  alias RentCars.Accounts.Avatar
   alias RentCars.Rentals.Rental
+  import Waffle.Ecto.Schema
 
   @role_values ~w/USER ADMIN/a
   @fields ~w/role/a
@@ -18,6 +20,8 @@
     field :email, :string
     field :drive_license, :string
     field :role, Ecto.Enum, values: @role_values, default: :USER
+    field :avatar, Avatar.Type
+
     has_many :rentals, Rental
 
     timestamps()
@@ -46,7 +50,7 @@
   end
 
   defp hash_password(%{valid?: true, changes: %{password: password}} = changeset) do
     change(changeset, Argon2.add_hash(password))
   end
 
   defp hash_password(changeset), do: changeset
@@ -54,4 +58,8 @@
   def update_user(user, params) do
     changeset(user, params)
   end
+
+  def update_photo(user, params) do
+    cast_attachments(user, params, [:avatar])
+  end
 end

lib/rent_cars_web/controllers/api/user_controller.ex

@@ -14,10 +14,20 @@ defmodule RentCarsWeb.Api.UserController do
   end
 
   def show(conn, %{"id" => id}) do
-    user = Accounts.get_user(id)
+    user = Accounts.get_user!(id)
     render(conn, :show, user: user)
   end
 
+  def upload_photo(conn, %{"avatar" => params}) do
+    [user_id] = get_req_header(conn, "user_id")
+
+    with {:ok, user} <- Accounts.upload_photo(user_id, params) do
+      conn
+      |> put_status(:created)
+      |> render(:show, user: user)
+    end
+  end
+
   # def update(conn, %{"id" => id, "user" => user_params}) do
   #   user = Accounts.get_user(id)
 

lib/rent_cars_web/controllers/api/user_json.ex

@@ -1,4 +1,5 @@
 defmodule RentCarsWeb.Api.UserJSON do
+  alias RentCars.Accounts.Avatar
   alias RentCars.Accounts.User
   # If you want to customize a particular status code,
   # you may add your own clauses, such as:
@@ -23,6 +24,8 @@ defmodule RentCarsWeb.Api.UserJSON do
       user_name: user.user_name,
       email: user.email,
       drive_license: user.drive_license,
+      role: user.role,
+      avatar: Avatar.url({user.avatar, user}),
       inserted_at: user.inserted_at,
       updated_at: user.updated_at
     }

lib/rent_cars_web/router.ex

@@ -49,7 +49,7 @@ defmodule RentCarsWeb.Router do
       pipe_through :authenticated
       post "/session/me", SessionController, :me
       get "/users/:id", UserController, :show
-
+      patch "/users/photo", UserController, :upload_photo
       post "/rentals", RentalController, :create
       post "/rentals/return/:id", RentalController, :return
       get "/rentals", RentalController, :index

mix.exs

@@ -67,7 +67,9 @@ defmodule RentCars.MixProject do
       {:argon2_elixir, "~> 4.0"},
       {:phoenix_swoosh, "~> 1.0"},
       {:money, "~> 1.12"},
-      {:timex, "~> 3.0"}
+      {:timex, "~> 3.0"},
+      {:waffle, "~> 1.1"},
+      {:waffle_ecto, "~> 0.0.11"}
     ]
   end
 

priv/repo/migrations/20240112052738_add_user_photo.exs

@@ -0,0 +1,9 @@
+defmodule RentCars.Repo.Migrations.AddUserPhoto do
+  use Ecto.Migration
+
+  def change do
+    alter table(:users) do
+      add :avatar, :string
+    end
+  end
+end

test/rent_cars/accounts_test.exs

@@ -3,11 +3,26 @@ defmodule RentCars.AccountsTest do
   alias RentCars.Accounts
   import RentCars.UserFixtures
 
+  describe "upload avatar" do
+    test "upload avatar image" do
+      user = user_fixture()
+
+      photo = %Plug.Upload{
+        content_type: "image/jpg",
+        filename: "1.jpg",
+        path: "test/support/fixtures/1.jpg"
+      }
+
+      assert {:ok, user_updated} = Accounts.upload_photo(user.id, photo)
+      assert user_updated.avatar.file_name == "1.jpg"
+    end
+  end
+
   describe "get user" do
     test "get_user/1" do
       user = user_fixture()
 
-      assert Accounts.get_user(user.id).email == user.email
+      assert Accounts.get_user!(user.id).email == user.email
     end
   end
 

test/rent_cars_web/controllers/api/user_controller_test.exs

@@ -16,4 +16,16 @@ defmodule RentCarsWeb.Api.AccountControllerTest do
              "email" => ^email
            } = json_response(conn, 200)["data"]
   end
+
+  test "upload user image", %{conn: conn, user: _user} do
+    photo = %Plug.Upload{
+      content_type: "image/jpg",
+      filename: "1.jpg",
+      path: "test/support/fixtures/1.jpg"
+    }
+
+    conn = patch(conn, Routes.api_user_path(conn, :upload_photo), avatar: photo)
+
+    assert json_response(conn, 201)["data"]["avatar"] |> String.contains?("1.jpg")
+  end
 end