Synnefo » Pithos MS Client

using System;

using System.Collections;

using System.Collections.Generic;

using System.Collections.Specialized;

using System.IO;

using System.Linq;

using System.Text;

using Caliburn.Micro;

namespace Pithos.Client.WPF.SelectiveSynch

{

    public class DirectoryRecord : PropertyChangedBase,IEnumerable<DirectoryRecord>

    {

        public DirectoryInfo Info { get; set; }

        DirectoryRecord _parent;

        public bool Added { get; set; }

        public bool Removed { get; set; }

        private bool? _isChecked;

        #region IsChecked

        /// <summary>

        /// Gets/sets the state of the associated UI toggle (ex. CheckBox).

        /// The return value is calculated based on the check state of all

        /// child FooViewModels.  Setting this property to true or false

        /// will set all children to the same check state, and setting it 

        /// to any value will cause the parent to verify its check state.

        /// </summary>

        public bool? IsChecked

        {

            get { return _isChecked; }

            set { this.SetIsChecked(value, true, true); }

        }

        void SetIsChecked(bool? value, bool updateChildren, bool updateParent)

        {

            if (value == _isChecked)

                return;

            _isChecked = value;

            //If the value is null both Added and Removed should be False

            Added = _isChecked??false;

            Removed = !(_isChecked??true);

            if (updateChildren && _isChecked.HasValue)

                this.Directories.ForEach(c => c.SetIsChecked(_isChecked, true, false));

            if (updateParent && _parent != null)

                _parent.VerifyCheckState();

            this.RaisePropertyChangedEventImmediately("IsChecked");

        }

        void VerifyCheckState()

        {

            bool? state = null;

            for (int i = 0; i < this.Directories.Count; ++i)

            {

                bool? current = this.Directories[i].IsChecked;

                if (i == 0)

                {

                    state = current;

                }

                else if (state != current)

                {

                    state = null;

                    break;

                }

            }

            this.SetIsChecked(state, false, true);

        }

        #endregion // IsChecked

        public bool IsInitiallySelected { get; private set; }

        readonly Lazy<List<DirectoryRecord>> _directories = new Lazy<List<DirectoryRecord>>();

        public List<DirectoryRecord> Directories

        {

            get

            {

                return _directories.Value;

            }

        }

        public DirectoryRecord(string ignorePath)

        {

            _directories = new Lazy<List<DirectoryRecord>>(() => 

                (from directory in Info.EnumerateDirectories("*", SearchOption.TopDirectoryOnly)

                where !directory.FullName.StartsWith(ignorePath)

                select new DirectoryRecord(ignorePath) { Info = directory }).ToList());

        }

/*

        public IEnumerable<DirectoryInfo> GetCheckedDirectories()

        {

            var q = from record in this

                    where record.IsChecked==true

                    select record.Info;

            return q;

        }

*/

/*

        public void SetSelections(StringCollection selections)

        {

            IsChecked=selections.Contains(Info.FullName);                

            foreach (var children in Directories)

            {

                children.SetSelections(selections);

            }

        }

*/

        public IEnumerator<DirectoryRecord> GetEnumerator()

        {

            yield return this;

            foreach (var children in Directories)

                foreach (var info in children)

                {

                    yield return info;

                }

        }

        IEnumerator IEnumerable.GetEnumerator()

        {

            return GetEnumerator();

        }

    }

}