From 8822432370e51537342671a08bd4b48be7bab2ed Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BE=90=E6=B6=9B?= <midnite@outlook.com>
Date: Thu, 26 Dec 2024 16:04:52 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=B8=80=E4=B8=AAHook?=
 =?UTF-8?q?=E4=BE=9B=E5=BF=AB=E9=80=9F=E8=8E=B7=E5=8F=96Color=E5=87=BD?=
 =?UTF-8?q?=E6=95=B0=E4=BD=BF=E7=94=A8=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/ColorFunctionContext.tsx | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/ColorFunctionContext.tsx b/src/ColorFunctionContext.tsx
index 06f1798..d15367d 100644
--- a/src/ColorFunctionContext.tsx
+++ b/src/ColorFunctionContext.tsx
@@ -1,4 +1,4 @@
-import { createContext, ReactNode, useEffect, useMemo, useState, useTransition } from 'react';
+import { createContext, ReactNode, use, useEffect, useMemo, useState, useTransition } from 'react';
 import { ColorFunctionContext } from './ColorFunctionContext';
 import init, * as funcs from './pkg/color_module';
 
@@ -7,7 +7,7 @@ export type ColorFunctionContextType = {
   isLoading: boolean;
   error: Error | null;
 };
-export const ColorFunctionContext = createContext<ColorFunctionContextType>({
+const ColorFunctionContext = createContext<ColorFunctionContextType>({
   colorFn: null,
   isLoading: true,
   error: null,
@@ -17,6 +17,12 @@ type WasmProviderProps = {
   children?: ReactNode;
 };
 
+export function useColorFunction(): ColorFunctionContextType {
+  const context = use<ColorFunctionContextType>(ColorFunctionContext);
+
+  return context;
+}
+
 export function ColorFunctionProvider({ children }: WasmProviderProps) {
   const [wasmInstance, setWasmInstance] = useState<Wasm.InitOutput | null>(null);
   const [isPending, startTransition] = useTransition();