Merge pull request #93 from epochtalk/polls-update

Polls update

Author: unenglishable

lib/epochtalk_server/models/poll.ex

@@ -124,10 +124,38 @@ defmodule EpochtalkServer.Models.Poll do
     if poll_answers_len > 0, do: poll_cs, else: add_error(poll_cs, :answers, "can't be blank")
   end
 
+  @doc """
+  Update changeset for `Poll` model
+  """
+  @spec update_changeset(
+          poll :: t(),
+          attrs :: map() | nil
+        ) :: Ecto.Changeset.t()
+  def update_changeset(poll, attrs \\ %{}) do
+    poll_answers_len = length(poll.poll_answers || [])
+
+    poll
+    |> cast(attrs, [
+      :max_answers,
+      :expiration,
+      :change_vote,
+      :display_mode
+    ])
+    |> validate_required([
+      :max_answers,
+      :change_vote,
+      :display_mode
+    ])
+    |> validate_naivedatetime(:expiration, after: :utc_now)
+    |> validate_number(:max_answers, greater_than: 0, less_than_or_equal_to: poll_answers_len)
+    |> validate_length(:question, min: 1, max: 255)
+    |> validate_display_mode()
+  end
+
   @doc """
   Creates a new `Poll` in the database
   """
-  @spec create(post_attrs :: map()) :: {:ok, post :: t()} | {:error, Ecto.Changeset.t()}
+  @spec create(poll_attrs :: map()) :: {:ok, poll :: t()} | {:error, Ecto.Changeset.t()}
   def create(poll_attrs) do
     Repo.transaction(fn ->
       post_cs = create_changeset(%Poll{}, poll_attrs)
@@ -148,6 +176,19 @@ defmodule EpochtalkServer.Models.Poll do
     end)
   end
 
+  @doc """
+  Updates an existing `Poll` in the database
+  """
+  @spec update(poll_attrs :: map()) ::
+          {:ok, poll :: t()} | {:error, Ecto.Changeset.t()}
+  def update(attrs) do
+    Poll
+    |> Repo.get_by(thread_id: attrs["thread_id"])
+    |> Repo.preload([:poll_answers])
+    |> update_changeset(attrs)
+    |> Repo.update()
+  end
+
   @doc """
   Returns boolean indicating if the specified `User` has voted on the specified `Thread`
   """
@@ -167,6 +208,96 @@ defmodule EpochtalkServer.Models.Poll do
     Repo.exists?(query)
   end
 
+  @doc """
+  Returns boolean indicating if the specified `User` has not voted on the specified `Thread`
+  """
+  @spec has_not_voted(thread_id :: integer, user_id :: integer) :: boolean
+  def has_not_voted(thread_id, user_id), do: !has_voted(thread_id, user_id)
+
+  @doc """
+  Returns boolean indicating if the specified `Poll` exists given a `Thread` id
+  """
+  @spec exists(thread_id :: integer) :: boolean
+  def exists(thread_id) do
+    query =
+      from p in Poll,
+        where: p.thread_id == ^thread_id,
+        select: p.id
+
+    Repo.exists?(query)
+  end
+
+  @doc """
+  Returns integer indicating the max number of answers for the `Poll` given a `Thread` id
+  """
+  @spec max_answers(thread_id :: integer) :: non_neg_integer
+  def max_answers(thread_id) do
+    query =
+      from p in Poll,
+        where: p.thread_id == ^thread_id,
+        select: p.max_answers
+
+    Repo.one(query)
+  end
+
+  @doc """
+  Sets boolean indicating if the specified `Poll` is locked given a `Thread` id
+  """
+  @spec set_locked(thread_id :: integer, locked :: boolean) :: {non_neg_integer, nil | [term()]}
+  def set_locked(thread_id, locked) when is_integer(thread_id) and is_boolean(locked) do
+    query = from p in Poll, where: p.thread_id == ^thread_id
+    Repo.update_all(query, set: [locked: locked])
+  end
+
+  @doc """
+  Returns boolean indicating if the specified `Poll` is locked given a `Thread` id
+  """
+  @spec locked(thread_id :: integer) :: boolean
+  def locked(thread_id) do
+    query =
+      from p in Poll,
+        where: p.thread_id == ^thread_id,
+        select: p.locked
+
+    Repo.one(query)
+  end
+
+  @doc """
+  Returns boolean indicating if the specified `Poll` is unlocked given a `Thread` id
+  """
+  @spec unlocked(thread_id :: integer) :: boolean
+  def unlocked(thread_id), do: !locked(thread_id)
+
+  @doc """
+  Returns boolean indicating if the specified `Poll` allows votes to be changed given a `Thread` id
+  """
+  @spec allow_change_vote(thread_id :: integer) :: boolean
+  def allow_change_vote(thread_id) do
+    query =
+      from p in Poll,
+        where: p.thread_id == ^thread_id,
+        select: p.change_vote
+
+    Repo.one(query)
+  end
+
+  @doc """
+  Returns boolean indicating if the specified `Poll` is currently running given a `Thread` id
+  """
+  @spec running(thread_id :: integer) :: boolean
+  def running(thread_id) do
+    query =
+      from p in Poll,
+        where: p.thread_id == ^thread_id,
+        select: p.expiration
+
+    expiration = Repo.one(query)
+
+    if expiration == nil,
+      do: true,
+      else: NaiveDateTime.compare(expiration, NaiveDateTime.utc_now()) == :gt
+  end
+
   @doc """
   Queries `Poll` Data by thread
   """
