Download data model files before editing data model

Author: Nfactor26

src/Pixel.Automation.Designer.ViewModels/AutomationBuilder/AutomationEditorViewModel.cs

@@ -99,6 +99,7 @@ public override async Task EditDataModelAsync()
             try
             {
                 logger.Information($"Opening code editor for editing data model for project : {this.CurrentProject.Name}");
+                await this.projectManager.DownloadDataModelFilesAsync();
                 var editorFactory = this.EntityManager.GetServiceOfType<ICodeEditorFactory>();
                 var projectFileSystem = this.projectManager.GetProjectFileSystem();
                 using (var editor = editorFactory.CreateMultiCodeEditorScreen())

src/Pixel.Automation.Designer.ViewModels/AutomationBuilder/AutomationProjectManager.cs

@@ -197,6 +197,12 @@ public override async Task DeleteDataFileAsync(string fileToDelete)
             await this.projectDataManager.DeleteDataFileAsync(this.activeProject, this.loadedVersion as ProjectVersion, fileToDelete);
         }
 
+        ///<inheritdoc/>
+        public override async Task DownloadDataModelFilesAsync()
+        {
+            await this.projectDataManager.DownloadDataModelFilesAsync(this.activeProject, this.loadedVersion as ProjectVersion);
+        }
+
         ///<inheritdoc/>
         protected override string GetProjectName()
         {

src/Pixel.Automation.Designer.ViewModels/AutomationBuilder/PrefabEditorViewModel.cs

@@ -61,6 +61,7 @@ public async Task DoLoad(PrefabProject prefabProject, VersionInfo versionToLoad
         /// <returns></returns>
         public override async Task EditDataModelAsync()
         {
+            await this.projectManager.DownloadDataModelFilesAsync();
             var editorFactory = this.EntityManager.GetServiceOfType<ICodeEditorFactory>();
             var prefabFileSystem = this.projectManager.GetProjectFileSystem();
             using (var editor = editorFactory.CreateMultiCodeEditorScreen())

src/Pixel.Automation.Designer.ViewModels/AutomationBuilder/PrefabProjectManager.cs

@@ -222,6 +222,12 @@ public override async Task DeleteDataFileAsync(string fileToDelete)
             await this.prefabDataManager.DeleteDataFileAsync(this.prefabProject, this.loadedVersion, fileToDelete);
         }
 
+        ///<inheritdoc/>
+        public override async Task DownloadDataModelFilesAsync()
+        {
+            await this.prefabDataManager.DownloadDataModelFilesAsync(this.prefabProject, this.loadedVersion);
+        }
+
         /// <inheritdoc/>       
         protected override string GetProjectName()
         {

src/Pixel.Automation.Designer.ViewModels/AutomationBuilder/ProjectManager.cs

@@ -73,6 +73,9 @@ public ProjectManager(ISerializer serializer, IEntityManager entityManager, IFil
         ///<inheritdoc/>
         public abstract Task DeleteDataFileAsync(string fileToDelete);
 
+        ///<inheritdoc/>
+        public abstract Task DownloadDataModelFilesAsync();
+
         ///<inheritdoc/>
         public abstract Task Save();
 

src/Pixel.Automation.Editor.Core/Interfaces/IProjectManager.cs

@@ -34,6 +34,12 @@ public interface IProjectManager
         /// <returns></returns>
         Task DeleteDataFileAsync(string fileToDelete);
 
+        /// <summary>
+        /// Download the data model files (*.cs) belonging the the loaded version of project
+        /// </summary>
+        /// <returns></returns>
+        Task DownloadDataModelFilesAsync();
+
         /// <summary>
         /// Deserialize the contents of a file to given type while ensuring that any references to data model assembly in file are replaced with current session's
         /// data model assembly. Use this to load a process file, prefab file or a test file.

src/Pixel.Persistence.Services.Client/DataManagers/PrefabDataManager.cs

@@ -134,6 +134,33 @@ bool HasMostRecentDataAlreadyAvailable()
         }
     }
 
+    /// <inheritdoc/> 
+    public async Task DownloadDataModelFilesAsync(PrefabProject prefabProject, PrefabVersion prefabVersion)
+    {
+        Guard.Argument(prefabProject, nameof(prefabProject)).NotNull();
+        Guard.Argument(prefabVersion, nameof(prefabVersion)).NotNull();
+
+        if (IsOnlineMode)
+        {
+            var dataModelsDirectory = Path.Combine(this.applicationFileSystem.GetPrefabProjectWorkingDirectory(prefabProject, prefabVersion), Constants.DataModelDirectory);
+            if (Directory.Exists(dataModelsDirectory))
+            {
+                Directory.Delete(dataModelsDirectory, true);
+            }
+            Directory.CreateDirectory(dataModelsDirectory);
+            var zippedContent = await this.filesClient.DownloadProjectDataFilesOfType(prefabProject.PrefabId, prefabVersion.ToString(), "cs");
+            if (zippedContent.Length > 0)
+            {
+                string versionDirectory = this.applicationFileSystem.GetPrefabProjectWorkingDirectory(prefabProject, prefabVersion);
+                using (var memoryStream = new MemoryStream(zippedContent, false))
+                {
+                    var zipArchive = new ZipArchive(memoryStream, ZipArchiveMode.Read);
+                    zipArchive.ExtractToDirectory(versionDirectory, true);
+                }
+            }
+        }
+    }
+
     ///<inheritdoc/>
     public async Task DownloadPrefabsAsync()
     {

src/Pixel.Persistence.Services.Client/DataManagers/ProjectDataManager.cs

@@ -104,6 +104,33 @@ public async Task DownloadProjectDataFilesAsync(AutomationProject automationProj
         }
     }
 
+    /// <inheritdoc/> 
+    public async Task DownloadDataModelFilesAsync(AutomationProject automationProject, ProjectVersion projectVersion)
+    {
+        Guard.Argument(automationProject, nameof(automationProject)).NotNull();
+        Guard.Argument(projectVersion, nameof(projectVersion)).NotNull();
+       
+        if (IsOnlineMode)
+        {
+            var dataModelsDirectory = Path.Combine(this.applicationFileSystem.GetAutomationProjectWorkingDirectory(automationProject, projectVersion), Constants.DataModelDirectory);
+            if (Directory.Exists(dataModelsDirectory))
+            {
+                Directory.Delete(dataModelsDirectory, true);
+            }
+            Directory.CreateDirectory(dataModelsDirectory);
+            var zippedContent = await this.filesClient.DownloadProjectDataFilesOfType(automationProject.ProjectId, projectVersion.ToString(), "cs");
+            if (zippedContent.Length > 0)
+            {
+                string versionDirectory = this.applicationFileSystem.GetAutomationProjectWorkingDirectory(automationProject, projectVersion);
+                using (var memoryStream = new MemoryStream(zippedContent, false))
+                {
+                    var zipArchive = new ZipArchive(memoryStream, ZipArchiveMode.Read);
+                    zipArchive.ExtractToDirectory(versionDirectory, true);
+                }
+            }
+        }
+    }
+
     /// <summary>
     /// Download all the files having specific tags belonging to the version of the AutomationProject being managed
     /// </summary>
@@ -141,7 +168,7 @@ async Task DownloadFilesWithTagsAsync(AutomationProject automationProject, Proje
             }
         }
     }
