- {% if not authenticated %}
-
Аутентификация через Telegram...
- '''
+ return render_template_string(html)
+
+@app.route('/auth/telegram', methods=['POST'])
+def auth_telegram():
+ telegram_user = None
+
+ # Mocking for local development
+ mock_user_id_str = request.args.get('mock_user_id')
+ if MOCK_TELEGRAM_MODE and mock_user_id_str:
+ try:
+ mock_user_id = int(mock_user_id_str)
+ telegram_user = {
+ "id": mock_user_id,
+ "first_name": "MockUser",
+ "username": f"mockuser{mock_user_id}",
+ "is_bot": False,
+ "language_code": "en"
+ }
+ logging.info(f"Using MOCK Telegram user: {telegram_user}")
+ except ValueError:
+ return jsonify({'status': 'error', 'message': 'Invalid mock_user_id.'}), 400
+ else:
+ # Actual Telegram authentication
+ try:
+ payload = request.get_json()
+ init_data_str = payload.get('initData')
+ except Exception as e:
+ logging.error(f"Failed to parse JSON payload: {e}")
+ return jsonify({'status': 'error', 'message': 'Invalid request format.'}), 400
+
+ if not init_data_str:
+ return jsonify({'status': 'error', 'message': 'initData is missing.'}), 400
+
+ telegram_user = check_telegram_authorization(init_data_str, BOT_API_TOKEN)
+
+ if telegram_user and telegram_user.get("id"):
+ telegram_user_id = telegram_user["id"]
+ telegram_user_id_str = str(telegram_user_id)
+
+ session['telegram_user_id'] = telegram_user_id
+ session['telegram_username'] = telegram_user.get('username', telegram_user.get('first_name', f'User{telegram_user_id}'))
+ session['telegram_first_name'] = telegram_user.get('first_name', '')
+ session['telegram_last_name'] = telegram_user.get('last_name', '')
+
+ data = load_data()
+ if telegram_user_id_str not in data['users']:
+ data['users'][telegram_user_id_str] = {
+ 'telegram_info': telegram_user, # Store full TG info
+ 'created_at': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+ 'filesystem': {
+ "type": "folder", "id": "root", "name": "root", "children": []
+ }
+ }
+ logging.info(f"New Telegram user created: {telegram_user_id_str}")
+ else:
+ # Update telegram_info if it changed
+ data['users'][telegram_user_id_str]['telegram_info'] = telegram_user
+ initialize_user_filesystem(data['users'][telegram_user_id_str]) # Ensure filesystem exists
+ logging.info(f"Existing Telegram user logged in: {telegram_user_id_str}")
+
+ try:
+ save_data(data)
+ return jsonify({'status': 'success', 'redirect_url': url_for('dashboard')})
+ except Exception as e:
+ logging.error(f"Error saving data during Telegram auth for {telegram_user_id_str}: {e}")
+ return jsonify({'status': 'error', 'message': 'Failed to save user data.'}), 500
+ else:
+ logging.warning("Telegram authentication failed or no user data returned from check.")
+ return jsonify({'status': 'error', 'message': 'Telegram authentication failed.'}), 403
+
+
+@app.route('/dashboard', methods=['GET', 'POST'])
+def dashboard():
+ if 'telegram_user_id' not in session:
+ flash('Пожалуйста, пройдите аутентификацию через Telegram.')
+ return redirect(url_for('index_page')) # Redirect to auth trigger page
+
+ telegram_user_id = str(session['telegram_user_id'])
+ telegram_username = session.get('telegram_username', f'User {telegram_user_id}')
+
+ data = load_data()
+ if telegram_user_id not in data['users']:
+ session.clear() # Clear corrupted session
+ flash('Пользователь не найден в базе данных! Пожалуйста, пройдите аутентификацию снова.')
+ return redirect(url_for('index_page'))
+
+ user_storage_entry = data['users'][telegram_user_id] # This is the dict value
+ # Ensure filesystem is initialized (should be done at auth, but double check)
+ if 'filesystem' not in user_storage_entry:
+ initialize_user_filesystem(user_storage_entry)
+
+
+ current_folder_id = request.args.get('folder_id', 'root')
+ # Filesystem is directly under user_storage_entry
+ current_folder, parent_folder = find_node_by_id(user_storage_entry['filesystem'], current_folder_id)
+
+ if not current_folder or current_folder.get('type') != 'folder':
+ flash('Папка не найдена!', 'error')
+ current_folder_id = 'root'
+ current_folder, parent_folder = find_node_by_id(user_storage_entry['filesystem'], current_folder_id)
+ if not current_folder: # Should not happen if initialized correctly
+ logging.error(f"CRITICAL: Root folder not found for user {telegram_user_id}")
+ flash('Критическая ошибка: корневая папка не найдена.', 'error')
+ session.clear()
+ return redirect(url_for('index_page'))
+
+ items_in_folder = sorted(current_folder.get('children', []), key=lambda x: (x['type'] != 'folder', x.get('name', x.get('original_filename', '')).lower()))
+
+ if request.method == 'POST':
+ if not HF_TOKEN_WRITE:
+ flash('Загрузка невозможна: токен для записи не настроен.', 'error')
+ return redirect(url_for('dashboard', folder_id=current_folder_id))
+
+ files = request.files.getlist('files')
+ if not files or all(not f.filename for f in files):
+ flash('Файлы для загрузки не выбраны.', 'error')
+ return redirect(url_for('dashboard', folder_id=current_folder_id))
+
+ if len(files) > 20:
+ flash('Максимум 20 файлов за раз!', 'error')
+ return redirect(url_for('dashboard', folder_id=current_folder_id))
+
+ target_folder_id = request.form.get('current_folder_id', 'root')
+ target_folder_node, _ = find_node_by_id(user_storage_entry['filesystem'], target_folder_id)
+
+ if not target_folder_node or target_folder_node.get('type') != 'folder':
+ flash('Целевая папка для загрузки не найдена!', 'error')
+ return redirect(url_for('dashboard'))
+
+ api = HfApi()
+ uploaded_count = 0
+ errors = []
+
+ for file_in_request in files:
+ if file_in_request and file_in_request.filename:
+ original_filename = secure_filename(file_in_request.filename)
+ name_part, ext_part = os.path.splitext(original_filename)
+ unique_suffix = uuid.uuid4().hex[:8]
+ unique_filename = f"{name_part}_{unique_suffix}{ext_part}"
+ file_id = uuid.uuid4().hex
+
+ # Use telegram_user_id in HF path
+ hf_path = f"cloud_files/{telegram_user_id}/{target_folder_id}/{unique_filename}"
+ temp_path = os.path.join(UPLOAD_FOLDER, f"{file_id}_{unique_filename}")
+
+ try:
+ file_in_request.save(temp_path)
+ commit_msg = f"User {telegram_username} (ID: {telegram_user_id}) uploaded {original_filename} to folder {target_folder_id}"
+ api.upload_file(
+ path_or_fileobj=temp_path,
+ path_in_repo=hf_path,
+ repo_id=REPO_ID,
+ repo_type="dataset",
+ token=HF_TOKEN_WRITE,
+ commit_message=commit_msg
+ )
+
+ file_info = {
+ 'type': 'file',
+ 'id': file_id,
+ 'original_filename': original_filename,
+ 'unique_filename': unique_filename,
+ 'path': hf_path,
+ 'file_type': get_file_type(original_filename),
+ 'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
}
- };
-
- {% else %}
-
- {{ content | safe if content else "" }}
- {% endif %}
-