further case fixes for empty list bug (#143)

* log

* fix empty case

* cleanup

* log

* simplify getting dep metadata

* cleanup

* log

* filter empty lists

* rework dep metadata with fallback

* update log

* cleanup

* cleanup

src/Database/Queries.hs

@@ -19,7 +19,7 @@ import Model (
     Metric (Metric),
     PkgDependency (..),
     PkgRecord (PkgRecord),
-    VersionRecord (VersionRecord), VersionPlatform (VersionPlatform), EntityField (VersionPlatformPkgId, VersionPlatformVersionNumber, VersionPlatformArch, AdminPkgsPkgId, AdminPkgsAdmin), PkgRecordId, AdminPkgs, AdminId,
+    VersionRecord (VersionRecord), VersionPlatform (VersionPlatform), EntityField (VersionPlatformPkgId, VersionPlatformVersionNumber, VersionPlatformArch, AdminPkgsPkgId, AdminPkgsAdmin, PkgDependencyDepId), PkgRecordId, AdminPkgs, AdminId,
  )
 import Orphans.Emver ()
 import Startlude (
@@ -73,6 +73,9 @@ import Database.Esqueleto.Experimental (
     (==.),
     (^.),
     (||.),
+    isNothing,
+    (<=.),
+    limit
  )
 import Database.Persist qualified as P
 import Database.Persist.Postgresql (
@@ -96,7 +99,7 @@ import Model (
         VersionRecordNumber,
         VersionRecordPkgId,
         VersionRecordTitle,
-        VersionRecordUpdatedAt, PkgRecordHidden, VersionPlatformRam, PkgRecordUpdatedAt
+        VersionRecordUpdatedAt, PkgRecordHidden, VersionPlatformRam, PkgRecordUpdatedAt, VersionRecordCreatedAt
     ),
     Key (unPkgRecordKey),
     PkgCategory,
@@ -113,10 +116,8 @@ import Startlude (
     snd,
     sortOn,
     ($>),
-    (<$>), Int,
+    (<$>), Int, listToMaybe,
  )
-import Database.Esqueleto.Experimental (isNothing)
-import Database.Esqueleto.Experimental ((<=.))
 
 serviceQuerySource ::
     (MonadResource m, MonadIO m) =>
@@ -182,16 +183,15 @@ getPkgDependencyData ::
     MonadIO m =>
     PkgId ->
     Version ->
-    ReaderT SqlBackend m [(P.Entity PkgDependency, P.Entity PkgRecord)]
+    ReaderT SqlBackend m [P.Entity PkgRecord]
 getPkgDependencyData pkgId pkgVersion =
     select $
         from $ do
             (pkgDepRecord :& pr) <- from $ table @PkgDependency
-                `innerJoin` table @PkgRecord `on` (\(v :& p) -> (PkgRecordId === PkgDependencyPkgId) (p :& v))
+                `innerJoin` table @PkgRecord `on` (\(pd :& pr) -> (PkgRecordId === PkgDependencyDepId) (pr :& pd))
             where_ (pkgDepRecord ^. PkgDependencyPkgId ==. val (PkgRecordKey pkgId))
             where_ (pkgDepRecord ^. PkgDependencyPkgVersion ==. val pkgVersion)
-            pure (pkgDepRecord, pr)
-
+            pure pr
 
 (===) ::
     (PersistEntity val1, PersistEntity val2, P.PersistField typ) =>
@@ -201,6 +201,19 @@ getPkgDependencyData pkgId pkgVersion =
     SqlExpr (Value Bool)
 (===) a' b' (a :& b) = a ^. a' ==. b ^. b'
 
+getLatestVersionRecord ::
+    MonadIO m =>
+    Key PkgRecord ->
+    ReaderT SqlBackend m (Maybe VersionRecord)
+getLatestVersionRecord pkgId = do
+    vrs <- select $ do
+        v <- from $ table @VersionRecord
+        where_ $ v ^. VersionRecordPkgId ==. val pkgId
+        orderBy [desc (v ^. VersionRecordCreatedAt)]
+        limit 1
+        pure v
+    pure $ entityVal <$> listToMaybe vrs
+
 
 getCategoriesFor ::
     MonadUnliftIO m =>

src/Handler/Package/V1/Index.hs

@@ -21,19 +21,18 @@ import Database.Persist.Sql (SqlBackend)
 import Database.Queries (
     collateVersions,
     getCategoriesFor,
-    getDependencyVersions,
     getPkgDataSource,
     getPkgDependencyData,
-    serviceQuerySource,
+    serviceQuerySource, getLatestVersionRecord,
  )
 import Foundation (Handler, Route (InstructionsR, LicenseR), RegistryCtx (appSettings))
 import Handler.Package.Api (DependencyRes (..), PackageListRes (..), PackageRes (..))
 import Handler.Types.Api (ApiVersion (..))
 import Handler.Util (basicRender, parseQueryParam, filterDeprecatedVersions, filterDevices, getPkgArch)
 import Lib.PkgRepository (PkgRepo, getIcon, getManifest)
-import Lib.Types.Core (PkgId)
-import Lib.Types.Emver (Version, VersionRange (..), parseRange, satisfies, (<||))
-import Model (Category (..), Key (..), PkgDependency (..), VersionRecord (..), PkgRecord (pkgRecordHidden))
+import Lib.Types.Core (PkgId, PkgId(PkgId))
+import Lib.Types.Emver (Version, Version(Version), VersionRange (..), parseRange, satisfies)
+import Model (Category (..), Key (..), VersionRecord (..), PkgRecord (..))
 import Protolude.Unsafe (unsafeFromJust)
 import Settings (AppSettings (communityVersion))
 import Startlude (
@@ -51,7 +50,6 @@ import Startlude (
     Num ((*), (-)),
     Show,
     Text,
-    Traversable (traverse),
     const,
     encodeUtf8,
     filter,
@@ -62,14 +60,12 @@ import Startlude (
     id,
     liftA2,
     mappend,
-    maximumOn,
     nonEmpty,
     note,
     pure,
     readMaybe,
     snd,
     sortOn,
-    zipWith,
     zipWithM,
     ($),
     (&&&),
@@ -77,8 +73,7 @@ import Startlude (
     (.*),
     (<$>),
     (<&>),
-    (=<<),
-    (>)
+    (=<<)
  )
 import UnliftIO (Concurrently (..), mapConcurrently)
 import Yesod (
@@ -89,15 +84,16 @@ import Yesod (
     lookupGetParam,
  )
 import Data.Tuple (fst)
-import Database.Persist.Postgresql (entityVal)
+import Database.Persist.Postgresql (entityVal, entityKey)
 import Yesod.Core (getsYesod)
-import Data.List (head)
 import Yesod (YesodRequest(reqGetParams))
 import Yesod (getRequest)
 import Data.List (last)
 import Data.Text (isPrefixOf)
-import Startlude (length)
-
+import Control.Monad.Logger (logWarn)
+import Data.String.Interpolate.IsString (
+    i,
+ )
 data PackageReq = PackageReq
     { packageReqId :: !PkgId
     , packageReqVersion :: !VersionRange
@@ -166,7 +162,7 @@ getPackageIndexR = do
                     .| sinkList
 
     -- NOTE: if a package's dependencies do not meet the system requirements, it is currently omitted from the list
-    pkgsWithDependencies <- runDB $ mapConcurrently (getPackageDependencies osPredicate) filteredPackages
+    pkgsWithDependencies <- runDB $ mapConcurrently getPackageDependencies filteredPackages
     PackageListRes <$> runConcurrently (zipWithM (Concurrently .* constructPackageListApiRes) filteredPackages pkgsWithDependencies)
 
 getPkgIdsQuery :: Handler (Maybe [PackageReq])
@@ -211,21 +207,24 @@ getRamQuery = parseQueryParam "hardware.ram" ((flip $ note . mappend "Invalid 'r
 
 getPackageDependencies ::
     (MonadIO m, MonadLogger m, MonadResource m, Has PkgRepo r, MonadReader r m) =>
-    (Version -> Bool) ->
     PackageMetadata ->
     ReaderT SqlBackend m (HashMap PkgId DependencyRes)
-getPackageDependencies osPredicate PackageMetadata{packageMetadataPkgId = pkg, packageMetadataPkgVersion = pkgVersion} =
+getPackageDependencies PackageMetadata{packageMetadataPkgId = pkg, packageMetadataPkgVersion = pkgVersion} =
     do
-        pkgDepInfo' <- getPkgDependencyData pkg pkgVersion
-        let pkgDepInfo = fmap (\a -> (entityVal $ fst a, entityVal $ snd a)) pkgDepInfo'
-        pkgDepInfoWithVersions <- traverse getDependencyVersions (fst <$> pkgDepInfo)
-        let depMetadata = zipWith (selectDependencyBestVersion osPredicate) pkgDepInfo pkgDepInfoWithVersions 
-        lift $
+        depPkgRecordEntities <- getPkgDependencyData pkg pkgVersion
         fmap HM.fromList $
-                for depMetadata $ \(depId, title, v, isLocal) -> do
-                    icon <- loadIcon depId v
-                    pure $ (depId, DependencyRes title icon isLocal)
-
+            for depPkgRecordEntities $ \(pr) -> do
+                let depId = unPkgRecordKey $ entityKey pr
+                let depPkgRecord = entityVal pr
+                mVersionRecord <- getLatestVersionRecord $ entityKey pr
+                case mVersionRecord of
+                    Just VersionRecord{..} -> do
+                        icon <- lift $ loadIcon depId versionRecordNumber
+                        pure $ (depId, DependencyRes versionRecordTitle icon $ pkgRecordHidden depPkgRecord)
+                    Nothing -> do
+                        $logWarn [i|No latest version record found for #{depId} while getting dependency metadata for #{pkg}@#{pkgVersion}. Using fallback package.|]
+                        icon <- lift $ loadIcon (PkgId "fallback") $ Version(1,0,0,0)
+                        pure $ (depId, DependencyRes "Unknown" icon $ pkgRecordHidden depPkgRecord)
 
 constructPackageListApiRes ::
     (MonadResource m, MonadReader r m, Has AppSettings r, Has PkgRepo r) =>
@@ -273,23 +272,3 @@ selectLatestVersionFromSpec pkgRanges vs =
     let pkgId = NE.head $ versionRecordPkgId <$> vs
         spec = pkgRanges (unPkgRecordKey pkgId)
      in headMay . sortOn (Down . versionRecordNumber) $ NE.filter ((`satisfies` spec) . versionRecordNumber) vs
-
-
--- get best version of the dependency based on what is specified in the db (ie. what is specified in the manifest for the package)
-selectDependencyBestVersion :: (Version -> Bool) -> (PkgDependency, PkgRecord) -> [VersionRecord] -> (PkgId, Text, Version, Bool)
-selectDependencyBestVersion osPredicate pkgDepInfo depVersions = do
-    let pkgDepRecord = fst pkgDepInfo
-    let isLocal = pkgRecordHidden $ snd pkgDepInfo
-    let depId = pkgDependencyDepId pkgDepRecord
-    let pkgId = unPkgRecordKey depId
-    let versionRequirement = pkgDependencyDepVersionRange pkgDepRecord
-    let latestDepVersion = head $ sortOn (Down . versionRecordNumber) depVersions
-    let compatiblePkgDepInfo = (filter (osPredicate . versionRecordOsVersion)) depVersions
-    if (length compatiblePkgDepInfo > 0)
-        then do
-            let satisfactory = filter ((<|| versionRequirement) . versionRecordNumber) compatiblePkgDepInfo
-            case maximumOn versionRecordNumber satisfactory of
-                Just bestVersion -> (pkgId, versionRecordTitle bestVersion, versionRecordNumber bestVersion, isLocal)
-                -- use latest version of dep for metadata info
-                Nothing -> (pkgId, versionRecordTitle latestDepVersion, versionRecordNumber latestDepVersion, isLocal)
-        else (pkgId, versionRecordTitle latestDepVersion, versionRecordNumber latestDepVersion, isLocal)

src/Handler/Util.hs

@@ -101,6 +101,7 @@ import Settings (AppSettings(whitelist))
 import Network.HTTP.Types (status200)
 import Database.Persist (insert_)
 import Yesod (lookupPostParam)
+import Data.Maybe (isNothing)
 
 orThrow :: MonadHandler m => m (Maybe a) -> m a -> m a
 orThrow action other =
@@ -240,7 +241,7 @@ getPkgArch = do
 filterDeprecatedVersions :: Version -> (Version -> Bool) -> [VersionRecord] -> [VersionRecord]
 filterDeprecatedVersions communityVersion osPredicate vrs = do
     if (osPredicate communityVersion)
-        then filter (\v -> not $ isJust $ versionRecordDeprecatedAt v) $ vrs
+        then filter (\v -> isNothing $ versionRecordDeprecatedAt v) $ vrs
         else vrs
 
 filterDevices :: (MonadUnliftIO m) => (MM.MultiMap Text Text) -> [(VersionRecord, VersionPlatform)] -> m [VersionRecord]