AI Assistant: Integration between AiIntegration, GridCommands, AiAssistant and AiChat (#33564)

Author: dmirgaev

packages/devextreme/js/__internal/grids/data_grid/ai_assistant/ai_assistant.ts

@@ -1,9 +1,9 @@
 import messageLocalization from '@js/common/core/localization/message';
-import { AIAssistantController } from '@ts/grids/grid_core/ai_assistant/ai_assistant_controller';
 import { AIAssistantView } from '@ts/grids/grid_core/ai_assistant/ai_assistant_view';
 import { AIAssistantViewController } from '@ts/grids/grid_core/ai_assistant/ai_assistant_view_controller';
 
 import gridCore from '../m_core';
+import { DataGridAIAssistantController } from './ai_assistant_controller';
 
 gridCore.registerModule('aiAssistant', {
   defaultOptions() {
@@ -15,7 +15,7 @@ gridCore.registerModule('aiAssistant', {
     };
   },
   controllers: {
-    aiAssistant: AIAssistantController,
+    aiAssistant: DataGridAIAssistantController,
     aiAssistantViewController: AIAssistantViewController,
   },
   views: {

packages/devextreme/js/__internal/grids/data_grid/ai_assistant/ai_assistant_controller.ts

@@ -0,0 +1,22 @@
+import { AIAssistantController } from '@ts/grids/grid_core/ai_assistant/ai_assistant_controller';
+import type { GridCommand } from '@ts/grids/grid_core/ai_assistant/types';
+
+import { DataGridAIAssistantIntegrationController } from './ai_assistant_integration_controller';
+import { dataGridCommands } from './commands/index';
+
+export class DataGridAIAssistantController extends AIAssistantController {
+  protected aiAssistantIntegrationController?: DataGridAIAssistantIntegrationController;
+
+  protected getAiAssistantIntegrationController(): DataGridAIAssistantIntegrationController {
+    return new DataGridAIAssistantIntegrationController(this.component);
+  }
+
+  protected getGridCommandList(): GridCommand[] {
+    const coreCommands = super.getGridCommandList();
+
+    return [
+      ...coreCommands,
+      ...dataGridCommands,
+    ];
+  }
+}

packages/devextreme/js/__internal/grids/data_grid/ai_assistant/ai_assistant_integration_controller.ts

@@ -0,0 +1,26 @@
+import { AIAssistantIntegrationController } from '@ts/grids/grid_core/ai_assistant/ai_assistant_integration_controller';
+import type { GridContext } from '@ts/grids/grid_core/ai_assistant/types';
+import type { Column } from '@ts/grids/grid_core/columns_controller/types';
+
+export class DataGridAIAssistantIntegrationController extends AIAssistantIntegrationController {
+  protected getGridExtraContext(): GridContext {
+    const context = super.getGridExtraContext();
+
+    context.summary = {
+      totalItems: this.option('summary.totalItems'),
+      groupItems: this.option('summary.groupItems'),
+    };
+
+    return context;
+  }
+
+  protected getGridColumnExtraContext(column: Column): GridContext {
+    const context = super.getGridColumnExtraContext(column);
+
+    context.summary = {
+      groupIndex: column.groupIndex,
+    };
+
+    return context;
+  }
+}

packages/devextreme/js/__internal/grids/data_grid/ai_assistant/commands/index.ts

@@ -0,0 +1,12 @@
+import type { GridCommand } from '@ts/grids/grid_core/ai_assistant/types';
+
+import { clearGroupingCommand, groupingCommand } from './grouping';
+import { clearSummaryCommand, summaryCommand } from './summary';
+
+export const dataGridCommands = [
+  groupingCommand,
+  clearGroupingCommand,
+  summaryCommand,
+  clearSummaryCommand,
+  // TODO: try to remove "as GridCommand[]"
+] as GridCommand[];