From 5d8d8fbdb59e57283e5b2763eb9555830840d8a4 Mon Sep 17 00:00:00 2001
From: Yunus Emre Kalkan <emre.kalkan@volosoft.com>
Date: Mon, 9 Mar 2020 13:07:18 +0300
Subject: [PATCH] Cli: changed nuget version info resource

---
 .../ProjectBuilding/AbpIoSourceCodeStore.cs   | 50 +++++++++++++++++--
 .../Steps/ProjectReferenceReplaceStep.cs      |  2 +-
 .../Abp/Cli/ProjectBuilding/TemplateFile.cs   |  5 +-
 3 files changed, 50 insertions(+), 7 deletions(-)

diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/AbpIoSourceCodeStore.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/AbpIoSourceCodeStore.cs
index 03504251f5..263b675a2a 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/AbpIoSourceCodeStore.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/AbpIoSourceCodeStore.cs
@@ -54,19 +54,21 @@ namespace Volo.Abp.Cli.ProjectBuilding
                 version = latestVersion;
             }
 
+            var nugetVersion = (await GetTemplateNugetVersionAsync(name, type, version)) ?? version;
+
             DirectoryHelper.CreateIfNotExists(CliPaths.TemplateCache);
 
             if (!string.IsNullOrWhiteSpace(templateSource) && !IsNetworkSource(templateSource))
             {
                 Logger.LogInformation("Using local " + type + ": " + name + ", version: " + version);
-                return new TemplateFile(File.ReadAllBytes(Path.Combine(templateSource, name + "-" + version + ".zip")), version, latestVersion);
+                return new TemplateFile(File.ReadAllBytes(Path.Combine(templateSource, name + "-" + version + ".zip")), version, latestVersion, nugetVersion);
             }
 
             var localCacheFile = Path.Combine(CliPaths.TemplateCache, name + "-" + version + ".zip");
             if (Options.CacheTemplates && File.Exists(localCacheFile) && templateSource.IsNullOrWhiteSpace())
             {
                 Logger.LogInformation("Using cached " + type + ": " + name + ", version: " + version);
-                return new TemplateFile(File.ReadAllBytes(localCacheFile), version, latestVersion);
+                return new TemplateFile(File.ReadAllBytes(localCacheFile), version, latestVersion, nugetVersion);
             }
 
             Logger.LogInformation("Downloading " + type + ": " + name + ", version: " + version);
@@ -86,7 +88,7 @@ namespace Volo.Abp.Cli.ProjectBuilding
                 File.WriteAllBytes(localCacheFile, fileContent);
             }
 
-            return new TemplateFile(fileContent, version, latestVersion);
+            return new TemplateFile(fileContent, version, latestVersion, nugetVersion);
 
         }
 
@@ -110,7 +112,38 @@ namespace Volo.Abp.Cli.ProjectBuilding
 
                 var result = await response.Content.ReadAsStringAsync();
 
-                return JsonSerializer.Deserialize<GetLatestSourceCodeVersionResultDto>(result).Version;
+                return JsonSerializer.Deserialize<GetVersionResultDto>(result).Version;
+            }
+        }
+
+        private async Task<string> GetTemplateNugetVersionAsync(string name, string type, string version)
+        {
+            try
+            {
+                using (var client = new CliHttpClient(TimeSpan.FromMinutes(10)))
+                {
+                    var response = await client.PostAsync(
+                        $"{CliUrls.WwwAbpIo}api/download/{type}/get-nuget-version/",
+                        new StringContent(
+                            JsonSerializer.Serialize(
+                                new GetTemplateNugetVersionDto { Name = name, Version = version }
+                            ),
+                            Encoding.UTF8,
+                            MimeTypes.Application.Json
+                        ),
+                        CancellationTokenProvider.Token
+                    );
+
+                    await RemoteServiceExceptionHandler.EnsureSuccessfulHttpResponseAsync(response);
+
+                    var result = await response.Content.ReadAsStringAsync();
+
+                    return JsonSerializer.Deserialize<GetVersionResultDto>(result).Version;
+                }
+            }
+            catch (Exception)
+            {
+                return null;
             }
         }
 
@@ -162,7 +195,14 @@ namespace Volo.Abp.Cli.ProjectBuilding
             public string Name { get; set; }
         }
 
-        public class GetLatestSourceCodeVersionResultDto
+        public class GetTemplateNugetVersionDto
+        {
+            public string Name { get; set; }
+
+            public string Version { get; set; }
+        }
+
+        public class GetVersionResultDto
         {
             public string Version { get; set; }
         }
diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ProjectReferenceReplaceStep.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ProjectReferenceReplaceStep.cs
index a7fd3f529f..5aaf4b1488 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ProjectReferenceReplaceStep.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ProjectReferenceReplaceStep.cs
@@ -29,7 +29,7 @@ namespace Volo.Abp.Cli.ProjectBuilding.Building.Steps
             }
             else
             {
-                var nugetPackageVersion = context.TemplateFile.Version;
+                var nugetPackageVersion = context.TemplateFile.RepositoryNugetVersion;
 
                 if (IsBranchName(nugetPackageVersion))
                 {
diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/TemplateFile.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/TemplateFile.cs
index c7f9e77cb2..fc514e8a74 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/TemplateFile.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/TemplateFile.cs
@@ -6,13 +6,16 @@
 
         public string LatestVersion { get; }
 
+        public string RepositoryNugetVersion { get; }
+
         public byte[] FileBytes { get; }
 
-        public TemplateFile(byte[] fileBytes, string version, string latestVersion)
+        public TemplateFile(byte[] fileBytes, string version, string latestVersion, string repositoryNugetVersion)
         {
             FileBytes = fileBytes;
             Version = version;
             LatestVersion = latestVersion;
+            RepositoryNugetVersion = repositoryNugetVersion;
         }
     }
 }
\ No newline at end of file