feat(transform-hook): UseTransformComponent added (#405)

Hook added to return an element, given the current state

Co-authored-by: Tyler Lawson <tyler@secondspectrum.com>

src/hooks/index.ts

@@ -2,3 +2,4 @@ export * from "./use-controls";
 export * from "./use-transform-init";
 export * from "./use-transform-effect";
 export * from "./use-transform-context";
+export * from "./use-transform-component";

src/hooks/use-transform-component.tsx

@@ -0,0 +1,16 @@
+import { useMemo } from "react";
+
+import { useTransformContext } from "./use-transform-context";
+import { getState } from "utils";
+import { ReactZoomPanPinchContextState } from "../models";
+
+export function useTransformComponent<T>(
+  callback: (state: ReactZoomPanPinchContextState) => T,
+): T {
+  const libraryContext = useTransformContext();
+
+  return useMemo(
+    () => callback(getState(libraryContext)),
+    [libraryContext, callback],
+  );
+}