openhardwaremonitor/Utilities/Logger.cs

/*
 
  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) 2013 Michael Möller <mmoeller@openhardwaremonitor.org>

*/

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using OpenHardwareMonitor.Hardware;

namespace OpenHardwareMonitor.Utilities {
  public class Logger {

    private const string fileNameFormat = 
      "OpenHardwareMonitorLog-{0:yyyy-MM-dd}.csv";

    private readonly IComputer computer;

    private DateTime day = DateTime.MinValue;
    private string fileName;
    private ISensor[] sensors;

    public Logger(IComputer computer) {
      this.computer = computer;
      this.computer.HardwareAdded += HardwareAdded;
      this.computer.HardwareRemoved += HardwareRemoved;    
    }

    private void HardwareRemoved(IHardware hardware) {
      hardware.SensorRemoved -= SensorRemoved;
      foreach (ISensor sensor in hardware.Sensors)
        SensorRemoved(sensor);
      foreach (IHardware subHardware in hardware.SubHardware)
        HardwareRemoved(subHardware);
    }

    private void HardwareAdded(IHardware hardware) {
      hardware.SensorRemoved += SensorRemoved;
      foreach (IHardware subHardware in hardware.SubHardware)
        HardwareAdded(subHardware);
    }

    private void SensorRemoved(ISensor sensor) {
      for (int i = 0; i < sensors.Length; i++) {
        if (sensor == sensors[i])
          sensors[i] = null;
      }
    }

    private static string GetFileName(DateTime date) {
      return AppDomain.CurrentDomain.BaseDirectory +
        Path.DirectorySeparatorChar + string.Format(fileNameFormat, date);
    }

    private void AddSensorRemovedHandler() {
      for (int i = 0; i < sensors.Length; i++) {
        ISensor sensor = sensors[i];
        int index = i;
        SensorEventHandler handler = null;        
        handler = s => {
          if (s != sensor) 
            return;

          sensors[index] = null;
          sensor.Hardware.SensorRemoved -= handler;          
        };
        sensor.Hardware.SensorRemoved += handler;
      }
    }

    private bool OpenExistingLogFile() {
      if (!File.Exists(fileName))
        return false;

      string[] identifiers;
      try {
        String line;
        using (StreamReader reader = new StreamReader(fileName)) 
          line = reader.ReadLine(); 
       
        if (string.IsNullOrEmpty(line))
          return false;
        
        identifiers = line.Split(',').Skip(1).ToArray();
      } catch {
        return false;
      }

      if (identifiers.Length == 0)
        return false;

      sensors = new ISensor[identifiers.Length];
      SensorVisitor visitor = new SensorVisitor(sensor => {
        for (int i = 0; i < identifiers.Length; i++)
          if (sensor.Identifier.ToString() == identifiers[i])
            sensors[i] = sensor;
      });
      visitor.VisitComputer(computer);
      return true;
    }

    private void CreateNewLogFile() {
      IList<ISensor> list = new List<ISensor>();
      SensorVisitor visitor = new SensorVisitor(sensor => {
        list.Add(sensor);
      });
      visitor.VisitComputer(computer);
      sensors = list.ToArray();      

      using (StreamWriter writer = new StreamWriter(fileName, false)) {
        writer.Write(",");
        for (int i = 0; i < sensors.Length; i++) {
          writer.Write(sensors[i].Identifier);
          if (i < sensors.Length - 1)
            writer.Write(",");
          else
            writer.WriteLine();
        }

        writer.Write("Time,");
        for (int i = 0; i < sensors.Length; i++) {
          writer.Write('"');
          writer.Write(sensors[i].Name);
          writer.Write('"');
          if (i < sensors.Length - 1)
            writer.Write(",");
          else
            writer.WriteLine();
        }
      }
    }

    public void Log() {
      var now = DateTime.Now;
      if (day != now.Date) {
        day = now.Date;
        fileName = GetFileName(day);

        if (!OpenExistingLogFile())
          CreateNewLogFile();
      }

      using (StreamWriter writer = new StreamWriter(fileName, true)) {
        writer.Write(now.ToString("G", CultureInfo.InvariantCulture));
        writer.Write(",");
        for (int i = 0; i < sensors.Length; i++) {
          if (sensors[i] != null) {
            float? value = sensors[i].Value;
            if (value.HasValue)
              writer.Write(
                value.Value.ToString("R", CultureInfo.InvariantCulture));
          }
          if (i < sensors.Length - 1)
            writer.Write(",");
          else
            writer.WriteLine();
        }
      }

    }
  }
}