import { t } from './trpc';
import initializeApollo from '@/apollo/client';
import { unlimitedRoles } from '@/config/users';
import * as CRMTypes from '@/graphql/crm.types';
import { Handlers } from '@sentry/node';
import { TRPCError } from '@trpc/server';

/**
 * @see https://trpc.io/docs/v10/middlewares
 */
export const userMiddleware = t.middleware(async ({ ctx, next }) => {
  if (process.env.NODE_ENV !== 'development' && !ctx.user) {
    throw new TRPCError({
      code: 'UNAUTHORIZED',
    });
  }

  const apolloClient = initializeApollo(null, ctx.headers);

  const {
    data: { systemusers },
  } = await apolloClient.query({
    query: CRMTypes.GetSystemUserDocument,
    variables: {
      domainname: ctx.user.domainName,
    },
  });

  const systemuser = systemusers?.at(0);

  const unlimited = systemuser?.roles?.some((x) => x?.name && unlimitedRoles.includes(x.name));

  return next({
    ctx: {
      unlimited,
    },
  });
});

export const sentryMiddleware = t.middleware(
  Handlers.trpcMiddleware({
    attachRpcInput: true,
  })
);

export const middleware = t.middleware;