Update main.py

main.py

@@ -1,66 +1,53 @@
-from flask import Flask, render_template, request, jsonify
-from flask_socketio import SocketIO, emit
+from flask import Flask, request, jsonify
 import uuid
 
 app = Flask(__name__)
-app.config['SECRET_KEY'] = 'secret!'
-socketio = SocketIO(app)
 
-# Словарь для хранения соответствия токена и идентификатора сессии (socket id)
-clients = {}  # {token: sid}
+# В памяти храним зарегистрированных пользователей и сессии передачи файлов
+users = {}
+transfers = {}
 
-@app.route('/')
-def index():
-    # Главная страница с базовой информацией
-    return render_template('index.html')
-
-@app.route('/register')
+@app.route('/register', methods=['POST'])
 def register():
-    # Генерация уникального токена для нового пользователя
-    token = str(uuid.uuid4())
-    # Изначально сессия не установлена
-    clients[token] = None
-    return render_template('register.html', token=token)
-
-@app.route('/file/<token>')
-def file_info(token):
-    """
-    Эндпоинт для проверки статуса клиента по токену.
-    Если клиент (отправитель) онлайн – возвращается статус "ready",
-    иначе – "offline".
-    """
-    if token in clients and clients[token] is not None:
-        return jsonify({"status": "ready"})
-    return jsonify({"status": "offline"})
-
-@socketio.on('register')
-def handle_register(data):
-    """
-    При подключении через WebSocket клиент отправляет свой токен.
-    Здесь мы сохраняем привязку токена к session id.
-    """
-    token = data.get('token')
-    if token:
-        clients[token] = request.sid
-        emit('registered', {'message': f'Токен {token} успешно зарегистрирован.'})
-    else:
-        emit('error', {'message': 'Токен отсутствует.'})
-
-@socketio.on('signal')
-def handle_signal(data):
-    """
-    Обработка сигналов для установления P2P-соединения.
-    Клиент передаёт объект с полями:
-      - target: токен целевого клиента
-      - signal: сигнальные данные (например, offer/answer ICE)
-    Сервер пересылает сигнал нужному клиенту по session id.
-    """
-    target = data.get('target')
-    signal_data = data.get('signal')
-    if target in clients and clients[target]:
-        socketio.emit('signal', {'signal': signal_data, 'from': request.sid}, room=clients[target])
-    else:
-        emit('error', {'message': 'Целевой клиент не доступен.'})
+    data = request.json
+    username = data.get('username')
+    password = data.get('password')
+    if not username or not password:
+        return jsonify({'error': 'Не переданы обязательные поля'}), 400
+    if username in users:
+        return jsonify({'error': 'Пользователь уже существует'}), 400
+    users[username] = {'password': password}
+    return jsonify({'message': 'Регистрация прошла успешно'})
+
+@app.route('/create_transfer', methods=['POST'])
+def create_transfer():
+    data = request.json
+    filename = data.get('filename')
+    ip = data.get('ip')
+    port = data.get('port')
+    filesize = data.get('filesize', 0)
+    if not filename or not ip or not port:
+        return jsonify({'error': 'Отсутствуют необходимые параметры'}), 400
+
+    transfer_id = str(uuid.uuid4())
+    transfers[transfer_id] = {
+        'filename': filename,
+        'ip': ip,
+        'port': port,
+        'filesize': filesize,
+        'downloaded': False
+    }
+    # Формируем ссылку для скачивания. request.host_url учитывает адрес сервера.
+    download_link = request.host_url + 'download/' + transfer_id
+    return jsonify({'download_link': download_link})
+
+@app.route('/download/<transfer_id>', methods=['GET'])
+def get_transfer(transfer_id):
+    transfer = transfers.get(transfer_id)
+    if not transfer:
+        return jsonify({'error': 'Неверный идентификатор передачи'}), 404
+    return jsonify(transfer)
 
 if __name__ == '__main__':
-    socketio.run(app, debug=True)
+    # Запускаем сервер на всех интерфейсах на порту 5000
+    app.run(host='0.0.0.0', port=5000)