Add a fw-all endpoint

Author: aveao

README.md

@@ -1,5 +1,7 @@
 # rebble-cohorts
-cohorts.rebble.io: The Rebble cohorts API
+cohorts.rebble.io: The Rebble cohorts API. It handles firmware delivery.
+
+For archival, use `/cohort?select=fw-all`, which returns a list of all stored firmware.
 
 ## Configuration
 

cohorts/api.py

@@ -33,6 +33,10 @@ def _latest_firmware(hardware, kind):
     )
 
 
+def _all_firmware():
+    return Firmware.query.order_by(Firmware.kind, Firmware.timestamp.desc()).all()
+
+
 FW_BEELINE_FIELDS = {
     "mobilePlatform": "user.mobile_platform",
     "mobileVersion": "user.mobile_version",
@@ -67,6 +71,12 @@ def generate_fw():
     return response
 
 
+def generate_fw_all():
+    _add_fw_beeline_context()
+    response = [row.to_json(archival=True) for row in _all_firmware()]
+    return response
+
+
 generators = {
     "pipeline-api": lambda: {"host": "pipeline-api.rebble.io"},
     "linked-services": lambda: {"enabled_providers": []},
@@ -75,6 +85,7 @@ def generate_fw():
         "version": 11,
     },
     "fw": generate_fw,
+    "fw-all": generate_fw_all,
 }
 
 

cohorts/models.py

@@ -16,14 +16,18 @@ class Firmware(db.Model):
     timestamp = db.Column(db.Integer, nullable=False)
     notes = db.Column(db.Text, nullable=True)
 
-    def to_json(self):
-        return {
+    def to_json(self, archival: bool = False):
+        result = {
             "url": self.url,
             "sha-256": self.sha256,
             "friendlyVersion": self.version,
             "timestamp": self.timestamp,
             "notes": self.notes if self.notes else self.version,
         }
+        if archival:
+            result["kind"] = self.kind
+            result["hardware"] = self.hardware
+        return result
 
 
 db.Index(