Manage Account: {{ user.username or user.phone }}
ID: {{ user.id }} | Telegram ID: {{ user.telegram_id }} | Phone: {{ user.phone }}
'''
@app.route('/')
def index():
if 'user_id' in session:
return redirect(url_for('blabla_gram_app'))
return render_template_string(LOGIN_TEMPLATE)
@app.route('/api/login', methods=['POST'])
def api_login():
data = request.json
phone = data.get('phone')
step = data.get('step')
if not phone:
return jsonify({'success': False, 'message': 'Phone number is required.'})
session_hash = hashlib.md5(phone.encode()).hexdigest()
session_file_path = str(Path(SESSION_DIR) / f"{session_hash}.session")
session['current_login_phone'] = phone
session['current_login_session_file'] = session_file_path
async def _login_async():
client = TelegramClient(session['current_login_session_file'], API_ID, API_HASH)
result = {}
try:
await client.connect()
if step == 'start':
if await client.is_user_authorized():
me = await client.get_me()
with sqlite3.connect(DB_PATH) as conn:
c = conn.cursor()
c.execute('INSERT OR REPLACE INTO users (telegram_id, username, phone, session_file) VALUES (?, ?, ?, ?)',
(str(me.id), me.username or '', session['current_login_phone'], session['current_login_session_file']))
conn.commit()
user_db_id = c.execute('SELECT id FROM users WHERE telegram_id = ?', (str(me.id),)).fetchone()[0]
session['user_id'] = user_db_id
result = {'success': True, 'message': 'Already logged in.', 'user_id': user_db_id}
else:
sent_code = await client.send_code_request(session['current_login_phone'])
session['phone_code_hash'] = sent_code.phone_code_hash
result = {'success': True, 'message': 'Code sent. Please check your Telegram app.', 'phone_code_hash': sent_code.phone_code_hash}
elif step == 'code':
code = data.get('code')
phone_code_hash = session.get('phone_code_hash')
if not phone_code_hash:
raise ValueError('Session expired, please try again.')
try:
me = await client.sign_in(phone=session['current_login_phone'], code=code, phone_code_hash=phone_code_hash)
with sqlite3.connect(DB_PATH) as conn:
c = conn.cursor()
c.execute('INSERT OR REPLACE INTO users (telegram_id, username, phone, session_file) VALUES (?, ?, ?, ?)',
(str(me.id), me.username or '', session['current_login_phone'], session['current_login_session_file']))
conn.commit()
user_db_id = c.execute('SELECT id FROM users WHERE telegram_id = ?', (str(me.id),)).fetchone()[0]
session['user_id'] = user_db_id
result = {'success': True, 'message': 'Logged in successfully.', 'user_id': user_db_id}
except SessionPasswordNeededError:
result = {'success': False, 'password_required': True, 'message': 'Cloud password required for 2FA.'}
except FloodWaitError as e:
result = {'success': False, 'message': f'Telegram flood wait: please try again in {e.seconds} seconds.'}
except Exception as e:
result = {'success': False, 'message': f'Invalid code or other error: {e}'}
elif step == 'password':
password = data.get('password')
try:
me = await client.sign_in(password=password)
with sqlite3.connect(DB_PATH) as conn:
c = conn.cursor()
c.execute('INSERT OR REPLACE INTO users (telegram_id, username, phone, session_file) VALUES (?, ?, ?, ?)',
(str(me.id), me.username or '', session['current_login_phone'], session['current_login_session_file']))
conn.commit()
user_db_id = c.execute('SELECT id FROM users WHERE telegram_id = ?', (str(me.id),)).fetchone()[0]
session['user_id'] = user_db_id
result = {'success': True, 'message': 'Logged in with password.', 'user_id': user_db_id}
except FloodWaitError as e:
result = {'success': False, 'message': f'Telegram flood wait: please try again in {e.seconds} seconds.'}
except Exception as e:
result = {'success': False, 'message': f'Invalid password or other error: {e}'}
else:
result = {'success': False, 'message': 'Invalid step.'}
except FloodWaitError as e:
result = {'success': False, 'message': f'Telegram flood wait: please try again in {e.seconds} seconds.'}
except Exception as e:
result = {'success': False, 'message': f'An unexpected error occurred during login: {e}'}
finally:
if client.is_connected():
await client.disconnect()
return result
return jsonify(asyncio.run(_login_async()))
@app.route('/api/logout', methods=['POST'])
def api_logout():
user_id = session.get('user_id')
if user_id:
async def _logout_async():
client, error = await get_user_client(user_id)
if error: return
try:
if client.is_connected():
await client.log_out()
except Exception:
pass
finally:
if client and client.is_connected():
await client.disconnect()
asyncio.run(_logout_async())
session.clear()
return jsonify({'success': True, 'message': 'Logged out successfully.'})
@app.route('/app')
def blabla_gram_app():
if 'user_id' not in session:
return redirect(url_for('index'))
return render_template_string(BLABLAGRAM_APP_TEMPLATE)
@app.route('/api/user_chats')
def api_user_chats():
user_id = session.get('user_id')
if not user_id:
return jsonify({'success': False, 'message': 'User not logged in.'}), 401
async def _get_chats_async():
client, error = await get_user_client(user_id)
if error: return None, error
chats_info = []
try:
async for dialog in client.iter_dialogs():
title = dialog.title
chat_type = 'User'
participants = None
if isinstance(dialog.entity, User):
chat_type = 'User'
full_name = f"{dialog.entity.first_name or ''} {dialog.entity.last_name or ''}".strip()
title = full_name if full_name else "Unnamed User"
if dialog.entity.username:
title += f" (@{dialog.entity.username})"
elif isinstance(dialog.entity, Channel):
chat_type = 'Channel'
if hasattr(dialog.entity, 'participants_count'):
participants = dialog.entity.participants_count
elif isinstance(dialog.entity, Chat):
chat_type = 'Group'
if hasattr(dialog.entity, 'participants_count'):
participants = dialog.entity.participants_count
else:
title = title if title else "Unknown Chat"
chat_type = "Unknown"
initial = title[0].upper() if title else '?'
chats_info.append({
'id': dialog.id,
'title': title,
'type': chat_type,
'participants': participants,
'avatar_initial': initial
})
except Exception as e:
return None, str(e)
finally:
if client and client.is_connected():
await client.disconnect()
return chats_info, None
chats, error = asyncio.run(_get_chats_async())
if error:
return jsonify({'success': False, 'message': f"Failed to load chats: {error}"}), 500
return jsonify({'success': True, 'chats': sorted(chats, key=lambda x: x['title'])})
@app.route('/api/chat_messages/