add request style

Author: alisonjlai

src/client/initialize.ts

@@ -37,9 +37,8 @@ export function initialize<T = {}>(): PluginInstance<T> {
     emit('config', pluginConfig.config ?? {});
   });
 
-  on('wb:plugin:style:update', (styleColors: any) => {
-    pluginConfig.styleColors = styleColors;
-    emit('style', styleColors);
+  on('wb:plugin:style:update', (style: any) => {
+    emit('style', style);
   });
 
   // send initialize event
@@ -229,29 +228,14 @@ export function initialize<T = {}>(): PluginInstance<T> {
       },
     },
 
-    // Style management functions for hooks
     style: {
-      /**
-       * Subscribe to style updates
-       * @param callback Function to call when style updates
-       * @returns Unsubscriber function
-       */
-      subscribe(callback: (style: any) => void) {
+      subscribeToStyle(callback: (style: any) => void) {
         on('style', callback);
         return () => off('style', callback);
       },
 
-      /**
-       * Request current style from workbook
-       * @returns Promise with current style
-       */
-      async getStyle() {
-        try {
-          return await execPromise('wb:plugin:style:get');
-        } catch (error) {
-          // Return default style if request fails
-          return { backgroundColor: 'transparent' };
-        }
+      getStyle() {
+        return execPromise('wb:plugin:style:get');
       },
     },
 

src/react/hooks.ts

@@ -257,11 +257,9 @@ export function usePluginStyle(): PluginStyle | undefined {
   const [style, setStyle] = useState<PluginStyle | undefined>(undefined);
 
   useEffect(() => {
-    const unsubscribe = client.style.subscribe(setStyle);
-    // Request initial style data on mount
+    // Request initial style data on mount and subscribe to updates
     void client.style.getStyle().then(setStyle);
-
-    return unsubscribe;
+    return client.style.subscribeToStyle(setStyle);
   }, [client]);
 
   return style;

src/types.ts

@@ -27,7 +27,6 @@ export interface PluginConfig<T> {
   id: string;
   config: T;
   screenshot: boolean;
-  // styleColors: PluginStyle;
   [key: string]: any;
 }
 
@@ -206,19 +205,6 @@ export type CustomPluginConfigOptions =
 export interface PluginInstance<T = any> {
   sigmaEnv: 'author' | 'viewer' | 'explorer';
 
-  // /**
-  //  * Plugin style colors from the workbook
-  //  * @returns {PluginStyle | undefined} Style colors if available
-  //  */
-  // styleColors?: PluginStyle | undefined;
-
-  // /**
-  //  * Listen to style color changes
-  //  * @param {Function} callback Function to call when style colors change
-  //  * @returns {Function} Unsubscriber function
-  //  */
-  // onStyleChange(callback: (styleColors: PluginStyle) => void): () => void;
-
   config: {
     /**
      * Getter for entire Plugin Config
@@ -378,16 +364,13 @@ export interface PluginInstance<T = any> {
     fetchMoreElementData(configId: string): void;
   };
 
-  /**
-   * Style management for plugins
-   */
   style: {
     /**
      * Subscribe to style updates
      * @param callback Function to call when style updates
      * @returns Unsubscriber function
      */
-    subscribe(callback: (style: PluginStyle) => void): () => void;
+    subscribeToStyle(callback: (style: PluginStyle) => void): () => void;
 
     /**
      * Request current style from workbook