tools/opentf/LsFilesCommand.cs

   1 //
   2 // LsFilesCommand.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.Security.Cryptography;
  31 using System.Collections.Generic;
  32 using System.IO;
  33 using System.Text;
  34 using Microsoft.TeamFoundation.Client;
  35 using Microsoft.TeamFoundation.VersionControl.Common;
  36 using Microsoft.TeamFoundation.VersionControl.Client;
  37 using Mono.GetOptions;
  38
  39 [Command("ls-files", "Shows known, deleted, modified, old, or unknown files under the given path.",
  40                                  "<path>")]
  41 class LsFilesCommand : Command
  42 {
  43         [Option("Get all files, not just those out of date", "", "all")]
  44                 public bool OptionAll = false;
  45
  46         [Option("Look for deleted files.", "D", "deleted")]
  47                 public bool OptionDeleted = false;
  48
  49         [Option("Look for modified files.", "", "modified")]
  50                 public bool OptionModified = false;
  51
  52         [Option("Show out of date files (newer version on server)", "", "old")]
  53                 public bool OptionOld = false;
  54
  55         [Option("Show other/unknown files", "", "others")]
  56                 public bool OptionOthers = false;
  57
  58         [Option("Show writable files", "", "writable")]
  59                 public bool OptionWritable;
  60
  61         public LsFilesCommand(Driver driver, string[] args): base(driver, args)
  62         {
  63         }
  64
  65         public void ShowModifiedFiles(string itemPath, SortedList<string, string> itemList)
  66         {
  67                 DirectoryInfo dir = new DirectoryInfo(itemPath);
  68                 MD5CryptoServiceProvider md5 = new MD5CryptoServiceProvider();
  69                 foreach (FileInfo file in dir.GetFiles("*", SearchOption.AllDirectories))
  70                         {
  71                                 if (!itemList.ContainsKey(file.FullName)) continue;
  72
  73                                 using (FileStream fileStream = new FileStream(file.FullName, FileMode.Open, FileAccess.Read))
  74                                         {
  75                                                 md5.ComputeHash(fileStream);
  76                                                 string hash1 = Convert.ToBase64String(md5.Hash);
  77
  78                                                 string hash2 = itemList[file.FullName];
  79                                                 if (hash1 != hash2)
  80                                                         Driver.WriteLine(file.FullName);
  81                                         }
  82                         }
  83         }
  84
  85         public void ShowOtherFiles(string itemPath, SortedList<string, string> itemList)
  86         {
  87                 DirectoryInfo dir = new DirectoryInfo(itemPath);
  88                 foreach (FileInfo file in dir.GetFiles("*", SearchOption.AllDirectories))
  89                         {
  90                                 if (!itemList.ContainsKey(file.FullName) && !IsExcludedFile(file.FullName))
  91                                         {
  92                                                 Driver.WriteLine(file.FullName);
  93                                         }
  94                         }
  95         }
  96
  97         public void ShowWritableFiles(string itemPath)
  98         {
  99                 DirectoryInfo dir = new DirectoryInfo(itemPath);
 100                 foreach (FileInfo file in dir.GetFiles("*", SearchOption.AllDirectories))
 101                         {
 102                                 if (IsExcludedFile(file.FullName)) continue;
 103
 104                                 bool isReadOnly = (FileAttributes.ReadOnly == (File.GetAttributes(file.FullName) & FileAttributes.ReadOnly));
 105                                 if (!isReadOnly) Driver.WriteLine(file.FullName);
 106                         }
 107         }
 108
 109         public void ShowDeletedFiles(string itemPath, SortedList<string, string> itemList)
 110         {
 111                 SortedList<string, bool> dirList = new SortedList<string, bool>();
 112                 DirectoryInfo dir = new DirectoryInfo(itemPath);
 113
 114                 foreach (FileInfo file in dir.GetFiles("*", SearchOption.AllDirectories))
 115                         {
 116                                 dirList.Add(file.FullName, true);
 117                         }
 118
 119                 foreach (DirectoryInfo di in dir.GetDirectories("*", SearchOption.AllDirectories))
 120                         {
 121                                 dirList.Add(di.FullName, true);
 122                         }
 123
 124                 foreach (string key in itemList.Keys)
 125                         {
 126                                 if (!dirList.ContainsKey(key))
 127                                         {
 128                                                 Driver.WriteLine(key);
 129                                         }
 130                         }
 131         }
 132
 133         public void ShowOldFiles(Workspace workspace, ItemSpec itemSpec)
 134         {
 135                 List<ItemSpec> itemSpecs = new List<ItemSpec>();
 136                 itemSpecs.Add(itemSpec);
 137
 138                 ExtendedItem[][] items = workspace.GetExtendedItems(itemSpecs.ToArray(),
 139                                                                                                                                                                                                                                 DeletedState.NonDeleted, ItemType.Any);
 140
 141                 StringBuilder sb = new StringBuilder();
 142                 foreach (ExtendedItem[] itemArray in items)
 143                         {
 144                                 foreach (ExtendedItem item in itemArray)
 145                                         {
 146                                                 if (item.IsLatest || item.ItemType == ItemType.Folder) continue;
 147
 148                                                 // new files will have a null local item
 149                                                 if (String.IsNullOrEmpty(item.LocalItem))
 150                                                         {
 151                                                                 if (OptionAll)
 152                                                                         sb.Append(String.Format("{0}\n", workspace.GetLocalItemForServerItem(item.SourceServerItem)));
 153                                                         }
 154                                                 else
 155                                                         Driver.WriteLine(item.LocalItem);
 156                                         }
 157                         }
 158
 159                 if (OptionAll)
 160                         {
 161                                 Driver.WriteLine();
 162                                 Driver.WriteLine("New files on the server:");
 163                                 Driver.WriteLine();
 164                                 Driver.WriteLine(sb.ToString());
 165                         }
 166         }
 167
 168         public override void Run()
 169         {
 170                 string path = Environment.CurrentDirectory;
 171                 if (Arguments.Length > 0)
 172                                 path = Path.GetFullPath(Arguments[0]);
 173
 174                 if (File.Exists(path))
 175                         {
 176                                 // would need to fixup dir.GetDirectories calls if we wanted to support filenames
 177                                 Console.WriteLine("Error: This command only takes paths as arguments, not file names.");
 178                                 Environment.Exit((int)ExitCode.Failure);
 179                         }
 180
 181                 char[] charsToTrim = { Path.DirectorySeparatorChar, Path.AltDirectorySeparatorChar};
 182                 string itemPath = path.TrimEnd(charsToTrim);
 183
 184                 if (OptionWritable)
 185                         {
 186                                 ShowWritableFiles(itemPath);
 187                                 Environment.Exit((int)ExitCode.Success);
 188                         }
 189
 190                 Workspace workspace = GetWorkspaceFromCache();
 191                 workspace.RefreshMappings();
 192                 string serverPath = workspace.GetServerItemForLocalItem(itemPath);
 193
 194                 // process command options
 195                 ItemSpec itemSpec = new ItemSpec(itemPath, RecursionType.Full);
 196
 197                 if (OptionOld)
 198                         {
 199                                 ShowOldFiles(workspace, itemSpec);
 200                                 Environment.Exit((int)ExitCode.Success);
 201                         }
 202
 203                 // pull item list based on WorkspaceVersion. otherwise might get
 204                 // new items on server that haven't been pulled yet in the list returned
 205                 WorkspaceVersionSpec version = new WorkspaceVersionSpec(workspace);
 206                 ItemSet itemSet = VersionControlServer.GetItems(itemSpec, version, DeletedState.NonDeleted, ItemType.Any, true);
 207
 208                 Item[] items = itemSet.Items;
 209                 SortedList<string, string > itemList = new SortedList<string, string >(PathComparer);
 210
 211                 foreach (Item item in items)
 212                         {
 213                                 if (item.ServerItem.Length == serverPath.Length) continue;
 214                                 string serverItem = item.ServerItem.Remove(0, serverPath.Length+1);
 215
 216                                 // server item paths are separated with '/', but on windows the file list below has '\' separated paths
 217                                 if (Path.DirectorySeparatorChar != '/')
 218                                         serverItem = serverItem.Replace('/', Path.DirectorySeparatorChar);
 219
 220                                 string fname = Path.Combine(itemPath, serverItem);
 221                                 string hash = "";
 222
 223                                 if (item.ItemType == ItemType.File && item.HashValue != null)
 224                                         hash = Convert.ToBase64String(item.HashValue);
 225
 226                                 itemList.Add(fname, hash);
 227                         }
 228
 229                 if (OptionOthers) ShowOtherFiles(itemPath, itemList);
 230                 else if (OptionDeleted) ShowDeletedFiles(itemPath, itemList);
 231                 else if (OptionModified) ShowModifiedFiles(itemPath, itemList);
 232                 else
 233                         {
 234                                 foreach (string key in itemList.Keys)
 235                                         {
 236                                                 Driver.WriteLine(key);
 237                                         }
 238                         }
 239         }
 240 }