Update server.js

server.js

@@ -5,101 +5,22 @@ import path from 'path';
 import fetch from 'node-fetch';
 import { glob } from 'glob';
 import { createProxyMiddleware } from 'http-proxy-middleware';
-// zlib пока не нужен, если не модифицируем сжатые ответы вручную
-// import zlib from 'zlib';
 
 // --- Конфигурация ---
-const APP_PORT = parseInt(process.env.PORT || '7860', 10);
+const APP_PORT = parseInt(process.env.PORT || '7860', 10); // HF устанавливает PORT
 const INTERNAL_TELEGRAM_API_PORT = parseInt(process.env.INTERNAL_TELEGRAM_API_PORT || '8081', 10);
 const TELEGRAM_DATA_DIR = process.env.TELEGRAM_DATA_DIR || '/var/lib/telegram-bot-api';
 const FILES_TTL_HOURS = parseInt(process.env.FILES_TTL || '-1', 10);
 const PROXY_TELEGRAM_PATH_PREFIX = process.env.PROXY_TELEGRAM_PATH_PREFIX || '/telegram';
 
+
 const GITHUB_USERNAME = process.env.GITHUB_USERNAME || '';
 const GITHUB_TOKEN = process.env.GITHUB_TOKEN || '';
 const ENV_GIST_ID = process.env.ENV_GIST_ID || '';
 
 const app = express();
 
