tools/tf/Driver.cs

   1 //
   2 // Driver.cs
   3 //
   4 // Authors:
   5 //      Joel Reed (joelwreed@gmail.com)
   6 //
   7
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 using System;
  30 using System.Reflection;
  31 using System.Collections.Generic;
  32 using System.Collections.Specialized;
  33 using System.Net;
  34 using System.IO;
  35 using System.Text;
  36 using System.Security.Cryptography.X509Certificates;
  37 using Mono.GetOptions;
  38 using Microsoft.TeamFoundation;
  39 using Microsoft.TeamFoundation.Client;
  40 using Microsoft.TeamFoundation.VersionControl.Client;
  41 using Microsoft.TeamFoundation.Server;
  42
  43 [assembly: AssemblyTitle ("tf.exe")]
  44 [assembly: AssemblyVersion ("1.0")]
  45 [assembly: AssemblyDescription ("Team Foundation Source Control Tool")]
  46 [assembly: AssemblyCopyright ("(c) Joel W. Reed")]
  47
  48 [assembly: Mono.UsageComplement ("")]
  49
  50 [assembly: Mono.About("Team Foundation Source Control Tool")]
  51 [assembly: Mono.Author ("Joel W. Reed")]
  52
  53 partial class Driver : ICertificatePolicy
  54 {
  55         private TeamFoundationServer _tfs;
  56         private VersionControlServer _versionControl;
  57         private TfOptions Options;
  58         private string[] Commands;
  59
  60         public Driver(TfOptions options, string[] commands)
  61                 {
  62                         Options = options;
  63                         Commands = commands;
  64                 }
  65
  66         static void ConflictEventHandler(Object sender, ConflictEventArgs e)
  67         {
  68                 Console.Error.WriteLine(e.Message);
  69         }
  70
  71         static void ExceptionEventHandler(Object sender, ExceptionEventArgs e)
  72         {
  73                 Console.Error.WriteLine(e.Failure.Message);
  74         }
  75
  76         public VersionControlServer ConnectToServer()
  77         {
  78                 if (null != _versionControl) return _versionControl;
  79                 string server = GetServerUrl();
  80                 string login = GetLogin(server);
  81
  82                 string userinfo = "";
  83                 string pwd = "";
  84
  85                 int comma = login.IndexOf(",");
  86                 if (comma != -1)
  87                         {
  88                                 userinfo = login.Substring(0, comma);
  89                                 pwd = login.Substring(comma+1);
  90                         }
  91                 else
  92                         {
  93                                 userinfo = login;
  94                                 Console.Write("Password: ");
  95                                 pwd = Console.ReadLine();
  96                         }
  97
  98                 int slash = userinfo.IndexOf('\\');
  99                 if (-1 == slash) Options.Username = userinfo;
 100                 else
 101                         {
 102                                 Options.Domain = userinfo.Substring(0, slash);
 103                                 Options.Username = userinfo.Substring(slash+1);
 104                         }
 105
 106                 _tfs = new TeamFoundationServer(server, new NetworkCredential(Options.Username, pwd, Options.Domain));
 107                 _versionControl = (VersionControlServer) _tfs.GetService(typeof(VersionControlServer));
 108
 109                 _versionControl.Conflict += ConflictEventHandler;
 110                 _versionControl.NonFatalError += ExceptionEventHandler;
 111
 112                 // save credentials if passed
 113                 bool saveSetting = Settings.Current.GetAsBool("Credentials.Save");
 114                 if (saveSetting && !String.IsNullOrEmpty(Options.Login))
 115                         TfsKeyring.SetCredentials(server, Options.Domain, Options.Username, pwd);
 116
 117                 return _versionControl;
 118         }
 119
 120         static public void ShowHelp()
 121         {
 122                 Console.WriteLine("usage: tf [subcommand] [arguments]");
 123                 Console.WriteLine();
 124                 Console.WriteLine("Available subcommands:");
 125                 Console.WriteLine();
 126                 Console.WriteLine("  add");
 127                 Console.WriteLine("  cache");
 128                 Console.WriteLine("  changeset");
 129                 Console.WriteLine("  checkin");
 130                 Console.WriteLine("  checkout (alias edit)");
 131                 Console.WriteLine("  configure (alias config)");
 132                 Console.WriteLine("  delete");
 133                 Console.WriteLine("  diff");
 134                 Console.WriteLine("  dir");
 135                 Console.WriteLine("  get");
 136                 Console.WriteLine("  history");
 137                 Console.WriteLine("  labels");
 138                 Console.WriteLine("  ls-files");
 139                 Console.WriteLine("  online");
 140                 Console.WriteLine("  permission");
 141                 Console.WriteLine("  properties");
 142                 Console.WriteLine("  rename");
 143                 Console.WriteLine("  status");
 144                 Console.WriteLine("  treeclean");
 145                 Console.WriteLine("  undo");
 146                 Console.WriteLine("  view");
 147                 Console.WriteLine("  workfold");
 148                 Console.WriteLine("  workspace");
 149                 Console.WriteLine("  workspaces");
 150         }
 151
 152         public void ProcessBranch() {}
 153         public void ProcessBranches() {}
 154         public void ProcessDifference() {}
 155         public void ProcessLock() {}
 156         public void ProcessMerge() {}
 157         public void ProcessMerges() {}
 158         public void ProcessResolve() {}
 159         public void ProcessShelve() {}
 160         public void ProcessShelvesets() {}
 161         public void ProcessUndelete() {}
 162         public void ProcessUnlabel() {}
 163         public void ProcessUnshelve() {}
 164
 165         public void ProcessAllCommands()
 166         {
 167                 // some commands don't need to connect to TFS server
 168                 string firstCmd = Commands[0].ToLower();
 169                 Command cmd = null;
 170
 171                 switch (firstCmd)
 172                         {
 173                         case "cache":
 174                                 cmd = new CacheCommand(Commands, Options);
 175                                 break;
 176                         case "config":
 177                         case "configure":
 178                                 cmd = new ConfigureCommand(Commands, Options);
 179                                 break;
 180                         case "help":
 181                                 ShowHelp();
 182                                 Environment.Exit(0);
 183                                 break;
 184                         }
 185
 186                 if (cmd != null)
 187                         {
 188                                 cmd.Run();
 189                                 Environment.Exit(0);
 190                         }
 191
 192                 VersionControlServer vcs = ConnectToServer();
 193
 194                 for (int i=0; i < Commands.Length; i++)
 195                         {
 196                                 string scmd = Commands[i].ToLower();
 197                                 List<string> cmdArgs = new List<string>();
 198                                 cmdArgs.Add(scmd);
 199
 200                                 int j;
 201                                 for (j=i+1; j < Commands.Length; j++)
 202                                         {
 203                                                 string arg = Commands[j];
 204                                                 if (arg == "%") break;
 205                                                 cmdArgs.Add(arg);
 206                                         }
 207
 208                                 ProcessCommand(vcs, scmd, cmdArgs.ToArray());
 209                                 i = j;
 210                         }
 211         }
 212
 213         public void ProcessCommand(VersionControlServer vcs, string cmd,
 214                                                                                                                  string[] cmdArgs)
 215         {
 216                 Command command = null;
 217
 218                 switch (cmd)
 219                         {
 220                         case "add":
 221                                 command = new AddCommand(cmdArgs, Options, vcs);
 222                                 break;
 223                         case "branch":
 224                                 ProcessBranch();
 225                                 break;
 226                         case "branches":
 227                                 ProcessBranches();
 228                                 break;
 229                         case "changeset":
 230                                 command = new ChangesetCommand(cmdArgs, Options, vcs);
 231                                 break;
 232                         case "checkin":
 233                                 command = new CheckinCommand(cmdArgs, Options, vcs);
 234                                 break;
 235                         case "checkout":
 236                                 command = new CheckoutCommand(cmdArgs, Options, vcs);
 237                                 break;
 238                         case "edit": // command alias
 239                                 command = new CheckoutCommand(cmdArgs, Options, vcs);
 240                                 break;
 241                         case "del": // command alias
 242                                 command = new DeleteCommand(cmdArgs, Options, vcs);
 243                                 break;
 244                         case "delete":
 245                                 command = new DeleteCommand(cmdArgs, Options, vcs);
 246                                 break;
 247                         case "diff": // command alias
 248                                 command = new DiffCommand(cmdArgs, Options, vcs);
 249                                 break;
 250                         case "difference":
 251                                 command = new DiffCommand(cmdArgs, Options, vcs);
 252                                 break;
 253                         case "dir":
 254                                 command = new DirCommand(cmdArgs, Options, vcs);
 255                                 break;
 256                         case "get":
 257                                 command = new GetCommand(cmdArgs, Options, vcs);
 258                                 break;
 259                         case "help":
 260                                 ShowHelp();
 261                                 break;
 262                         case "hist": // command alias
 263                                 command = new HistoryCommand(cmdArgs, Options, vcs);
 264                                 break;
 265                         case "history":
 266                                 command = new HistoryCommand(cmdArgs, Options, vcs);
 267                                 break;
 268                         case "label":
 269                                 command = new LabelCommand(cmdArgs, Options, vcs);
 270                                 break;
 271                         case "labels":
 272                                 command = new LabelsCommand(cmdArgs, Options, vcs);
 273                                 break;
 274                         case "lock":
 275                                 ProcessLock();
 276                                 break;
 277                         case "ls-files":
 278                                 command = new LsFilesCommand(cmdArgs, Options, vcs);
 279                                 break;
 280                         case "online":
 281                                 command = new OnlineCommand(cmdArgs, Options, vcs);
 282                                 break;
 283                         case "merge":
 284                                 ProcessMerge();
 285                                 break;
 286                         case "merges":
 287                                 ProcessMerges();
 288                                 break;
 289                         case "move":
 290                                 command = new RenameCommand(cmdArgs, Options, vcs);
 291                                 break;
 292                         case "perm":
 293                                 command = new PermissionCommand(cmdArgs, Options, vcs);
 294                                 break;
 295                         case "permission":
 296                                 command = new PermissionCommand(cmdArgs, Options, vcs);
 297                                 break;
 298                         case "prop":
 299                                 command = new PropertiesCommand(cmdArgs, Options, vcs);
 300                                 break;
 301                         case "properties":
 302                                 command = new PropertiesCommand(cmdArgs, Options, vcs);
 303                                 break;
 304                         case "ren":
 305                                 command = new RenameCommand(cmdArgs, Options, vcs);
 306                                 break;
 307                         case "rename":
 308                                 command = new RenameCommand(cmdArgs, Options, vcs);
 309                                 break;
 310                         case "resolve":
 311                                 ProcessResolve();
 312                                 break;
 313                         case "shelve":
 314                                 ProcessShelve();
 315                                 break;
 316                         case "shelvesets":
 317                                 ProcessShelvesets();
 318                                 break;
 319                         case "stat":
 320                                 command = new StatusCommand(cmdArgs, Options, vcs);
 321                                 break;
 322                         case "status":
 323                                 command = new StatusCommand(cmdArgs, Options, vcs);
 324                                 break;
 325                         case "treeclean":
 326                                 command = new TreeCleanCommand(cmdArgs, Options, vcs);
 327                                 break;
 328                         case "treediff":
 329                                 command = new TreeDiffCommand(cmdArgs, Options, vcs);
 330                                 break;
 331                         case "undel": // alias for undelete
 332                                 ProcessUndelete();
 333                                 break;
 334                         case "undelete":
 335                                 ProcessUndelete();
 336                                 break;
 337                         case "undo":
 338                                 command = new UndoCommand(cmdArgs, Options, vcs);
 339                                 break;
 340                         case "unlabel":
 341                                 command = new UnlabelCommand(cmdArgs, Options, vcs);
 342                                 break;
 343                         case "unshelve":
 344                                 ProcessUnshelve();
 345                                 break;
 346                         case "view":
 347                                 command = new ViewCommand(cmdArgs, Options, vcs);
 348                                 break;
 349                         case "workfold":
 350                                 command = new WorkfoldCommand(cmdArgs, Options, vcs);
 351                                 break;
 352                         case "workspace":
 353                                 command = new WorkspaceCommand(cmdArgs, Options, vcs);
 354                                 break;
 355                         case "workspaces":
 356                                 command = new WorkspacesCommand(cmdArgs, Options, vcs);
 357                                 break;
 358                         default:
 359                                 Console.WriteLine("Unknown command: " + cmd);
 360                                 break;
 361                         }
 362
 363                 if (command != null) command.Run();
 364         }
 365
 366         public string GetServerUrl()
 367         {
 368                 if (!String.IsNullOrEmpty(Options.Server))
 369                         {
 370                                 if (Options.Server.StartsWith("http://") || Options.Server.StartsWith("https://"))
 371                                         return Options.Server;
 372                                 else
 373                                         return String.Format("http://{0}:8080/", Options.Server);
 374                         }
 375
 376                 WorkspaceInfo info = Workstation.Current.GetLocalWorkspaceInfo(Environment.CurrentDirectory);
 377                 if (info == null)
 378                         {
 379                                 Console.WriteLine("Unable to determine the team foundation server");
 380                                 Console.WriteLine("      hint: try adding /server:<ip|name>");
 381                                 Environment.Exit(1);
 382                         }
 383
 384                 return info.ServerUri.ToString();
 385         }
 386
 387         public string GetLogin(string server)
 388         {
 389                 if (!String.IsNullOrEmpty(Options.Login))
 390                         return Options.Login;
 391
 392                 // check the keyring
 393                 string login = TfsKeyring.GetCredentials(server);
 394                 if (!String.IsNullOrEmpty(login)) return login;
 395
 396                 // finally prompt
 397                 return PromptForLogin();
 398         }
 399
 400         public VersionControlServer VersionControlServer
 401         {
 402                 get { return _versionControl; }
 403         }
 404
 405         // ignoring certificate errors
 406         public bool CheckValidationResult (ServicePoint sp,
 407                                                                                                                                                  X509Certificate certificate, WebRequest request, int error)
 408         {
 409                 return true;
 410         }
 411
 412         public static void Main(string[] args)
 413         {
 414                 if (args.Length == 0) {
 415                         ShowHelp();
 416                         return;
 417                 }
 418
 419                 TfOptions options = new TfOptions();
 420                 options.ProcessArgs(args);
 421
 422                 Driver driver = new Driver(options, options.RemainingArguments);
 423                 ServicePointManager.CertificatePolicy = driver;
 424
 425                 // basic auth doesn't seem to work well on mono when POSTing
 426                 // large data sets via a webservice.
 427                 AuthenticationManager.Unregister("Basic");
 428
 429                 try
 430                         {
 431                                 driver.ProcessAllCommands();
 432                         }
 433                 catch (TeamFoundationServerException e)
 434                         {
 435                                 Console.Error.WriteLine(e.Message);
 436                                 // Console.WriteLine(options.Domain + "\\" + options.Username);
 437                         }
 438         }
 439 }