fix: pr feedback and small refactoring

flora-pm · Nov 21, 2024 · 97d19ef · 97d19ef
1 parent 5d3ca82
commit 97d19ef
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 34 deletions.
diff --git a/changelog.d/63 b/changelog.d/63
@@ -3,17 +3,5 @@ prs: #785
 issues: #556
 
 description: {
-
-- Migration for `create_package_groups` & `create_package_group_packages`
-- Updates to various files as a result of the work done
-- Created:
-    - `Flora/Model/PackageGroup/Query.hs`
-    - `Flora/Model/PackageGroup/Types.hs`
-    - `Flora/Model/PackageGroup/Update.hs`
-- Created:
-    - `Flora/Model/PackageGroupPackage/Query.hs`
-    - `Flora/Model/PackageGroupPackage/Types.hs`
-    - `Flora/Model/PackageGroupPackage/Update.hs`
-- Created:
-    - test/Flora/PackageGroupSpec.hs`
+  Migration for `create_package_groups` & `create_package_group_packages`
 }
diff --git a/src/core/Flora/Model/PackageGroup/Query.hs b/src/core/Flora/Model/PackageGroup/Query.hs
@@ -7,30 +7,24 @@ module Flora.Model.PackageGroup.Query
 
 import Data.Text (Text)
 import Data.Vector (Vector)
-import Database.PostgreSQL.Entity (selectOneByField)
-import Database.PostgreSQL.Entity.DBT
-  ( QueryNature (Select)
-  , query
-  )
+import Database.PostgreSQL.Entity (joinSelectOneByField, selectOneByField)
 import Database.PostgreSQL.Entity.Types (field)
 import Database.PostgreSQL.Simple (Only (..))
-import Database.PostgreSQL.Simple.SqlQQ (sql)
 import Effectful (Eff, type (:>))
 import Effectful.PostgreSQL.Transact.Effect (DB, dbtToEff)
 import Flora.Model.Package.Types (Package)
-import Flora.Model.PackageGroup.Types (PackageGroup, PackageGroupId)
+import Flora.Model.PackageGroup.Types (PackageGroup (..), PackageGroupId (..))
+import Flora.Model.PackageGroupPackage.Types (PackageGroupPackage (..))
 
 getPackagesByPackageGroupId :: DB :> es => PackageGroupId -> Eff es (Vector Package)
-getPackagesByPackageGroupId packageGroupId = dbtToEff $ query Select q (Only packageGroupId)
-  where
-    q =
-      [sql|
-        SELECT p.*
-        FROM packages AS p
-        JOIN package_group_packages AS pgp
-        ON p.package_id = pgp.package_id
-        WHERE pgp.package_group_id = ?
-      |]
+getPackagesByPackageGroupId packageGroupId =
+  dbtToEff $
+    joinSelectOneByField
+      @Package
+      @PackageGroupPackage
+      [field| package_id |]
+      [field| package_group_id |]
+      packageGroupId
 
 getPackageGroupByPackageGroupName :: DB :> es => Text -> Eff es (Maybe PackageGroup)
 getPackageGroupByPackageGroupName groupName = dbtToEff $ selectOneByField [field| group_name |] (Only groupName)
diff --git a/src/core/Flora/Model/PackageGroup/Types.hs b/src/core/Flora/Model/PackageGroup/Types.hs
@@ -1,7 +1,6 @@
 module Flora.Model.PackageGroup.Types where
 
 import GHC.Generics
-
 import Control.DeepSeq (NFData)
 import Data.Aeson
 import Data.Text (Text)

diff --git a/test/Flora/PackageGroupSpec.hs b/test/Flora/PackageGroupSpec.hs
@@ -52,7 +52,7 @@ testAddPackageToPackageGroup = do
         .~ pure user.userId
   packageGroup <-
     instantiatePackageGroup randomPackageGroupTemplate
-  packageGroupPackage <-
+  void $
     instantiatePackageGroupPackage $
       randomPackageGroupPackageTemplate
         & #packageGroupId
@@ -75,7 +75,7 @@ testRemovePackageFromPackageGroup = do
         .~ pure user.userId
   packageGroup <-
     instantiatePackageGroup randomPackageGroupTemplate
-  packageGroupPackage <-
+  void $
     instantiatePackageGroupPackage $
       randomPackageGroupPackageTemplate
         & #packageGroupId
@@ -99,7 +99,7 @@ testGetPackagesByPackageGroupId = do
         .~ pure user.userId
   packageGroup <-
     instantiatePackageGroup randomPackageGroupTemplate
-  packageGroupPackage <-
+  void $
     instantiatePackageGroupPackage $
       randomPackageGroupPackageTemplate
         & #packageGroupId