// ManageDrivers.cs

using System;
using System.Collections.Generic;
using System.Text;
using Microsoft.ComponentStudio.ComponentPlatformInterface;

namespace MyDriverNameSpace
{
	public class MyDriverClass
	{
		static void Main()
		{
			MyDriverClass mDC = new MyDriverClass();
			mDC.AddDriverPath();
			Console.WriteLine(
				"The driver is added to the answer file.");
			mDC.RemoveDriverPath();
			Console.WriteLine(
				"The driver has been removed from the answer file.");
	}

		private const string DmiName =
		"Microsoft-Windows-PnpCustomizationsNonWinPE";

		private const string ansFile =
		@"C:\myWorkingFolder\My Unattend.xml";

		private const string wimPath = 
		@"C:\myWorkingFolder\install.wim";

		public void AddDriverPath()
		{
			DriverPath nvidiaPath;
			AnswerFile unattend;

			// Create a CPI instance:
			Cpi n = Cpi.Instance;

			// Check if the catalog file is created, if not create one:
			WimInfo wInfo = n.OpenWim(wimPath);
			if (!wInfo.Images[0].ValidateCatalog().Succeeded)
				wInfo.Images[0].CreateCatalog();

			// Open the offline image:
			OfflineImage image = n.OpenOfflineImage(wInfo.Images[0]);

			// Create the answer file:
			unattend = n.CreateAnswerFile(image);
			// Add a driver action to add nvidia driver:
			DriverPathCollection driverCollection = 
			unattend.GetDriverPaths(ConfigurationPass.auditSystem);
			nvidiaPath = driverCollection.Create(
			@"\\server1\share1\drivers\nvidia\GForce4",
			CreateOptions.ReturnExisting);

			// Optional: add credential to access the driver:
			nvidiaPath.DynamicKey = "nvidia1";
			nvidiaPath.Domain = "domain1";
			nvidiaPath.UserName = "user1";
			nvidiaPath.Password = "password1";

			// Save the file:
			unattend.SaveAs(ansFile);
	}

		public void RemoveDriverPath()
		{
			DriverPath nvidiaPath;
			DriverPath ati;
			AnswerFile unattend;
			string wimPath = @"C:\myWorkingFolder\install.wim";

			// Create a CPI instance:
			Cpi n = Cpi.Instance;

			// Check if the catalog file is created, if not create one:
			WimInfo wInfo = n.OpenWim(wimPath);
			if (!wInfo.Images[0].ValidateCatalog().Succeeded)
				wInfo.Images[0].CreateCatalog();

			// Open the offline image:
			OfflineImage image = n.OpenOfflineImage(wInfo.Images[0]);

			// Create the answer file:
			unattend = n.CreateAnswerFile(image);

			// Add a driver action to add nvidia driver.
			DriverPathCollection driverPaths =
		 unattend.GetDriverPaths(ConfigurationPass.auditSystem);
			nvidiaPath = driverPaths.Create(
			@"\\server1\share1\drivers\nvidia\GForce4",
			CreateOptions.ReturnExisting);

			// Optional: add credential to access the driver:
			nvidiaPath.DynamicKey = "nvidia1";
			nvidiaPath.Domain = "domain1";
			nvidiaPath.UserName = "user1";
			nvidiaPath.Password = "password1";
			ati = driverPaths.Create(
			@"\\server1\share1\drivers\ati\radeon",
			CreateOptions.ReturnExisting);
			ati.DynamicKey = "ati1";

			// Try to remove the nvidia driver via answer file instance 
			// and verify that the driver path is updated accordingly:
			Identity dmiId = image.FindComponentIdentity(DmiName);
			AnswerFileSettingOverride so =
			unattend.SettingOverrides.FindAny(dmiId,
			nvidiaPath.Path, ConfigurationPass.auditSystem);
			so.Remove();
	}
}
}