import {
  Args,
  Resolver,
  Query,
  Parent,
  ResolveField,
  Context,
} from '@nestjs/graphql';
import { Plan } from './types';
import { Price } from '@ultimatebackend/contracts';
import { GqlContext } from '@ultimatebackend/core';
import {
  StripePlan as StripePlanRpc,
  Plan as PlanRpc,
} from '@ultimatebackend/proto-schema/billing';

@Resolver(() => Plan)
export class PlansResolver {
  @Query(() => Plan)
  async plan(
    @Args('id') id: string,
    @Context() ctx: GqlContext,
  ): Promise<PlanRpc> {
    const result = await ctx?.rpc?.billing?.svc.readPlan({ id }).toPromise();
    return result.plan;
  }

  @Query(() => [Plan])
  async plans(@Context() ctx: GqlContext): Promise<PlanRpc[]> {
    const result = await ctx?.rpc?.billing?.svc.findPlans({}).toPromise();
    return result.plans;
  }

  @ResolveField(() => [Price])
  async prices(
    @Context() ctx: GqlContext,
    @Parent() parent: Plan,
  ): Promise<StripePlanRpc[]> {
    const result = await ctx?.rpc?.billing?.svc
      .findStripePlans({ productId: parent.normalizedName })
      .toPromise();
    return result.plans;
  }
}