/*

   This Source Code Form is subject to the terms of the Mozilla Public

   License, v. 2.0. If a copy of the MPL was not distributed with this

   file, You can obtain one at http://mozilla.org/MPL/2.0/.

   Copyright (C) 2009-2011 Michael Möller <mmoeller@openhardwaremonitor.org>

 */

 using System;

 using System.Collections.Generic;

 using System.Drawing;

 using System.Text;

 using System.Windows.Forms;

 using OpenHardwareMonitor.Hardware;

 using OpenHardwareMonitor.Utilities;

 namespace OpenHardwareMonitor.GUI {

   public class SystemTray : IDisposable {

     private IComputer computer;

     private PersistentSettings settings;

     private List<SensorNotifyIcon> list = new List<SensorNotifyIcon>();

     private bool mainIconEnabled = false;

     private NotifyIcon mainIcon;

     public SystemTray(IComputer computer, PersistentSettings settings) {

       this.computer = computer;

       this.settings = settings;

       computer.HardwareAdded += new HardwareEventHandler(HardwareAdded);

       computer.HardwareRemoved += new HardwareEventHandler(HardwareRemoved);

       this.mainIcon = new NotifyIcon();

       ContextMenu contextMenu = new ContextMenu();

       MenuItem hideShowItem = new MenuItem("Hide/Show");

       hideShowItem.Click += delegate(object obj, EventArgs args) {

         SendHideShowCommand();

       };

       contextMenu.MenuItems.Add(hideShowItem);

       contextMenu.MenuItems.Add(new MenuItem("-"));      

       MenuItem exitItem = new MenuItem("Exit");

       exitItem.Click += delegate(object obj, EventArgs args) {

         SendExitCommand();

       };

       contextMenu.MenuItems.Add(exitItem);

       this.mainIcon.ContextMenu = contextMenu;

       this.mainIcon.DoubleClick += delegate(object obj, EventArgs args) {

         SendHideShowCommand();

       };

       this.mainIcon.Icon = EmbeddedResources.GetIcon("smallicon.ico");

       this.mainIcon.Text = "Open Hardware Monitor";

     }

     private void HardwareRemoved(IHardware hardware) {

       hardware.SensorAdded -= new SensorEventHandler(SensorAdded);

       hardware.SensorRemoved -= new SensorEventHandler(SensorRemoved);

       foreach (ISensor sensor in hardware.Sensors) 

         SensorRemoved(sensor);

       foreach (IHardware subHardware in hardware.SubHardware)

         HardwareRemoved(subHardware);

     }

     private void HardwareAdded(IHardware hardware) {

       foreach (ISensor sensor in hardware.Sensors)

         SensorAdded(sensor);

       hardware.SensorAdded += new SensorEventHandler(SensorAdded);

       hardware.SensorRemoved += new SensorEventHandler(SensorRemoved);

       foreach (IHardware subHardware in hardware.SubHardware)

         HardwareAdded(subHardware);

     }

     private void SensorAdded(ISensor sensor) {

       if (settings.GetValue(new Identifier(sensor.Identifier, 

         "tray").ToString(), false)) 

         Add(sensor, false);   

     }

     private void SensorRemoved(ISensor sensor) {

       if (Contains(sensor)) 

         Remove(sensor, false);

     }

     public void Dispose() {

       foreach (SensorNotifyIcon icon in list)

         icon.Dispose();

       mainIcon.Dispose();

     }

     public void Redraw() {

       foreach (SensorNotifyIcon icon in list)

         icon.Update();

     }

     public bool Contains(ISensor sensor) {

       foreach (SensorNotifyIcon icon in list)

         if (icon.Sensor == sensor)

           return true;

       return false;

     }

     public void Add(ISensor sensor, bool balloonTip) {

       if (Contains(sensor)) {

         return;

       } else {        

         list.Add(new SensorNotifyIcon(this, sensor, balloonTip, settings));

         UpdateMainIconVisibilty();

         settings.SetValue(new Identifier(sensor.Identifier, "tray").ToString(), true);

       }

     }

     public void Remove(ISensor sensor) {

       Remove(sensor, true);

     }

     private void Remove(ISensor sensor, bool deleteConfig) {

       if (deleteConfig) {

         settings.Remove(

           new Identifier(sensor.Identifier, "tray").ToString());

         settings.Remove(

           new Identifier(sensor.Identifier, "traycolor").ToString());

       }

       SensorNotifyIcon instance = null;

       foreach (SensorNotifyIcon icon in list)

         if (icon.Sensor == sensor)

           instance = icon;

       if (instance != null) {

         list.Remove(instance);

         UpdateMainIconVisibilty();

         instance.Dispose();        

       }

     }

     public event EventHandler HideShowCommand;

     public void SendHideShowCommand() {

       if (HideShowCommand != null)

         HideShowCommand(this, null);

     }

     public event EventHandler ExitCommand;

     public void SendExitCommand() {

       if (ExitCommand != null)

         ExitCommand(this, null);

     }

     private void UpdateMainIconVisibilty() {

       if (mainIconEnabled) {

         mainIcon.Visible = list.Count == 0;

       } else {

         mainIcon.Visible = false;

       }

     }

     public bool IsMainIconEnabled {

       get { return mainIconEnabled; }

       set {

         if (mainIconEnabled != value) {

           mainIconEnabled = value;

           UpdateMainIconVisibilty();

         }

       }

     }

   }

 }