-// --- Вспомогательные функции (остаются без изменений) ---
-function formatBytes(bytes, decimals = 2) { /* ... */ }
-async function getDirectoryStats(dirPath) { /* ... */ }
-async function cleanupOldFiles(dirPath, ttlHours) { /* ... */ }
-async function updateEnvGistInGithub() { /* ... */ }
-// Копипаст функций formatBytes, getDirectoryStats, cleanupOldFiles, updateEnvGistInGithub из предыдущего ответа
-
-// --- Маршруты для инструментов (остаются без изменений) ---
-const toolsRouter = express.Router();
-toolsRouter.get('/', (req, res) => { /* ... */ });
-toolsRouter.get('/stats', async (req, res) => { /* ... */ });
-toolsRouter.get('/file/:filepath(*)', (req, res) => { /* ... */ });
-app.use('/tools', toolsRouter);
-// Копипаст маршрутов toolsRouter из предыдущего ответа
-
-
-// --- Обратный прокси для Telegram Bot API (МАКСИМАЛЬНО УПРОЩЕННЫЙ) ---
-const telegramApiProxy = createProxyMiddleware({
-  target: `http://localhost:${INTERNAL_TELEGRAM_API_PORT}`,
-  changeOrigin: true,
-  // selfHandleResponse: false, // Убираем или комментируем
-  pathRewrite: (path, req) => {
-    const newPath = path.startsWith(PROXY_TELEGRAM_PATH_PREFIX) ? path.substring(PROXY_TELEGRAM_PATH_PREFIX.length) : path;
-    console.log(`[Proxy PathRewrite] Original: ${path}, New: ${newPath}`);
-    return newPath;
-  },
-  onProxyReq: (proxyReq, req, res) => {
-    console.log(`[Proxy Req] To Telegram API: ${req.method} ${proxyReq.path}`); // proxyReq.path уже будет после pathRewrite
-  },
-  onProxyRes: (proxyRes, req, res) => {
-    // Этот обработчик теперь будет получать ответ, который УЖЕ отправляется клиенту.
-    // Мы не можем здесь легко модифицировать тело, если selfHandleResponse: false.
-    // Но мы можем логировать.
-    console.log(`[Proxy Res] From Telegram API: Status ${proxyRes.statusCode}`);
-  },
-  onError: (err, req, res) => {
-    console.error('[Proxy] onError:', err);
-    if (res && !res.headersSent) { // Добавим проверку на res
-        res.writeHead(502, { 'Content-Type': 'application/json' });
-        res.end(JSON.stringify({ error: 'Proxy Connection Error', message: err.message }));
-    } else if (res && !res.writableEnded) { // Добавим проверку на res
-        res.end();
-    }
-  }
-});
-app.use(PROXY_TELEGRAM_PATH_PREFIX, telegramApiProxy);
-
-
-// --- Корневой маршрут ---
-app.get('/', (req, res) => {
-  res.redirect('/tools');
-});
-
-// --- Запуск сервера ---
-app.listen(APP_PORT, () => {
-  console.log(`Main app (Reverse Proxy & Tools) listening on port ${APP_PORT}`);
-  console.log(`Telegram Bot API (internal) should be on port ${INTERNAL_TELEGRAM_API_PORT}`);
-  console.log(`Tools available at /tools`);
-  console.log(`Telegram API proxied from ${PROXY_TELEGRAM_PATH_PREFIX}/`);
-
-  updateEnvGistInGithub().catch(console.error);
-
-  if (FILES_TTL_HOURS > 0) {
-    const ttlIntervalMs = FILES_TTL_HOURS * 60 * 60 * 1000;
-    console.log(`[TTL] Scheduling cleanup every ${FILES_TTL_HOURS} hours.`);
-    setTimeout(() => {
-        cleanupOldFiles(TELEGRAM_DATA_DIR, FILES_TTL_HOURS).catch(console.error);
-        setInterval(() => cleanupOldFiles(TELEGRAM_DATA_DIR, FILES_TTL_HOURS).catch(console.error), ttlIntervalMs);
-    }, 5 * 60 * 1000);
-  }
-});
-
-function gracefulShutdown() { console.log('Received shutdown signal. Exiting...'); process.exit(0); }
-process.on('SIGINT', gracefulShutdown);
-process.on('SIGTERM', gracefulShutdown);
-
-// --- КОПИПАСТ ВСПОМОГАТЕЛЬНЫХ ФУНКЦИЙ И РОУТОВ TOOLS ---
-// (чтобы не потерять их, если вы копируете только этот блок)
-
+// --- Вспом��гательные функции (formatBytes, getDirectoryStats, cleanupOldFiles) ---
 function formatBytes(bytes, decimals = 2) {
   if (bytes === 0) return '0 Bytes';
   const k = 1024;
@@ -161,6 +82,7 @@ async function cleanupOldFiles(dirPath, ttlHours) {
   return result;
 }
 
+// --- Обновление Gist ---
 async function updateEnvGistInGithub() {
   if (!GITHUB_USERNAME || !GITHUB_TOKEN || !ENV_GIST_ID) {
     console.warn('Gist update skipped: GITHUB_USERNAME, GITHUB_TOKEN, or ENV_GIST_ID is not set.');
@@ -168,8 +90,9 @@ async function updateEnvGistInGithub() {
   }
   try {
     const spaceId = process.env.SPACE_ID || 'N/A';
-    const spaceHost = process.env.SPACE_HOST || '';
+    const spaceHost = process.env.SPACE_HOST || 'N/A'; // e.g., your-user-your-space.hf.space
     const publicSpaceUrl = spaceHost ? `https://${spaceHost}` : 'N/A';
+
     const gistFilename = `hf_space_info.json`;
     const content = {
       last_updated: new Date().toISOString(),
@@ -199,7 +122,10 @@ async function updateEnvGistInGithub() {
   } catch (error) { console.error('Error updating Gist:', error.message); }
 }
 
+// --- Маршруты для инструментов ---
+const toolsRouter = express.Router();
 toolsRouter.get('/', (req, res) => {
+  const spaceHost = process.env.SPACE_HOST || 'your-space.hf.space';
   res.send(`Telegram API Proxy & Tools. Stats: /tools/stats. File base: /tools/file/. Telegram API via: ${PROXY_TELEGRAM_PATH_PREFIX}/bot<TOKEN>/...`);
 });
 toolsRouter.get('/stats', async (req, res) => {
@@ -217,10 +143,57 @@ toolsRouter.get('/file/:filepath(*)', (req, res) => {
   if (!absoluteRequestedPath.startsWith(path.resolve(TELEGRAM_DATA_DIR))) return res.status(403).send('Forbidden.');
   if (fssync.existsSync(absoluteRequestedPath) && fssync.statSync(absoluteRequestedPath).isFile()) {
     res.sendFile(absoluteRequestedPath, err => {
-      if (err) {
-        console.error(`Error sending file ${absoluteRequestedPath}:`, err);
-        if (!res.headersSent) res.status(500).send('Error sending file.');
-      }
+      if (err && !res.headersSent) res.status(500).send('Error sending file.');
     });
   } else res.status(404).send('File not found.');
-});
+});
+app.use('/tools', toolsRouter);
+
+// --- Обратный прокси для Telegram Bot API ---
+app.use(PROXY_TELEGRAM_PATH_PREFIX, createProxyMiddleware({
+  target: `http://localhost:${INTERNAL_TELEGRAM_API_PORT}`,
+  changeOrigin: true,
+  pathRewrite: (path, req) => {
+    // Убираем префикс PROXY_TELEGRAM_PATH_PREFIX из пути перед отправкой на Telegram API
+    // Например, /telegram/bot<TOKEN>/getMe -> /bot<TOKEN>/getMe
+    return path.startsWith(PROXY_TELEGRAM_PATH_PREFIX) ? path.substring(PROXY_TELEGRAM_PATH_PREFIX.length) : path;
+  },
+  onProxyReq: (proxyReq, req, res) => {
+    console.log(`[Proxy] Request to Telegram API: ${req.method} ${proxyReq.path}`);
+  },
+  onError: (err, req, res) => {
+    console.error('[Proxy] Error:', err);
+    if (!res.headersSent) {
+        res.writeHead(500, { 'Content-Type': 'application/json' });
+    }
+    res.end(JSON.stringify({ error: 'Proxy Error', message: err.message }));
+  }
+}));
+
+// --- Корневой маршрут ---
+app.get('/', (req, res) => {
+  res.redirect('/tools');
+});
+
+// --- Запуск сервера ---
+app.listen(APP_PORT, () => {
+  console.log(`Main app (Reverse Proxy & Tools) listening on port ${APP_PORT}`);
+  console.log(`Telegram Bot API (internal) should be on port ${INTERNAL_TELEGRAM_API_PORT}`);
+  console.log(`Tools available at /tools`);
+  console.log(`Telegram API proxied from ${PROXY_TELEGRAM_PATH_PREFIX}/`);
+
+  updateEnvGistInGithub().catch(console.error);
+
+  if (FILES_TTL_HOURS > 0) {
+    const ttlIntervalMs = FILES_TTL_HOURS * 60 * 60 * 1000;
+    console.log(`[TTL] Scheduling cleanup every ${FILES_TTL_HOURS} hours.`);
+    setTimeout(() => {
+        cleanupOldFiles(TELEGRAM_DATA_DIR, FILES_TTL_HOURS).catch(console.error);
+        setInterval(() => cleanupOldFiles(TELEGRAM_DATA_DIR, FILES_TTL_HOURS).catch(console.error), ttlIntervalMs);
+    }, 5 * 60 * 1000);
+  }
+});
+
+function gracefulShutdown() { console.log('Received shutdown signal. Exiting...'); process.exit(0); }
+process.on('SIGINT', gracefulShutdown);
+process.on('SIGTERM', gracefulShutdown);