feat: add community flags to repos / videos / jobs

Author: mydearxym

lib/mastani_server/cms/delegates/article_curd.ex

@@ -30,7 +30,7 @@ defmodule MastaniServer.CMS.Delegate.ArticleCURD do
   defp flag_query(queryable, filter, flag \\ %{}) do
     flag = %{pin: false, trash: false} |> Map.merge(flag)
 
-    # TODO: remove case judge ?
+    # NOTE: this case judge is used for test case
     case filter |> Map.has_key?(:community) do
       true ->
         queryable
@@ -40,7 +40,6 @@ defmodule MastaniServer.CMS.Delegate.ArticleCURD do
         |> where([q, f, c], c.raw == ^filter.community)
 
       false ->
-        # for runing tests usage, no flag when db_insert
         queryable
     end
   end
@@ -128,7 +127,7 @@ defmodule MastaniServer.CMS.Delegate.ArticleCURD do
             {:ok, "pass"}
         end
 
-        # Repo.insert(%CMS.PostCommunityFlags{post_id: content.id, community_id: community.id, })
+        # Repo.insert(%CMS.PostCommunityFlag{post_id: content.id, community_id: community.id, })
       end)
       |> Multi.run(:set_tag, fn %{add_content_author: content} ->
         case attrs |> Map.has_key?(:tags) do

lib/mastani_server/cms/delegates/article_operation.ex

@@ -5,35 +5,73 @@ defmodule MastaniServer.CMS.Delegate.ArticleOperation do
   import MastaniServer.CMS.Utils.Matcher
   import Ecto.Query, warn: false
   import Helper.ErrorCode
-  # import ShortMaps
+  import ShortMaps
 
   alias Helper.ORM
   alias MastaniServer.Accounts.User
-  alias MastaniServer.CMS.{Community, PinState, Post, PostCommunityFlags, Tag}
+
+  alias MastaniServer.CMS.{
+    Community,
+    PinState,
+    Post,
+    PostCommunityFlag,
+    Job,
+    JobCommunityFlag,
+    RepoCommunityFlag,
+    Video,
+    VideoCommunityFlag,
+    Tag
+  }
+
+  alias MastaniServer.CMS.Repo, as: CMSRepo
   alias MastaniServer.Repo
 
   @doc """
   pin / unpin, trash / untrash articles
   """
-  def set_community_flags(%Post{id: post_id}, community_id, attrs) do
-    with {:ok, post} <- ORM.find(Post, post_id),
+  def set_community_flags(%Post{id: _} = content, community_id, attrs),
+    do: do_set_flag(content, community_id, attrs)
+
+  def set_community_flags(%Job{id: _} = content, community_id, attrs),
+    do: do_set_flag(content, community_id, attrs)
+
+  def set_community_flags(%CMSRepo{id: _} = content, community_id, attrs),
+    do: do_set_flag(content, community_id, attrs)
+
+  def set_community_flags(%Video{id: _} = content, community_id, attrs),
+    do: do_set_flag(content, community_id, attrs)
+
+  defp do_set_flag(content, community_id, attrs) do
+    with {:ok, content} <- ORM.find(content.__struct__, content.id),
          {:ok, community} <- ORM.find(Community, community_id),
-         {:ok, _} <- insert_flag_record(post, community_id, attrs) do
-      ORM.find(Post, post.id)
+         {:ok, record} <- insert_flag_record(content, community_id, attrs) do
+      {:ok, struct(content, %{pin: record.pin, trash: record.trash})}
     end
   end
 
-  defp insert_flag_record(%Post{id: id}, community_id, attrs) do
-    clauses = %{
-      post_id: id,
-      community_id: community_id
-    }
+  defp insert_flag_record(%Post{id: post_id}, community_id, attrs) do
+    clauses = ~m(post_id community_id)a
+    PostCommunityFlag |> ORM.upsert_by(clauses, Map.merge(attrs, clauses))
+  end
+
+  defp insert_flag_record(%Job{id: job_id}, community_id, attrs) do
+    clauses = ~m(job_id community_id)a
+    JobCommunityFlag |> ORM.upsert_by(clauses, Map.merge(attrs, clauses))
+  end
 
-    attrs = attrs |> Map.merge(clauses)
+  defp insert_flag_record(%CMSRepo{id: repo_id}, community_id, attrs) do
+    clauses = ~m(repo_id community_id)a
+    RepoCommunityFlag |> ORM.upsert_by(clauses, Map.merge(attrs, clauses))
+  end
 
-    PostCommunityFlags |> ORM.upsert_by(clauses, attrs)
+  defp insert_flag_record(%Video{id: video_id}, community_id, attrs) do
+    clauses = ~m(video_id community_id)a
+    VideoCommunityFlag |> ORM.upsert_by(clauses, Map.merge(attrs, clauses))
   end
 
+  @doc """
+  set content to diffent community
+  """
   def set_community(%Community{id: community_id}, thread, content_id) when valid_thread(thread) do
     with {:ok, action} <- match_action(thread, :community),
          {:ok, content} <- ORM.find(action.target, content_id, preload: :communities),

lib/mastani_server/cms/job.ex

@@ -4,7 +4,7 @@ defmodule MastaniServer.CMS.Job do
 
   use Ecto.Schema
   import Ecto.Changeset
-  alias MastaniServer.CMS.{Author, Community, JobFavorite, Tag}
+  alias MastaniServer.CMS.{Author, Community, JobFavorite, JobCommunityFlag, Tag}
 
   @required_fields ~w(title company company_logo location body digest length)a
   @optional_fields ~w(link_addr link_source min_education)a
@@ -35,6 +35,12 @@ defmodule MastaniServer.CMS.Job do
     field(:digest, :string)
     field(:length, :integer)
 
+    has_many(:community_flags, {"jobs_communities_flags", JobCommunityFlag})
+
+    # NOTE: this one is tricky, pin is dynamic changed when return by func: add_pin_contents_ifneed
+    field(:pin, :boolean, default_value: false, virtual: true)
+    field(:trash, :boolean, default_value: false, virtual: true)
+
     # has_many(:comments, {"jobs_comments", JobComment})
     has_many(:favorites, {"jobs_favorites", JobFavorite})
     # has_many(:stars, {"posts_stars", PostStar})

lib/mastani_server/cms/job_community_flag.ex

@@ -0,0 +1,35 @@
+defmodule MastaniServer.CMS.JobCommunityFlag do
+  @moduledoc false
+  alias __MODULE__
+
+  use Ecto.Schema
+  import Ecto.Changeset
+  alias Helper.Certification
+  alias MastaniServer.Accounts
+  alias MastaniServer.CMS.{Community, Job}
+
+  @required_fields ~w(job_id community_id)a
+  @optional_fields ~w(pin trash)a
+
+  @type t :: %JobCommunityFlag{}
+
+  schema "jobs_communities_flags" do
+    belongs_to(:job, Job, foreign_key: :job_id)
+    belongs_to(:community, Community, foreign_key: :community_id)
+
+    field(:pin, :boolean)
+    field(:trash, :boolean)
+
+    timestamps(type: :utc_datetime)
+  end
+
+  @doc false
+  def changeset(%JobCommunityFlag{} = job_community_flag, attrs) do
+    job_community_flag
+    |> cast(attrs, @optional_fields ++ @required_fields)
+    |> validate_required(@required_fields)
+    |> foreign_key_constraint(:job_id)
+    |> foreign_key_constraint(:community_id)
+    |> unique_constraint(:job_id, name: :jobs_communities_flags_job_id_community_id_index)
+  end
+end

lib/mastani_server/cms/post.ex

@@ -9,7 +9,7 @@ defmodule MastaniServer.CMS.Post do
     Author,
     Community,
     PostComment,
-    PostCommunityFlags,
+    PostCommunityFlag,
     PostFavorite,
     PostStar,
     Tag
@@ -27,8 +27,7 @@ defmodule MastaniServer.CMS.Post do
     field(:length, :integer)
     field(:views, :integer, default: 0)
 
-    # has_many(:pins, {"posts_pins", PostPin})
-    has_many(:community_flags, {"posts_communities_flags", PostCommunityFlags})
+    has_many(:community_flags, {"posts_communities_flags", PostCommunityFlag})
 
     # NOTE: this one is tricky, pin is dynamic changed when return by func: add_pin_contents_ifneed
     field(:pin, :boolean, default_value: false, virtual: true)

lib/mastani_server/cms/post_community_flag.ex

@@ -1,4 +1,4 @@
-defmodule MastaniServer.CMS.PostCommunityFlags do
+defmodule MastaniServer.CMS.PostCommunityFlag do
   @moduledoc false
   alias __MODULE__
 
@@ -11,7 +11,7 @@ defmodule MastaniServer.CMS.PostCommunityFlags do
   @required_fields ~w(post_id community_id)a
   @optional_fields ~w(pin trash refined)a
 
-  @type t :: %PostCommunityFlags{}
+  @type t :: %PostCommunityFlag{}
 
   schema "posts_communities_flags" do
     belongs_to(:post, Post, foreign_key: :post_id)
@@ -25,8 +25,8 @@ defmodule MastaniServer.CMS.PostCommunityFlags do
   end
 
   @doc false
-  def changeset(%PostCommunityFlags{} = post_community_flags, attrs) do
-    post_community_flags
+  def changeset(%PostCommunityFlag{} = post_community_flag, attrs) do
+    post_community_flag
     |> cast(attrs, @optional_fields ++ @required_fields)
     |> validate_required(@required_fields)
     |> foreign_key_constraint(:post_id)

lib/mastani_server/cms/repo.ex

@@ -4,7 +4,7 @@ defmodule MastaniServer.CMS.Repo do
 
   use Ecto.Schema
   import Ecto.Changeset
-  alias MastaniServer.CMS.{Author, Community, RepoBuilder, Tag}
+  alias MastaniServer.CMS.{Author, Community, RepoBuilder, RepoCommunityFlag, Tag}
 
   @required_fields ~w(repo_name desc readme language producer producer_link repo_link repo_star_count repo_fork_count repo_watch_count)a
   @optional_fields ~w(views pin trash last_fetch_time)
@@ -26,6 +26,10 @@ defmodule MastaniServer.CMS.Repo do
     field(:repo_watch_count, :integer)
 
     field(:views, :integer, default: 0)
+
+    has_many(:community_flags, {"repos_communities_flags", RepoCommunityFlag})
+
+    # NOTE: this one is tricky, pin is dynamic changed when return by func: add_pin_contents_ifneed
     field(:pin, :boolean, default_value: false)
     field(:trash, :boolean, default_value: false)
 

lib/mastani_server/cms/repo_community_flag.ex

@@ -0,0 +1,35 @@
+defmodule MastaniServer.CMS.RepoCommunityFlag do
+  @moduledoc false
+  alias __MODULE__
+
+  use Ecto.Schema
+  import Ecto.Changeset
+  alias Helper.Certification
+  alias MastaniServer.Accounts
+  alias MastaniServer.CMS.{Community, Repo}
+
+  @required_fields ~w(repo_id community_id)a
+  @optional_fields ~w(pin trash)a
+
+  @type t :: %RepoCommunityFlag{}
+
+  schema "repos_communities_flags" do
+    belongs_to(:repo, Repo, foreign_key: :repo_id)
+    belongs_to(:community, Community, foreign_key: :community_id)
+
+    field(:pin, :boolean)
+    field(:trash, :boolean)
+
+    timestamps(type: :utc_datetime)
+  end
+
+  @doc false
+  def changeset(%RepoCommunityFlag{} = repo_community_flag, attrs) do
+    repo_community_flag
+    |> cast(attrs, @optional_fields ++ @required_fields)
+    |> validate_required(@required_fields)
+    |> foreign_key_constraint(:repo_id)
+    |> foreign_key_constraint(:community_id)
+    |> unique_constraint(:repo_id, name: :repos_communities_flags_repo_id_community_id_index)
+  end
+end

lib/mastani_server/cms/utils/matcher.ex

@@ -20,7 +20,10 @@ defmodule MastaniServer.CMS.Utils.Matcher do
     Community,
     PostCommentLike,
     PostCommentDislike,
-    PostCommunityFlags
+    PostCommunityFlag,
+    JobCommunityFlag,
+    RepoCommunityFlag,
+    VideoCommunityFlag
   }
 
   @support_thread [:post, :video, :repo, :job]
