import fs from 'node:fs/promises';
import { URL } from 'node:url';
import { Transform } from 'node:stream';
import { serve } from '@hono/node-server';
import { Hono } from 'hono';
import { createServer as createNodeServer } from 'node:http';

// 创建 Hono 应用
const app = new Hono();

// Constants
const isProduction = process.env.NODE_ENV === 'production';
const port = process.env.PORT || 5173;
const base = process.env.BASE || '/';
const ABORT_DELAY = 10000;

// 解析基础路径为 URL 对象，方便后续处理
const baseUrl = new URL(base, `http://localhost:${port}`);

// Cached production assets
let templateHtml = '';
if (isProduction) {
  templateHtml = await fs.readFile('./dist/client/index.html', 'utf-8');
}

// 生产环境中间件
let compressionMiddleware;
let sirvMiddleware;
if (isProduction) {
  compressionMiddleware = (await import('compression')).default();
  sirvMiddleware = (await import('sirv')).default('./dist/client', { 
    extensions: [],
    baseUrl: base 
  });
}

// Vite 开发服务器
/** @type {import('vite').ViteDevServer | undefined} */
let vite;
if (!isProduction) {
  const { createServer } = await import('vite');
  vite = await createServer({
    server: { middlewareMode: true },
    appType: 'custom',
    base,
  });
}

// 将原有请求处理逻辑转换为 Hono 中间件
app.use(async (c, next) => {
  try {
    // 使用 c.env 获取原生请求响应对象（关键修复）
    const req = c.env.incoming;
    const res = c.env.outgoing;
    const url = new URL(req.url, `http://${req.headers.host}`);
    const path = url.pathname;

    // 检查是否匹配基础路径
    if (!path.startsWith(baseUrl.pathname)) {
      return c.text('Not found', 404);
    }

    // 处理基础路径
    const normalizedUrl = path.replace(baseUrl.pathname, '/') || '/';

    // 开发环境：使用 Vite 中间件
    if (!isProduction && vite) {
      // 使用 Vite 中间件处理请求
      const handled = await new Promise((resolve) => {
        vite.middlewares(req, res, () => resolve(false));
      });
      
      // 如果 Vite 中间件已经处理了请求，直接返回
      if (handled) {
        return;
      }
    } 
    // 生产环境：使用 compression 和 sirv 中间件
    else if (isProduction) {
      // 先尝试 compression 中间件
      const compressed = await new Promise((resolve) => {
        compressionMiddleware(req, res, () => resolve(false));
      });
      
      if (compressed) {
        return;
      }
      
      // 再尝试 sirv 中间件处理静态文件
      const served = await new Promise((resolve) => {
        sirvMiddleware(req, res, () => resolve(false));
      });
      
      if (served) {
        return;
      }
    }

    // 处理所有其他请求的 SSR 逻辑
    /** @type {string} */
    let template;
    /** @type {import('./src/entry-server.ts').render} */
    let render;
    
    if (!isProduction && vite) {
      // 开发环境：读取最新模板并转换
      template = await fs.readFile('./index.html', 'utf-8');
      template = await vite.transformIndexHtml(normalizedUrl, template);
      render = (await vite.ssrLoadModule('/src/entry-server.tsx')).render;
    } else {
      // 生产环境：使用缓存的模板
      template = templateHtml;
      render = (await import('./dist/server/entry-server.js')).render;
    }

    let didError = false;

    const { pipe, abort } = render(normalizedUrl, {
      onShellError() {
        res.writeHead(500, { 'Content-Type': 'text/html' });
        res.end('<h1>Something went wrong</h1>');
      },
      onShellReady() {
        res.writeHead(didError ? 500 : 200, { 'Content-Type': 'text/html' });

        const transformStream = new Transform({
          transform(chunk, encoding, callback) {
            res.write(chunk, encoding);
            callback();
          },
        });

        const [htmlStart, htmlEnd] = template.split(`<!--app-html-->`);

        res.write(htmlStart);

        transformStream.on('finish', () => {
          res.end(htmlEnd);
        });

        pipe(transformStream);
      },
      onError(error) {
        didError = true;
        console.error(error);
      },
    });

    // 设置超时中止
    const abortTimeout = setTimeout(() => {
      abort();
    }, ABORT_DELAY);

    // 清理超时
    res.on('finish', () => {
      clearTimeout(abortTimeout);
    });

    // 告诉 Hono 我们已经手动处理了响应
    c.res = new Response(null, { status: 200 });
  } catch (e) {
    if (!isProduction && vite) {
      vite.ssrFixStacktrace(e);
    }
    console.error(e.stack);
    return c.text(e.stack, 500);
  }
});

// 使用 Hono 的 serve 启动服务器
serve({
  fetch: app.fetch,
  createServer: createNodeServer,
  port: port,
}, () => {
  console.log(`Server started at http://localhost:${port}`);
});