feat: viewmodel class generator

Author: raideer

.cursor/rules/commands.mdc

@@ -4,7 +4,7 @@ globs: src/command/*.ts
 alwaysApply: false
 ---
 - Commands are located in `src/command` directory
-- Commands must extend the [Command.ts](mdc:src/command/Command.ts) abstract class or any other class that inherits it
+- Commands must extend the [Command.ts](mdc:magento-toolbox/src/command/Command.ts) abstract class or any other class that inherits it
 - Command name must be defined in format: `magentoToolbox.commandName`
-- New commands must be added to [index.ts](mdc:src/command/index.ts) file so that they are loaded. Commands also need to be added to [package.json](mdc:package.json) under contributes -> commands
-- Commands that only generate files based on a pre-defined template can extend the [SimpleTemplateGeneratorCommand.ts](mdc:src/command/SimpleTemplateGeneratorCommand.ts) class
+- New commands must be added to [index.ts](mdc:magento-toolbox/src/command/index.ts) file so that they are loaded. Commands also need to be added to [package.json](mdc:magento-toolbox/package.json) under contributes -> commands
+- Commands that only generate files based on a pre-defined template can extend the [SimpleTemplateGeneratorCommand.ts](mdc:magento-toolbox/src/command/SimpleTemplateGeneratorCommand.ts) class

CHANGELOG.md

@@ -4,6 +4,9 @@ All notable changes to the "magento-toolbox" extension will be documented in thi
 
 Check [Keep a Changelog](http://keepachangelog.com/) for recommendations on how to structure this file.
 
+## [Unreleased]
+- Added: Generator command for a ViewModel class
+
 ## [1.3.1] - 2025-03-23
 - Fixed: Generated plugin class arguments contain an incorrect namespace
 

package.json

@@ -65,6 +65,11 @@
         "title": "Generate Block",
         "category": "Magento Toolbox"
       },
+      {
+        "command": "magento-toolbox.generateViewModel",
+        "title": "Generate ViewModel",
+        "category": "Magento Toolbox"
+      },
       {
         "command": "magento-toolbox.indexWorkspace",
         "title": "Index Workspace",
@@ -244,6 +249,10 @@
           "command": "magento-toolbox.generateBlock",
           "when": "resourcePath =~ /app\\/code\\/.+\\/.+/i"
         },
+        {
+          "command": "magento-toolbox.generateViewModel",
+          "when": "resourcePath =~ /app\\/code\\/.+\\/.+/i"
+        },
         {
           "command": "magento-toolbox.generateEventsXml",
           "when": "resourcePath =~ /app\\/code\\/.+\\/.+/i"

src/command/GenerateViewModelCommand.ts

@@ -0,0 +1,58 @@
+import { Command } from 'command/Command';
+import ViewModelClassGenerator from 'generator/viewModel/ViewModelClassGenerator';
+import ViewModelWizard, { ViewModelWizardData } from 'wizard/ViewModelWizard';
+import FileGeneratorManager from 'generator/FileGeneratorManager';
+import { Uri, window } from 'vscode';
+import Common from 'util/Common';
+import WizzardClosedError from 'webview/error/WizzardClosedError';
+import IndexManager from 'indexer/IndexManager';
+import ModuleIndexer from 'indexer/module/ModuleIndexer';
+
+export default class GenerateViewModelCommand extends Command {
+  constructor() {
+    super('magento-toolbox.generateViewModel');
+  }
+
+  public async execute(uri?: Uri): Promise<void> {
+    const moduleIndex = IndexManager.getIndexData(ModuleIndexer.KEY);
+    let contextModule: string | undefined;
+
+    const contextUri = uri || window.activeTextEditor?.document.uri;
+
+    if (moduleIndex && contextUri) {
+      const module = moduleIndex.getModuleByUri(contextUri);
+
+      if (module) {
+        contextModule = module.name;
+      }
+    }
+
+    const viewModelWizard = new ViewModelWizard();
+
+    let data: ViewModelWizardData;
+
+    try {
+      data = await viewModelWizard.show(contextModule);
+    } catch (error) {
+      if (error instanceof WizzardClosedError) {
+        return;
+      }
+
+      throw error;
+    }
+
+    const manager = new FileGeneratorManager([new ViewModelClassGenerator(data)]);
+
+    const workspaceFolder = Common.getActiveWorkspaceFolder();
+
+    if (!workspaceFolder) {
+      window.showErrorMessage('No active workspace folder');
+      return;
+    }
+
+    await manager.generate(workspaceFolder.uri);
+    await manager.writeFiles();
+    await manager.refreshIndex(workspaceFolder);
+    manager.openFirstFile();
+  }
+}

src/command/index.ts

@@ -5,6 +5,7 @@ export { default as CopyMagentoPathCommand } from './CopyMagentoPathCommand';
 export { default as GenerateXmlCatalogCommand } from './GenerateXmlCatalogCommand';
 export { default as GenerateObserverCommand } from './GenerateObserverCommand';
 export { default as GenerateBlockCommand } from './GenerateBlockCommand';
+export { default as GenerateViewModelCommand } from './GenerateViewModelCommand';
 export { default as GenerateEventsXmlCommand } from './GenerateEventsXmlCommand';
 export { default as GenerateGraphqlSchemaFileCommand } from './GenerateGraphqlSchemaFile';
 export { default as GenerateRoutesXmlFileCommand } from './GenerateRoutesXmlFileCommand';

src/generator/viewModel/ViewModelClassGenerator.ts

@@ -0,0 +1,45 @@
+import FileHeader from 'common/php/FileHeader';
+import FileGenerator from 'generator/FileGenerator';
+import GeneratedFile from 'generator/GeneratedFile';
+import { PhpFile, PsrPrinter } from 'node-php-generator';
+import Magento from 'util/Magento';
+import { Uri } from 'vscode';
+import { ViewModelWizardData } from 'wizard/ViewModelWizard';
+
+export default class ViewModelClassGenerator extends FileGenerator {
+  private static readonly ARGUMENT_INTERFACE =
+    'Magento\\Framework\\View\\Element\\Block\\ArgumentInterface';
+
+  public constructor(protected data: ViewModelWizardData) {
+    super();
+  }
+
+  public async generate(workspaceUri: Uri): Promise<GeneratedFile> {
+    const [vendor, module] = this.data.module.split('_');
+    const pathParts = this.data.directory.split('/');
+    const namespaceParts = [vendor, module, ...pathParts];
+    const moduleDirectory = Magento.getModuleDirectory(vendor, module, workspaceUri);
+
+    const header = FileHeader.getHeader(this.data.module);
+
+    const phpFile = new PhpFile();
+    if (header) {
+      phpFile.addComment(header);
+    }
+    phpFile.setStrictTypes(true);
+
+    const namespace = phpFile.addNamespace(namespaceParts.join('\\'));
+    namespace.addUse(ViewModelClassGenerator.ARGUMENT_INTERFACE);
+
+    const viewModelClass = namespace.addClass(this.data.className);
+
+    viewModelClass.setImplements([ViewModelClassGenerator.ARGUMENT_INTERFACE]);
+
+    const printer = new PsrPrinter();
+
+    return new GeneratedFile(
+      Uri.joinPath(moduleDirectory, this.data.directory, `${this.data.className}.php`),
+      printer.printFile(phpFile)
+    );
+  }
+}

src/wizard/ViewModelWizard.ts

@@ -0,0 +1,63 @@
+import IndexManager from 'indexer/IndexManager';
+import ModuleIndexer from 'indexer/module/ModuleIndexer';
+import { GeneratorWizard } from 'webview/GeneratorWizard';
+import { WizardFieldBuilder } from 'webview/WizardFieldBuilder';
+import { WizardFormBuilder } from 'webview/WizardFormBuilder';
+import { WizardTabBuilder } from 'webview/WizardTabBuilder';
+
+export interface ViewModelWizardData {
+  module: string;
+  className: string;
+  directory: string;
+}
+
+export default class ViewModelWizard extends GeneratorWizard {
+  public async show(contextModule?: string): Promise<ViewModelWizardData> {
+    const moduleIndexData = IndexManager.getIndexData(ModuleIndexer.KEY);
+
+    if (!moduleIndexData) {
+      throw new Error('Module index data not found');
+    }
+
+    const modules = moduleIndexData.getModuleOptions(m => m.location === 'app');
+
+    const builder = new WizardFormBuilder();
+
+    builder.setTitle('Generate a new ViewModel');
+    builder.setDescription('Generates a new Magento2 ViewModel class.');
+
+    const tab = new WizardTabBuilder();
+    tab.setId('viewModel');
+    tab.setTitle('ViewModel');
+
+    tab.addField(
+      WizardFieldBuilder.select('module', 'Module*')
+        .setOptions(modules)
+        .setInitialValue(contextModule || modules[0].value)
+        .build()
+    );
+
+    tab.addField(
+      WizardFieldBuilder.text('className', 'Class Name*')
+        .setPlaceholder('ViewModel class name')
+        .build()
+    );
+
+    tab.addField(
+      WizardFieldBuilder.text('directory', 'Directory*')
+        .setPlaceholder('ViewModel/Path')
+        .setInitialValue('ViewModel')
+        .build()
+    );
+
+    builder.addTab(tab.build());
+
+    builder.addValidation('module', 'required');
+    builder.addValidation('className', 'required|min:1');
+    builder.addValidation('directory', 'required|min:1');
+
+    const data = await this.openWizard<ViewModelWizardData>(builder.build());
+
+    return data;
+  }
+}