import { Hono } from 'hono';
import { logger } from 'hono/logger';
import { contextStorage, getContext } from 'hono/context-storage';
import { prettyJSON } from 'hono/pretty-json';
import { cors } from 'hono/cors';

import { trpcServer } from '@hono/trpc-server';
import Redis from 'ioredis';
import redis from './redis';
import minioClient from './minio';
import { Client } from 'minio';
import oidc from './oidc/provider';
import { appRouter } from './trpc';

import { createBunWebSocket } from 'hono/bun';
import { wsHandler, wsConfig } from './socket';

type Env = {
	Variables: {
		redis: Redis;
		minio: Client;
	};
};

const app = new Hono<Env>();

app.use(
	'*',
	cors({
		origin: 'http://localhost:3001',
		credentials: true,
	}),
);

app.use('*', async (c, next) => {
	c.set('redis', redis);
	c.set('minio', minioClient);
	await next();
});
app.use('*', async (c, next) => {
	c.set('redis', redis);
	await next();
});
app.use(contextStorage());
app.use(prettyJSON()); // With options: prettyJSON({ space: 4 })
app.use(logger());
app.use(
	'/trpc/*',
	trpcServer({
		router: appRouter,
	}),
);

app.use('/oidc/*', async (c, next) => {
	// @ts-ignore
	await oidc.callback(c.req.raw, c.res.raw);
	// return void 也可以
	return;
});

// 添加 WebSocket 路由
app.get('/ws', wsHandler);
const bunServerConfig = {
	port: 3000,
	fetch: app.fetch,
	...wsConfig,
};
// 启动 Bun 服务器
Bun.serve(bunServerConfig);
export default app;