tools/tf/Command.cs

   1 //
   2 // Command.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.IO;
  31 using System.Text;
  32 using System.Text.RegularExpressions;
  33 using System.Collections.Generic;
  34 using Microsoft.TeamFoundation.Client;
  35 using Microsoft.TeamFoundation.VersionControl.Client;
  36
  37 abstract class Command
  38 {
  39         private static bool runningOnUnix = true;
  40         private string[] arguments;
  41         private CommandOptions options = new CommandOptions();
  42         private VersionControlServer vcs;
  43         protected Driver driver;
  44
  45         public string[] Arguments
  46         {
  47                 get { return arguments; }
  48         }
  49
  50         public CommandOptions Options
  51         {
  52                 get { return options; }
  53         }
  54
  55         static Command()
  56         {
  57                 int p = (int) Environment.OSVersion.Platform;
  58                 if (!((p == 4) || (p == 128))) runningOnUnix = false;
  59         }
  60
  61         public static StringComparer PathComparer
  62         {
  63                 get {
  64                         if (!runningOnUnix) return StringComparer.CurrentCultureIgnoreCase;
  65                         return StringComparer.CurrentCulture;
  66                 }
  67         }
  68
  69         public Command(Driver driver, string[] args)
  70         {
  71                 this.driver = driver;
  72                 this.vcs = driver.VersionControlServer;
  73                 options.ProcessArgs(args);
  74                 this.arguments = Options.RemainingArguments;
  75         }
  76
  77         public Workspace GetWorkspaceFromServer()
  78         {
  79                 string name = Options.Workspace;
  80
  81                 // if option passed use it
  82                 if (String.IsNullOrEmpty(name))
  83                         {
  84                                 // guess based on current working directory
  85                                 WorkspaceInfo info = Workstation.Current.GetLocalWorkspaceInfo(Environment.CurrentDirectory);
  86                                 if (info != null) name = info.Name;
  87                         }
  88
  89                 if (String.IsNullOrEmpty(name))
  90                         {
  91                                 Console.WriteLine("Unable to determine the workspace");
  92                                 Console.WriteLine("      hint: try adding /workspace:<name>");
  93                                 Environment.Exit(1);
  94                         }
  95
  96                 return VersionControlServer.GetWorkspace(name, driver.Username);
  97         }
  98
  99         public Workspace GetWorkspaceFromCache()
 100         {
 101                 string path = Environment.CurrentDirectory;
 102                 if (Arguments.Length > 1)
 103                         {
 104                                 path = Path.GetFullPath(Arguments[1]);
 105                         }
 106
 107                 WorkspaceInfo info = Workstation.Current.GetLocalWorkspaceInfo(path);
 108                 if (info == null && (!String.IsNullOrEmpty(Options.Workspace)))
 109                         {
 110                                 string ownerName = String.Format("{0}\\{1}", driver.Domain, driver.Username).ToUpper();
 111                                 info = Workstation.Current.GetLocalWorkspaceInfo(vcs, Options.Workspace,
 112                                                                                                                                                                                                                                  ownerName);
 113                         }
 114
 115                 if (info == null)
 116                         {
 117                                 Console.WriteLine("Unable to determine the workspace.");
 118                                 Console.WriteLine("  Path: " + path);
 119                                 Console.WriteLine();
 120                                 Console.WriteLine("Hints:");
 121                                 Console.WriteLine("  Try adding /workspace:<name>");
 122                                 Console.WriteLine("  Review command options prefixed with '/'. Invalid options are mistaken for paths.");
 123                                 Environment.Exit(1);
 124                         }
 125
 126                 return vcs.GetWorkspace(info);
 127         }
 128
 129         public VersionSpec Version
 130         {
 131                 get {
 132                         if (!String.IsNullOrEmpty(Options.Version))
 133                                 return VersionSpec.ParseSingleSpec(Options.Version, driver.Username);
 134                         return VersionSpec.Latest;
 135                 }
 136         }
 137
 138         public int WindowWidth
 139         {
 140                 get {
 141                         // if output piped to a file, we don't want 0!
 142                         // this also throws on weird terminals on msclr
 143                         try
 144                                 {
 145                                         int w = Console.WindowWidth;
 146                                         if (w != 0) return w - 1;
 147                                 }
 148                         catch (IOException) {}
 149                         return 143;
 150                 }
 151         }
 152
 153         public string Computer
 154         {
 155                 get {
 156                         if (String.IsNullOrEmpty(Options.Computer)) return Environment.MachineName;
 157                         if (Options.Computer == "*") return null;
 158                         return Options.Computer;
 159                 }
 160         }
 161
 162         public string Owner
 163         {
 164                 get {
 165                         if (String.IsNullOrEmpty(Options.Owner)) return driver.Username;
 166                         if (Options.Owner == "*") return null;
 167                         return Options.Owner;
 168                 }
 169         }
 170
 171         public VersionControlServer VersionControlServer
 172         {
 173                 get { return vcs; }
 174         }
 175
 176         public List<string> UnVerifiedFullPaths(string[] args)
 177         {
 178                 List<string> paths = new List<string>();
 179                 for (int i = 1; i < args.Length; i++)
 180                         {
 181                                 string fullPath = Path.GetFullPath(args[i]);
 182                                 paths.Add(fullPath);
 183                         }
 184
 185                 return paths;
 186         }
 187
 188         public List<string> VerifiedFullPaths(string[] args)
 189         {
 190                 List<string> paths = UnVerifiedFullPaths(args);
 191                 foreach (string path in paths)
 192                         {
 193                                 if (!File.Exists(path) && !Directory.Exists(path))
 194                                         {
 195                                                 Console.WriteLine("{0}: No such file or directory.", path);
 196                                                 Environment.Exit(-1);
 197                                         }
 198                         }
 199
 200                 return paths;
 201         }
 202
 203         public void DeleteReadOnlyFile(string fullName)
 204         {
 205                 File.SetAttributes(fullName, FileAttributes.Normal);
 206                 File.Delete(fullName);
 207         }
 208
 209         public Regex WildcardToRegex(string exclusionList)
 210         {
 211                 string[] wildcards = exclusionList.Split(',');
 212                 StringBuilder sb = new StringBuilder();
 213
 214                 foreach (string wildcard in wildcards)
 215                         {
 216                                 if (0 != sb.Length) sb.Append("|");
 217                                 string regex = "^" + Regex.Escape(wildcard).Replace("\\*", ".*").Replace("\\?", ".") + "$";
 218                                 sb.Append(regex);
 219                         }
 220
 221                 return new Regex(sb.ToString());
 222         }
 223
 224         public void ConfirmFilesSpecified()
 225         {
 226                 if (Arguments.Length < 2)
 227                         {
 228                                 Console.WriteLine("No files specified.");
 229                                 Environment.Exit(0);
 230                         }
 231         }
 232
 233         public abstract void Run ();
 234 }
 235