Merge pull request #11476 from MarcCoquand/add-profile-and-bookshelf

Add bookshelf and profile

Author: mekarpeles

openlibrary/plugins/openlibrary/api.py

@@ -828,6 +828,16 @@ def GET(self):
                     type="application/opds+json",
                     templated=True,
                 ),
+                Link(
+                    rel="http://opds-spec.org/shelf",
+                    href="https://archive.org/services/loans/loan/?action=user_bookshelf",
+                    type="application/opds+json",
+                ),
+                Link(
+                    rel="profile",
+                    href="https://archive.org/services/loans/loan/?action=user_profile",
+                    type="application/opds-profile+json",
+                ),
             ],
         )
         web.header('Content-Type', 'application/opds+json')
@@ -838,6 +848,8 @@ class opds_books(delegate.page):
     path = r"/opds/books/(OL\d+M)"
 
     def GET(self, edition_olid: str):
+        from pyopds2 import Link
+
         provider = get_opds_data_provider()
         resp = provider.search(query=f'edition_key:{edition_olid}')
         web.header('Content-Type', 'application/opds-publication+json')
@@ -847,9 +859,20 @@ def GET(self, edition_olid: str):
                 data=json.dumps({'error': 'Edition not found'}),
             )
 
-        return delegate.RawText(
-            json.dumps(resp.records[0].to_publication().model_dump())
-        )
+            pub = resp.records[0].to_publication()
+            pub.links += [
+                Link(
+                    rel="http://opds-spec.org/shelf",
+                    href="https://archive.org/services/loans/loan/?action=user_bookshelf",
+                    type="application/opds+json",
+                ),
+                Link(
+                    rel="profile",
+                    href="https://archive.org/services/loans/loan/?action=user_profile",
+                    type="application/opds-profile+json",
+                ),
+            ]
+            return delegate.RawText(json.dumps(pub.model_dump()))
 
 
 class opds_home(delegate.page):
@@ -934,6 +957,16 @@ def build_homepage():
                         type="application/opds+json",
                         templated=True,
                     ),
+                    Link(
+                        rel="http://opds-spec.org/shelf",
+                        href="https://archive.org/services/loans/loan/?action=user_bookshelf",
+                        type="application/opds+json",
+                    ),
+                    Link(
+                        rel="profile",
+                        href="https://archive.org/services/loans/loan/?action=user_profile",
+                        type="application/opds-profile+json",
+                    ),
                 ],
             )
             return catalog.model_dump()