From 764a0311876973d19a7b9185fa67cc9ca87b5059 Mon Sep 17 00:00:00 2001
From: Josh Brown <jb1@microsoft.com>
Date: Wed, 20 Dec 2023 10:15:30 -0800
Subject: [PATCH] make_stubs_nuget internal

---
 csharp/ql/src/Stubs/make_stubs_nuget.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/csharp/ql/src/Stubs/make_stubs_nuget.py b/csharp/ql/src/Stubs/make_stubs_nuget.py
index 28126b0cf7b2..5cf8e1a48a85 100644
--- a/csharp/ql/src/Stubs/make_stubs_nuget.py
+++ b/csharp/ql/src/Stubs/make_stubs_nuget.py
@@ -15,6 +15,14 @@ def write_csproj_prefix(ioWrapper):
         '    <AppendTargetFrameworkToOutputPath>false</AppendTargetFrameworkToOutputPath>\n')
     ioWrapper.write('  </PropertyGroup>\n\n')
 
+def write_nuget_config(ioWrapper):
+    ioWrapper.write(b"""<?xml version="1.0" encoding="utf-8"?>
+    <configuration>
+        <packageSources>
+          <clear />
+          <add key="IDDP" value="https://identitydivision.pkgs.visualstudio.com/_packaging/IDDP/nuget/v3/index.json" />
+        </packageSources>
+    </configuration>""")
 
 print('Script to generate stub file from a nuget package')
 print(' Usage: python3 ' + sys.argv[0] +
@@ -70,6 +78,10 @@ def run_cmd(cmd, msg="Failed to run command"):
                  projectNameIn, '--output', projectDirIn])
 helpers.remove_files(projectDirIn, '.cs')
 
+nugetConfigFile = open(os.path.join(projectDirIn, 'nuget.config'), 'wb')
+write_nuget_config(nugetConfigFile)
+nugetConfigFile.close()
+
 print("\n* Adding reference to package: " + nuget)
 cmd = ['dotnet', 'add', projectDirIn, 'package', nuget]
 if (version != "latest"):
@@ -77,7 +89,7 @@ def run_cmd(cmd, msg="Failed to run command"):
     cmd.append(version)
 run_cmd(cmd)
 
-sdk_version = '7.0.102'
+sdk_version = '7.0.403'
 print("\n* Creating new global.json file and setting SDK to " + sdk_version)
 run_cmd(['dotnet', 'new', 'globaljson', '--force', '--sdk-version', sdk_version, '--output', workDir])