main.js

const { app, Tray, Menu, BrowserWindow, ipcMain } = require('electron');
const { generateSensorIcon } = require('./iconGenerator');
const path = require('path');
const fs = require('fs');
const axios = require('axios');

const sensorIcons = {
  'sensor.living_room_temperature': path.join(__dirname, 'assets/temp.png'),
  //'sensor.humidity': path.join(__dirname, 'assets/humidity.png'),
  // Add more mappings as needed
};


let tray, configWindow;
let config = JSON.parse(fs.readFileSync('config.json', 'utf8'));
let currentSensorIndex = 0;

function saveConfig() {
  fs.writeFileSync('config.json', JSON.stringify(config, null, 2));
}

async function fetchSensors() {
  
  try {
    const res = await axios.get(`${config.haUrl}/api/states`, {
      headers: { Authorization: `Bearer ${config.token}` },
    });

    return res.data
      .filter(s => config.sensors.includes(s.entity_id))
      .map(s => {
        const name = s.attributes.friendly_name || s.entity_id;
        const value = `${s.state} ${s.attributes.unit_of_measurement || ''}`;
        return {
          label: `${name}: ${value}`,
          //icon: sensorIcons[s.entity_id] || undefined,
          enabled: false
        };
      });

  } catch {
    return ['Error fetching data'];
  }
}

function createTray() {
  tray = new Tray('assets/ha-taskbar.png');
  tray.setToolTip('Home Assistant Tray');
  
  updateTray();

  tray.setContextMenu(Menu.buildFromTemplate([
    { label: 'Configure', click: createConfigWindow },
    { type: 'separator' },
    { label: 'Quit', role: 'quit' }
  ]));
}

async function updateTray() {
  const sensorData = await fetchSensors();

  if (sensorData.length === 0) {
    tray.setImage(generateSensorIcon({ text: 'ERR', color: '#FF0000', bg: '#000000' }));
    return;
  }

  const { label, icon } = sensorData[currentSensorIndex];
  const parts = label.split(': ');
  const displayValue = parts[1] || parts[0];

  // Optional: color-code by sensor
  const colorMap = ['#007bff', '#28a745', '#ffc107', '#dc3545'];
  const bgColor = '#ffffff';
  const textColor = colorMap[currentSensorIndex % colorMap.length];

  tray.setImage(generateSensorIcon({ text: displayValue, color: textColor, bg: bgColor }));

  tray.setContextMenu(Menu.buildFromTemplate([
    ...sensorData,
    { type: 'separator' },
    { label: 'Configure', click: createConfigWindow },
    { label: 'Quit', role: 'quit' }
  ]));

  tray.setToolTip(sensorData.map(s => s.label).join('\n'));

  // Next sensor in rotation
  currentSensorIndex = (currentSensorIndex + 1) % sensorData.length;

  // Rotate every 10 seconds
  setTimeout(updateTray, 10000);
}




function createConfigWindow() {
  if (configWindow) return;

  configWindow = new BrowserWindow({
    width: 500,
    height: 600,
    webPreferences: {
      preload: path.join(__dirname, 'preload.js')
    }
  });

  configWindow.loadFile('renderer.html');
  configWindow.on('closed', () => { configWindow = null; });
}

ipcMain.handle('get-config', () => config);

ipcMain.handle('get-all-sensors', async () => {
  const res = await axios.get(`${config.haUrl}/api/states`, {
    headers: { Authorization: `Bearer ${config.token}` },
  });

  return res.data
    .filter(s => s.entity_id.startsWith('sensor.'))
    .map(s => ({ id: s.entity_id, name: s.attributes.friendly_name || s.entity_id }));
});

ipcMain.on('save-config', (event, newConfig) => {
  config = newConfig;
  saveConfig();
  updateTray();
});

app.whenReady().then(createTray);