Get Go SDK version from pulumi-language-dotnet/go.mod

Author: julienp

build/Program.fs

@@ -1,6 +1,7 @@
 open System
 open System.IO
 open System.Linq
+open System.Text.RegularExpressions
 open Fake.IO
 open Fake.Core
 open Publish
@@ -27,6 +28,24 @@ let pulumiFSharp = Path.Combine(sdk, "Pulumi.FSharp")
 let integrationTests = Path.Combine(repositoryRoot, "integration_tests")
 let pulumiLanguageDotnet = Path.Combine(repositoryRoot, "pulumi-language-dotnet")
 
+// Find the version of the Pulumi Go SDK that we are using for the language plugin.
+let findGoSDKVersion =
+    let goMod = Path.Combine(pulumiLanguageDotnet, "go.mod")
+    try
+        let lines = File.ReadAllLines(goMod)
+        let patternRegex = new Regex("^\\s*github.com/pulumi/pulumi/sdk", RegexOptions.IgnoreCase)
+        match Array.tryFind (patternRegex.IsMatch) lines with
+        | Some(matchingLine) ->
+            let version = matchingLine.Split(' ')[1]
+            let version = version.TrimStart('v')
+            Some(version)
+        | None ->
+            None    
+    with
+    | ex ->
+        printfn "Error while trying to find the GO SDK version: %s" ex.Message
+        None
+
 /// Runs `dotnet clean` command against the solution file,
 /// then proceeds to delete the `bin` and `obj` directory of each project in the solution
 let cleanSdk() =
@@ -79,9 +98,12 @@ let listIntegrationTests() =
 let buildSdk() =
     cleanSdk()
     restoreSdk()
-    printfn "Building Pulumi SDK"
-    if Shell.Exec("dotnet", "build --configuration Release", sdk) <> 0
-    then failwith "build failed"
+    match findGoSDKVersion with
+    | None -> failwith "Could not find the Pulumi SDK version in go.mod"
+    | Some(version) ->
+        printfn "Building Pulumi SDK"
+        if Shell.Exec("dotnet", "build --configuration Release -p:PulumiSDKVersion=" + version, sdk) <> 0
+        then failwith "build failed"
 
 /// Publishes packages for Pulumi, Pulumi.Automation and Pulumi.FSharp to nuget.
 /// Requires NUGET_PUBLISH_KEY and PULUMI_VERSION environment variables.
@@ -148,10 +170,13 @@ let testPulumiSdk coverage =
 let testPulumiAutomationSdk coverage =
     cleanSdk()
     restoreSdk()
-    printfn "Testing Pulumi Automation SDK"
-    let coverageArgs = if coverage then $" -p:CollectCoverage=true -p:CoverletOutputFormat=cobertura -p:CoverletOutput={coverageDir}/coverage.pulumi.automation.xml" else ""
-    if Shell.Exec("dotnet", "test --configuration Release" + coverageArgs, pulumiAutomationSdkTests) <> 0
-    then failwith "automation tests failed"
+    match findGoSDKVersion with
+    | None -> failwith "Could not find the Pulumi SDK version in go.mod"
+    | Some(version) ->
+        printfn "Testing Pulumi Automation SDK"
+        let coverageArgs = if coverage then $" -p:CollectCoverage=true -p:CoverletOutputFormat=cobertura -p:CoverletOutput={coverageDir}/coverage.pulumi.automation.xml" else ""
+        if Shell.Exec("dotnet", "test --configuration Release -p:PulumiSDKVersion=" + version + " --filter InstallDefaultRoot" + coverageArgs, pulumiAutomationSdkTests) <> 0
+        then failwith "automation tests failed"
 
 let syncProtoFiles() = GitSync.repository {
     remoteRepository = "https://github.com/pulumi/pulumi.git"

sdk/Pulumi.Automation/Commands/LocalPulumiCommand.cs

@@ -110,13 +110,11 @@ private LocalPulumiCommand(string command, SemVersion? version)
         /// <param name="cancellationToken">A cancellation token.</param>
         public static async Task<LocalPulumiCommand> Install(LocalPulumiCommandOptions? options = null, CancellationToken cancellationToken = default)
         {
-            var _assemblyVersion = Assembly.GetExecutingAssembly().GetName().Version;
-            if (_assemblyVersion == null)
-            {
-                throw new Exception("Failed to get assembly version.");
-            }
-            var assemblyVersion = new SemVersion(_assemblyVersion.Major, _assemblyVersion.Minor, _assemblyVersion.Build);
-            var version = options?.Version ?? assemblyVersion;
+            var sdkVersion = Assembly.GetExecutingAssembly()
+                .GetCustomAttributes(typeof(PulumiSDKVersion), false)
+                .Cast<PulumiSDKVersion>().First().Version;
+
+            var version = options?.Version ?? sdkVersion;
             var home = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
             var optionsWithDefaults = new LocalPulumiCommandOptions
             {

sdk/Pulumi.Automation/Pulumi.Automation.csproj

@@ -58,6 +58,12 @@
     </None>
   </ItemGroup>
 
+  <ItemGroup>
+    <AssemblyAttribute Include="Pulumi.Automation.PulumiSDKVersion">
+      <_Parameter1>$(PulumiSDKVersion)</_Parameter1>
+    </AssemblyAttribute>
+  </ItemGroup>
+
   <ItemGroup>
     <ProjectReference Include="..\Pulumi\Pulumi.csproj" />
   </ItemGroup>

sdk/Pulumi.Automation/PulumiSDKVersion.cs

@@ -0,0 +1,23 @@
+using System;
+using Semver;
+
+namespace Pulumi.Automation
+{
+    [AttributeUsage(AttributeTargets.Assembly)]
+    public class PulumiSDKVersion : Attribute
+    {
+        public SemVersion Version { get; private set; }
+
+        public PulumiSDKVersion() : this("") { }
+
+        public PulumiSDKVersion(string value)
+        {
+            value = value.Trim();
+            if (value.StartsWith("v"))
+            {
+                value = value.Substring(1);
+            }
+            Version = SemVersion.Parse(value, SemVersionStyles.Strict);
+        }
+    }
+}