38501-vm/backend/src/db/api/leads.js

const db = require('../models');
const FileDBApi = require('./file');
const crypto = require('crypto');
const Utils = require('../utils');

const Sequelize = db.Sequelize;
const Op = Sequelize.Op;

module.exports = class LeadsDBApi {
    static async create(data, options) {
        const currentUser = (options && options.currentUser) || { id: null };
        const transaction = (options && options.transaction) || undefined;

        const leads = await db.leads.create(
            {
                id: data.id || undefined,
                keyword: data.keyword || null,
                description: data.description || null,
                urgency: data.urgency || null,
                status: data.status || 'SUBMITTED',
                contact_name: data.contact_name || null,
                contact_phone: data.contact_phone || null,
                contact_email: data.contact_email || null,
                address: data.address || null,
                city: data.city || null,
                state: data.state || null,
                zip: data.zip || null,
                lat: data.lat || null,
                lng: data.lng || null,
                inferred_tags_json: data.inferred_tags_json || null,
                tenant_key: data.tenant_key || null,
                created_at_ts: data.created_at_ts || null,
                updated_at_ts: data.updated_at_ts || null,
                importHash: data.importHash || null,
                createdById: currentUser.id,
                updatedById: currentUser.id,
            },
            { transaction },
        );

        await leads.setUser( data.user || currentUser.id, { transaction });
        await leads.setCategory( data.category || null, { transaction });

        return leads;
    }

    static async findAll(filter, options) {
        const limit = filter.limit || 0;
        let offset = 0;
        let where = {};
        const currentPage = +filter.page;
        offset = currentPage * limit;

        const currentUser = options?.currentUser;
        const transaction = (options && options.transaction) || undefined;

        // Data Isolation
        if (currentUser && currentUser.app_role) {
            const roleName = currentUser.app_role.name;
            if (roleName === 'Consumer') {
                where.userId = currentUser.id;
            } else if (roleName === 'Verified Business Owner') {
                // Business owners see leads matched to them
                if (currentUser.businessId) {
                    where['$lead_matches_lead.businessId$'] = currentUser.businessId;
                } else {
                    where['$lead_matches_lead.business.owner_userId$'] = currentUser.id;
                }
            }
        }

        let include = [
            { model: db.users, as: 'user' },
            { model: db.categories, as: 'category' },
            { 
                model: db.lead_matches, 
                as: 'lead_matches_lead',
                include: [{ model: db.businesses, as: 'business' }]
            }
        ];

        // Apply filters
        if (filter) {
            if (filter.id) where.id = Utils.uuid(filter.id);
            if (filter.keyword) where.keyword = { [Op.iLike]: `%${filter.keyword}%` };
            if (filter.status) where.status = filter.status;
        }

        const queryOptions = {
            where,
            include,
            distinct: true,
            subQuery: false,
            limit: options?.countOnly ? undefined : (limit ? Number(limit) : undefined),
            offset: options?.countOnly ? undefined : (offset ? Number(offset) : undefined),
            order: [['createdAt', 'desc']],
            transaction
        };

        const { rows, count } = await db.leads.findAndCountAll(queryOptions);

        return {
            rows: options?.countOnly ? [] : rows,
            count: count
        };
    }

    static async findAllAutocomplete(query, limit, offset) {
        let where = {};
        if (query) {
            where = {
                [Op.or]: [
                    { ['id']: Utils.uuid(query) },
                    { ['keyword']: { [Op.iLike]: `%${query}%` } },
                ],
            };
        }

        const records = await db.leads.findAll({
            attributes: ['id', 'keyword'],
            where,
            limit: limit ? Number(limit) : undefined,
            offset: offset ? Number(offset) : undefined,
            order: [['keyword', 'ASC']],
        });

        return records.map((record) => ({
            id: record.id,
            label: record.keyword,
        }));
    }

    static async findBy(where, options) {
        const transaction = (options && options.transaction) || undefined;
        const leads = await db.leads.findOne({ 
            where,
            include: [
                { model: db.users, as: 'user' },
                { model: db.categories, as: 'category' },
                { 
                    model: db.lead_matches, 
                    as: 'lead_matches_lead',
                    include: [{ model: db.businesses, as: 'business' }]
                },
                { model: db.messages, as: 'messages_lead' },
                { 
                  model: db.lead_photos, 
                  as: 'lead_photos_lead',
                  include: [{ model: db.file, as: 'photos' }]
                },
                { 
                  model: db.lead_events, 
                  as: 'lead_events_lead',
                  include: [{ model: db.users, as: 'actor_user' }]
                }
            ],
            transaction 
        });
        return leads ? leads.get({plain: true}) : null;
    }

    static async update(id, data, options) {
        const currentUser = (options && options.currentUser) || {id: null};
        const transaction = (options && options.transaction) || undefined;
        const leads = await db.leads.findByPk(id, {transaction});
        if (!leads) return null;

        const updatePayload = { ...data, updatedById: currentUser.id };
        await leads.update(updatePayload, {transaction});

        if (data.user) await leads.setUser(data.user, { transaction });
        if (data.category) await leads.setCategory(data.category, { transaction });

        return leads;
    }

    static async remove(id, options) {
        const currentUser = (options && options.currentUser) || {id: null};
        const transaction = (options && options.transaction) || undefined;
        const leads = await db.leads.findByPk(id, options);
        await leads.update({ deletedBy: currentUser.id }, { transaction });
        await leads.destroy({ transaction });
        return leads;
    }

    static async deleteByIds(ids, options) {
        const currentUser = (options && options.currentUser) || { id: null };
        const transaction = (options && options.transaction) || undefined;
        const leads = await db.leads.findAll({ where: { id: { [Op.in]: ids } }, transaction });
        for (const record of leads) {
            await record.update({deletedBy: currentUser.id}, {transaction});
            await record.destroy({transaction});
        }
        return leads;
    }
};