git-tfs/GitTfs/Commands/ResetRemote.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using NDesk.Options;
using Sep.Git.Tfs.Core;
using StructureMap;

namespace Sep.Git.Tfs.Commands
{
    [Pluggable("reset-remote")]
    [Description("reset-remote commit-sha1-ref\n ex : git tfs reset-remote 3dcce821d7a20e6b2499cdd6f2f52ffbe8507be7")]
    [RequiresValidGitRepository]
    public class ResetRemote : GitTfsCommand
    {
        private readonly Globals globals;
        private readonly TextWriter stdout;
        private bool ForceResetRemote;

        public ResetRemote(Globals globals, TextWriter stdout)
        {
            this.globals = globals;
            this.stdout = stdout;
        }

        public virtual OptionSet OptionSet
        {
            get
            {
                return new OptionSet
                    {
                        { "force", "Force reset remote (when current commit do not belong to the remote to reset)", v => ForceResetRemote = v != null },
                    };
            }
        }

        public int Run(string commitRef)
        {
            var targetCommit = globals.Repository.GetTfsCommit(commitRef);
            if (targetCommit == null)
                throw new GitTfsException("error : the commit where you want to reset the tfs remote does not belong to a tfs remote!");

            if (!ForceResetRemote)
            {
                var currentTfsCommit = globals.Repository.GetCurrentTfsCommit();
                if (currentTfsCommit == null)
                    throw new GitTfsException("error : the current commit does not belong to a tfs remote!",
                        new List<string> { "Use '--force' option to reset a remote from a commit not belonging a tfs remote"});

                if (targetCommit.Remote.Id != currentTfsCommit.Remote.Id)
                    throw new GitTfsException("error : the commit where you want to reset the tfs remote does not belong to the current tfs remote \""
                                              + currentTfsCommit.Remote.Id + "\"!",
                        new List<string> { "Use '--force' option to reset the remote \""
                                              + targetCommit.Remote.Id + "\" to a commit not belonging to the current remote \""
                                              + currentTfsCommit.Remote.Id + "\"" });
            }

            globals.Repository.ResetRemote(targetCommit.Remote, targetCommit.GitCommit);

            stdout.WriteLine("Remote 'tfs/" + targetCommit.Remote.Id + "' reset successfully.\n");
            stdout.WriteLine("Note: remember to use the '--force' option when doing the next 'fetch' to force git-tfs to fetch again the changesets!");
            return GitTfsExitCodes.OK;
        }
    }
}