Avalonia/Perspex.Controls/TabStrip.cs

// -----------------------------------------------------------------------
// <copyright file="TabStrip.cs" company="Steven Kirk">
// Copyright 2014 MIT Licence. See licence.md for more information.
// </copyright>
// -----------------------------------------------------------------------

namespace Perspex.Controls
{
    using System;
    using System.Collections;
    using System.Linq;
    using Perspex.Controls.Generators;
    using Perspex.Input;

    public class TabStrip : SelectingItemsControl
    {
        private static readonly ItemsPanelTemplate PanelTemplate = new ItemsPanelTemplate(
            () => new StackPanel());

        static TabStrip()
        {
            ItemsPanelProperty.OverrideDefaultValue(typeof(TabStrip), PanelTemplate);
        }

        public TabStrip()
        {
            this.PointerPressed += this.OnPointerPressed;
            this.GetObservable(SelectedItemProperty).Subscribe(this.SelectedItemChanged);
        }

        protected override ItemContainerGenerator CreateItemContainerGenerator()
        {
            TabControl tabControl = this.TemplatedParent as TabControl;
            ItemContainerGenerator result;

            if (tabControl != null)
            {
                result = tabControl.ItemContainerGenerator;
            }
            else
            {
                result = new TypedItemContainerGenerator<TabItem>(this);
            }

            result.StateChanged += ItemsContainerGeneratorStateChanged;

            return result;
        }

        private void ItemsContainerGeneratorStateChanged(object sender, EventArgs e)
        {
            if (this.ItemContainerGenerator.State == ItemContainerGeneratorState.Generated)
            {
                var tabs = this.ItemContainerGenerator.GetAll()
                    .Select(x => x.Item2)
                    .OfType<TabItem>()
                    .ToList();

                this.SelectedItem = tabs.FirstOrDefault(x => x.IsSelected) ?? tabs.FirstOrDefault();
            }
        }

        private void OnPointerPressed(object sender, PointerEventArgs e)
        {
            IVisual source = (IVisual)e.Source;
            ContentPresenter presenter = source.GetVisualAncestors()
                .OfType<ContentPresenter>()
                .FirstOrDefault();

            if (presenter !=  null)
            {
                TabItem item = presenter.TemplatedParent as TabItem;

                if (item != null)
                {
                    this.SelectedItem = item;
                }
            }
        }

        private void SelectedItemChanged(object selected)
        {
            foreach (TabItem item in this.ItemContainerGenerator.GetAll().Select(x => x.Item2))
            {
                item.IsSelected = selected == item;
            }
        }
    }
}