import {
	EntityExpand,
	EntityResponseSchema,
	LegacyVersion,
} from "@autumn/shared";
import { invoicesToResponse } from "@/internal/invoices/invoiceUtils.js";
import { OrgService } from "@/internal/orgs/OrgService.js";
import { routeHandler } from "@/utils/routerUtils.js";
import { orgToVersion } from "@/utils/versionUtils/legacyVersionUtils.js";
import { parseEntityExpand } from "../entityUtils.js";
import { getEntityResponse } from "../getEntityUtils.js";

export const handleGetEntity = async (req: any, res: any) =>
	routeHandler({
		req,
		res,
		action: "getEntity",
		handler: async (req, res) => {
			const entityId = req.params.entity_id as string;
			const customerId = req.params.customer_id as string;
			const expand = parseEntityExpand(req.query.expand);

			const { orgId, env, db, logger, features } = req;

			const org = await OrgService.getFromReq(req);
			const apiVersion = orgToVersion({
				org,
				reqApiVersion:
					req.apiVersion >= LegacyVersion.v1_1
						? req.apiVersion
						: LegacyVersion.v1_2,
			});

			// const start = performance.now();
			const { entities, customer, fullEntities, invoices } =
				await getEntityResponse({
					db,
					entityIds: [entityId],
					org,
					env,
					customerId,
					expand,
					entityId,
					apiVersion,
					features,
					logger,
				});

			const entity = entities[0];
			const withInvoices = expand.includes(EntityExpand.Invoices);

			res.status(200).json(
				EntityResponseSchema.parse({
					...entity,
					invoices: withInvoices
						? invoicesToResponse({
								invoices: invoices || [],
								logger,
							})
						: undefined,
				}),
			);
		},
	});