Add missing endpoint urls

kiliczsh · Apr 22, 2023 · 6b8512f · 6b8512f
1 parent 2cca134
commit 6b8512f
Show file tree

Hide file tree

Showing 3 changed files with 101 additions and 5 deletions.
diff --git a/gokyuzu/BlueskyEndpoints.py b/gokyuzu/BlueskyEndpoints.py
@@ -10,19 +10,49 @@ def get_url(self, path):
     # com.atproto.server
     def createAccount(self):
         return self.get_url("/xrpc/com.atproto.server.createAccount")
-
+
+    def getAccountInviteCodes(self):
+        return self.get_url("/xrpc/com.atproto.server.getAccountInviteCodes")
+
+    def createInviteCode(self):
+        return self.get_url("/xrpc/com.atproto.server.createInviteCode")
+
+    def createInviteCodes(self):
+        return self.get_url("/xrpc/com.atproto.server.createInviteCodes")
+
+    def describeServer(self):
+        return self.get_url("/xrpc/com.atproto.server.describeServer")
+
+    def deleteAccount(self):
+        return self.get_url("/xrpc/com.atproto.server.deleteAccount")
+
+    def resetPassword(self):
+        return self.get_url("/xrpc/com.atproto.server.resetPassword")
+
+    def requestPasswordReset(self):
+        return self.get_url("/xrpc/com.atproto.server.requestPasswordReset")
+
     def createSession(self):
         return self.get_url("/xrpc/com.atproto.server.createSession")
 
+    def deleteSession(self):
+        return self.get_url("/xrpc/com.atproto.server.deleteSession")
+
     def getSession(self):
         return self.get_url("/xrpc/com.atproto.server.getSession")
 
-    def getAccountInviteCodes(self):
-        return self.get_url("/xrpc/com.atproto.server.getAccountInviteCodes")
+    def refreshSession(self):
+        return self.get_url("/xrpc/com.atproto.server.refreshSession")
 
-    def describeServer(self):
-        return self.get_url("/xrpc/com.atproto.server.describeServer")
+    def createAppPassword(self):
+        return self.get_url("/xrpc/com.atproto.server.createAppPassword")
+
+    def listAppPassword(self):
+        return self.get_url("/xrpc/com.atproto.server.listAppPassword")
 
+    def revokeAppPassword(self):
+        return self.get_url("/xrpc/com.atproto.server.revokeAppPassword")
+
     # app.bsky.actor
     def getProfile(self):
         return self.get_url("/xrpc/app.bsky.actor.getProfile")
@@ -51,27 +81,62 @@ def getFollowers(self):
     def getFollows(self):
         return self.get_url("/xrpc/app.bsky.graph.getFollows")
 
+    def getMutes(self):
+        return self.get_url("/xrpc/app.bsky.graph.getMutes")
+
+    def muteActor(self):
+        return self.get_url("/xrpc/app.bsky.graph.muteActor")
+
+    def unmuteActor(self):
+        return self.get_url("/xrpc/app.bsky.graph.unmuteActor")
+
     # app.bsky.notification
     def listNotifications(self):
         return self.get_url("/xrpc/app.bsky.notification.listNotifications")
 
     def updateSeen(self):
         return self.get_url("/xrpc/app.bsky.notification.updateSeen")
 
+    def getUnreadCount(self):
+        return self.get_url("/xrpc/app.bsky.notification.getUnreadCount")
+
     # com.atproto.repo
+    def applyWrites(self):
+        return self.get_url("/xrpc/com.atproto.repo.applyWrites")
+
     def createRecord(self):
         return self.get_url("/xrpc/com.atproto.repo.createRecord")
 
     def deleteRecord(self):
         return self.get_url("/xrpc/com.atproto.repo.deleteRecord")
 
+    def describeRepo(self):
+        return self.get_url("/xrpc/com.atproto.repo.describeRepo")
+
+    def getRecord(self):
+        return self.get_url("/xrpc/com.atproto.repo.getRecord")
+
+    def listRecords(self):
+        return self.get_url("/xrpc/com.atproto.repo.listRecords")
+
+    def putRecord(self):
+        return self.get_url("/xrpc/com.atproto.repo.putRecord")
+
+    def uploadBlob(self):
+        return self.get_url("/xrpc/com.atproto.repo.uploadBlob")
+
     # com.atproto.identity
     def resolveHandle(self):
         return self.get_url("/xrpc/com.atproto.identity.resolveHandle")
 
     def updateHandle(self):
         return self.get_url("/xrpc/com.atproto.identity.updateHandle")
 
+    # health
+    def health(self):
+        return self.get_url("/xrpc/health")
+
+    # options
     def getAllEndpoints(self):
         all = []
 
@@ -93,6 +158,28 @@ def getAllEndpoints(self):
         all.append(self.deleteRecord())
         all.append(self.resolveHandle())
         all.append(self.updateHandle())
+        all.append(self.getRecord())
+        all.append(self.listRecords())
+        all.append(self.putRecord())
+        all.append(self.uploadBlob())
+        all.append(self.applyWrites())
+        all.append(self.describeRepo())
+        all.append(self.getUnreadCount())
+        all.append(self.getFollowers())
+        all.append(self.getFollows())
+        all.append(self.getMutes())
+        all.append(self.muteActor())
+        all.append(self.unmuteActor())
+        all.append(self.describeServer())
+        all.append(self.deleteAccount())
+        all.append(self.resetPassword())
+        all.append(self.requestPasswordReset())
+        all.append(self.deleteSession())
+        all.append(self.refreshSession())
+        all.append(self.createAppPassword())
+        all.append(self.listAppPassword())
+        all.append(self.revokeAppPassword())
+        all.append(self.health())
 
         return all
 

diff --git a/gokyuzu/__init__.py b/gokyuzu/__init__.py
@@ -192,3 +192,8 @@ def updateHandle(self, handle):
 
         response = self.SESSION.postJson(self.ENDPOINTS.updateHandle(), json=request_data)
         return response
+
+    def health(self):
+        request_url = self.ENDPOINTS.health()
+        response = self.SESSION.get(request_url)
+        return response
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -6,6 +6,10 @@
 BSKY_SOCIAL_PASSWORD = str(os.environ.get('BSKY_SOCIAL_PASSWORD'))
 
 class TestBluesky(unittest.TestCase):
+    def test_health(self):
+        bsky = Bluesky(BSKY_SOCIAL_HANDLE, BSKY_SOCIAL_PASSWORD)
+        bsky.health()
+
     def test_env(self):
         self.assertNotEqual(BSKY_SOCIAL_HANDLE, '')
         self.assertNotEqual(BSKY_SOCIAL_PASSWORD, '')