use-delete-entity-public.tsx

import { Graph, type Op } from '@graphprotocol/grc-20';
import type { Connect, Entity } from '@graphprotocol/hypergraph';
import { useQueryClient } from '@tanstack/react-query';
import request, { gql } from 'graphql-request';
import { publishOps } from '../publish-ops.js';

type DeleteEntityPublicParams = {
  space: string;
};

const deleteEntityQueryDocument = gql`
query entityToDelete($entityId: UUID!, $spaceId: UUID!) {
  entity(id: $entityId) {
    valuesList(filter: {spaceId: {is: $spaceId}}) {
      propertyId
    }
    relationsList(filter: {spaceId: {is: $spaceId}}) {
      id
    }
  }
}
`;

type EntityToDeleteQueryResult = {
  entity: {
    valuesList: {
      propertyId: string;
    }[];
    relationsList: {
      id: string;
    }[];
  };
} | null;

export const useDeleteEntityPublic = <S extends Entity.AnyNoContext>(type: S, { space }: DeleteEntityPublicParams) => {
  const queryClient = useQueryClient();

  return async ({ id, walletClient }: { id: string; walletClient: Connect.SmartSessionClient }) => {
    try {
      const result = await request<EntityToDeleteQueryResult>(Graph.TESTNET_API_ORIGIN, deleteEntityQueryDocument, {
        spaceId: space,
        entityId: id,
      });
      if (!result) {
        return { success: false, error: 'Entity not found' };
      }
      const { valuesList, relationsList } = result.entity;
      const ops: Op[] = [];
      const { ops: unsetEntityValuesOps } = Graph.unsetEntityValues({
        id,
        properties: valuesList.map(({ propertyId }) => propertyId),
      });
      ops.push(...unsetEntityValuesOps);
      for (const relation of relationsList) {
        const { ops: deleteRelationOps } = Graph.deleteRelation({ id: relation.id });
        ops.push(...deleteRelationOps);
      }

      const { cid, txResult } = await publishOps({
        ops,
        space,
        name: `Delete entity ${id}`,
        walletClient,
      });
      // TODO: temporary fix until we get the information from the API when a transaction is confirmed
      await new Promise((resolve) => setTimeout(resolve, 2000));
      queryClient.invalidateQueries({
        queryKey: [
          'hypergraph-public-entities',
          // @ts-expect-error - TODO: find a better way to access the type.name
          type.name,
          space,
        ],
      });

      return { success: true, cid, txResult };
    } catch (error) {
      return { success: false, error: 'Failed to delete entity' };
    }
  };
};