1
from http.server import BaseHTTPRequestHandler
2
import os
3
import json
4
import asyncio
5
import requests
6
import datetime
7
import io
8
from telebot.async_telebot import AsyncTelebot
9
from supabase import create_client, Client
10
from telebot import types
11
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
12

13
# Initialize Bot
14
BOT_TOKEN = os.environ.get('BOT_TOKEN')
15
bot = AsyncTelebot(BOT_TOKEN)
16

17
# Initialize Supabase
18
SUPABASE_URL = os.environ.get("SUPABASE_URL")
19
SUPABASE_KEY = os.environ.get("SUPABASE_SERVICE_ROLE_KEY")
20
supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
21

22
# Supabase storage bucket
23
bucket_name = "william"
24

25
def generate_start_keyboard():
26
    keyboard = InlineKeyboardMarkup()
27
    keyboard.add(InlineKeyboardButton("Open William app", web_app=WebAppInfo(url="https://williamfrontend.netlify.app/")))
28
    return keyboard
29

30
@bot.message_handler(commands=['start'])
31
async def start(message):
32
    user_id = str(message.from_user.id)
33
    user_first_name = str(message.from_user.first_name)
34
    user_last_name = message.from_user.last_name
35
    user_username = message.from_user.username
36
    user_language_code = str(message.from_user.language_code)
37
    is_premium = getattr(message.from_user, 'is_premium', False)
38
    text = message.text.split()
39

40
    welcome_message = (
41
        f"Hi, {user_first_name}! 👋
42

43
"
44
        f"Welcome to William Drop Memecoin Airdrop!
45

46
"
47
        f"Here you can earn coins by mining or by spending!
48

49
"
50
        f"Invite friends to earn more coins and level up faster! 🚀"
51
    )
52

53
    try:
54
        user_data = {
55
            'userImage': None,
56
            'firstName': user_first_name,
57
            'lastName': user_last_name,
58
            'username': user_username,
59
            'languageCode': user_language_code,
60
            'isPremium': is_premium,
61
            'referrals': {},
62
            'balance': 0,
63
            'mineRate': 0.001,
64
            'isMining': False,
65
            'miningStartedTime': None,
66
            'daily': {
67
                'claimedTime': None,
68
                'claimedDay': 0,
69
            },
70
            'links': None,
71
        }
72

73
        photos = await bot.get_user_profile_photos(user_id, limit=1)
74
        if photos.total_count > 0:
75
            file_id = photos.photos[0][-1].file_id
76
            file_info = await bot.get_file(file_id)
77
            file_path = file_info.file_path
78
            file_url = f"https://api.telegram.org/file/bot{BOT_TOKEN}/{file_path}"
79

80
            # Download the image
81
            response = requests.get(file_url)
82
            if response.status_code == 200:
83
                # Wrap content in a file-like object
84
                file_data = io.BytesIO(response.content)
85
                file_name = f"user_images/{user_id}.jpg"
86
                supabase.storage.from_(bucket_name).upload(file_name, file_data)
87

88
                # Generate public URL
89
                user_image = supabase.storage.from_(bucket_name).get_public_url(file_name)
90
                user_data['userImage'] = user_image
91

92
        # Handle referral code
93
        if len(text) > 1 and text[1].startswith('ref_'):
94
            referrer_id = text[1][4:]
95
            referrer = supabase.table('users').select("*").eq('id', referrer_id).execute()
96

97
            if referrer.data:
98
                referrer_data = referrer.data[0]
99
                user_data['referredBy'] = referrer_id
100

101
                bonus_amount = 500 if is_premium else 100
102
                new_balance = referrer_data['balance'] + bonus_amount
103
                referrals = referrer_data.get('referrals', {}) or {}
104

105
                referrals[user_id] = {
106
                    'addedValue': bonus_amount,
107
                    'firstName': user_first_name,
108
                    'lastName': user_last_name,
109
                    'userImage': user_data['userImage'],
110
                }
111

112
                # Update referrer in the database
113
                supabase.table('users').update({
114
                    'balance': new_balance,
115
                    'referrals': referrals,
116
                }).eq('id', referrer_id).execute()
117

118
        # Save user data to the database
119
        supabase.table('users').upsert(user_data).execute()
120

121
        keyboard = generate_start_keyboard()
122
        await bot.reply_to(message, welcome_message, reply_markup=keyboard)
123

124
    except Exception as e:
125
        error_message = "Error. Please try again!"
126
        await bot.reply_to(message, error_message)
127
        print(f"Error: {str(e)}")
128

129
class Handler(BaseHTTPRequestHandler):
130
    def do_POST(self):
131
        content_length = int(self.headers['Content-Length'])
132
        post_data = self.rfile.read(content_length)
133
        update_dict = json.loads(post_data.decode('utf-8'))
134

135
        asyncio.run(self.process_update(update_dict))
136

137
        self.send_response(200)
138
        self.end_headers()
139

140
    async def process_update(self, update_dict):
141
        update = types.Update.de_json(update_dict)
142
        await bot.process_new_updates([update])
143

144
    def do_GET(self):
145
        self.send_response(200)
146
        self.end_headers()
147
        self.wfile.write("Bot is running".encode())