From b03da62a5caa0c2562390da49feb3beb6bba8c3d Mon Sep 17 00:00:00 2001
From: Artur Arseniev <artur.catch@hotmail.it>
Date: Mon, 19 Feb 2024 22:05:30 +0400
Subject: [PATCH] Add `trait:select` event

---
 src/trait_manager/index.ts | 24 +++++++++++++++++++++---
 src/trait_manager/types.ts |  7 +++++++
 2 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/src/trait_manager/index.ts b/src/trait_manager/index.ts
index 8d034c612..6551f32b9 100644
--- a/src/trait_manager/index.ts
+++ b/src/trait_manager/index.ts
@@ -28,7 +28,7 @@
  * @module Traits
  */
 
-import { debounce } from 'underscore';
+import { bindAll, debounce } from 'underscore';
 import { Module } from '../abstract';
 import { Model } from '../common';
 import Component from '../dom_components/model/Component';
@@ -85,11 +85,13 @@ export default class TraitManager extends Module<TraitManagerConfigModule> {
     const { state, config, events } = this;
     const ppfx = config.pStylePrefix;
     ppfx && (config.stylePrefix = `${ppfx}${config.stylePrefix}`);
+    bindAll(this, '__onSelect');
 
     const upAll = debounce(() => this.__upSel(), 0);
     const update = debounce(() => this.__onUp(), 0);
     state.listenTo(em, 'component:toggled', upAll);
     state.listenTo(em, events.value, update);
+    state.on('change:traits', this.__onSelect);
 
     this.debounced = [upAll, update];
   }
@@ -114,7 +116,7 @@ export default class TraitManager extends Module<TraitManagerConfigModule> {
    *
    */
   getCategories(): Category[] {
-    const cmp = this.state.get('component');
+    const cmp = this.getComponent();
     const categories = cmp?.traits.categories?.models || [];
     return [...categories];
   }
@@ -146,6 +148,16 @@ export default class TraitManager extends Module<TraitManagerConfigModule> {
     return getItemsByCategory<Trait>(traits || this.getTraits());
   }
 
+  /**
+   * Get component from the currently selected traits.
+   * @example
+   * tm.getComponent();
+   * // Component {}
+   */
+  getComponent() {
+    return this.state.attributes.component;
+  }
+
   /**
    * Add new trait type.
    * More about it here: [Define new Trait type](https://grapesjs.com/docs/modules/Traits.html#define-new-trait-type).
@@ -211,6 +223,12 @@ export default class TraitManager extends Module<TraitManagerConfigModule> {
     this.__appendTo();
   }
 
+  __onSelect() {
+    const { em, events, state } = this;
+    const { component, traits } = state.attributes;
+    em.trigger(events.select, { component, traits });
+  }
+
   __trgCustom(opts: TraitCustomData = {}) {
     const { em, events, __ctn } = this;
     this.__ctn = __ctn || opts.container;
@@ -226,6 +244,6 @@ export default class TraitManager extends Module<TraitManagerConfigModule> {
   }
 
   __onUp() {
-    this.select(this.state.get('component'));
+    this.select(this.getComponent());
   }
 }
diff --git a/src/trait_manager/types.ts b/src/trait_manager/types.ts
index 93919252a..ef5f58d7b 100644
--- a/src/trait_manager/types.ts
+++ b/src/trait_manager/types.ts
@@ -174,6 +174,13 @@ export type TraitsEvent = `${TraitsEvents}`;
 
 /**{START_EVENTS}*/
 export enum TraitsEvents {
+  /**
+   * @event `trait:select` New traits selected (eg. by changing a component).
+   * @example
+   * editor.on('trait:select', ({ traits, component }) => { ... });
+   */
+  select = 'trait:select',
+
   /**
    * @event `trait:value` Trait value updated.
    * @example