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                 // sometimes e.Failure is null, not sure why yet
  74                 if (e.Failure != null)
  75                         Console.Error.WriteLine(e.Failure.Message);
  76         }
  77
  78         public VersionControlServer ConnectToServer()
  79         {
  80                 if (null != _versionControl) return _versionControl;
  81                 string server = GetServerUrl();
  82                 string login = GetLogin(server);
  83
  84                 string userinfo = "";
  85                 string pwd = "";
  86
  87                 int comma = login.IndexOf(",");
  88                 if (comma != -1)
  89                         {
  90                                 userinfo = login.Substring(0, comma);
  91                                 pwd = login.Substring(comma+1);
  92                         }
  93                 else
  94                         {
  95                                 userinfo = login;
  96                                 Console.Write("Password: ");
  97                                 pwd = Console.ReadLine();
  98                         }
  99
 100                 int slash = userinfo.IndexOf('\\');
 101                 if (-1 == slash) Options.Username = userinfo;
 102                 else
 103                         {
 104                                 Options.Domain = userinfo.Substring(0, slash);
 105                                 Options.Username = userinfo.Substring(slash+1);
 106                         }
 107
 108                 _tfs = new TeamFoundationServer(server, new NetworkCredential(Options.Username, pwd, Options.Domain));
 109                 _versionControl = (VersionControlServer) _tfs.GetService(typeof(VersionControlServer));
 110
 111                 _versionControl.Conflict += ConflictEventHandler;
 112                 _versionControl.NonFatalError += ExceptionEventHandler;
 113
 114                 // save credentials if passed
 115                 bool saveSetting = Settings.Current.GetAsBool("Credentials.Save");
 116                 if (saveSetting && !String.IsNullOrEmpty(Options.Login))
 117                         TfsKeyring.SetCredentials(server, Options.Domain, Options.Username, pwd);
 118
 119                 return _versionControl;
 120         }
 121
 122         static public void ShowHelp()
 123         {
 124                 Console.WriteLine("usage: tf [subcommand] [arguments]");
 125                 Console.WriteLine();
 126                 Console.WriteLine("Available subcommands:");
 127                 Console.WriteLine();
 128
 129                 CommandRegistry.ShowHelp();
 130         }
 131
 132         public void ProcessBranch() {}
 133         public void ProcessBranches() {}
 134         public void ProcessDifference() {}
 135         public void ProcessLock() {}
 136         public void ProcessMerge() {}
 137         public void ProcessMerges() {}
 138         public void ProcessResolve() {}
 139         public void ProcessShelve() {}
 140         public void ProcessShelvesets() {}
 141         public void ProcessUndelete() {}
 142         public void ProcessUnlabel() {}
 143         public void ProcessUnshelve() {}
 144
 145         public void ProcessAllCommands()
 146         {
 147                 // some commands don't need to connect to TFS server
 148                 string firstCmd = Commands[0].ToLower();
 149                 Command cmd = null;
 150
 151                 switch (firstCmd)
 152                         {
 153                         case "cache":
 154                                 cmd = new CacheCommand(Commands, Options);
 155                                 break;
 156                         case "config":
 157                         case "configure":
 158                                 cmd = new ConfigureCommand(Commands, Options);
 159                                 break;
 160                         case "help":
 161                                 ShowHelp();
 162                                 Environment.Exit(0);
 163                                 break;
 164                         }
 165
 166                 if (cmd != null)
 167                         {
 168                                 cmd.Run();
 169                                 Environment.Exit(0);
 170                         }
 171
 172                 VersionControlServer vcs = ConnectToServer();
 173
 174                 for (int i=0; i < Commands.Length; i++)
 175                         {
 176                                 string scmd = Commands[i].ToLower();
 177                                 List<string> cmdArgs = new List<string>();
 178                                 cmdArgs.Add(scmd);
 179
 180                                 int j;
 181                                 for (j=i+1; j < Commands.Length; j++)
 182                                         {
 183                                                 string arg = Commands[j];
 184                                                 if (arg == "%") break;
 185                                                 cmdArgs.Add(arg);
 186                                         }
 187
 188                                 ProcessCommand(vcs, scmd, cmdArgs.ToArray());
 189                                 i = j;
 190                         }
 191         }
 192
 193         public void ProcessCommand(VersionControlServer vcs, string cmd,
 194                                                                                                                  string[] cmdArgs)
 195         {
 196                 Type commandType = CommandRegistry.GetCommandType(cmd);
 197                 if (commandType == null)
 198                         {
 199                                 Console.WriteLine("Unknown command: " + cmd);
 200                                 return;
 201                         }
 202
 203                 Command command = (Command) Activator.CreateInstance(commandType, new object[]{cmdArgs, Options, vcs});
 204                 command.Run();
 205         }
 206
 207         public string GetServerUrl()
 208         {
 209                 if (!String.IsNullOrEmpty(Options.Server))
 210                         {
 211                                 if (Options.Server.StartsWith("http://") || Options.Server.StartsWith("https://"))
 212                                         return Options.Server;
 213                                 else
 214                                         return String.Format("http://{0}:8080/", Options.Server);
 215                         }
 216
 217                 WorkspaceInfo info = Workstation.Current.GetLocalWorkspaceInfo(Environment.CurrentDirectory);
 218                 if (info == null)
 219                         {
 220                                 Console.WriteLine("Unable to determine the team foundation server");
 221                                 Console.WriteLine("      hint: try adding /server:<ip|name>");
 222                                 Environment.Exit(1);
 223                         }
 224
 225                 return info.ServerUri.ToString();
 226         }
 227
 228         public string GetLogin(string server)
 229         {
 230                 if (!String.IsNullOrEmpty(Options.Login))
 231                         return Options.Login;
 232
 233                 // check the keyring
 234                 string login = TfsKeyring.GetCredentials(server);
 235                 if (!String.IsNullOrEmpty(login)) return login;
 236
 237                 // finally prompt
 238                 return PromptForLogin();
 239         }
 240
 241         public VersionControlServer VersionControlServer
 242         {
 243                 get { return _versionControl; }
 244         }
 245
 246         // ignoring certificate errors
 247         public bool CheckValidationResult (ServicePoint sp,
 248                                                                                                                                                  X509Certificate certificate, WebRequest request, int error)
 249         {
 250                 return true;
 251         }
 252
 253         public static void Main(string[] args)
 254         {
 255                 if (args.Length == 0) {
 256                         ShowHelp();
 257                         return;
 258                 }
 259
 260                 TfOptions options = new TfOptions();
 261                 options.ProcessArgs(args);
 262
 263                 Driver driver = new Driver(options, options.RemainingArguments);
 264                 ServicePointManager.CertificatePolicy = driver;
 265
 266                 // basic auth doesn't seem to work well on mono when POSTing
 267                 // large data sets via a webservice.
 268                 AuthenticationManager.Unregister("Basic");
 269
 270                 try
 271                         {
 272                                 driver.ProcessAllCommands();
 273                         }
 274                 catch (TeamFoundationServerException e)
 275                         {
 276                                 Console.Error.WriteLine(e.Message);
 277                                 // Console.WriteLine(options.Domain + "\\" + options.Username);
 278                         }
 279         }
 280 }