pleroma/app/soapbox/entity-store/hooks/useCreateEntity.ts

import { z } from 'zod';

import { useAppDispatch } from 'soapbox/hooks';

import { importEntities } from '../actions';

import { useEntityRequest } from './useEntityRequest';
import { parseEntitiesPath, toAxiosRequest } from './utils';

import type { Entity } from '../types';
import type { EntityCallbacks, EntityRequest, EntitySchema, ExpandedEntitiesPath } from './types';

interface UseCreateEntityOpts<TEntity extends Entity = Entity> {
  schema?: EntitySchema<TEntity>
}

function useCreateEntity<TEntity extends Entity = Entity, Data = any>(
  expandedPath: ExpandedEntitiesPath,
  entityRequest: EntityRequest,
  opts: UseCreateEntityOpts<TEntity> = {},
) {
  const dispatch = useAppDispatch();

  const { request, isLoading } = useEntityRequest();
  const { entityType, listKey } = parseEntitiesPath(expandedPath);

  async function createEntity(data: Data, callbacks: EntityCallbacks<TEntity> = {}): Promise<void> {
    try {
      const result = await request({
        ...toAxiosRequest(entityRequest),
        data,
      });

      const schema = opts.schema || z.custom<TEntity>();
      const entity = schema.parse(result.data);

      // TODO: optimistic updating
      dispatch(importEntities([entity], entityType, listKey));

      if (callbacks.onSuccess) {
        callbacks.onSuccess(entity);
      }
    } catch (error) {
      if (callbacks.onError) {
        callbacks.onError(error);
      }
    }
  }

  return {
    createEntity,
    isLoading,
  };
}

export { useCreateEntity };