@@ -180,6 +311,8 @@ defmodule EpochtalkServer.Models.Poll do
     Repo.one(query)
   end
 
+  # var q = 'UPDATE polls SET (max_answers, change_vote, expiration, display_mode) = ($1, $2, $3, $4) WHERE id = $5';
+
   # === Private Helper Functions ===
 
   defp validate_display_mode(changeset) do

lib/epochtalk_server/models/poll_response.ex

@@ -1,10 +1,11 @@
 defmodule EpochtalkServer.Models.PollResponse do
   use Ecto.Schema
   import Ecto.Changeset
-  # import Ecto.Query
+  import Ecto.Query
   alias EpochtalkServer.Repo
   alias EpochtalkServer.Models.PollResponse
   alias EpochtalkServer.Models.PollAnswer
+  alias EpochtalkServer.Models.Poll
   alias EpochtalkServer.Models.User
 
   @moduledoc """
@@ -54,17 +55,43 @@ defmodule EpochtalkServer.Models.PollResponse do
   ## === Database Functions ===
 
   @doc """
-  Create on `PollResponse` for specific poll. Used to vote for a `Poll`
+  Create on `PollResponse` for specific `Poll`. Used to vote for a `Poll`
   """
-  @spec create(attrs :: map, user_id :: integer) :: :ok
+  @spec create(attrs :: map, user_id :: integer) :: {:ok, any()} | {:error, any()}
   def create(%{"answer_ids" => answer_ids} = attrs, user_id)
       when is_map(attrs) and is_integer(user_id),
-      do: Enum.each(answer_ids, &PollResponse.create(&1, user_id))
+      do:
+        Repo.transaction(fn ->
+          Enum.map(answer_ids, &PollResponse.create(&1, user_id))
+        end)
 
   def create(answer_id, user_id) do
     poll_response_cs =
       create_changeset(%PollResponse{}, %{answer_id: answer_id, user_id: user_id})
 
-    Repo.insert(poll_response_cs)
+    case Repo.insert(poll_response_cs) do
+      {:ok, poll_response} -> poll_response
+      {:error, cs} -> Repo.rollback(cs)
+    end
+  end
+
+  @doc """
+  Delete `PollResponse` for specific `Poll` and `User`. Used to remove vote from a `Poll`
+  """
+  @spec delete(thread_id :: non_neg_integer, user_id :: non_neg_integer) ::
+          {non_neg_integer(), nil | [t()]}
+  def delete(thread_id, user_id) do
+    poll_answer_ids =
+      from pa in PollAnswer,
+        join: p in Poll,
+        on: p.thread_id == ^thread_id and pa.poll_id == p.id,
+        select: pa.id
+
+    poll_responses_by_user =
+      from pr in PollResponse,
+        where: pr.answer_id in subquery(poll_answer_ids) and pr.user_id == ^user_id,
+        select: pr
+
+    Repo.delete_all(poll_responses_by_user)
   end
 end

lib/epochtalk_server/models/thread.ex

@@ -419,10 +419,13 @@ defmodule EpochtalkServer.Models.Thread do
           body: p.content["body"],
           thread_id: p.thread_id,
           thread_slug: t.slug,
-          username: u.username
+          username: u.username,
+          user_id: u.id,
+          user: u
         }
 
-    Repo.one(query_first_thread_post_data)
+    first_post = Repo.one(query_first_thread_post_data)
+    Map.put(first_post, :user, Repo.preload(first_post.user, :roles))
   end
 
   @doc """
@@ -446,10 +449,13 @@ defmodule EpochtalkServer.Models.Thread do
           body: p.content["body"],
           thread_id: p.thread_id,
           thread_slug: t.slug,
-          username: u.username
+          username: u.username,
+          user_id: u.id,
+          user: u
         }
 
-    Repo.one(query_first_thread_post_data)
+    first_post = Repo.one(query_first_thread_post_data)
+    Map.put(first_post, :user, Repo.preload(first_post.user, :roles))
   end
 
   ## === Private Helper Functions ===