chore: Clean up worklet directive usage (#447)

## Description

This PR moves the `'worklet'` directive to the top of the file where
possible (in files which contain only worklet functions).

Author: MatiPl01

packages/react-native-sortables/src/constants/layoutAnimations.ts

@@ -1,9 +1,9 @@
+'worklet';
 import { type LayoutAnimation, withTiming } from 'react-native-reanimated';
 
 const ITEM_LAYOUT_ANIMATION_DURATION = 300;
 
 export const SortableItemExiting = (): LayoutAnimation => {
-  'worklet';
   const animations = {
     opacity: withTiming(0, {
       duration: ITEM_LAYOUT_ANIMATION_DURATION
@@ -27,7 +27,6 @@ export const SortableItemExiting = (): LayoutAnimation => {
 };
 
 export const SortableItemEntering = (): LayoutAnimation => {
-  'worklet';
   const animations = {
     opacity: withTiming(1, {
       duration: ITEM_LAYOUT_ANIMATION_DURATION

packages/react-native-sortables/src/integrations/reanimated/utils/animatedTimeout.ts

@@ -1,3 +1,4 @@
+'worklet';
 import { makeMutable } from 'react-native-reanimated';
 
 import type { AnyFunction } from '../../../helperTypes';
@@ -8,7 +9,6 @@ const TIMEOUT_ID = makeMutable(0);
 export type AnimatedTimeoutID = number;
 
 function removeFromPendingTimeouts(id: AnimatedTimeoutID): void {
-  'worklet';
   PENDING_TIMEOUTS.modify(pendingTimeouts => {
     'worklet';
     delete pendingTimeouts[id];
@@ -20,7 +20,6 @@ export function setAnimatedTimeout<F extends AnyFunction>(
   callback: F,
   delay: number
 ): AnimatedTimeoutID {
-  'worklet';
   let startTimestamp: number;
 
   const currentId = TIMEOUT_ID.value;
@@ -46,6 +45,5 @@ export function setAnimatedTimeout<F extends AnyFunction>(
 }
 
 export function clearAnimatedTimeout(handle: AnimatedTimeoutID): void {
-  'worklet';
   removeFromPendingTimeouts(handle);
 }

packages/react-native-sortables/src/providers/flex/layout/updates/insert/utils.ts

@@ -1,3 +1,4 @@
+'worklet';
 import type { ControlledSizes } from '../../../../../types';
 import { reorderInsert, resolveDimension } from '../../../../../utils';
 
@@ -27,7 +28,6 @@ const getGroupItemIndex = (
   group: Array<string>,
   keyToIndex: Record<string, number>
 ) => {
-  'worklet';
   const key = group[inGroupIndex];
   if (key === undefined) return null;
   return keyToIndex[key] ?? null;
@@ -38,8 +38,6 @@ export const getTotalGroupSize = (
   mainItemSizes: ControlledSizes,
   gap: number
 ) => {
-  'worklet';
-
   const sizesSum = group.reduce(
     (total, key) => total + (resolveDimension(mainItemSizes, key) ?? 0),
     0
@@ -64,7 +62,6 @@ const getIndexesWhenSwappedToGroupBefore = ({
   mainGap,
   mainItemSizes
 }: ItemGroupSwapProps): SwappedGroupIndexesResult => {
-  'worklet';
   if (groupSizeLimit === Infinity) {
     return null;
   }
@@ -149,7 +146,6 @@ const getIndexesWhenSwappedToGroupAfter = ({
   mainGap,
   mainItemSizes
 }: ItemGroupSwapProps): SwappedGroupIndexesResult => {
-  'worklet';
   const activeGroup = itemGroups[currentGroupIndex];
   if (groupSizeLimit === Infinity || activeGroup === undefined) {
     return null;
@@ -231,7 +227,6 @@ const getIndexesWhenSwappedToGroupAfter = ({
 export const getSwappedToGroupBeforeIndices = (
   props: ItemGroupSwapProps
 ): ItemGroupSwapResult | null => {
-  'worklet';
   const indexes = getIndexesWhenSwappedToGroupBefore(props);
   if (indexes === null) return null;
 
@@ -252,7 +247,6 @@ export const getSwappedToGroupBeforeIndices = (
 export const getSwappedToGroupAfterIndices = (
   props: ItemGroupSwapProps
 ): ItemGroupSwapResult | null => {
-  'worklet';
   const indexes = getIndexesWhenSwappedToGroupAfter(props);
   if (indexes === null) return null;
 

packages/react-native-sortables/src/providers/flex/layout/utils/layout.ts

@@ -1,3 +1,4 @@
+'worklet';
 import { IS_WEB } from '../../../../constants';
 import type {
   AlignContent,
@@ -24,7 +25,6 @@ const createGroups = (
   groups: Array<Array<string>>;
   crossAxisGroupSizes: Array<number>;
 } => {
-  'worklet';
   const groups: Array<Array<string>> = [];
   const crossAxisGroupSizes: Array<number> = [];
 
@@ -77,7 +77,6 @@ const calculateAlignment = (
   totalSize: number;
   adjustedGap: number;
 } => {
-  'worklet';
   let startOffset = 0;
   let adjustedGap = providedGap;
 
@@ -151,7 +150,6 @@ const handleLayoutCalculation = (
   isReverse: boolean,
   shouldWrap: boolean
 ) => {
-  'worklet';
   const isRow = axisDirections.main === 'row';
   const expandMultiGroup = !IS_WEB && groups.length > 1; // expands to max height/width
   const paddingHorizontal = paddings.left + paddings.right;
@@ -334,7 +332,6 @@ export const calculateLayout = ({
   limits,
   paddings
 }: FlexLayoutProps): FlexLayout | null => {
-  'worklet';
   if (!limits) {
     return null;
   }

packages/react-native-sortables/src/providers/grid/layout/utils/helpers.ts

@@ -1,9 +1,9 @@
+'worklet';
+
 export const getMainIndex = (index: number, numGroups: number): number => {
-  'worklet';
   return +index % numGroups;
 };
 
 export const getCrossIndex = (index: number, numGroups: number): number => {
-  'worklet';
   return Math.floor(+index / numGroups);
 };

packages/react-native-sortables/src/providers/shared/AutoScrollProvider/utils.ts

@@ -1,3 +1,4 @@
+'worklet';
 import type { MeasuredDimensions } from 'react-native-reanimated';
 
 import type { DebugLineUpdater, DebugRectUpdater } from '../../../types';
@@ -16,7 +17,6 @@ export const handleMeasurementsVertical = (
   debugRects: Record<'end' | 'start', DebugRectUpdater> | undefined,
   debugLine: DebugLineUpdater | undefined
 ) => {
-  'worklet';
   const { height: sH, pageY: sY } = scrollableMeasurements;
   const { height: cH, pageY: cY } = containerMeasurements;
   maxOverScrollOffset ??= threshold;
@@ -65,7 +65,6 @@ export const handleMeasurementsHorizontal = (
   debugRects: Record<'end' | 'start', DebugRectUpdater> | undefined,
   debugLine: DebugLineUpdater | undefined
 ) => {
-  'worklet';
   const { pageX: sX, width: sW } = scrollableMeasurements;
   const { pageX: cX, width: cW } = containerMeasurements;
   maxOverScrollOffset ??= threshold;

packages/react-native-sortables/src/providers/shared/utils.ts

@@ -1,7 +1,7 @@
+'worklet';
 import { EXTRA_SWAP_OFFSET } from '../../constants';
 import type { Maybe } from '../../helperTypes';
 
 export const getAdditionalSwapOffset = (size?: Maybe<number>) => {
-  'worklet';
   return size ? Math.min(EXTRA_SWAP_OFFSET, size / 2) : EXTRA_SWAP_OFFSET;
 };

packages/react-native-sortables/src/utils/arrays.ts

@@ -1,10 +1,8 @@
-export const zipArrays = <T, U>(a: Array<T>, b: Array<U>): Array<[T, U]> => {
-  'worklet';
-  return a.slice(0, b.length).map((_, i) => [a[i], b[i]]) as Array<[T, U]>;
-};
+'worklet';
+export const zipArrays = <T, U>(a: Array<T>, b: Array<U>) =>
+  a.slice(0, b.length).map((_, i) => [a[i], b[i]]) as Array<[T, U]>;
 
 export const reverseArray = <T>(array: Array<T>): void => {
-  'worklet';
   for (let i = 0; i < array.length / 2; i++) {
     [array[i], array[array.length - i - 1]] = [
       array[array.length - i - 1]!,

packages/react-native-sortables/src/utils/dimensions.ts

@@ -1,19 +1,15 @@
+'worklet';
 import type { ControlledSizes, Dimensions } from '../types';
 
-export const resolveDimension = (dimension: ControlledSizes, key: string) => {
-  'worklet';
-  return (
-    dimension &&
-    (typeof dimension === 'number' ? dimension : (dimension[key] ?? null))
-  );
-};
+export const resolveDimension = (dimension: ControlledSizes, key: string) =>
+  dimension &&
+  (typeof dimension === 'number' ? dimension : (dimension[key] ?? null));
 
 export const getItemDimensions = (
   key: string,
   itemWidths: ControlledSizes,
   itemHeights: ControlledSizes
 ): Dimensions | null => {
-  'worklet';
   const itemWidth = resolveDimension(itemWidths, key);
   const itemHeight = resolveDimension(itemHeights, key);
 

packages/react-native-sortables/src/utils/equality.ts

@@ -1,34 +1,23 @@
+'worklet';
 import type { AnyRecord, Maybe } from '../helperTypes';
 import type { Vector } from '../types';
 
-export function lt(a: number, b: number): boolean {
-  'worklet';
-  return a < b;
-}
-
-export function gt(a: number, b: number): boolean {
-  'worklet';
-  return a > b;
-}
+export const lt = (a: number, b: number): boolean => a < b;
+export const gt = (a: number, b: number): boolean => a > b;
 
 export const areArraysDifferent = <T>(
   arr1: Array<T>,
   arr2: Array<T>,
   areEqual = (a: T, b: T): boolean => a === b
-): boolean => {
-  'worklet';
-  return (
-    arr1.length !== arr2.length ||
-    arr1.some((item, index) => !areEqual(item, arr2[index] as T))
-  );
-};
+): boolean =>
+  arr1.length !== arr2.length ||
+  arr1.some((item, index) => !areEqual(item, arr2[index] as T));
 
 export const areValuesDifferent = (
   dim1: number | undefined,
   dim2: number | undefined,
   eps?: number
 ): boolean => {
-  'worklet';
   if (dim1 === undefined) {
     return dim2 !== undefined;
   }
@@ -47,19 +36,14 @@ export const areVectorsDifferent = (
   vec1: Vector,
   vec2: Vector,
   eps?: number
-): boolean => {
-  'worklet';
-  return (
-    areValuesDifferent(vec1.x, vec2.x, eps) ||
-    areValuesDifferent(vec1.y, vec2.y, eps)
-  );
-};
+): boolean =>
+  areValuesDifferent(vec1.x, vec2.x, eps) ||
+  areValuesDifferent(vec1.y, vec2.y, eps);
 
 export const haveEqualPropValues = <T extends AnyRecord>(
   obj1: Maybe<T>,
   obj2: Maybe<T>
 ): boolean => {
-  'worklet';
   if (!obj1 || !obj2) {
     return false;
   }