Electron_generator/main.js

// main.js - DEFINITIVE VERSION - Dynamic Line-by-Line Generation

const { app, BrowserWindow, ipcMain, dialog } = require('electron');
const path = require('path');
const fs = require('fs');

const createWindow = () => {
  const win = new BrowserWindow({
    width: 900,
    height: 750,
    webPreferences: {
      preload: path.join(__dirname, 'preload.js'),
      contextIsolation: true,
      nodeIntegration: false
    }
  });

  win.maximize();

  win.loadFile('index.html');
  win.webContents.openDevTools();
};

app.whenReady().then(() => {
  ipcMain.handle('form:submit', (event, data) => {
    console.log('--- Data received in main.js ---', data);

    // --- YAML Generation by Building an Array of Lines ---

    // Define spacing constants for indentation
    const sp2 = '  ';
    const sp4 = '    ';
    const sp6 = '      ';
    const sp8 = '        ';

    // Create an array to hold each line of the YAML file
    const yamlLines = [];

    // Build the 'machine' section line by line
    const machineType = data.machineType || 'Unspecified';
    const machineModel = data.model || '';
    const serialNo = data.serialNo || '';
    const VendorName = data.vendor || '';
    yamlLines.push('machine:');
    yamlLines.push(`${sp2}profile: ${machineType}`);
    yamlLines.push(`${sp2}identification:`);
    yamlLines.push(`${sp4}namespace: http://${VendorName}/${machineType}/${machineModel}/${serialNo}`);
    yamlLines.push(`${sp4}browseName: ${data.browseName || ''}`);
    yamlLines.push(`${sp4}manufacturer: "[${VendorName}]"`);
    yamlLines.push(`${sp4}model: "${machineModel}"`);
    yamlLines.push(`${sp4}serialNumber: "${serialNo}"`);
    yamlLines.push(`${sp4}yearofconstruction: "${data.buildYear || ''}"`);
    yamlLines.push(`${sp2}version: "${data.version || 'Unset'}"`);

    // 2. Build the 'alarms' section from the alarms data
    if (data.alarms) {
      yamlLines.push(''); // Blank line for separation
      yamlLines.push(`${sp2}alarms:`);
      yamlLines.push(`${sp4}packing: OBJECTS`); // As per PowerShell script logic
      yamlLines.push(`${sp4}behavior: ${data.alarmBehavior || 'STATIC'}`);
      yamlLines.push(`${sp4}count: ${data.alarmCount}`);
      if (data.alarms.length > 0) {
        yamlLines.push(`${sp4}alarms: `); // Note the trailing space
        data.alarms.forEach(alarm => {
          const id = alarm.id || 0;
          const name = alarm.name || `Alarm_${id}`;
          const message = alarm.message.replace(/:/g, ';'); // Sanitize message as in PowerShell

          yamlLines.push(`${sp6}- ID: ${id}`);
          yamlLines.push(`${sp8}name: ${name}`);
          yamlLines.push(`${sp8}message: ${message}`);
        });
      }
    }

    //3. Build the 'warning' section from the warnings data
    if (data.warnings && data.warnings.length > 0) {
      yamlLines.push(''); // Blank line for separation
      yamlLines.push(`${sp2}warnings:`);
      yamlLines.push(`${sp4}count: ${data.warnings.length}`);
      yamlLines.push(`${sp4}warnings: `); // Note the trailing space

      data.warnings.forEach(warning => {
        const id = warning.id || 0;
        const name = warning.name || `Warning_${id}`;
        const message = warning.message.replace(/:/g, ';'); // Sanitize message as in PowerShell

        yamlLines.push(`${sp6}- ID: ${id}`);
        yamlLines.push(`${sp8}name: ${name}`);
        yamlLines.push(`${sp8}message: ${message}`);
      });
    }

    //Build the 'counters' section from the counters data
    if (data.counters && data.counters.length > 0) {
      yamlLines.push(''); // Blank line for separation
      yamlLines.push(`${sp2}counters:`);

      data.counters.forEach(counter => {
        const id = counter.id || 0;
        const name = counter.name || `Counter_${id}`;
        const type = counter.type || '';
        const description = counter.description || '';

        yamlLines.push(`${sp4}- name: ${name}`);
        yamlLines.push(`${sp6}ID: ${id}`);
        yamlLines.push(`${sp6}type: ${type}`);
        yamlLines.push(`${sp6}description: ${description}`);
      });
    }

    // Build and Append the 'optionals' section if needed
    if (data.stopReason === 'Yes') {
      yamlLines.push(''); // Add a blank line for spacing
      yamlLines.push(`${sp2}optionals:`);
      yamlLines.push(`${sp4}- StopReason`);
    }

    // Build and Append the 'addins' section if needed
    const selectedAddins = data.addins || []; // Default to an empty array

    if (selectedAddins.length > 0) {
      yamlLines.push('');
      yamlLines.push(`${sp2}addins:`);
      
      if (selectedAddins.includes('AcPower')) {
        yamlLines.push(`${sp4}- name: AcPower`);
      }
      if (selectedAddins.includes('Utilities')) {
        yamlLines.push(`${sp4}- name: EnergyUtilitiesAddin`);
        yamlLines.push(`${sp6}# Requires data from a future 'Utilities ADDIN' UI.`);
      }
      if (selectedAddins.includes('TrackAdvance')) {
        yamlLines.push(`${sp4}- name: TrackAdvance`);
        yamlLines.push(`${sp6}# Requires data from a future 'TrackAdvance ADDIN' UI.`);
      }
      if (selectedAddins.includes('AGV')) {
          yamlLines.push(`${sp4}- name: AGV`);
          yamlLines.push(`${sp6}# Requires data from a future 'AGV ADDIN' UI.`);
      }
    }

    // 4. Join all the generated lines into a single string
    const generatedContent = yamlLines.join('\n');
    
    // --- End of YAML Generation Logic ---

    console.log('--- Generated YAML Content ---\n', generatedContent);

    // File saving logic 
    const nameParts = [
        data.machineType,
        data.vendor,
        data.model,
        data.serialNo,
        data.browseName
    ];

    // Filter out any empty or null values
    const nonEmptyParts = nameParts.filter(Boolean);

    // Join the parts with an underscore
    const baseName = nonEmptyParts.join('_');
    
    // Sanitize the final string to remove characters invalid for filenames, but keep underscores and hyphens.
    const safeBaseName = baseName.replace(/[/\\?%*:|"<>]/g, '-');
    
    const yamlFileName = `${safeBaseName}.yml`;
    const documentsPath = app.getPath('documents');
    const yamlFilePath = path.join(documentsPath, yamlFileName);

    try {
      fs.writeFileSync(yamlFilePath, generatedContent, 'utf8');
      dialog.showMessageBox({
        type: 'info',
        title: 'Success',
        message: 'YAML File Generated Successfully!',
        detail: `The file was saved to:\n${yamlFilePath}`
      });
      return { success: true, filePath: yamlFilePath };
    } catch (error) {
      console.error("Failed to write YAML file:", error);
      dialog.showErrorBox('File Save Error', `Failed to save the YAML file.\nError: ${error.message}`);
      return { success: false, error: error.message };
    }
  });

  createWindow();
  app.on('activate', () => { if (BrowserWindow.getAllWindows().length === 0) createWindow(); });
});
app.on('window-all-closed', () => { if (process.platform !== 'darwin') app.quit(); });