Virton Messenger

import os import json import uuid from datetime import datetime from flask import Flask, Response, request, jsonify from werkzeug.security import generate_password_hash, check_password_hash app = Flask(__name__) DB_FILE = 'db.json' def init_db(): if not os.path.exists(DB_FILE): with open(DB_FILE, 'w') as f: json.dump({ "users": {}, "chatrooms": {}, "messages": {} }, f, indent=4) def read_db(): with open(DB_FILE, 'r') as f: return json.load(f) def write_db(data): with open(DB_FILE, 'w') as f: json.dump(data, f, indent=4) @app.route('/') def index(): html_content = ''' Virton Messenger

Чаты

Пользователи

Тёмная тема

Выберите чат

Начните общение или просмотрите список пользователей

''' return Response(html_content, mimetype='text/html') @app.route('/api/users', methods=['GET']) def get_users(): db = read_db() users_list = [] for address, user_data in db['users'].items(): users_list.append({ 'address': address, 'username': user_data.get('username') }) return jsonify({'users': sorted(users_list, key=lambda x: x.get('username') or 'zzzz')}) @app.route('/api/user_data', methods=['POST']) def get_user_data(): data = request.get_json() address = data.get('address') if not address: return jsonify({'error': 'Address is required'}), 400 db = read_db() user_info = db['users'].get(address) username = user_info.get('username') if user_info else None return jsonify({'username': username}) @app.route('/api/set_username', methods=['POST']) def set_username(): data = request.get_json() address = data.get('address') username = data.get('username') if not address or not username: return jsonify({'error': 'Address and username are required'}), 400 if len(username) < 3 or len(username) > 20: return jsonify({'error': 'Username must be between 3 and 20 characters'}), 400 db = read_db() if address not in db['users']: db['users'][address] = {} db['users'][address]['username'] = username write_db(db) return jsonify({'success': True}) @app.route('/api/chatrooms', methods=['GET']) def get_chatrooms(): db = read_db() chatrooms_list = [] for room_id, room_data in db['chatrooms'].items(): chatrooms_list.append({ 'id': room_id, 'name': room_data['name'], 'is_private': room_data['is_private'] }) return jsonify({'chatrooms': sorted(chatrooms_list, key=lambda x: x['name'])}) @app.route('/api/create_chatroom', methods=['POST']) def create_chatroom(): data = request.get_json() name = data.get('name') password = data.get('password') creator_address = data.get('creator_address') if not name or not creator_address: return jsonify({'error': 'Name and creator address are required'}), 400 db = read_db() room_id = str(uuid.uuid4()) db['chatrooms'][room_id] = { 'name': name, 'creator': creator_address, 'is_private': bool(password), 'password_hash': generate_password_hash(password) if password else None } db['messages'][room_id] = [] write_db(db) return jsonify({'success': True, 'chatroom_id': room_id}) @app.route('/api/join_chatroom', methods=['POST']) def join_chatroom(): data = request.get_json() chatroom_id = data.get('chatroom_id') password = data.get('password') db = read_db() chatroom = db['chatrooms'].get(chatroom_id) if not chatroom: return jsonify({'error': 'Chatroom not found'}), 404 if chatroom['is_private']: if not password or not check_password_hash(chatroom['password_hash'], password): return jsonify({'error': 'Invalid password'}), 403 return jsonify({'success': True}) @app.route('/api/messages/', methods=['GET']) def get_messages(chatroom_id): db = read_db() if chatroom_id not in db['messages']: return jsonify({'error': 'Chatroom not found'}), 404 messages_with_names = [] room_messages = db['messages'].get(chatroom_id, []) for msg in room_messages: sender_address = msg['sender_address'] user_info = db['users'].get(sender_address) display_name = (user_info.get('username') if user_info and user_info.get('username') else f"{sender_address[:4]}...{sender_address[-4:]}") msg_copy = msg.copy() msg_copy['display_name'] = display_name messages_with_names.append(msg_copy) return jsonify({'messages': messages_with_names}) @app.route('/api/send_message', methods=['POST']) def send_message(): data = request.get_json() chatroom_id = data.get('chatroom_id') sender_address = data.get('sender_address') text = data.get('text') if not all([chatroom_id, sender_address, text]): return jsonify({'error': 'Missing data'}), 400 if len(text) > 1000: return jsonify({'error': 'Message is too long'}), 400 db = read_db() if chatroom_id not in db['messages']: return jsonify({'error': 'Chatroom not found'}), 404 message = { 'id': str(uuid.uuid4()), 'sender_address': sender_address, 'text': text, 'timestamp': datetime.utcnow().isoformat() + "Z" } if len(db['messages'][chatroom_id]) >= 100: db['messages'][chatroom_id].pop(0) db['messages'][chatroom_id].append(message) write_db(db) return jsonify({'success': True}) if __name__ == '__main__': init_db() app.run(host='0.0.0.0', port=7860)

Virton

Чаты

Пользователи

Выберите чат

Создать новый чат

Вход в приватный чат

Профиль

Сканировать QR-код