Add details to OpMode

Fixes #90
Fixe #91

Author: alan412

src/blocks/mrc_opmode_details.ts

@@ -0,0 +1,99 @@
+/**
+ * @license
+ * Copyright 2025 Porpoiseful LLC
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * @fileoverview Block for Opmode details
+ * @author [email protected] (Alan Smith)
+ */
+
+/**
+ * @license
+ * Copyright 2025 Porpoiseful LLC
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * @fileoverview Create a component with a name of a certain type
+ * @author [email protected] (Alan Smith)
+ */
+import * as Blockly from 'blockly';
+
+import { ExtendedPythonGenerator, OpModeDetails } from '../editor/extended_python_generator';
+import { createFieldDropdown } from '../fields/FieldDropdown';
+import { MRC_STYLE_CLASS_BLOCKS } from '../themes/styles';
+
+export const BLOCK_NAME = 'mrc_opmode_details';
+
+type OpmodeDetailsBlock = Blockly.Block & OpmodeDetailsMixin;
+interface OpmodeDetailsMixin extends OpmodeDetailsMixinType {
+}
+type OpmodeDetailsMixinType = typeof OPMODE_DETAILS;
+
+const OPMODE_DETAILS = {
+  /**
+    * Block initialization.
+    */
+  init: function (this: OpmodeDetailsBlock): void {
+    this.setStyle(MRC_STYLE_CLASS_BLOCKS);
+    this.appendDummyInput()
+      .appendField('Type')
+      .appendField(createFieldDropdown(['Auto', 'Teleop', 'Test']), 'TYPE')
+      .appendField('    ')
+      .appendField('Enabled')
+      .appendField(new Blockly.FieldCheckbox(true), 'ENABLED');
+
+    this.appendDummyInput()
+        .appendField('Display Name')
+        .appendField(new Blockly.FieldTextInput(''), 'NAME')
+    this.appendDummyInput()
+        .appendField('Display Group')
+        .appendField(new Blockly.FieldTextInput(''), 'GROUP');
+
+    this.getField('TYPE')?.setTooltip('What sort of OpMode this is');
+    this.getField('ENABLED')?.setTooltip('Whether the OpMode is shown on Driver Station');
+    this.getField('NAME')?.setTooltip('The name shown on the Driver Station.  If blank will use the class name.');
+    this.getField('GROUP')?.setTooltip('An optional group to group OpModes on Driver Station');
+  },
+}
+
+export const setup = function () {
+  Blockly.Blocks[BLOCK_NAME] = OPMODE_DETAILS;
+}
+
+export const pythonFromBlock = function (
+  block: OpmodeDetailsBlock,
+  generator: ExtendedPythonGenerator,
+) {
+    generator.setOpModeDetails(new OpModeDetails(
+                                block.getFieldValue('NAME'),
+                                block.getFieldValue('GROUP'),
+                                block.getFieldValue('ENABLED') == 'TRUE',
+                                block.getFieldValue('TYPE')
+                              ));
+    return '';
+}

src/blocks/setup_custom_blocks.ts

@@ -12,6 +12,7 @@ import * as Mechanism from './mrc_mechanism';
 import * as Component from './mrc_component';
 import * as MechanismContainerHolder from './mrc_mechanism_component_holder';
 import * as Port from './mrc_port';
+import * as OpModeDetails from './mrc_opmode_details';
 
 const customBlocks = [
   CallPythonFunction,
@@ -27,6 +28,7 @@ const customBlocks = [
   Component,
   MechanismContainerHolder,
   Port,
+  OpModeDetails
 ];
 
 export const setup = function(forBlock: any) {

src/editor/extended_python_generator.ts

@@ -27,6 +27,24 @@ import { FunctionArg } from '../blocks/mrc_call_python_function';
 import * as MechanismContainerHolder from '../blocks/mrc_mechanism_component_holder';
 import * as commonStorage from '../storage/common_storage';
 
+export class OpModeDetails {
+  constructor(private name: string, private group : string, private enabled : boolean, private type : string) {}
+  annotations() : string{
+    let code = '';
+
+    if(this.enabled){
+      code += '@' + this.type + "\n";
+      if(this.name){
+        code += '@name("' + this.name + '")\n';
+      }
+      if(this.group){
+        code += '@group("' + this.group + '")\n';
+      }
+    }
+    return code;
+  }
+}
+
 // Extends the python generator to collect some information about functions and
 // variables that have been defined so they can be used in other modules.
 
@@ -36,7 +54,9 @@ export class ExtendedPythonGenerator extends PythonGenerator {
 
   private classMethods: {[key: string]: string} = Object.create(null);
   private ports: {[key: string]: string} = Object.create(null);
-
+    // Opmode details
+  private details : OpModeDetails | null  = null;
+  
   constructor() {
     super('Python');
   }
@@ -137,15 +157,17 @@ export class ExtendedPythonGenerator extends PythonGenerator {
     if (this.context && this.workspace) {
       const className = this.context.getClassName();
       const classParent = this.context.getClassParent();
+      const annotations = this.details?.annotations();
       this.addImport(classParent);
+
       const classDef = 'class ' + className + '(' + classParent + '):\n';
       const classMethods = [];
       for (const name in this.classMethods) {
         classMethods.push(this.classMethods[name])
       }
       this.classMethods = Object.create(null);
       this.ports = Object.create(null);
-      code = classDef + this.prefixLines(classMethods.join('\n\n'), this.INDENT);
+      code = annotations + classDef + this.prefixLines(classMethods.join('\n\n'), this.INDENT);
 
       this.context.setExportedBlocks(this.produceExportedBlocks(this.workspace));
     }
@@ -262,6 +284,13 @@ export class ExtendedPythonGenerator extends PythonGenerator {
     }
     return exportedBlocks;
   }
+  setOpModeDetails(details : OpModeDetails) {
+    this.details = details;
+  }
+  
+  getOpModeDetails() : OpModeDetails | null{
+    return this.details;
+  }
 }
 
 export const extendedPythonGenerator = new ExtendedPythonGenerator();

src/modules/opmode_start.json

@@ -2,11 +2,24 @@
     "blocks": {
         "languageVersion": 0,
         "blocks": [
+            {
+                "type": "mrc_opmode_details",
+                "id": "ElbONc{)gh,UprTW(|1C",
+                "x": 10,
+                "y": 10,
+                "deletable": false,
+                "fields": {
+                    "NAME": "",
+                    "GROUP": "",
+                    "TYPE": "Auto",
+                    "ENABLED": true 
+                }
+            },
             {
                 "type": "mrc_class_method_def",
                 "id": "ElbONc{)s(,UprTW(|1C",
                 "x": 10,
-                "y": 10,
+                "y": 120,
                 "deletable": false,
                 "extraState": {
                     "canChangeSignature": false,
@@ -29,7 +42,7 @@
                 "type": "mrc_class_method_def",
                 "id": "wxFAh6eaR1|3fTuV:UAd",
                 "x": 10,
-                "y": 200,
+                "y": 300,
                 "deletable": false,
                 "extraState": {
                     "canChangeSignature": false,