-
+   
     /// <inheritdoc/>  
     public IEnumerable<AutomationProject> GetAllProjects()
     {

src/Pixel.Persistence.Services.Client/Interfaces/IPrefabDataManager.cs

@@ -44,6 +44,14 @@ public interface IPrefabDataManager
     /// <returns></returns>
     Task DownloadPrefabDataAsync(PrefabProject prefabProject, PrefabVersion prefabVersion);
 
+    /// <summary>
+    /// Download dta model files (*.cs) for a given version of prefab
+    /// </summary>
+    /// <param name="prefabProject"></param>
+    /// <param name="prefabVersion"></param>
+    /// <returns></returns>
+    Task DownloadDataModelFilesAsync(PrefabProject prefabProject, PrefabVersion prefabVersion);
+
     /// <summary>
     /// Add a new prefab
     /// </summary>

src/Pixel.Persistence.Services.Client/Interfaces/IProjectDataManager.cs

@@ -66,11 +66,19 @@ public interface IProjectDataManager
 
 
     /// <summary>
-    /// Download all the data files belong to the version of AutomationProject being managed
+    /// Download all the data files belonging to the version of AutomationProject being managed
     /// </summary>
     /// <returns></returns>
     Task DownloadProjectDataFilesAsync(AutomationProject automationProject, ProjectVersion projectVersion);
 
+    /// <summary>
+    /// Download data model files (*.cs) belonging to the version of AutomationProject being managed
+    /// </summary>
+    /// <param name="automationProject"></param>
+    /// <param name="projectVersion"></param>
+    /// <returns></returns>
+    Task DownloadDataModelFilesAsync(AutomationProject automationProject, ProjectVersion projectVersion);
+
     /// <summary>
     /// Save the data files belonging to the version of AutomationProject being managed
     /// </summary>