From e2221066e817fddd83576f113ec6181eb5f100d4 Mon Sep 17 00:00:00 2001
From: OuzelTheBird <archallica@gmail.com>
Date: Sun, 16 Feb 2025 19:54:29 +0300
Subject: [PATCH] improvement: add VLESS key

---
 telegram/utils.py | 5 ++++-
 vpnworks/api.py   | 7 +++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/telegram/utils.py b/telegram/utils.py
index e4c02d6..1314ba2 100644
--- a/telegram/utils.py
+++ b/telegram/utils.py
@@ -120,6 +120,7 @@ async def send_configs(client: VpnWorksApi, message: Message, bot: Bot):
     outline_key = results.get('outline')
     amnezia_filename = results.get('amnezia')
     wireguard_filename = results.get('wireguard')
+    vless_key = results.get('vless')
     username = results.get('username')
     users_dict = await client.get_users_dict()
     person_name = users_dict.get(f'{username}').get('PersonName')
@@ -128,7 +129,9 @@ async def send_configs(client: VpnWorksApi, message: Message, bot: Bot):
 
     caption_message = (
         f'Outline key:\n'
-        f'<code>{outline_key}</code>\n'
+        f'<code>{outline_key}</code>\n\n'
+        f'VLESS key:\n'
+        f'<code>{vless_key}</code>\n'
         f'\n\n<a href="{person_link}">{person_name}</a>'
         f'\n{person_desc}\n\n'
         f'<code>{username}</code>'
diff --git a/vpnworks/api.py b/vpnworks/api.py
index d34f243..d444e85 100644
--- a/vpnworks/api.py
+++ b/vpnworks/api.py
@@ -79,6 +79,7 @@ async def create_conf_file(self):
             'amnezia': '',
             'wireguard': '',
             'outline': '',
+            'vless': '',
         }
 
         amnezia_config = data.get('AmnzOvcConfig')
@@ -105,6 +106,12 @@ async def create_conf_file(self):
             if outline_key:
                 results.update({'outline': outline_key})
 
+        vless_config = data.get('Proto0Config')
+        if vless_config:
+            vless_key = vless_config.get('AccessKey')
+            if vless_key:
+                results.update({'vless': vless_key})
+
         if not results:
             return 'No configurations provided'
         return results