# -*- coding: utf-8 -*- import os import json import uuid import hashlib import datetime import mimetypes from functools import wraps from flask import Flask, render_template_string, request, jsonify, session, redirect, url_for, make_response from werkzeug.utils import secure_filename from flask_socketio import SocketIO, emit, join_room, leave_room # NEW: برای چت آنلاین app = Flask(__name__) app.secret_key = 'chatme_secret_key_instagram_like_app_2026' app.config['SESSION_TYPE'] = 'filesystem' app.config['MAX_CONTENT_LENGTH'] = 50 * 1024 * 1024 # 50MB max # NEW: SocketIO برای چت لحظه‌ای socketio = SocketIO(app, cors_allowed_origins="*", ping_timeout=60) # ======================== # DATA STORAGE (In-memory + JSON backup) - همینطور که بود حفظ شده # ======================== DATA_FILE = 'chatme_data.json' def load_data(): default_data = { 'users': {}, 'posts': [], 'stories': [], 'chats': {}, 'notifications': [], 'viewed_stories': {}, 'post_views': {} } if os.path.exists(DATA_FILE): try: with open(DATA_FILE, 'r', encoding='utf-8') as f: data = json.load(f) for key in default_data: if key not in data: data[key] = default_data[key] return data except: return default_data return default_data def save_data(data): with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump(data, f, ensure_ascii=False, indent=2) # Initialize data DATA = load_data() if not DATA['users']: # Seed users (کامل حفظ شده) DATA['users'] = { 'gods._legend': { 'email': 'demo@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'gods._legend', 'fullName': 'Nazanin zahra', 'bio': '✨', 'avatar': None, 'private': False, 'followers': ['B.enayati', 'moien.s', 'مهدی مرادی', 'امیرحسین سرباز'], 'following': ['ferioonazi', 'B.enayati', 'moien.s', 'مهدی مرادی', 'امیرحسین سرباز'] }, 'ferioonazi': { 'email': 'feri@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'ferioonazi', 'fullName': 'Feri', 'bio': '🎬', 'avatar': None, 'private': False, 'followers': [], 'following': [] }, 'B.enayati': { 'email': 'b@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'B.enayati', 'fullName': 'B.enayati', 'bio': '', 'avatar': None, 'private': False, 'followers': [], 'following': [] }, 'moien.s': { 'email': 'moien@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'moien.s', 'fullName': 'moien.s', 'bio': '', 'avatar': None, 'private': False, 'followers': [], 'following': [] }, 'مهدی مرادی': { 'email': 'mahdi@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'مهدی مرادی', 'fullName': 'مهدی مرادی', 'bio': 'wfldingdob', 'avatar': None, 'private': False, 'followers': [], 'following': [] }, 'امیرحسین سرباز': { 'email': 'amir@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'امیرحسین سرباز', 'fullName': 'امیرحسین سرباز', 'bio': '', 'avatar': None, 'private': False, 'followers': [], 'following': [] }, 'chanelandrichie': { 'email': 'channel@chat.me', 'password': hashlib.md5('123456'.encode()).hexdigest(), 'username': 'chanelandrichie', 'fullName': 'chanelandrichie', 'bio': '...Ani, are you ok?', 'avatar': None, 'private': False, 'followers': [], 'following': [] } } DATA['posts'] = [ {'id': 1, 'user': 'chanelandrichie', 'caption': '...Ani, are you ok', 'likes': [], 'comments': [], 'imageColor': '#e84393', 'timestamp': int(datetime.datetime.now().timestamp() * 1000) - 100000, 'media': None}, {'id': 2, 'user': 'gods._legend', 'caption': '✨', 'likes': [], 'comments': [], 'imageColor': '#6c5ce7', 'timestamp': int(datetime.datetime.now().timestamp() * 1000) - 50000, 'media': None} ] DATA['notifications'] = [ {'user': 'gods._legend', 'type': 'story_share', 'text': 'Nazanin zahra حلقه فیلمی از ferioonazi ارسال کرد', 'time': int(datetime.datetime.now().timestamp() * 1000) - 600000, 'read': False}, {'user': 'gods._legend', 'type': 'story_share', 'text': 'B.enayati حلقه فیلمی از marziyeh_zand ارسال کرد', 'time': int(datetime.datetime.now().timestamp() * 1000) - 500000, 'read': False}, {'user': 'gods._legend', 'type': 'like', 'text': 'امیرحسین سرباز پیامی را پسندید', 'time': int(datetime.datetime.now().timestamp() * 1000) - 200000, 'read': True} ] save_data(DATA) def save_data_to_file(): save_data(DATA) # ======================== # NEW: WebSocket Events برای چت آنلاین و لحظه‌ای # ======================== @socketio.on('connect') def handle_connect(): if 'username' in session: join_room(session['username']) print(f"🔵 {session['username']} آنلاین شد") @socketio.on('disconnect') def handle_disconnect(): if 'username' in session: print(f"⚫ {session['username']} آفلاین شد") @socketio.on('send_message') def handle_send_message(data): """دریافت پیام و ارسال لحظه‌ای به گیرنده""" if 'username' not in session: return sender = session['username'] receiver = data.get('receiver') text = data.get('text', '').strip() if not receiver or not text: return # ذخیره پیام در DATA (همان ساختار قبلی) chat_id = f"{min(sender, receiver)}_{max(sender, receiver)}" if chat_id not in DATA['chats']: DATA['chats'][chat_id] = [] message = { 'sender': sender, 'receiver': receiver, 'text': text, 'time': int(datetime.datetime.now().timestamp() * 1000) } DATA['chats'][chat_id].append(message) # ذخیره نوتیفیکیشن برای پیام add_notification(receiver, sender, 'msg', f'{sender} به شما پیام داد 💬') save_data_to_file() # ارسال لحظه‌ای پیام به گیرنده (بدون رفرش صفحه) emit('new_message', message, room=receiver) emit('new_message', message, room=sender) @socketio.on('typing') def handle_typing(data): """نمایش وضعیت تایپ کردن در چت""" if 'username' not in session: return sender = session['username'] receiver = data.get('receiver') is_typing = data.get('is_typing', False) if receiver: emit('user_typing', {'sender': sender, 'is_typing': is_typing}, room=receiver) # ======================== # Helper Functions (همینطور که بودند حفظ شده) # ======================== def login_required(f): @wraps(f) def decorated_function(*args, **kwargs): if 'username' not in session: return jsonify({'error': 'Unauthorized', 'redirect': '/'}), 401 return f(*args, **kwargs) return decorated_function def get_current_user(): if 'username' in session: return DATA['users'].get(session['username']) return None def format_count(n): if not n: return '0' if n >= 1000000: return f"{n/1000000:.1f} میلیون" if n >= 1000: return f"{n/1000:.1f} هزار" return str(n) def add_notification(target_user, from_user, notif_type, text): if target_user == from_user: return DATA['notifications'].insert(0, { 'user': target_user, 'type': notif_type, 'text': text, 'time': int(datetime.datetime.now().timestamp() * 1000), 'read': False }) if len(DATA['notifications']) > 100: DATA['notifications'] = DATA['notifications'][:100] save_data_to_file() # ======================== # API Routes (همه APIهای اصلی حفظ شده) # ======================== @app.route('/') def index(): if 'username' in session: return render_template_string(HTML_TEMPLATE) return render_template_string(HTML_TEMPLATE) @app.route('/api/me', methods=['GET']) @login_required def get_me(): user = get_current_user() if not user: return jsonify({'error': 'User not found'}), 404 return jsonify({ 'username': user['username'], 'fullName': user['fullName'], 'bio': user['bio'], 'avatar': user['avatar'], 'private': user['private'], 'followers': user.get('followers', []), 'following': user.get('following', []), 'followersCount': len(user.get('followers', [])), 'followingCount': len(user.get('following', [])) }) @app.route('/api/login', methods=['POST']) def login(): data = request.get_json() email = data.get('email', '').strip().lower() password = data.get('password', '') hashed = hashlib.md5(password.encode()).hexdigest() for username, user in DATA['users'].items(): if user['email'] == email and user['password'] == hashed: session['username'] = username return jsonify({'success': True, 'username': username}) return jsonify({'error': 'ایمیل یا رمز عبور اشتباه است'}), 401 @app.route('/api/register', methods=['POST']) def register(): data = request.get_json() email = data.get('email', '').strip().lower() fullName = data.get('fullName', '').strip() username = data.get('username', '').strip() password = data.get('password', '') avatar = data.get('avatar', None) if not email or not fullName or not username or not password: return jsonify({'error': 'همه فیلدها الزامی است'}), 400 if username in DATA['users']: return jsonify({'error': 'نام کاربری تکراری است'}), 400 DATA['users'][username] = { 'email': email, 'password': hashlib.md5(password.encode()).hexdigest(), 'username': username, 'fullName': fullName, 'bio': '', 'avatar': avatar, 'private': False, 'followers': [], 'following': [] } save_data_to_file() return jsonify({'success': True}) @app.route('/api/logout', methods=['POST']) @login_required def logout(): session.pop('username', None) return jsonify({'success': True}) @app.route('/api/posts', methods=['GET']) @login_required def get_posts(): posts = sorted(DATA['posts'], key=lambda x: x['timestamp'], reverse=True) for post in posts: if str(post['id']) not in DATA['post_views']: DATA['post_views'][str(post['id'])] = 1000 return jsonify({ 'posts': [{ 'id': p['id'], 'user': p['user'], 'caption': p['caption'], 'likes': p['likes'], 'likesCount': len(p['likes']), 'commentsCount': len(p.get('comments', [])), 'media': p.get('media'), 'imageColor': p.get('imageColor'), 'timestamp': p['timestamp'], 'isLiked': session['username'] in p['likes'], 'views': DATA['post_views'].get(str(p['id']), 0) } for p in posts] }) @app.route('/api/posts', methods=['POST']) @login_required def create_post(): data = request.get_json() caption = data.get('caption', '') media = data.get('media', None) new_post = { 'id': int(datetime.datetime.now().timestamp() * 1000), 'user': session['username'], 'caption': caption, 'likes': [], 'comments': [], 'imageColor': '#' + format(int(uuid.uuid4().hex[:6], 16), '06x'), 'timestamp': int(datetime.datetime.now().timestamp() * 1000), 'media': media } DATA['posts'].insert(0, new_post) DATA['post_views'][str(new_post['id'])] = 0 save_data_to_file() return jsonify({'success': True, 'postId': new_post['id']}) @app.route('/api/posts//like', methods=['POST']) @login_required def like_post(post_id): post = next((p for p in DATA['posts'] if p['id'] == post_id), None) if not post: return jsonify({'error': 'Post not found'}), 404 username = session['username'] if username in post['likes']: post['likes'].remove(username) liked = False else: post['likes'].append(username) liked = True if post['user'] != username: add_notification(post['user'], username, 'like', f'{username} پست شما را ❤️ پسندید') save_data_to_file() return jsonify({'success': True, 'liked': liked, 'likesCount': len(post['likes'])}) @app.route('/api/posts//comment', methods=['POST']) @login_required def add_comment(post_id): post = next((p for p in DATA['posts'] if p['id'] == post_id), None) if not post: return jsonify({'error': 'Post not found'}), 404 data = request.get_json() text = data.get('text', '').strip() if not text: return jsonify({'error': 'کامنت خالی است'}), 400 if 'comments' not in post: post['comments'] = [] post['comments'].append({ 'user': session['username'], 'text': text, 'time': int(datetime.datetime.now().timestamp() * 1000) }) if post['user'] != session['username']: add_notification(post['user'], session['username'], 'comment', f'{session["username"]} روی پست شما کامنت گذاشت 💬') save_data_to_file() return jsonify({'success': True, 'commentsCount': len(post['comments'])}) @app.route('/api/posts//comments', methods=['GET']) @login_required def get_comments(post_id): post = next((p for p in DATA['posts'] if p['id'] == post_id), None) if not post: return jsonify({'error': 'Post not found'}), 404 return jsonify({'comments': post.get('comments', [])}) @app.route('/api/posts/', methods=['DELETE']) @login_required def delete_post(post_id): post = next((p for p in DATA['posts'] if p['id'] == post_id), None) if not post or post['user'] != session['username']: return jsonify({'error': 'Unauthorized'}), 403 DATA['posts'] = [p for p in DATA['posts'] if p['id'] != post_id] save_data_to_file() return jsonify({'success': True}) @app.route('/api/stories', methods=['GET']) @login_required def get_stories(): current_user = session['username'] following = DATA['users'][current_user].get('following', []) now = int(datetime.datetime.now().timestamp() * 1000) twenty_four_hours = 24 * 60 * 60 * 1000 # Clean old stories DATA['stories'] = [s for s in DATA['stories'] if now - s['timestamp'] < twenty_four_hours] user_stories = {} for username in [current_user] + following: user_stories_list = [s for s in DATA['stories'] if s['user'] == username] if user_stories_list: user_stories[username] = sorted(user_stories_list, key=lambda x: x['timestamp'], reverse=True)[0] result = [] for username, story in user_stories.items(): viewed_key = f"{story['user']}_{story['timestamp']}" result.append({ 'user': username, 'media': story.get('media'), 'timestamp': story['timestamp'], 'viewed': DATA['viewed_stories'].get(viewed_key, False), 'likesCount': len(story.get('likes', [])), 'isLiked': session['username'] in story.get('likes', []) }) return jsonify({'stories': result}) @app.route('/api/stories', methods=['POST']) @login_required def create_story(): data = request.get_json() media = data.get('media') new_story = { 'user': session['username'], 'timestamp': int(datetime.datetime.now().timestamp() * 1000), 'likes': [], 'media': media } DATA['stories'].append(new_story) save_data_to_file() return jsonify({'success': True}) @app.route('/api/stories///view', methods=['POST']) @login_required def view_story(story_user, story_time): viewed_key = f"{story_user}_{story_time}" DATA['viewed_stories'][viewed_key] = True save_data_to_file() return jsonify({'success': True}) @app.route('/api/stories///like', methods=['POST']) @login_required def like_story(story_user, story_time): story = next((s for s in DATA['stories'] if s['user'] == story_user and s['timestamp'] == story_time), None) if not story: return jsonify({'error': 'Story not found'}), 404 username = session['username'] if 'likes' not in story: story['likes'] = [] if username in story['likes']: story['likes'].remove(username) liked = False else: story['likes'].append(username) liked = True if story['user'] != username: add_notification(story['user'], username, 'story_like', f'{username} استوری شما را ❤️ پسندید') save_data_to_file() return jsonify({'success': True, 'liked': liked, 'likesCount': len(story['likes'])}) @app.route('/api/users/', methods=['GET']) @login_required def get_user(username): user = DATA['users'].get(username) if not user: return jsonify({'error': 'User not found'}), 404 current_user = session['username'] user_posts = [p for p in DATA['posts'] if p['user'] == username] return jsonify({ 'username': user['username'], 'fullName': user['fullName'], 'bio': user['bio'], 'avatar': user['avatar'], 'private': user['private'], 'followers': user.get('followers', []), 'following': user.get('following', []), 'followersCount': len(user.get('followers', [])), 'followingCount': len(user.get('following', [])), 'postsCount': len(user_posts), 'isFollowing': current_user in user.get('followers', []), 'posts': [{ 'id': p['id'], 'caption': p['caption'], 'media': p.get('media'), 'imageColor': p.get('imageColor'), 'likesCount': len(p['likes']), 'commentsCount': len(p.get('comments', [])), 'views': DATA['post_views'].get(str(p['id']), 0) } for p in user_posts] }) @app.route('/api/users//follow', methods=['POST']) @login_required def follow_user(username): if username == session['username']: return jsonify({'error': 'Cannot follow yourself'}), 400 current_user_data = DATA['users'][session['username']] target_user_data = DATA['users'].get(username) if not target_user_data: return jsonify({'error': 'User not found'}), 404 if 'following' not in current_user_data: current_user_data['following'] = [] if session['username'] in target_user_data.get('followers', []): # Unfollow current_user_data['following'].remove(username) target_user_data['followers'].remove(session['username']) following = False else: # Follow current_user_data['following'].append(username) if 'followers' not in target_user_data: target_user_data['followers'] = [] target_user_data['followers'].append(session['username']) following = True add_notification(username, session['username'], 'follow', f'{session["username"]} شما را دنبال کرد') save_data_to_file() return jsonify({'success': True, 'following': following, 'followersCount': len(target_user_data['followers'])}) @app.route('/api/user/update', methods=['POST']) @login_required def update_profile(): data = request.get_json() user = DATA['users'][session['username']] if 'fullName' in data: user['fullName'] = data['fullName'] if 'bio' in data: user['bio'] = data['bio'] if 'avatar' in data: user['avatar'] = data['avatar'] if 'private' in data: user['private'] = data['private'] save_data_to_file() return jsonify({'success': True}) @app.route('/api/notifications', methods=['GET']) @login_required def get_notifications(): user_notifs = [n for n in DATA['notifications'] if n['user'] == session['username']] for n in user_notifs: n['read'] = True save_data_to_file() return jsonify({'notifications': user_notifs}) @app.route('/api/notifications/count', methods=['GET']) @login_required def get_notification_count(): count = len([n for n in DATA['notifications'] if n['user'] == session['username'] and not n['read']]) return jsonify({'count': count}) @app.route('/api/chats', methods=['GET']) @login_required def get_chats(): user_chats = {} for chat_user, messages in DATA['chats'].items(): if any(m['sender'] == session['username'] or m['receiver'] == session['username'] for m in messages): user_chats[chat_user] = messages result = [] for other_user, messages in user_chats.items(): last_msg = messages[-1] if messages else None result.append({ 'user': other_user, 'lastMessage': last_msg['text'] if last_msg else '', 'lastTime': last_msg['time'] if last_msg else 0 }) return jsonify({'chats': result}) @app.route('/api/chats/', methods=['GET']) @login_required def get_chat_messages(other_user): chat_id = f"{min(session['username'], other_user)}_{max(session['username'], other_user)}" messages = DATA['chats'].get(chat_id, []) return jsonify({'messages': messages}) @app.route('/api/chats/', methods=['POST']) @login_required def send_message(other_user): data = request.get_json() text = data.get('text', '').strip() if not text: return jsonify({'error': 'پیام خالی است'}), 400 chat_id = f"{min(session['username'], other_user)}_{max(session['username'], other_user)}" if chat_id not in DATA['chats']: DATA['chats'][chat_id] = [] message = { 'sender': session['username'], 'receiver': other_user, 'text': text, 'time': int(datetime.datetime.now().timestamp() * 1000) } DATA['chats'][chat_id].append(message) add_notification(other_user, session['username'], 'msg', f'{session["username"]} به شما پیام داد 💬') save_data_to_file() # NEW: ارسال لحظه‌ای پیام از طریق WebSocket socketio.emit('new_message', message, room=other_user) socketio.emit('new_message', message, room=session['username']) return jsonify({'success': True, 'message': message}) @app.route('/api/explore', methods=['GET']) @login_required def get_explore(): posts = sorted(DATA['posts'], key=lambda x: x['timestamp'], reverse=True)[:30] return jsonify({ 'posts': [{ 'id': p['id'], 'user': p['user'], 'media': p.get('media'), 'imageColor': p.get('imageColor'), 'views': DATA['post_views'].get(str(p['id']), 0) } for p in posts] }) @app.route('/api/search', methods=['GET']) @login_required def search_users(): query = request.args.get('q', '').strip().lower() if not query: return jsonify({'users': []}) results = [] for username, user in DATA['users'].items(): if query in username.lower() and username != session['username']: results.append({ 'username': username, 'fullName': user['fullName'], 'avatar': user['avatar'] }) return jsonify({'users': results[:20]}) # ============= NEW: ارسال پست به کاربر دیگر ============= @app.route('/api/send_post_to_user', methods=['POST']) @login_required def send_post_to_user(): """ارسال یک پست برای کاربر دیگر (مثل اشتراک‌گذاری پست در چت)""" data = request.get_json() target_username = data.get('target_username', '').strip() post_id = data.get('post_id') if not target_username or not post_id: return jsonify({'error': 'نام کاربری مقصد و آیدی پست الزامی است'}), 400 if target_username not in DATA['users']: return jsonify({'error': 'کاربر مورد نظر یافت نشد'}), 404 # پیدا کردن پست post = next((p for p in DATA['posts'] if p['id'] == post_id), None) if not post: return jsonify({'error': 'پست یافت نشد'}), 404 # ارسال پیام حاوی لینک پست به کاربر chat_id = f"{min(session['username'], target_username)}_{max(session['username'], target_username)}" if chat_id not in DATA['chats']: DATA['chats'][chat_id] = [] post_link_message = f"📷 {session['username']} یک پست برای شما ارسال کرد: [مشاهده پست] - {post['caption'][:50]}..." message = { 'sender': session['username'], 'receiver': target_username, 'text': post_link_message, 'post_id': post_id, # ذخیره آیدی پست برای نمایش 'time': int(datetime.datetime.now().timestamp() * 1000) } DATA['chats'][chat_id].append(message) add_notification(target_username, session['username'], 'post_share', f'{session["username"]} یک پست برای شما ارسال کرد') save_data_to_file() # ارسال لحظه‌ای socketio.emit('new_message', message, room=target_username) return jsonify({'success': True, 'message': 'پست با موفقیت ارسال شد'}) # ============= NEW: گرفتن اطلاعات کامل یک کاربر برای جستجو ============= @app.route('/api/user_full/', methods=['GET']) @login_required def get_user_full(username): """گرفتن اطلاعات کامل یک کاربر برای نمایش پیج""" user = DATA['users'].get(username) if not user: return jsonify({'error': 'User not found'}), 404 current_user = session['username'] user_posts = [p for p in DATA['posts'] if p['user'] == username] return jsonify({ 'username': user['username'], 'fullName': user['fullName'], 'bio': user['bio'], 'avatar': user['avatar'], 'private': user['private'], 'followersCount': len(user.get('followers', [])), 'followingCount': len(user.get('following', [])), 'postsCount': len(user_posts), 'isFollowing': current_user in user.get('followers', []), 'posts': [{ 'id': p['id'], 'caption': p['caption'], 'media': p.get('media'), 'imageColor': p.get('imageColor'), 'likesCount': len(p['likes']), 'timestamp': p['timestamp'] } for p in user_posts[:9]] # آخرین 9 پست }) # ======================== # HTML Template (کامل و بدون تغییر - فقط اضافه شدن بخش چت آنلاین و جستجو) # ======================== HTML_TEMPLATE = ''' ChatMe | چت‌می - اینستاگرام کامل
یا
رمز عبور را فراموش کرده‌اید؟
حساب ندارید؟ ثبت نام کنید
دریافت برنامه
Google Play
App Store
اعلان‌ها
عمومی
درخواست‌ها
اصلی
🌍 کاوش کردن
💬 پیام‌ها
0پست
0دنبال‌کننده
0دنبال‌شونده
⚙️ تنظیمات
🌙 حالت تاریک
🔒 حساب خصوصی
🗑️ پاک کردن چت‌ها
ℹ️ درباره ChatMe
''' if __name__ == '__main__': app.run()