classroomio/packages/db/src/queries/notifications/organizationmember-email-notifications.ts

import {
  DEFAULT_MEMBER_EMAIL_NOTIFICATION_ROW,
  memberEmailNotificationRowToApiResponse,
  memberEmailNotificationRowToPersonalSettings,
  personalSettingsToMemberEmailNotificationRow,
  type MemberEmailNotificationRow
} from '@cio/utils/notifications';
import type { PersonalEmailNotificationSettings } from '@cio/utils/notifications';
import type { TMemberEmailNotificationPreferencesUpdate } from '@cio/utils/validation/notifications';

import { getOrganizationMemberIdByOrgAndProfile } from '../organization/organization';
import { db } from '@db/drizzle';
import * as schema from '@db/schema';
import { eq } from 'drizzle-orm';

function rowFromRecord(
  record: typeof schema.organizationmemberEmailNotifications.$inferSelect | undefined
): MemberEmailNotificationRow | null {
  if (!record) {
    return null;
  }

  return {
    newStudent: record.newStudent,
    newSubmission: record.newSubmission,
    gradingResult: record.gradingResult,
    newsfeed: record.newsfeed,
    quizAssigned: record.quizAssigned,
    cohortReminder: record.cohortReminder,
    session: record.session,
    courseCompletion: record.courseCompletion
  };
}

export async function getOrganizationMemberEmailNotifications(
  profileId: string,
  organizationId: string
): Promise<MemberEmailNotificationRow | null> {
  try {
    const organizationMemberId = await getOrganizationMemberIdByOrgAndProfile(organizationId, profileId);

    if (!organizationMemberId) {
      return null;
    }

    const [record] = await db
      .select()
      .from(schema.organizationmemberEmailNotifications)
      .where(eq(schema.organizationmemberEmailNotifications.organizationMemberId, organizationMemberId))
      .limit(1);

    return rowFromRecord(record);
  } catch (error) {
    console.error('getOrganizationMemberEmailNotifications error:', error);
    throw new Error('Failed to get organization member email notification preferences');
  }
}

export async function getOrganizationMemberEmailNotificationsForApi(profileId: string, organizationId: string) {
  const row = await getOrganizationMemberEmailNotifications(profileId, organizationId);

  return memberEmailNotificationRowToApiResponse(row);
}

export async function getPersonalEmailNotificationSettingsForOrg(
  profileId: string,
  organizationId: string
): Promise<PersonalEmailNotificationSettings | undefined> {
  const row = await getOrganizationMemberEmailNotifications(profileId, organizationId);

  return memberEmailNotificationRowToPersonalSettings(row);
}

export async function upsertOrganizationMemberEmailNotifications(
  profileId: string,
  organizationId: string,
  preferences: TMemberEmailNotificationPreferencesUpdate
): Promise<MemberEmailNotificationRow> {
  try {
    const organizationMemberId = await getOrganizationMemberIdByOrgAndProfile(organizationId, profileId);

    if (!organizationMemberId) {
      throw new Error('Organization member not found');
    }

    const existing = await getOrganizationMemberEmailNotifications(profileId, organizationId);
    const merged = personalSettingsToMemberEmailNotificationRow({
      ...(existing ?? DEFAULT_MEMBER_EMAIL_NOTIFICATION_ROW),
      ...preferences
    });
    const updatedAt = new Date().toISOString();

    const [record] = await db
      .insert(schema.organizationmemberEmailNotifications)
      .values({
        organizationMemberId,
        ...merged,
        updatedAt
      })
      .onConflictDoUpdate({
        target: schema.organizationmemberEmailNotifications.organizationMemberId,
        set: {
          ...merged,
          updatedAt
        }
      })
      .returning();

    const row = rowFromRecord(record);

    if (!row) {
      throw new Error('Failed to upsert organization member email notification preferences');
    }

    return row;
  } catch (error) {
    console.error('upsertOrganizationMemberEmailNotifications error:', error);
    throw new Error('Failed to update organization member email notification preferences');
  }
}