TryGhost/Ghost

const TableImporter = require('./TableImporter');
const {faker} = require('@faker-js/faker');
const {luck} = require('../utils/random');

class MembersLabelsImporter extends TableImporter {
    static table = 'members_labels';
    static dependencies = ['labels', 'members'];

    constructor(knex, transaction, {labels}) {
        super(MembersLabelsImporter.table, knex, transaction);
        this.labels = labels;
    }

    async import(quantity) {
        const members = await this.transaction.select('id').from('members');
        this.labels = await this.transaction.select('id').from('labels');

        await this.importForEach(members, quantity ? quantity / members.length : 1);
    }

    generate() {
        if (luck(90)) {
            // 90% of members don't have labels
            return;
        }
        // TODO: Ensure we don't generate the same member label twice
        return {
            id: this.fastFakeObjectId(),
            member_id: this.model.id,
            label_id: this.labels[faker.datatype.number({
                min: 0,
                max: this.labels.length - 1
            })].id,
            sort_order: 0
        };
    }
}

module.exports = MembersLabelsImporter;