using System;
using System.Diagnostics;

namespace Developpez.Dotnet.Diagnostics
{
    /// <summary>
    /// Permet de chronométrer un traitement effectué dans un bloc <c>using</c>.
    /// </summary>
    public class AutoStopwatch : IDisposable
    {
        private Stopwatch _stopwatch;
        private Action<TimeSpan> _action;

        /// <summary>
        /// Initialise une nouvelle instance de <c>AutoStopwatch</c>
        /// </summary>
        public AutoStopwatch()
        {
            _stopwatch = Stopwatch.StartNew();
        }

        /// <summary>
        /// Initialise une nouvelle instance de <c>AutoStopwatch</c>
        /// en spécifiant l'action à effectuer quand le traitement se termine
        /// </summary>
        /// <param name="action">Action a exécuter quand le traitement est terminé (prend en paramètre le temps écoulé)</param>
        public AutoStopwatch(Action<TimeSpan> action)
        {
            _action = action;
            _stopwatch = Stopwatch.StartNew();
        }

        /// <summary>
        /// Renvoie le temps écoulé depuis le début du traitement
        /// </summary>
        public TimeSpan Elapsed
        {
            get { return _stopwatch.Elapsed; }
        }

        /// <summary>
        /// Renvoie le nombre de millisecondes écoulées depuis le début du traitement
        /// </summary>
        public long ElapsedMilliseconds
        {
            get { return _stopwatch.ElapsedMilliseconds; }
        }

        /// <summary>
        /// Renvoie le nombre de ticks écoulés depuis le début du traitement
        /// </summary>
        public long ElapsedTicks
        {
            get { return _stopwatch.ElapsedTicks; }
        }

        /// <summary>
        /// Arrête le chronomètre.
        /// </summary>
        public void Stop()
        {
            if (_stopwatch.IsRunning)
                _stopwatch.Stop();
        }

        #region IDisposable Members

        /// <summary>
        /// Libère les ressources utilisées par l'objet et appelle l'action de fin de traitement.
        /// </summary>
        void IDisposable.Dispose()
        {
            Stop();
            if (_action != null)
            {
                var tmp = _action;
                _action = null;
                tmp(_stopwatch.Elapsed);
            }
        }

        #endregion

        /// <summary>
        /// Démarre une nouvelle AutoStopwatch avec l'action indiquée
        /// </summary>
        /// <param name="action">Action a exécuter quand le traitement est terminé (prend en paramètre le temps écoulé)</param>
        /// <returns>Une nouvelle instance de AutoStopwatch</returns>
        public static AutoStopwatch StartNew(Action<TimeSpan> action)
        {
            return new AutoStopwatch(action);
        }
    }
}