lasthourcloud/apps/server/src/server.ts

import { fastifyTRPCPlugin } from '@trpc/server/adapters/fastify';
import fastify from 'fastify';
import { appRouter } from './trpc';
import { createContext } from './trpc/context';
import cors from '@fastify/cors';
import * as path from 'node:path';
import serve from '@fastify/static';
import autoLoad from '@fastify/autoload';
// import { prisma } from './prisma';
import Graceful from '@ladjs/graceful';
import { scheduler } from './scheduler';

const isDev = process.env['NODE_ENV'] === 'development';

export interface ServerOptions {
	dev?: boolean;
	port?: number;
	prefix?: string;
}

export function createServer(opts: ServerOptions) {
	const dev = opts.dev ?? true;
	const port = opts.port ?? 3000;
	const prefix = opts.prefix ?? '/trpc';
	const server = fastify({ logger: dev, trustProxy: true });
	server.register(cors);
	server.register(fastifyTRPCPlugin, {
		prefix,
		trpcOptions: {
			router: appRouter,
			createContext,
			onError({ error, type, path, input, ctx, req }) {
				console.error('Error:', error);
				if (error.code === 'INTERNAL_SERVER_ERROR') {
					// send to bug reporting
				}
			}
		}
	});
	// Serve static files in production. Static files are generated by `yarn build` in the client folder by SvelteKit.
	if (!isDev) {
		server.register(serve, {
			root: path.join(__dirname, './public'),
			preCompressed: true
		});
		server.setNotFoundHandler(async function (request, reply) {
			if (request.raw.url && request.raw.url.startsWith('/api')) {
				return reply.status(404).send({
					success: false
				});
			}
			return reply.status(200).sendFile('index.html');
		});
	}
	server.register(autoLoad, {
		dir: path.join(__dirname, 'api'),
		options: { prefix: '/api' }
	});

	const stop = () => server.close();
	const start = async () => {
		try {
			await server.listen({ host: '0.0.0.0', port });
			console.log('Coolify server is listening on port', port, 'at 0.0.0.0 🚀');
			const graceful = new Graceful({ brees: [scheduler] });
			graceful.listen();
			scheduler.run('worker');
		} catch (err) {
			server.log.error(err);
			process.exit(1);
		}
	};

	return { server, start, stop };
}