import { DateTime } from 'luxon';
import Billing from '../../helpers/billing/index.ee.js';
import ExecutionStep from '../../models/execution-step.js';

const getBillingAndUsage = async (_parent, _params, context) => {
  const persistedSubscription = await context.currentUser.$relatedQuery(
    'currentSubscription'
  );

  const subscription = persistedSubscription
    ? paidSubscription(persistedSubscription)
    : freeTrialSubscription();

  return {
    subscription,
    usage: {
      task: executionStepCount(context),
    },
  };
};

const paidSubscription = (subscription) => {
  const currentPlan = Billing.paddlePlans.find(
    (plan) => plan.productId === subscription.paddlePlanId
  );

  return {
    status: subscription.status,
    monthlyQuota: {
      title: currentPlan.limit,
      action: {
        type: 'link',
        text: 'Cancel plan',
        src: subscription.cancelUrl,
      },
    },
    nextBillAmount: {
      title: subscription.nextBillAmount
        ? '€' + subscription.nextBillAmount
        : '---',
      action: {
        type: 'link',
        text: 'Update payment method',
        src: subscription.updateUrl,
      },
    },
    nextBillDate: {
      title: subscription.nextBillDate ? subscription.nextBillDate : '---',
      action: {
        type: 'text',
        text: '(monthly payment)',
      },
    },
  };
};

const freeTrialSubscription = () => {
  return {
    status: null,
    monthlyQuota: {
      title: 'Free Trial',
      action: {
        type: 'link',
        text: 'Upgrade plan',
        src: '/settings/billing/upgrade',
      },
    },
    nextBillAmount: {
      title: '---',
      action: null,
    },
    nextBillDate: {
      title: '---',
      action: null,
    },
  };
};

const executionIds = async (context) => {
  return (
    await context.currentUser
      .$relatedQuery('executions')
      .select('executions.id')
  ).map((execution) => execution.id);
};

const executionStepCount = async (context) => {
  const executionStepCount = await ExecutionStep.query()
    .whereIn('execution_id', await executionIds(context))
    .andWhere(
      'created_at',
      '>=',
      DateTime.now().minus({ days: 30 }).toISODate()
    )
    .count()
    .first();

  return executionStepCount.count;
};

export default getBillingAndUsage;