@@ -47,7 +50,7 @@ defmodule MastaniServer.CMS.Utils.Matcher do
   def match_action(:post, :tag), do: {:ok, %{target: Post, reactor: Tag}}
 
   def match_action(:post, :community),
-    do: {:ok, %{target: Post, reactor: Community, flag: PostCommunityFlags}}
+    do: {:ok, %{target: Post, reactor: Community, flag: PostCommunityFlag}}
 
   def match_action(:post, :comment),
     do: {:ok, %{target: Post, reactor: PostComment, preload: :author}}
@@ -59,14 +62,19 @@ defmodule MastaniServer.CMS.Utils.Matcher do
     do: {:ok, %{target: PostComment, reactor: PostCommentDislike}}
 
   # videos ...
-  def match_action(:video, :community), do: {:ok, %{target: Video, reactor: Community}}
+  def match_action(:video, :community),
+    do: {:ok, %{target: Video, reactor: Community, flag: VideoCommunityFlag}}
 
   # repos ...
-  def match_action(:repo, :community), do: {:ok, %{target: Repo, reactor: Community}}
+  def match_action(:repo, :community),
+    do: {:ok, %{target: Repo, reactor: Community, flag: RepoCommunityFlag}}
 
   # jobs ...
   def match_action(:job, :self), do: {:ok, %{target: Job, reactor: Job, preload: :author}}
