iMessages.ts

import { settings } from '$lib/config'
import type { MessageRow } from '$lib/types'
import os from 'node:os'
import path from 'node:path'
import { open } from 'sqlite'
import sqlite3 from 'sqlite3'
import { logger } from './logger'
import { hasPartnerMessages, isoToAppleNanoseconds } from './utils'

const CHAT_DB_PATH = path.join(os.homedir(), 'Library', 'Messages', 'chat.db')

const buildQuery = (startDate: string, endDate: string) => {
    logger.debug({ startDate, endDate }, 'Getting messages')
    const params = [settings.PARTNER_PHONE, isoToAppleNanoseconds(startDate), isoToAppleNanoseconds(endDate)]

    const query = `
        SELECT
            datetime(message.date / 1000000000 + strftime('%s', '2001-01-01'), 'unixepoch') AS timestamp,
            message.text AS text,
            handle.id AS contact_id,
            message.is_from_me
        FROM message
        JOIN handle ON message.handle_id = handle.ROWID
        WHERE message.text IS NOT NULL
            AND handle.id = ?
            AND message.date >= ?
            AND message.date <= ?
        ORDER BY message.date DESC`

    return { query, params }
}

const formatMessages = (rows: MessageRow[]) => {
    return rows
        .map((row) => ({
            sender: row.is_from_me
                ? 'me'
                : row.contact_id === settings.PARTNER_PHONE
                  ? 'partner'
                  : 'unknown',
            text: row.text,
            timestamp: new Date(
                row.timestamp.endsWith('Z') ? row.timestamp : `${row.timestamp}Z`
            ).toISOString(),
        }))
        .reverse() // Reverse to get chronological order
}

export const queryMessagesDb = async (startDate: string, endDate: string) => {
    if (!settings.PARTNER_PHONE) {
        logger.warn('PARTNER_PHONE setting not configured')
        return { messages: [] }
    }

    const db = await open({ filename: CHAT_DB_PATH, driver: sqlite3.Database })
    const { query, params } = buildQuery(startDate, endDate)

    try {
        const rows = (await db.all(query, params)) as MessageRow[]
        logger.info({ count: rows.length, handleId: settings.PARTNER_PHONE }, 'Fetched messages')

        const formattedMessages = formatMessages(rows)

        // Return empty array if all messages are from me (no partner messages)
        return { messages: hasPartnerMessages(formattedMessages) ? formattedMessages : [] }
    } catch (error) {
        logger.error({ error }, 'Error querying messages database')
        return { messages: [] }
    } finally {
        await db.close()
    }
}