overleaf-cep/services/web/app/src/Features/Subscription/SubscriptionLocator.js

const { promisify } = require('util')
const { Subscription } = require('../../models/Subscription')
const { DeletedSubscription } = require('../../models/DeletedSubscription')
const logger = require('@overleaf/logger')
const { SSOConfig } = require('../../models/SSOConfig')
require('./GroupPlansData') // make sure dynamic group plans are loaded

const SubscriptionLocator = {
  getUsersSubscription(userOrId, callback) {
    const userId = SubscriptionLocator._getUserId(userOrId)
    Subscription.findOne({ admin_id: userId }, function (err, subscription) {
      logger.debug({ userId }, 'got users subscription')
      callback(err, subscription)
    })
  },

  getUserIndividualSubscription(userOrId, callback) {
    const userId = SubscriptionLocator._getUserId(userOrId)
    Subscription.findOne(
      { admin_id: userId, groupPlan: false },
      function (err, subscription) {
        logger.debug({ userId }, 'got users individual subscription')
        callback(err, subscription)
      }
    )
  },

  getManagedGroupSubscriptions(userOrId, callback) {
    Subscription.find({
      manager_ids: userOrId,
      groupPlan: true,
    })
      .populate('admin_id')
      .exec(callback)
  },

  getMemberSubscriptions(userOrId, callback) {
    const userId = SubscriptionLocator._getUserId(userOrId)
    Subscription.find({ member_ids: userId })
      .populate('admin_id')
      .exec(callback)
  },

  getAdminEmail(subscriptionId, callback) {
    Subscription.findById(subscriptionId)
      .populate('admin_id', 'email')
      .exec((err, subscription) => {
        if (err) {
          return callback(err)
        }
        callback(err, subscription?.admin_id?.email)
      })
  },

  getAdminEmailAndName(subscriptionId, callback) {
    Subscription.findById(subscriptionId)
      .populate('admin_id', ['email', 'first_name', 'last_name'])
      .exec((err, subscription) => {
        if (err) {
          return callback(err)
        }
        callback(err, subscription?.admin_id)
      })
  },

  hasRecurlyGroupSubscription(userOrId, callback) {
    const userId = SubscriptionLocator._getUserId(userOrId)
    Subscription.exists(
      {
        groupPlan: true,
        recurlySubscription_id: { $exists: true },
        $or: [
          { member_ids: userId },
          { manager_ids: userId },
          { admin_id: userId },
        ],
      },
      callback
    )
  },

  getSubscription(subscriptionId, callback) {
    Subscription.findOne({ _id: subscriptionId }, callback)
  },

  getSubscriptionByMemberIdAndId(userId, subscriptionId, callback) {
    Subscription.findOne(
      { member_ids: userId, _id: subscriptionId },
      { _id: 1 },
      callback
    )
  },

  getGroupSubscriptionsMemberOf(userId, callback) {
    Subscription.find({ member_ids: userId }, { _id: 1, planCode: 1 }, callback)
  },

  getGroupsWithEmailInvite(email, callback) {
    Subscription.find({ invited_emails: email }, callback)
  },

  getGroupWithV1Id(v1TeamId, callback) {
    Subscription.findOne({ 'overleaf.id': v1TeamId }, callback)
  },

  getUserDeletedSubscriptions(userId, callback) {
    DeletedSubscription.find({ 'subscription.admin_id': userId }, callback)
  },

  getDeletedSubscription(subscriptionId, callback) {
    DeletedSubscription.findOne(
      {
        'subscription._id': subscriptionId,
      },
      callback
    )
  },

  _getUserId(userOrId) {
    if (userOrId && userOrId._id) {
      return userOrId._id
    } else if (userOrId) {
      return userOrId
    }
  },
}

async function hasSSOEnabled(subscription) {
  const ssoConfig = await SSOConfig.findById(subscription.ssoConfig).exec()
  return ssoConfig?.enabled
}

SubscriptionLocator.promises = {
  getUsersSubscription: promisify(SubscriptionLocator.getUsersSubscription),
  getUserIndividualSubscription: promisify(
    SubscriptionLocator.getUserIndividualSubscription
  ),
  getManagedGroupSubscriptions: promisify(
    SubscriptionLocator.getManagedGroupSubscriptions
  ),
  getMemberSubscriptions: promisify(SubscriptionLocator.getMemberSubscriptions),
  getAdminEmail: promisify(SubscriptionLocator.getAdminEmail),
  getAdminEmailAndName: promisify(SubscriptionLocator.getAdminEmailAndName),
  getSubscription: promisify(SubscriptionLocator.getSubscription),
  getSubscriptionByMemberIdAndId: promisify(
    SubscriptionLocator.getSubscriptionByMemberIdAndId
  ),
  getGroupSubscriptionsMemberOf: promisify(
    SubscriptionLocator.getGroupSubscriptionsMemberOf
  ),
  getGroupsWithEmailInvite: promisify(
    SubscriptionLocator.getGroupsWithEmailInvite
  ),
  getGroupWithV1Id: promisify(SubscriptionLocator.getGroupWithV1Id),
  getUserDeletedSubscriptions: promisify(
    SubscriptionLocator.getUserDeletedSubscriptions
  ),
  getDeletedSubscription: promisify(SubscriptionLocator.getDeletedSubscription),
  hasRecurlyGroupSubscription: promisify(
    SubscriptionLocator.hasRecurlyGroupSubscription
  ),
  hasSSOEnabled,
}
module.exports = SubscriptionLocator