From 795aa3ff903f2dcdd5905ee19a015fdbe23d6758 Mon Sep 17 00:00:00 2001
From: Who? <155328415+coddrago@users.noreply.github.com>
Date: Thu, 26 Dec 2024 11:43:56 +0700
Subject: [PATCH] Update heroku_info.py

---
 hikka/modules/heroku_info.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/hikka/modules/heroku_info.py b/hikka/modules/heroku_info.py
index f44e6c4..9669f55 100644
--- a/hikka/modules/heroku_info.py
+++ b/hikka/modules/heroku_info.py
@@ -33,6 +33,12 @@ class HerokuInfoMod(loader.Module):
                 validator=loader.validators.Link(),
             ),
 
+            loader.ConfigValue(
+                "pp_to_banner",
+                False,
+                validator=loader.validators.Boolean(),
+            ),
+
             loader.ConfigValue(
                 "show_heroku",
                 True,
@@ -40,6 +46,26 @@ class HerokuInfoMod(loader.Module):
             ),
         )
 
+    async def upload_pp_to_oxo(self, photo):
+        save_path = "profile_photo.jpg"
+        await self._client.download_media(photo, file=save_path)
+        try:
+            oxo = await utils.run_sync(
+                requests.post,
+                "https://0x0.st",
+                files={"file": open(save_path, 'rb')},
+                data={"secret": True},
+            )
+            return oxo.text.strip()
+        except Exception:
+            return ""
+
+    async def get_pp_for_banner(self):
+        photos = await self._client.get_profile_photos('me')
+        if photos:
+            return await self.upload_pp_to_oxo(photos[0])
+        return ""
+
     def _render_info(self, inline: bool) -> str:
         try:
             repo = git.Repo(search_parent_directories=True)
@@ -154,6 +180,8 @@ class HerokuInfoMod(loader.Module):
 
     @loader.command()
     async def infocmd(self, message: Message):
+        if self.config['pp_to_banner']:
+            self.config['banner_url'] = await self.get_pp_for_banner()
         await utils.answer_file(
             message,
             self.config["banner_url"],