export interface RefObject extends Function {
  current?: any;
}

function createRef(): RefObject {
  const func: RefObject = (node: any) => {
    func.current = node;
  };
  return func;
}

export function fillRef<T>(ref, node: T) {
  if (typeof ref === 'function') {
    ref(node);
  } else if (typeof ref === 'object' && ref && 'current' in ref) {
    (ref as any).current = node;
  }
}

/**
 * Merge refs into one ref function to support ref passing.
 */
export function composeRef<T>(...refs: any[]) {
  return (node: T) => {
    refs.forEach(ref => {
      fillRef(ref, node);
    });
  };
}

export default createRef;