feat: use github search for issue and repo linking

Author: onerandomusername

monty/exts/info/github/_handlers.py

@@ -259,17 +259,24 @@ def render_full(
         return obj.name or obj.login, [text_display]
 
 
-class RepoRenderer(GitHubRenderer[githubkit.rest.Repository | githubkit.rest.FullRepository, ghretos.Repo]):
+class RepoRenderer(
+    GitHubRenderer[
+        githubkit.rest.Repository | githubkit.rest.FullRepository | githubkit.rest.RepoSearchResultItem, ghretos.Repo
+    ]
+):
     def render_tiny(
         self,
-        obj: githubkit.rest.Repository | githubkit.rest.FullRepository,
+        obj: githubkit.rest.Repository | githubkit.rest.FullRepository | githubkit.rest.RepoSearchResultItem,
         *,
         context: ghretos.Repo,
     ) -> str:
         return f"📦 [{obj.name}](<{obj.html_url}>)"
 
     def render_ogp_cv2(
-        self, obj: githubkit.rest.Repository | githubkit.rest.FullRepository, *, context: ghretos.Repo
+        self,
+        obj: githubkit.rest.Repository | githubkit.rest.FullRepository | githubkit.rest.RepoSearchResultItem,
+        *,
+        context: ghretos.Repo,
     ) -> disnake.ui.Container:
         container = disnake.ui.Container()
         text_display = disnake.ui.TextDisplay("")
@@ -290,7 +297,7 @@ def render_ogp_cv2(
 
     def render_full(
         self,
-        obj: githubkit.rest.Repository | githubkit.rest.FullRepository,
+        obj: githubkit.rest.Repository | githubkit.rest.FullRepository | githubkit.rest.RepoSearchResultItem,
         *,
         context: ghretos.Repo,
     ) -> tuple[str, list[disnake.ui.TextDisplay]]:

monty/exts/info/github/cog.py

@@ -1,17 +1,21 @@
 import asyncio
 import functools
+import json
 import operator
+import pathlib
 import random
 import re
 from collections.abc import Mapping
-from typing import Any
+from typing import Any, TypedDict
 
 import cachingutils
 import disnake
 import ghretos
 import githubkit
 import githubkit.exception
+import githubkit.rest
 from disnake.ext import commands
+from typing_extensions import Required
 
 import monty.utils.services
 from monty import constants
@@ -46,6 +50,13 @@
 log = get_logger(__name__)
 
 
+class GitHubShorthandAliases(TypedDict, total=False):
+    owner: Required[str]
+    repo: Required[str]
+    alias_for_issues: bool
+    """Whether this shorthand can be used for issues/PRs in addition to repos."""
+
+
 class GithubInfo(
     commands.Cog,
     name="GitHub Information",
@@ -59,6 +70,15 @@ class GithubInfo(
     def __init__(self, bot: Monty) -> None:
         self.bot = bot
         self.client: GitHubFetcher = GitHubFetcher(bot.github)
+        self.short_repos: dict[str, GitHubShorthandAliases] = json.loads(
+            pathlib.Path("monty/resources/repo_aliases.json").read_text()
+        )
+
+        # Validate casefold.
+        assert all(
+            repo == repo.casefold() and value.get("repo") and value.get("owner")
+            for repo, value in self.short_repos.items()
+        ), "Repository shorthand keys must be casefolded and include exactly one `/` in the data file."
 
         self.autolink_cache: cachingutils.MemoryCache[
             int, tuple[disnake.Message, dict[ghretos.GitHubResource, github_handlers.InfoSize]]
@@ -81,6 +101,40 @@ async def fetch_resource(
         """Fetch a GitHub resource."""
         return await self.client.fetch_resource(resource)
 
+    async def resolve_repo(
+        self,
+        repo: ghretos.Repo,
+        *,
+        default_user: str | None = None,
+    ) -> ghretos.Repo:
+        """Resolve the owner of a GitHub repository."""
+        if repo.owner:
+            return repo
+        if repo.name in self.short_repos:
+            return ghretos.Repo(
+                owner=self.short_repos[repo.name]["owner"],
+                name=self.short_repos[repo.name]["repo"],
+            )
+        r = await self.bot.github.rest.search.async_repos(q=(repo.name + " is:public"), per_page=20, order="desc")
+        for repo_data in r.parsed_data.items:
+            if repo_data.name.casefold() == repo.name.casefold():
+                break
+
+        else:
+            # TODO: Check if the repository belongs to the default user before returning
+            # Fallback to the default user if provided
+            if default_user:
+                return ghretos.Repo(owner=default_user, name=repo.name)
+            msg = "GitHub repository not found."
+            raise commands.UserInputError(msg)
+
+        user, repo_name = repo_data.full_name.split("/", 1)
+
+        if not isinstance(repo_data, (githubkit.rest.RepoSearchResultItem)):
+            msg = "Could not resolve repository owner."
+            raise ValueError(msg)
+        return ghretos.Repo(owner=user, name=repo_name)
+
     async def get_full_reply(
         self,
         resource: ghretos.GitHubResource,
@@ -287,11 +341,22 @@ async def parse_contents(
         for segment in context.text.split():
             match = ghretos.parse_shorthand(
                 segment,
-                default_user=default_user,
+                allow_optional_user=True,
                 settings=settings,
             )
-            if match is not None:
-                matches[match] = github_handlers.InfoSize.TINY
+            if match is None:
+                continue
+            # resolve the repo owner if needed
+            try:
+                if isinstance(match, ghretos.Repo) and not match.owner:
+                    match = await self.resolve_repo(match, default_user=default_user)
+                elif getattr(match, "repo", None) and not match.repo.owner:
+                    object.__setattr__(match, "repo", await self.resolve_repo(match.repo, default_user=default_user))
+            except commands.UserInputError:
+                continue
+
+            matches[match] = github_handlers.InfoSize.TINY
+
         for url in context.urls:
             match = ghretos.parse_url(
                 url,
@@ -403,14 +468,21 @@ async def github_user(self, ctx: commands.Context, user: str) -> None:
         )
         await ctx.send(components=components)
 
-    @github_group.command(name="repo", description="Fetch GitHub repository information.")
+    @github_group.command(name="repo", aliases=("repository", "repo_info"))
     async def github_repo(self, ctx: commands.Context, user_and_repo: str, repo: str = "") -> None:
         """Fetch GitHub repository information."""
         # validate the repo
+        await ctx.trigger_typing()
+        obj: githubkit.rest.FullRepository | githubkit.rest.RepoSearchResultItem | None = None
         if user_and_repo.count("/") == 1 and not repo:
             user, repo = user_and_repo.split("/", 1)
         else:
-            user = user_and_repo
+            repo = user_and_repo
+            # Resolve the user from the repo name when possible
+            repo_shorthand = await self.resolve_repo(ghretos.Repo(owner="", name=repo))
+            user = repo_shorthand.owner
+            repo = repo_shorthand.name
+
         if not repo:
             msg = "Repository name is required."
             raise commands.UserInputError(msg)
@@ -429,6 +501,7 @@ async def github_repo(self, ctx: commands.Context, user_and_repo: str, repo: str
                 msg = "GitHub repository not found."
                 raise commands.UserInputError(msg) from e
             raise
+
         components: list[disnake.ui.Container | disnake.ui.ActionRow] = []
         components.append(github_handlers.RepoRenderer().render_ogp_cv2(obj, context=context))
         components.append(

monty/resources/repo_aliases.json

@@ -0,0 +1,22 @@
+{
+    "k8s": {
+        "owner": "kubernetes",
+        "repo": "kubernetes",
+        "alias_for_issues": true
+    },
+    "python": {
+        "owner": "python",
+        "repo": "cpython",
+        "alias_for_issues": true
+    },
+    "monty": {
+        "owner": "onerandomusername",
+        "repo": "monty",
+        "alias_for_issues": false
+    },
+    "monty-python": {
+        "owner": "onerandomusername",
+        "repo": "monty-python",
+        "alias_for_issues": true
+    }
+}