import { shallow } from 'zustand/shallow';

import { useStore } from './useStore';
import type { ReactFlowStore } from '../types/store';
import { ConnectionState } from '@xyflow/system';

const selector = (s: ReactFlowStore) => ({
  ...s.connection,
  inProgress: s.connection.fromHandle !== null,
});

/**
 * Hook for accessing the ongoing connection.
 *
 * @public
 * @returns ongoing connection
 */
export function useConnection(): ConnectionState & { inProgress: boolean } {
  return useStore(selector, shallow);
}