Ghost/ghost/member-attribution/lib/service.js

const MemberAttributionEventHandler = require('./event-handler');
const DomainEvents = require('@tryghost/domain-events');
const UrlTranslator = require('./url-translator');
const AttributionBuilder = require('./attribution');
const UrlHistory = require('./history');

class MemberAttributionService {
    constructor({Post, User, Tag, MemberCreatedEvent, SubscriptionCreatedEvent, urlService, labsService}) {
        const eventHandler = new MemberAttributionEventHandler({MemberCreatedEvent, SubscriptionCreatedEvent, DomainEvents, labsService});
        eventHandler.subscribe();

        this.urlService = urlService;
        this.models = {MemberCreatedEvent, SubscriptionCreatedEvent};

        const urlTranslator = new UrlTranslator({
            urlService, 
            models: {
                Post, User, Tag
            }
        });
        this.attributionBuilder = new AttributionBuilder({urlTranslator});
    }

    /**
     * 
     * @param {import('./history').UrlHistoryArray} historyArray 
     * @returns {import('./attribution').Attribution}
     */
    getAttribution(historyArray) {
        const history = new UrlHistory(historyArray);
        return this.attributionBuilder.getAttribution(history);
    }

    /**
     * Returns the parsed attribution for a member creation event
     * @param {string} memberId 
     * @returns {Promise<import('./attribution').AttributionResource|null>}
     */
    async getMemberCreatedAttribution(memberId) {
        const memberCreatedEvent = await this.models.MemberCreatedEvent.findOne({member_id: memberId}, {require: false});
        if (!memberCreatedEvent || !memberCreatedEvent.get('attribution_type')) {
            return null;
        }
        const attribution = this.attributionBuilder.build({
            id: memberCreatedEvent.get('attribution_id'),
            url: memberCreatedEvent.get('attribution_url'),
            type: memberCreatedEvent.get('attribution_type')
        });
        return await attribution.getResource();
    }

    /**
     * Returns the last attribution for a given subscription ID
     * @param {string} subscriptionId 
     * @returns {Promise<import('./attribution').AttributionResource|null>}
     */
    async getSubscriptionCreatedAttribution(subscriptionId) {
        const subscriptionCreatedEvent = await this.models.SubscriptionCreatedEvent.findOne({subscription_id: subscriptionId}, {require: false});
        if (!subscriptionCreatedEvent || !subscriptionCreatedEvent.get('attribution_type')) {
            return null;
        }
        const attribution = this.attributionBuilder.build({
            id: subscriptionCreatedEvent.get('attribution_id'),
            url: subscriptionCreatedEvent.get('attribution_url'),
            type: subscriptionCreatedEvent.get('attribution_type')
        });
        return await attribution.getResource();
    }
}

module.exports = MemberAttributionService;