pankod/refine

import { MetaQuery, BaseKey } from "@refinedev/core";
import * as gql from "gql-query-builder";
import pluralize from "pluralize";
import camelCase from "camelcase";

type GenerateUseOneSubscriptionParams = {
  resource: string;
  meta: MetaQuery;
  id?: BaseKey;
};

type GenerateUseOneSubscriptionReturnValues = {
  variables: any;
  query: string;
  operation: string;
};

export const generateUseOneSubscription = ({
  resource,
  meta,
  id,
}: GenerateUseOneSubscriptionParams): GenerateUseOneSubscriptionReturnValues => {
  if (!id) {
    console.error(
      "[useSubscription]: `id` is required in `params` for graphql subscriptions",
    );
  }

  const singularResource = pluralize.singular(resource);
  const camelResource = camelCase(singularResource);

  const operation = meta.operation ?? camelResource;

  const { query, variables } = gql.subscription({
    operation,
    variables: {
      id: { value: id, type: "ID", required: true },
    },
    fields: meta.fields,
  });

  return { query, variables, operation };
};