From 7fc21f4dadb9c365b9c487694621f7aa43f37b84 Mon Sep 17 00:00:00 2001
From: Xavier Mitault <xavier.mitault@epitech.eu>
Date: Sat, 23 Nov 2024 15:48:01 +0000
Subject: [PATCH] Add command for admin

---
 worldtimezone/extensions/world_clock.py | 39 +++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/worldtimezone/extensions/world_clock.py b/worldtimezone/extensions/world_clock.py
index cdff405..d0ce1ab 100644
--- a/worldtimezone/extensions/world_clock.py
+++ b/worldtimezone/extensions/world_clock.py
@@ -45,6 +45,45 @@ async def setIt_autocomplete_timezone(
     )
 
 
+@lightbulb.add_checks(lightbulb.human_only)
+@lightbulb.add_checks(lightbulb.owner_only)
+@plugin.command
+@lightbulb.add_cooldown(5, 1, lightbulb.UserBucket)
+@lightbulb.option("timezone", "TimeZone", type=str, required=True, autocomplete=True)
+@lightbulb.option("member", "Member", type=hikari.Member, required=True)
+@lightbulb.command("set_owner", description="Set your TimeZone", pass_options=True)
+@lightbulb.implements(lightbulb.SlashCommand)
+async def setItOwner(
+    ctx: lightbulb.SlashContext, timezone: str, member: hikari.Member
+) -> None:
+    wcd = ctx.bot.d.world_clock_data  # pyright: ignore[reportAny]
+    assert isinstance(wcd, world_clock_data.WorldClockData)
+    assert ctx.guild_id is not None
+    user = wcd.get_member(ctx.guild_id, member.id)
+    if user is None:
+        guild = wcd.get_guild(ctx.guild_id)
+        if guild is None:
+            guild = wcd.create_guild(ctx.guild_id)
+        user = wcd.create_member(guild, member.id)
+    res = wcd.set_member_tz(user, timezone)
+    if res is False:
+        _ = await ctx.respond(
+            f"Failed, please provide a working timezone ('{timezone}' is unknow)"
+        )
+        return
+    _ = await ctx.respond(f"Your TimeZone as been set for {member.display_name}!")
+
+
+@setItOwner.autocomplete("timezone")
+async def setIt_autocomplete_timezone(
+    opt,  # pyright: ignore[reportUnknownParameterType, reportMissingParameterType]
+    inter,  # pyright: ignore[reportUnknownParameterType, reportMissingParameterType, reportUnusedParameter]
+):
+    return world_clock_data.match_timezone(
+        opt.value  # pyright: ignore[reportUnknownArgumentType, reportUnknownMemberType]
+    )
+
+
 @lightbulb.add_checks(lightbulb.human_only)
 @plugin.command
 @lightbulb.add_cooldown(5, 1, lightbulb.UserBucket)