fix: remove erroneous get() API from usePuck

Author: chrisvxd

apps/demo/app/custom-ui/[...puckPath]/client.tsx

@@ -3,7 +3,7 @@
 
 "use client";
 
-import { ActionBar, Button, Data, Puck, Render } from "@/core";
+import { ActionBar, Button, Data, Puck, Render, useGetPuck } from "@/core";
 import { HeadingAnalyzer } from "@/plugin-heading-analyzer/src/HeadingAnalyzer";
 import config from "../../../config";
 import { UserConfig } from "../../../config/types";
@@ -16,7 +16,7 @@ import { ChevronUp, ChevronDown, Globe, Lock, Unlock } from "lucide-react";
 const usePuck = createUsePuck<UserConfig>();
 
 const CustomHeader = ({ onPublish }: { onPublish: (data: Data) => void }) => {
-  const get = usePuck((s) => s.get);
+  const getPuck = useGetPuck();
   const dispatch = usePuck((s) => s.dispatch);
   const previewMode = usePuck((s) => s.appState.ui.previewMode);
 
@@ -50,7 +50,7 @@ const CustomHeader = ({ onPublish }: { onPublish: (data: Data) => void }) => {
             Switch to {previewMode === "edit" ? "interactive" : "edit"}
           </Button>
           <Button
-            onClick={() => onPublish(get().appState.data)}
+            onClick={() => onPublish(getPuck().appState.data)}
             icon={<Globe size="14" />}
           >
             Publish

packages/core/lib/use-puck.ts

@@ -10,8 +10,6 @@ import { createStore, StoreApi, useStore } from "zustand";
 import { makeStatePublic } from "./data/make-state-public";
 import { getItem, ItemSelector } from "./data/get-item";
 
-type WithGet<T> = T & { get: () => T };
-
 export type UsePuckData<
   UserConfig extends Config = Config,
   G extends UserGenerics<UserConfig> = UserGenerics<UserConfig>
@@ -40,9 +38,7 @@ export type UsePuckData<
 export type PuckApi<UserConfig extends Config = Config> =
   UsePuckData<UserConfig>;
 
-type UsePuckStore<UserConfig extends Config = Config> = WithGet<
-  PuckApi<UserConfig>
->;
+type UsePuckStore<UserConfig extends Config = Config> = PuckApi<UserConfig>;
 
 type PickedStore = Pick<
   AppStore,
@@ -85,9 +81,7 @@ export const generateUsePuck = (store: PickedStore): UsePuckStore => {
     },
   };
 
-  const get = () => storeData;
-
-  return { ...storeData, get };
+  return storeData;
 };
 
 export const UsePuckStoreContext = createContext<StoreApi<UsePuckStore> | null>(