fix(mkpkglists): add all info fields to packages-meta-v1

This provides **complete** RPC-like info access to every
package in the database.

The SQL logic in this file for package metadata now exactly
reflects RPC's info logic, without searching for specific
packages.

Signed-off-by: Kevin Morris <kevr@0cost.org>

aurweb/scripts/mkpkglists.py

@@ -5,6 +5,8 @@ import gzip
 import json
 import os
 
+from collections import defaultdict
+from decimal import Decimal
 from typing import Tuple
 
 import aurweb.config
@@ -53,6 +55,67 @@ def update_state(state: str, update_time: int) -> None:
         f.write(str(update_time))
 
 
+TYPE_MAP = {
+    "depends": "Depends",
+    "makedepends": "MakeDepends",
+    "checkdepends": "CheckDepends",
+    "optdepends": "OptDepends",
+    "conflicts": "Conflicts",
+    "provides": "Provides",
+    "replaces": "Replaces",
+    "groups": "Groups",
+    "license": "License",
+    "keyword": "Keywords"
+}
+
+
+def get_extended_fields(package_id: int, pkgbase_id: int):
+    query = """
+    SELECT DependencyTypes.Name AS Type,
+    PackageDepends.DepName AS Name,
+    PackageDepends.DepCondition AS Cond
+    FROM PackageDepends
+    LEFT JOIN DependencyTypes
+    ON DependencyTypes.ID = PackageDepends.DepTypeID
+    WHERE PackageDepends.PackageID = ?
+    UNION SELECT RelationTypes.Name AS Type,
+    PackageRelations.RelName AS Name,
+    PackageRelations.RelCondition AS Cond
+    FROM PackageRelations
+    LEFT JOIN RelationTypes
+    ON RelationTypes.ID = PackageRelations.RelTypeID
+    WHERE PackageRelations.PackageID = ?
+    UNION SELECT 'groups' AS Type, Groups.Name, '' AS Cond
+    FROM Groups INNER JOIN PackageGroups
+    ON PackageGroups.PackageID = ?
+    AND PackageGroups.GroupID = Groups.ID
+    UNION SELECT 'license' AS Type, Licenses.Name, '' as Cond
+    FROM Licenses INNER JOIN PackageLicenses
+    ON PackageLicenses.PackageID = ?
+    AND PackageLicenses.LicenseID = Licenses.ID
+    UNION SELECT 'keyword' AS Type, PackageKeywords.Keyword AS Name, '' as Cond
+    FROM PackageKeywords WHERE PackageBaseID = ?
+    """
+
+    conn = aurweb.db.Connection()
+
+    args = [package_id] * 4 + [pkgbase_id]
+    cursor = conn.execute(query, args)
+
+    data = defaultdict(list)
+    data["License"] = []
+    data["Keywords"] = []
+    for result in cursor.fetchall():
+        key = TYPE_MAP.get(result[0])
+        output = result[1]
+        if result[2]:
+            output += result[2]
+        data[key].append(output)
+
+    conn.close()
+    return data
+
+
 def main():
     conn = aurweb.db.Connection()
 
@@ -64,11 +127,23 @@ def main():
     updated, update_time = should_update(packages_state, "Packages")
     if not updated:
         print("Updating Packages...")
-        columns = ("Packages.ID, PackageBaseID, Packages.Name, "
-                   "Version, Description, URL")
+
+        # Query columns; copied from RPC.
+        columns = ("Packages.ID, Packages.Name, "
+                   "PackageBases.ID AS PackageBaseID, "
+                   "PackageBases.Name AS PackageBase, "
+                   "Version, Description, URL, NumVotes, "
+                   "Popularity, OutOfDateTS AS OutOfDate, "
+                   "Users.UserName AS Maintainer, "
+                   "SubmittedTS AS FirstSubmitted, "
+                   "ModifiedTS AS LastModified")
+
+        # Perform query.
         cur = conn.execute(f"SELECT {columns} FROM Packages "
-                           "INNER JOIN PackageBases "
+                           "LEFT JOIN PackageBases "
                            "ON PackageBases.ID = Packages.PackageBaseID "
+                           "LEFT JOIN Users "
+                           "ON PackageBases.MaintainerUID = Users.ID "
                            "WHERE PackageBases.PackagerUID IS NOT NULL")
 
         # Store JSON-data in `output`, which can be reused for the
@@ -94,10 +169,23 @@ def main():
                     ]
                 }
             """
-            output = [{
-                column[0]: result[i]
-                for i, column in enumerate(cur.description)
-            } for result in cur.fetchall()]
+
+            def is_decimal(column):
+                if isinstance(column, Decimal):
+                    return float(column)
+                return column
+
+            output = []
+            for result in cur.fetchall():
+                extended_fields = get_extended_fields(result[0], result[2])
+
+                item = {
+                    column[0]: is_decimal(result[i])
+                    for i, column in enumerate(cur.description)
+                }
+                item.update(extended_fields)
+                output.append(item)
+
             json.dump({
                 "warning": ("This is a experimental! It can be removed "
                             "or modified without warning!"),