SourceTreeViewController.cs

using System;
using System.Reactive.Linq;
using CodeHub.Core;
using CodeHub.Core.Services;
using CodeHub.iOS.DialogElements;
using CodeHub.iOS.Views;
using ReactiveUI;
using Splat;
using UIKit;
using System.Linq;
using System.Collections.Generic;
using CodeHub.Core.Utils;
using CodeHub.iOS.Utilities;
using Humanizer;
using System.Reactive;

namespace CodeHub.iOS.ViewControllers.Source
{
    public class SourceTreeViewController : DialogViewController
    {
        private static string LoadErrorMessage = "Unable to load source tree.";
        private readonly SourceTitleView _titleView = new SourceTitleView();
        private bool _branchSelectorShowsBranches = true;
        private readonly string _username;
        private readonly string _repository;
        private readonly string _path;
        private string _sha;

        private readonly ReactiveCommand<Unit, Unit> _addFileCommand;
        private readonly ReactiveCommand<string, IReadOnlyList<Octokit.RepositoryContent>> _loadContents;

        private bool _canAddFile;
        private bool CanAddFile
        {
            get { return _canAddFile; }
            set { this.RaiseAndSetIfChanged(ref _canAddFile, value); }
        }

        private ShaType _shaType;
        private ShaType ShaType
        {
            get { return _shaType; }
            set { this.RaiseAndSetIfChanged(ref _shaType, value); }
        }

        public SourceTreeViewController(
            string username,
            string repository,
            string path,
            string sha,
            ShaType shaType,
            IApplicationService applicationService = null,
            IFeaturesService featuresService = null)
            : base(style: UITableViewStyle.Plain)
        {
            _username = username;
            _repository = repository;
            _path = path;
            _sha = sha;
            _shaType = shaType;

            _addFileCommand = ReactiveCommand.Create(
                ShowAddSource, 
                this.WhenAnyValue(x => x.CanAddFile, x => x.ShaType)
                    .Select(x => x.Item1 && x.Item2 == ShaType.Branch));

            applicationService = applicationService ?? Locator.Current.GetService<IApplicationService>();
            featuresService = featuresService ?? Locator.Current.GetService<IFeaturesService>();

            _loadContents = ReactiveCommand.CreateFromTask(async (string shaRef) =>
            {
                if (ShaType == ShaType.Branch)
                {
                    var repo = await applicationService.GitHubClient.Repository.Get(username, repository);
                    CanAddFile = repo.Permissions.Push;
                }
                else
                {
                    CanAddFile = false;
                }

                var encodedShaRef = System.Web.HttpUtility.UrlEncode(shaRef);

                if (string.IsNullOrEmpty(path))
                {
                    return await applicationService
                        .GitHubClient.Repository.Content
                        .GetAllContentsByRef(username, repository, encodedShaRef);
                }

                var encodedPath = path == null ? null : Uri.EscapeDataString(path);

                return await applicationService
                    .GitHubClient.Repository.Content
                    .GetAllContentsByRef(username, repository, encodedPath, encodedShaRef);
            });

            var addFileButton = new UIBarButtonItem(UIBarButtonSystemItem.Add);
            NavigationItem.RightBarButtonItem = addFileButton;

            _loadContents
                .ThrownExceptions
                .Do(_ => SetErrorView())
                .Select(HandleLoadError)
                .SelectMany(Interactions.Errors.Handle)
                .Subscribe();

            OnActivation(d =>
            {
                d(_titleView
                  .GetClickedObservable()
                  .Subscribe(_ => ShowBranchSelector()));

                d(_addFileCommand
                  .CanExecute
                  .Subscribe(x => addFileButton.Enabled = x));

                d(addFileButton
                  .GetClickedObservable()
                  .Select(_ => Unit.Default)
                  .InvokeReactiveCommand(_addFileCommand));
            });

            Appearing
                .Select(_ => _sha)
                .Where(x => !string.IsNullOrEmpty(x))
                .DistinctUntilChanged()
                .Do(_ => SetLoading(true))
                .InvokeReactiveCommand(_loadContents);

            _loadContents
                .Do(_ => SetLoading(false))
                .Subscribe(SetElements);

            NavigationItem.TitleView = _titleView;
        }

        private void SetLoading(bool isLoading)
        {
            Root.Reset();
            TableView.BackgroundView = null;
            TableView.TableFooterView = isLoading ? new LoadingIndicatorView() : null;
        }

        private void SetErrorView()
        {
            var emptyListView = new EmptyListView(Octicon.Code.ToEmptyListImage(), LoadErrorMessage)
            {
                Alpha = 0
            };

            TableView.TableFooterView = new UIView();
            TableView.BackgroundView = emptyListView;

            UIView.Animate(0.8, 0, UIViewAnimationOptions.CurveEaseIn,
                           () => emptyListView.Alpha = 1, null);
        }

