1 files changed, 72 insertions, 0 deletions
diff --git a/lib/prserv/client.py b/lib/prserv/client.py
new file mode 100644
index 000000000..9f5794c43
--- /dev/null
+++ b/lib/prserv/client.py
@@ -0,0 +1,72 @@
+#
+# Copyright BitBake Contributors
+#
+# SPDX-License-Identifier: GPL-2.0-only
+#
+
+import logging
+import bb.asyncrpc
+from . import create_async_client
+
+logger = logging.getLogger("BitBake.PRserv")
+
+class PRAsyncClient(bb.asyncrpc.AsyncClient):
+    def __init__(self):
+        super().__init__("PRSERVICE", "1.0", logger)
+
+    async def getPR(self, version, pkgarch, checksum, history=False):
+        response = await self.invoke(
+            {"get-pr": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "history": history}}
+        )
+        if response:
+            return response["value"]
+
+    async def test_pr(self, version, pkgarch, checksum, history=False):
+        response = await self.invoke(
+            {"test-pr": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "history": history}}
+        )
+        if response:
+            return response["value"]
+
+    async def test_package(self, version, pkgarch):
+        response = await self.invoke(
+            {"test-package": {"version": version, "pkgarch": pkgarch}}
+        )
+        if response:
+            return response["value"]
+
+    async def max_package_pr(self, version, pkgarch):
+        response = await self.invoke(
+            {"max-package-pr": {"version": version, "pkgarch": pkgarch}}
+        )
+        if response:
+            return response["value"]
+
+    async def importone(self, version, pkgarch, checksum, value):
+        response = await self.invoke(
+            {"import-one": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "value": value}}
+        )
+        if response:
+            return response["value"]
+
+    async def export(self, version, pkgarch, checksum, colinfo, history=False):
+        response = await self.invoke(
+            {"export": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "colinfo": colinfo, "history": history}}
+        )
+        if response:
+            return (response["metainfo"], response["datainfo"])
+
+    async def is_readonly(self):
+        response = await self.invoke(
+            {"is-readonly": {}}
+        )
+        if response:
+            return response["readonly"]
+
+class PRClient(bb.asyncrpc.Client):
+    def __init__(self):
+        super().__init__()
+        self._add_methods("getPR", "test_pr", "test_package", "max_package_pr", "importone", "export", "is_readonly")
+
+    def _get_async_client(self):
+        return PRAsyncClient()