From 02cf4c0cf23b14f8de1be3469162793f2f4f51e8 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Mon, 21 Nov 2022 21:33:02 -0800
Subject: [PATCH] Register as a Datasette plugin too, refs #4

---
 dclient/plugin.py | 8 ++++++++
 setup.py          | 8 ++++----
 2 files changed, 12 insertions(+), 4 deletions(-)
 create mode 100644 dclient/plugin.py

diff --git a/dclient/plugin.py b/dclient/plugin.py
new file mode 100644
index 0000000..7990c7f
--- /dev/null
+++ b/dclient/plugin.py
@@ -0,0 +1,8 @@
+from datasette import hookimpl
+
+
+@hookimpl
+def register_commands(cli):
+    from .cli import cli as dclient_cli
+
+    cli.add_command(dclient_cli, name="client")
diff --git a/setup.py b/setup.py
index 64cadf1..b2382c8 100644
--- a/setup.py
+++ b/setup.py
@@ -27,10 +27,10 @@ setup(
     license="Apache License, Version 2.0",
     version=VERSION,
     packages=["dclient"],
-    entry_points="""
-        [console_scripts]
-        dclient=dclient.cli:cli
-    """,
+    entry_points={
+        "datasette": ["client = dclient.plugin"],
+        "console_scripts": ["dclient = dclient.cli:cli"],
+    },
     install_requires=["click", "httpx", "appdirs"],
     extras_require={"test": ["pytest", "pytest-httpx", "cogapp", "pytest-mock"]},
     python_requires=">=3.7",