Skip to content

Commit

Permalink
chore: rewrite function => arrow function
Browse files Browse the repository at this point in the history
  • Loading branch information
li-jia-nan committed May 4, 2024
1 parent ffc47e2 commit 954d637
Showing 1 changed file with 12 additions and 15 deletions.
27 changes: 12 additions & 15 deletions src/ref.ts
Original file line number Diff line number Diff line change
@@ -1,43 +1,41 @@
/* eslint-disable no-param-reassign */
import type * as React from 'react';
import { isValidElement } from 'react';
import { ForwardRef, isFragment, isMemo } from 'react-is';
import useMemo from './hooks/useMemo';

export function fillRef<T>(ref: React.Ref<T>, node: T) {
export const fillRef = <T>(ref: React.Ref<T>, 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: React.Ref<T>[]): React.Ref<T> {
const refList = refs.filter(ref => ref);
export const composeRef = <T>(...refs: React.Ref<T>[]): React.Ref<T> => {
const refList = refs.filter(Boolean);
if (refList.length <= 1) {
return refList[0];
}

return (node: T) => {
refs.forEach(ref => {
fillRef(ref, node);
});
};
}
};

export function useComposeRef<T>(...refs: React.Ref<T>[]): React.Ref<T> {
export const useComposeRef = <T>(...refs: React.Ref<T>[]): React.Ref<T> => {
return useMemo(
() => composeRef(...refs),
refs,
(prev, next) =>
prev.length !== next.length || prev.every((ref, i) => ref !== next[i]),
);
}
};

export function supportRef(nodeOrComponent: any): boolean {
export const supportRef = (nodeOrComponent: any): boolean => {
const type = isMemo(nodeOrComponent)
? nodeOrComponent.type.type
: nodeOrComponent.type;
Expand All @@ -60,17 +58,16 @@ export function supportRef(nodeOrComponent: any): boolean {
return false;
}
return true;
}
};

export function supportNodeRef<T = any>(
export const supportNodeRef = <T = any>(
node: React.ReactNode,
): node is React.ReactElement & React.RefAttributes<T> {
): node is React.ReactElement & React.RefAttributes<T> => {
if (!isValidElement(node)) {
return false;
}
if (isFragment(node)) {
return false;
}
return supportRef(node);
}
/* eslint-enable */
};

0 comments on commit 954d637

Please sign in to comment.