Initial commit

Author: Jay24004

.gitignore

@@ -0,0 +1,5 @@
+.vercel
+.env
+.venv
+__pycache__
+.pyc

.vercelignore

@@ -0,0 +1,4 @@
+.env
+__pycache__
+.vercel
+.venv

main.py

@@ -0,0 +1,47 @@
+from flask import Flask, request, abort
+from utils.discord import create_dm, send_dm, send_webhook
+from dotenv import load_dotenv
+import os
+import pymongo
+import datetime
+
+load_dotenv()
+app = Flask(__name__)
+app.mongo = pymongo.MongoClient(os.environ['MONGO_URL'])
+app.db = app.mongo.tgk_database
+app.outh = app.db.outh
+app.votes = app.db.Votes
+
+@app.route('/')
+def index():
+    abort(404)
+
+@app.route('/api/vote', methods=['POST'])
+def vote():
+    if request.method != 'POST' or request.content_type != 'application/json' or request.headers['Authorization'] != os.environ['VOTE_PASSWORD']:
+        return "Unauthorized", 401
+    
+    user_data = app.votes.find_one({'_id': int(request.json['user'])})
+    if user_data is None:
+        user_data = {'_id': int(request.json['user']),'dm_id': None, 'votes': 0,'streak': 0,'last_vote': datetime.datetime.now(),'reminded': False}
+        dm_id = create_dm(user_data['_id'])
+        user_data['dm_id'] = int(dm_id)
+        app.votes.insert_one(user_data)
+    
+    user_data['votes'] += 1
+    if (datetime.datetime.now() - user_data['last_vote']).total_seconds() > 108000:
+        user_data['streak'] = 0
+    else:
+        user_data['streak'] += 1
+    
+    user_data['last_vote'] = datetime.datetime.now()
+    user_data['reminded'] = False
+
+    app.votes.update_one({'_id': int(request.json['user'])}, {'$set': user_data})
+    if user_data['dm_id'] is None:
+        dm_id = create_dm(user_data['_id'])
+        user_data['dm_id'] = int(dm_id)
+        app.votes.update_one({'_id': int(request.json['user'])}, {'$set': user_data})
+    send_dm(dm_id=user_data['dm_id'], total_votes=user_data['votes'], streak=user_data['streak'])
+    send_webhook(user_data['_id'], user_data['votes'], user_data['streak'])
+    return "OK", 200

requirements.txt

@@ -0,0 +1,14 @@
+certifi==2022.9.24
+charset-normalizer==2.1.1
+click==8.1.3
+colorama==0.4.6
+dnspython==2.2.1
+Flask==2.2.2
+idna==3.4
+itsdangerous==2.1.2
+Jinja2==3.1.2
+MarkupSafe==2.1.1
+pymongo==4.3.3
+requests==2.28.1
+urllib3==1.26.13
+Werkzeug==2.2.2

utils/__init__.py

@@ -0,0 +1,66 @@
+import requests
+import json
+import os
+import datetime
+
+
+headers = {
+    'Authorization': f"Bot {os.environ['DISCORD_TOKEN']}",
+    'Content-Type': 'application/json'
+}
+
+webhook_headers = {
+    'Content-Type': 'application/json'
+}
+
+base_url = 'https://discord.com/api/v10'
+
+def create_dm(user_id):
+    data = {
+        'recipient_id': user_id
+    }
+    r = requests.post(f'{base_url}/users/@me/channels', headers=headers, data=json.dumps(data))
+    return r.json()['id']
+
+def send_dm(dm_id:int, total_votes:int, streak:int):
+    embed = {
+        'type': 'rich',
+        'title': '<a:tgk_redcrown:1005473874693079071> Thank you for voting!',
+        'description': f"You have voted for [The Gambler's Kingdom](https://discord.gg/tgk) and have gotten the `࿔･ﾟ♡ TGK Voter's ♡ ࿔･ﾟ` role for 12 hours!" +"\nYou will be reminded to vote again in 12 hours (<t:" + str(int(datetime.datetime.timestamp(datetime.datetime.now() + datetime.timedelta(hours=12)))) + ":R>).",
+        'color': 8257405,
+        'url': 'https://discord.gg/yEPYYDZ3dD',
+        'footer':{
+            'text': "The Gambler's Kingdom - We appreciate your kind support!💖", 'icon_url': 'https://cdn.discordapp.com/icons/785839283847954433/a_23007c59f65faade4c973506d9e66224.gif?size=1024',
+        },
+        'fields': [
+            {'name': "Total Votes: ", 'value': f"{total_votes}", 'inline': True},
+            {'name': "Streak: ", 'value': f"{streak}", 'inline': True}
+        ]
+    }
+    payload = {'embeds': [embed]}
+
+    r = requests.post(f'{base_url}/channels/{dm_id}/messages', headers=headers, data=json.dumps(payload))
+    return r.status_code
+
+def send_webhook(user_id, total_votes, streak):
+    user = requests.get(f'{base_url}/users/{user_id}', headers=headers).json()
+    user_name = user['username'] + '#' + user['discriminator']
+    embed = {
+        'type': 'rich',
+        'title': f'<a:tgk_redcrown:1005473874693079071> Thank you for voting! {user_name}',
+        "description": f"You have voted for [The Gambler's Kingdom](https://discord.gg/tgk) and have gotten the <@&786884615192313866> role for 12 hours!" +"\nYou will be reminded to vote again in 12 hours (<t:" + str(int(datetime.datetime.timestamp(datetime.datetime.now() + datetime.timedelta(hours=12)))) + ":R>).",
+        "color": int("7dff7d", 16),
+        "url": "https://discord.gg/yEPYYDZ3dD",
+        "thumbnail": {
+                "url": "https://cdn.discordapp.com/emojis/830519601384128523.gif?v=1"
+            },
+        "fields": [
+            {"name": "Total Votes: ", "value": f"{total_votes}", "inline": True},
+            {"name": "Streak: ", "value": f"{streak}", "inline": True}
+        ],
+        "footer": {"text": "The Gambler's Kingdom - We appreciate your kind support!💖", "icon_url": "https://cdn.discordapp.com/icons/785839283847954433/a_23007c59f65faade4c973506d9e66224.gif?size=1024"},
+    }
+    payload = {'embeds': [embed], 'username': f'OCT∆NΞ Logging', 'avatar_url': 'https://cdn.discordapp.com/avatars/816699167824281621/1bf01631b86f25cb052d64b69759e8d4.png?size=4096'}
+
+    r = requests.post(os.environ['WEBHOOK_URL'], headers=webhook_headers, data=json.dumps(payload))
+    return r.status_code

utils/discord.py

@@ -0,0 +1,15 @@
+{
+    "version": 2,
+    "builds": [
+        {
+            "src": "./main.py",
+            "use": "@vercel/python"
+        }
+    ],
+    "routes": [
+        {
+            "src": "/(.*)",
+            "dest": "/"
+        }
+    ]
+}