Merge pull request #2 from mercurialworld/main

add /gh user

Author: object-Object

bot/src/ghutils/cogs/app_commands/github.py

@@ -33,7 +33,7 @@
     IssueReference,
     PRReference,
 )
-from ghutils.utils.discord.transformers import FullRepositoryOption
+from ghutils.utils.discord.transformers import FullRepositoryOption, UserOption
 from ghutils.utils.discord.visibility import MessageVisibility, respond_with_visibility
 from ghutils.utils.github import (
     CommitCheckState,
@@ -161,6 +161,55 @@ async def repo(
 
         await respond_with_visibility(interaction, visibility, embed=embed)
 
+    @app_commands.command()
+    @l10n.describe_common("visibility")
+    async def user(
+        self,
+        interaction: Interaction,
+        user: UserOption,
+        visibility: MessageVisibility = "private",
+    ):
+        # Use GraphQL to get # of repos starred by user...?
+        async with self.bot.github_app(interaction) as (github, _):
+            result = await github.async_graphql(
+                """
+                query($username: String!) {
+                    user(login: $username) {
+                        starredRepositories {
+                        totalCount
+                        }
+                    }
+                }
+                """,
+                {
+                    "username": user.login,
+                },
+            )
+            num_stars: int = result["user"]["starredRepositories"]["totalCount"]
+
+        embed = (
+            Embed(
+                description=user.bio,
+                url=user.html_url,
+            )
+            .set_thumbnail(url=user.avatar_url)
+            .add_field(name="Repositories", value=user.public_repos, inline=True)
+            .add_field(name="Stars", value=num_stars, inline=True)
+        )
+
+        # In case user doesn't have a display name
+        if user.name is None:
+            embed.title = user.login
+        else:
+            embed.title = user.name
+            embed.set_author(name=user.login)
+
+        embed.set_footer(
+            text=f"{user.followers} followers • {user.following} following"
+        )
+
+        await respond_with_visibility(interaction, visibility, embed=embed)
+
     @app_commands.command()
     async def login(self, interaction: Interaction):
         user_id = interaction.user.id

bot/src/ghutils/resources/l10n/en-US/main.ftl

@@ -38,6 +38,11 @@ parameter-description_gh-commit_reference =
 command-description_gh-repo =
     Get a link to a GitHub repository.
 
+# /gh user
+
+command-description_gh-user =
+    Get a link to a GitHub user.
+
 # /gh login
 
 command-description_gh-login =

bot/src/ghutils/utils/discord/transformers.py

@@ -8,7 +8,7 @@
 from discord.app_commands.models import Choice
 from githubkit import Response
 from githubkit.exception import GitHubException, RequestFailed
-from githubkit.rest import FullRepository
+from githubkit.rest import FullRepository, PrivateUser, PublicUser
 
 from ghutils.core.bot import GHUtilsBot
 from ghutils.core.types import LoginState
@@ -81,6 +81,49 @@ async def transform(self, interaction: Interaction, value: str):
                         raise ValueError(f"Failed to get repository: {e}")
 
 
+class UserTransformer(Transformer):
+    async def transform(self, interaction: Interaction, value: str):
+        async with GHUtilsBot.github_app_of(interaction) as (github, _):
+            try:
+                return await gh_request(github.rest.users.async_get_by_username(value))
+            except GitHubException as e:
+                match e:
+                    case RequestFailed(response=Response(status_code=404)):
+                        raise ValueError("User not found")
+                    case _:
+                        logger.warning(e)
+                        raise ValueError(f"Failed to get user: {e}")
+
+    async def autocomplete(  # pyright: ignore[reportIncompatibleMethodOverride]
+        self, interaction: Interaction, value: str
+    ) -> list[Choice[str]]:
+        async with GHUtilsBot.github_app_of(interaction) as (github, state):
+            if state != LoginState.LOGGED_IN:
+                return []
+
+            # If no value given, assume the user wants their own profile
+            if not value:
+                user = await gh_request(github.rest.users.async_get_authenticated())
+                return [Choice(name=user.login, value=user.login)]
+
+            try:
+                result = await gh_request(
+                    github.rest.search.async_users(
+                        value,
+                        per_page=25,
+                    )
+                )
+            except RequestFailed:
+                return []
+            except GitHubException as e:
+                logger.warning(e)
+                return []
+
+            return [Choice(name=user.login, value=user.login) for user in result.items]
+
+
+UserOption = Transform[PrivateUser | PublicUser, UserTransformer]
+
 RepositoryNameOption = Transform[RepositoryName, RepositoryNameTransformer]
 
 FullRepositoryOption = Transform[FullRepository, FullRepositoryTransformer]