Updated Attachement downloading function to use the WorkItemServer. (#42

)
nkdAgility · Aug 12, 2017 · ec2e194 · ec2e194
1 parent a579ad7
commit ec2e194
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/VstsSyncMigrator.Core/Execution/MigrationContext/AttachementExportMigrationContext.cs b/src/VstsSyncMigrator.Core/Execution/MigrationContext/AttachementExportMigrationContext.cs
@@ -1,4 +1,5 @@
 using Microsoft.TeamFoundation.WorkItemTracking.Client;
+using Microsoft.TeamFoundation.WorkItemTracking.Proxy;
 using System;
 using System.Collections.Generic;
 using System.Diagnostics;
@@ -40,9 +41,9 @@ internal override void InternalExecute()
             tfsqc.Query = string.Format(@"SELECT [System.Id], [System.Tags] FROM WorkItems WHERE [System.TeamProject] = @TeamProject {0} ORDER BY [System.ChangedDate] desc", _config.QueryBit);
             WorkItemCollection sourceWIS = tfsqc.Execute();
 
-            WebClient webClient = new WebClient();
-            webClient.Credentials = CredentialCache.DefaultNetworkCredentials;
             int current = sourceWIS.Count;
+            var workItemServer = me.Source.Collection.GetService<WorkItemServer>();
+
             foreach (WorkItem wi in sourceWIS)
             {
                 Trace.Write(string.Format("Attachement Export: {0} of {1} - {2}", current, sourceWIS.Count, wi.Id));
@@ -58,13 +59,14 @@ internal override void InternalExecute()
                         Trace.Write("...downloading");
                         try
                         {
-                            webClient.DownloadFile(wia.Uri, fpath);
+                            var fileLocation = workItemServer.DownloadFile(wia.Id);
+                            File.Copy(fileLocation, fpath, true);
                             Trace.Write("...done");
                         }
                         catch (Exception ex)
                         {
                             Telemetry.Current.TrackException(ex);
-                            Trace.Write("...failed");
+                            Trace.Write($"\r\nException downloading attachements {ex.Message}");
                         }
 
                     }