Add command to queue packages

src/bot/dragonfly_services.py

@@ -157,3 +157,12 @@ async def report_package(
         """Report a package to Dragonfly."""
         data = dataclasses.asdict(report)
         await self.make_request("POST", "/report", json=data)
+
+    async def queue_package(self: Self, name: str, version: str) -> None:
+        """Add a package to the Dragonfly scan queue."""
+        data = {
+            "name": name,
+            "version": version,
+        }
+
+        await self.make_request("POST", "/package", json=data)

src/bot/exts/dragonfly/dragonfly.py

@@ -2,6 +2,7 @@
 
 import logging
 from datetime import UTC, datetime, timedelta
+from http import HTTPStatus
 from logging import getLogger
 from typing import Self
 
@@ -582,6 +583,34 @@ async def before_scan_loop(self: Self) -> None:
         """Wait until the bot is ready."""
         await self.bot.wait_until_ready()
 
+    @commands.has_role(Roles.vipyr_security)
+    @commands.hybrid_command()
+    async def queue(self: Self, ctx: commands.Context, name: str, version: str) -> None:
+        """Add a package to the Dragonfly scan queue."""
+        try:
+            await self.bot.dragonfly_services.queue_package(name=name, version=version)
+            await ctx.send(f"Successfully queued package `{name} v{version}`")
+        except aiohttp.ClientResponseError as http_error:
+            status_code = http_error.status
+
+            if status_code == HTTPStatus.NOT_FOUND:
+                await ctx.send(f"Package `{name} v{version}` was not found on PyPI")
+
+            if status_code == HTTPStatus.CONFLICT:
+                scan_results = await self.bot.dragonfly_services.get_scanned_packages(
+                    name=name,
+                    version=version,
+                )
+
+                if scan_results:
+                    embed = _build_package_scan_result_embed(scan_results[0])
+                    await ctx.send(f"Package `{name} v{version}` has already been scanned.", embed=embed)
+                else:
+                    await ctx.send(f"Package `{name} v{version}` is already waiting to be scanned.")
+        except Exception as e:
+            await ctx.send(str(e))
+            raise
+
     @commands.has_role(Roles.vipyr_security)
     @commands.command()
     async def start(self: Self, ctx: commands.Context) -> None:  # type: ignore[type-arg]