From 23b055416efc76d7e74d73cc06f78ab1f417f407 Mon Sep 17 00:00:00 2001
From: Sunghwan Bang <sunghwan2789@gmail.com>
Date: Tue, 4 Apr 2023 18:50:24 +0900
Subject: [PATCH] Fixed parsing DotNetMajor with preview sdks (#6012)

---
 .../Common/MSBuild/StrawberryShake.targets    | 47 ++++---------------
 1 file changed, 8 insertions(+), 39 deletions(-)

diff --git a/src/StrawberryShake/MetaPackages/Common/MSBuild/StrawberryShake.targets b/src/StrawberryShake/MetaPackages/Common/MSBuild/StrawberryShake.targets
index cb9d642ed00..bb8a04c2c47 100644
--- a/src/StrawberryShake/MetaPackages/Common/MSBuild/StrawberryShake.targets
+++ b/src/StrawberryShake/MetaPackages/Common/MSBuild/StrawberryShake.targets
@@ -30,9 +30,14 @@
     Outputs="$(MSBuildProjectDirectory)\$(IntermediateOutputPath)berry\.build.info"
     Condition="'@(GraphQL)' != ''">
 
-    <DotNetMajor>
-      <Output TaskParameter="Major" PropertyName="DotNetMajor" />
-    </DotNetMajor>
+    <Exec Command="dotnet --version" StandardOutputImportance="Low" ConsoleToMsBuild="true">
+      <Output TaskParameter="ConsoleOutput" PropertyName="DotNetVersion" />
+    </Exec>
+
+    <PropertyGroup>
+      <DotNetMajor Condition="$([MSBuild]::VersionGreaterThanOrEquals($(DotNetVersion), 6))">6</DotNetMajor>
+      <DotNetMajor Condition="$([MSBuild]::VersionGreaterThanOrEquals($(DotNetVersion), 7))">7</DotNetMajor>
+    </PropertyGroup>
 
     <Error
       Text="The Strawberry Shake code generation requires .NET SDK 6 or .NET SDK 7 to work."
@@ -66,40 +71,4 @@
     <Touch Files="$(MSBuildProjectDirectory)\$(IntermediateOutputPath)berry\.build.info" AlwaysCreate="true" />
   </Target>
 
-  <UsingTask
-    TaskName="DotNetMajor"
-    TaskFactory="RoslynCodeTaskFactory"
-    AssemblyFile="$(MSBuildToolsPath)\Microsoft.Build.Tasks.Core.dll" >
-    <ParameterGroup>
-      <Major ParameterType="System.String" Output="true" />
-    </ParameterGroup>
-    <Task>
-      <Using Namespace="System"/>
-      <Using Namespace="System.Diagnostics"/>
-      <Code Type="Fragment" Language="cs">
-      <![CDATA[
-      var dotnetVersion = new ProcessStartInfo("dotnet", "--version")
-      {
-          UseShellExecute = false,
-          CreateNoWindow = true,
-          RedirectStandardOutput = true
-      };
-
-      var process = Process.Start(dotnetVersion);
-
-      if (process is not null)
-      {
-          var output = process.StandardOutput.ReadToEnd();
-          process.WaitForExit();
-
-          if (process.ExitCode == 0 && Version.TryParse(output.Trim(), out var version))
-          {
-              Major = version.Major.ToString();
-          }
-      }
-      ]]>
-      </Code>
-    </Task>
-  </UsingTask>
-
 </Project>