-  def match_action(:job, :community), do: {:ok, %{target: Job, reactor: Community}}
+
+  def match_action(:job, :community),
+    do: {:ok, %{target: Job, reactor: Community, flag: JobCommunityFlags}}
+
   def match_action(:job, :star), do: {:ok, %{target: Job, reactor: JobStar, preload: :user}}
   def match_action(:job, :tag), do: {:ok, %{target: Job, reactor: Tag}}
 

lib/mastani_server/cms/video.ex

@@ -4,7 +4,7 @@ defmodule MastaniServer.CMS.Video do
 
   use Ecto.Schema
   import Ecto.Changeset
-  alias MastaniServer.CMS.{Author, Community, Tag}
+  alias MastaniServer.CMS.{Author, Community, VideoCommunityFlag, Tag}
 
   @required_fields ~w(title poster desc duration duration_sec source)a
   @optional_fields ~w(link original_author original_author_link publish_at pin trash)
@@ -24,6 +24,10 @@ defmodule MastaniServer.CMS.Video do
     field(:original_author_link, :string)
 
     field(:views, :integer, default: 0)
+
+    has_many(:community_flags, {"videos_communities_flags", VideoCommunityFlag})
+
+    # NOTE: this one is tricky, pin is dynamic changed when return by func: add_pin_contents_ifneed
     field(:pin, :boolean, default_value: false)
     field(:trash, :boolean, default_value: false)