import { Prisma } from '@repo/database';
import { prisma } from './prisma';
import { logger } from '../logger';

export type ResolvedUser = Prisma.UserGetPayload<{ include: { organization: true } }>;
export type ResolvedContact = Prisma.ContactGetPayload<Record<string, never>>;
export type ResolvedOrganization = Prisma.OrganizationGetPayload<Record<string, never>>;
export type ResolvedEnrollment = Prisma.EnrollmentGetPayload<{ include: { track: true } }>;

export interface ResolvedEntity {
    user: ResolvedUser | null;
    contact: ResolvedContact | null;
    organization: ResolvedOrganization | null;
    activeEnrollment: ResolvedEnrollment | null;
}

export class EntityResolver {
    /**
     * Resolves User, Contact, Organization and Active Enrollment for a given phone number and organization.
     */
    static async resolve(phone: string, organizationId: string): Promise<ResolvedEntity> {
        try {
            const [user, contact, organization] = await Promise.all([
                prisma.user.findFirst({ 
                    where: { phone, organizationId },
                    include: { organization: true } 
                }),
                prisma.contact.findFirst({ 
                    where: { phoneNumber: phone, organizationId } 
                }),
                prisma.organization.findUnique({ 
                    where: { id: organizationId } 
                })
            ]);

            let activeEnrollment = null;
            if (user) {
                activeEnrollment = await prisma.enrollment.findFirst({
                    where: { userId: user.id, status: 'ACTIVE' },
                    include: { track: true }
                });
            }

            return {
                user,
                contact,
                organization: organization || user?.organization || null,
                activeEnrollment
            };
        } catch (err) {
            logger.error({ err, phone, organizationId }, '[EntityResolver] Failed to resolve entity');
            return { user: null, contact: null, organization: null, activeEnrollment: null };
        }
    }
}