        private UserError HandleLoadError(Exception error)
        {
            Root.Reset();

            var apiException = error as Octokit.ApiException;
            if (apiException?.StatusCode == System.Net.HttpStatusCode.NotFound)
                return new UserError($"The current directory does not exist under the selected Git reference ({_sha})");

            return new UserError(LoadErrorMessage, error);
        }

        public override void ViewWillAppear(bool animated)
        {
            base.ViewWillAppear(animated);

            _titleView.SubText = ShaType == ShaType.Hash
                ? _sha.Substring(0, Math.Min(_sha.Length, 7))
                : _sha;

            if (string.IsNullOrEmpty(_path))
                _titleView.Text = _repository;
            else
            {
                var path = _path.TrimEnd('/');
                _titleView.Text = path.Substring(path.LastIndexOf('/') + 1);
            }
        }

        private void SetElements(IEnumerable<Octokit.RepositoryContent> items)
        {
            var elements = items
                .OrderByDescending(x => x.Type.Value != Octokit.ContentType.File || 
                                   (x.Type.Value == Octokit.ContentType.File && x.DownloadUrl == null))
                .ThenBy(x => x.Name)
                .Select(CreateElement);
            
            Root.Reset(new Section { elements });
        }

        private void ShowBranchSelector()
        {
            var view = _branchSelectorShowsBranches
                ? BranchesAndTagsViewController.SelectedView.Branches
                : BranchesAndTagsViewController.SelectedView.Tags;

            var viewController = new BranchesAndTagsViewController(_username, _repository, view);

            viewController.TagSelected.Take(1).Subscribe(tag =>
            {
                _sha = tag.Name;
                ShaType = ShaType.Tag;
                _branchSelectorShowsBranches = false;
                this.DismissViewController(true, null);
            });

            viewController.BranchSelected.Take(1).Subscribe(branch =>
            {
                _sha = branch.Name;
                ShaType = ShaType.Branch;
                _branchSelectorShowsBranches = true;
                this.DismissViewController(true, null);
            });

            this.PresentModalViewController(viewController);
        }

        private void GoToSourceTree(Octokit.RepositoryContent content)
        {
            this.PushViewController(new SourceTreeViewController(
                _username, _repository, content.Path, _sha, ShaType));
        }

        private void GoToSubModule(Octokit.RepositoryContent content)
        {
            if (content == null)
                return;

            var gitUrl = content.GitUrl;
            if (string.IsNullOrEmpty(gitUrl))
                return;

            var repoDelimIndex = gitUrl.IndexOf("/repos/", StringComparison.Ordinal);
            if (repoDelimIndex < 0 || repoDelimIndex + 7 > gitUrl.Length)
                return;

            var nameAndSlug = gitUrl.Substring(repoDelimIndex + 7);
            var indexOfGit = nameAndSlug.LastIndexOf("/git", StringComparison.Ordinal);
            indexOfGit = indexOfGit < 0 ? 0 : indexOfGit;

            var repoId = RepositoryIdentifier.FromFullName(nameAndSlug.Substring(0, indexOfGit));
            if (repoId == null)
                return;

            var sha = gitUrl.Substring(gitUrl.LastIndexOf("/", StringComparison.Ordinal) + 1);

            this.PushViewController(new SourceTreeViewController(
                repoId.Owner, repoId.Name, null, sha, ShaType.Hash));
        }

        private void GoToFile(Octokit.RepositoryContent content)
        {
            var viewController = new FileSourceViewController(
                _username, _repository, content.Path, _sha, ShaType)
            {
                Content = content
            };

            this.PushViewController(viewController);
        }

        private void ShowAddSource()
        {
            var viewController = new AddSourceViewController(
                _username, _repository, _path, _sha);

            viewController
                .Success
                .Select(_ => _sha)
                .InvokeReactiveCommand(_loadContents);

            this.PresentModalViewController(viewController);
        }

        private Element CreateElement(Octokit.RepositoryContent content)
        {
            var weakRef = new WeakReference<SourceTreeViewController>(this);

            if (content.Type == Octokit.ContentType.Dir)
            {
                var e = new StringElement(content.Name, Octicon.FileDirectory.ToImage());
                e.Clicked.Subscribe(_ => weakRef.Get()?.GoToSourceTree(content));
                return e;
            }

            if (content.Type == Octokit.ContentType.File)
            {
                if (content.DownloadUrl != null)
                {
                    var e = new StringElement(content.Name, Octicon.FileCode.ToImage());
                    e.Style = UITableViewCellStyle.Subtitle;
                    e.Value = content.Size.Bytes().ToString("#.#");
                    e.Clicked.Subscribe(_ => weakRef.Get()?.GoToFile(content));
                    return e;
                }
                else
                {
                    var e = new StringElement(content.Name, Octicon.FileSubmodule.ToImage());
                    e.Clicked.Subscribe(_ => weakRef.Get()?.GoToSubModule(content));
                    return e;
                }
            }

            return new StringElement(content.Name) { Image = Octicon.FileMedia.ToImage() };
        }
    }
}