Update app.py

app.py

@@ -1,58 +1,68 @@
-import os
+#!/usr/bin/env python3
 import asyncio
 import asyncssh
-from flask import Flask
+import os
+import sys
+
+# Класс сессии: обрабатывает подключение и ввод данных
+class MySSHServerSession(asyncssh.SSHServerSession):
+    def connection_made(self, chan):
+        self.chan = chan
+
+    def session_started(self):
+        self.chan.write("Добро пожаловать в SSH-сервис на Hugging Face Spaces!\n")
+        self.chan.write("Введите команду (для выхода наберите 'exit'):\n")
 
-app = Flask(__name__)
+    def data_received(self, data, datatype):
+        # Если пользователь ввёл 'exit', закрываем сессию
+        if data.strip() == 'exit':
+            self.chan.write("До свидания!\n")
+            self.chan.exit(0)
+        else:
+            # Эхо-ответ: выводим, что ввёл пользователь
+            self.chan.write("Вы ввели: " + data)
 
-SSH_PORT = 2222  # Локальный порт для SSH
-SSH_USER = "user"
-SSH_PASSWORD = "password"
+    def connection_lost(self, exc):
+        pass
 
+# Класс SSH-сервера
+class MySSHServer(asyncssh.SSHServer):
+    def session_requested(self):
+        return MySSHServerSession()
+
+# Функция для запуска сервера
 async def start_ssh_server():
-    """Запускаем SSH-сервер на Python (без root-прав)"""
-    async def handle_client(process):
-        process.stdout.write("Welcome to Python SSH Server!\n")
-        while not process.stdin.at_eof():
-            line = await process.stdin.readexactly(1)
-            process.stdout.write(line)
+    # Читаем порт из переменной окружения PORT (по умолчанию 2222)
+    port = int(os.environ.get('PORT', 2222))
+
+    # Если задан путь к файлу ключа (переменная HOST_KEY_FILE) и он существует, используем его.
+    # Иначе генерируем временный ключ.
+    host_key_file = os.environ.get('HOST_KEY_FILE')
+    if host_key_file and os.path.exists(host_key_file):
+        host_keys = [host_key_file]
+    else:
+        print("Генерация временного ключа сервера...", file=sys.stderr)
+        host_keys = [asyncssh.generate_private_key('ssh-rsa')]
 
     try:
-        await asyncssh.listen(
-            "", SSH_PORT,
-            server_host_keys=["ssh_host_key"],
-            process_factory=handle_client,
-            authorized_client_keys=None,
-            password_auth=True,
-            username_passwords={SSH_USER: SSH_PASSWORD}
+        server = await asyncssh.create_server(
+            MySSHServer, '', port,
+            server_host_keys=host_keys,
+            # Для демонстрации принимаем любые учетные данные.
+            password_authenticator=lambda username, password: True,
+            allow_pty=True
         )
-        print(f"SSH server running on port {SSH_PORT}")
-    except Exception as e:
-        print(f"Error starting SSH server: {e}")
+    except (OSError, asyncssh.Error) as exc:
+        sys.exit('Ошибка при запуске сервера: ' + str(exc))
+    
+    print('SSH-сервер запущен на порту', port)
+    await server.wait_closed()
 
-@app.route('/')
-def home():
-    """Пробрасываем SSH через Ngrok"""
+def main():
     try:
-        import pyngrok
-        ssh_tunnel = pyngrok.ngrok.connect(SSH_PORT, "tcp")
-        print(f"SSH tunnel established: {ssh_tunnel.public_url}")
-        return f"SSH доступ: {ssh_tunnel.public_url.replace('tcp://', '')}"
-    except Exception as e:
-        print(f"Error starting Ngrok: {e}")
-        return "Error starting Ngrok tunnel."
-
-if __name__ == "__main__":
-    # Генерируем SSH-ключи (если их нет)
-    if not os.path.exists("ssh_host_key"):
-        os.system("ssh-keygen -b 2048 -t rsa -f ssh_host_key -N ''")
-
-    # Запускаем SSH сервер асинхронно
-    loop = asyncio.get_event_loop()
-    loop.create_task(start_ssh_server())
-
-    # Запускаем Flask
-    try:
-        app.run(host="0.0.0.0", port=7860)
-    except Exception as e:
-        print(f"Error starting Flask app: {e}")
+        asyncio.run(start_ssh_server())
+    except (OSError, asyncssh.Error) as exc:
+        sys.exit('Ошибка: ' + str(exc))
+
+if __name__ == '__main__':
+    main()