public static async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > Disk(
VirtualMachineDiskConfig diskConfig,
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
MachineConfig vmConfig,
Func <string, Task> reportProgress)
{
if (!File.Exists(diskConfig.Path))
{
await reportProgress($"Create VHD: {diskConfig.Name}").ConfigureAwait(false);
await engine.RunAsync(PsCommandBuilder.Create().Script(
$"New-VHD -Path \"{diskConfig.Path}\" -ParentPath \"{diskConfig.Template}\" -Differencing"),
reportProgress : p => ReportPowershellProgress($"Create VHD {diskConfig.Name}", p, reportProgress)).ConfigureAwait(false);
}
await GetOrCreateInfoAsync(vmInfo,
i => i.HardDrives,
disk => diskConfig.Path.Equals(disk.Path, StringComparison.OrdinalIgnoreCase),
async() =>
{
await reportProgress($"Add VHD: {diskConfig.Name}").ConfigureAwait(false);
return(await engine.GetObjectsAsync <HardDiskDriveInfo>(PsCommandBuilder.Create()
.AddCommand("Add-VMHardDiskDrive")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("Path", diskConfig.Path)).ConfigureAwait(false));
}).ConfigureAwait(false);
return(vmInfo.Recreate());
}
private static Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > InsertConfigDriveDisk(
string configDriveIsoPath,
TypedPsObject <VirtualMachineInfo> vmInfo,
IPowershellEngine engine)
{
return
(from dvdDrive in GetOrCreateInfoAsync(vmInfo,
l => l.DVDDrives,
drive => drive.ControllerLocation == 63 && drive.ControllerNumber == 0,
() => engine.GetObjectsAsync <DvdDriveInfo>(
PsCommandBuilder.Create().AddCommand("Add-VMDvdDrive")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("ControllerNumber", 0)
.AddParameter("ControllerLocation", 63)
.AddParameter("PassThru"))
)
from _ in engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMDvdDrive")
.AddParameter("VMDvdDrive", dvdDrive.PsObject)
.AddParameter("Path", configDriveIsoPath))
from vmInfoRecreated in vmInfo.RecreateOrReload(engine)
select vmInfoRecreated);
}
private static Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > EjectConfigDriveDisk(
string configDriveIsoPath,
TypedPsObject <VirtualMachineInfo> vmInfo,
IPowershellEngine engine)
{
var res = vmInfo.GetList(l => l.DVDDrives,
drive => configDriveIsoPath.Equals(drive.Path, StringComparison.OrdinalIgnoreCase))
.Apply(dvdDriveList =>
{
var array = dvdDriveList.ToArray();
if (array.Length > 1)
{
array.Take(array.Length - 1).Iter(r =>
engine.Run(PsCommandBuilder.Create()
.AddCommand("Remove-VMDvdDrive")
.AddParameter("VMDvdDrive", r.PsObject)));
}
return(vmInfo.GetList(l => l.DVDDrives, drive => configDriveIsoPath.Equals(drive.Path, StringComparison.OrdinalIgnoreCase)));
}).Apply(driveInfos =>
{
return(driveInfos.Map(driveInfo => engine.Run(PsCommandBuilder.Create()
.AddCommand("Set-VMDvdDrive")
.AddParameter("VMDvdDrive", driveInfo.PsObject)
.AddParameter("Path", null)))
.Traverse(x => x).Map(
x => vmInfo.Recreate()));
});
return(res);
}
public async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > GetInfoAsync(string vmName) =>
(await _engine.GetObjectsAsync <VirtualMachineInfo>(PsCommandBuilder.Create()
.AddCommand("get-vm").AddArgument(vmName)).ConfigureAwait(false))
.Apply(
queryResult =>
{
return(queryResult.Match(
Left: l => l,
Right: r =>
{
var vmArray = r.ToArray();
if (vmArray.Length > 1)
{
return Prelude.Left <PowershellFailure, TypedPsObject <VirtualMachineInfo> >(
new PowershellFailure {
Message = $"VM name '{vmName}' is not unique."
});
}
if (vmArray.Length == 0)
{
return Prelude.Left <PowershellFailure, TypedPsObject <VirtualMachineInfo> >(
new PowershellFailure {
Message = $"VM name '{vmName}' not found"
});
}
var vmInfo = vmArray[0];
SetInfo(vmInfo);
return vmInfo;
}
));
});
public static Task <Either <PowershellFailure, Unit> > DetachUndefinedDvdDrives(
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
Seq <VMDriveStorageSettings> plannedStorageSettings,
Func <string, Task> reportProgress)
{
var controllersAndLocations = plannedStorageSettings.Where(x => x.Type == VirtualMachineDriveType.DVD)
.Map(x => new { x.ControllerNumber, x.ControllerLocation })
.GroupBy(x => x.ControllerNumber)
.ToImmutableDictionary(x => x.Key, x => x.Map(y => y.ControllerLocation).ToImmutableArray());
return(FindAndApply(vmInfo,
i => i.DVDDrives,
hd =>
{
//ignore cloud init drive, will be handled later
if (hd.ControllerLocation == 63 && hd.ControllerNumber == 0)
{
return false;
}
var detach = !controllersAndLocations.ContainsKey(hd.ControllerNumber) ||
!controllersAndLocations[hd.ControllerNumber].Contains(hd.ControllerLocation);
return detach;
},
i => engine.RunAsync(PsCommandBuilder.Create().AddCommand("Remove-VMDvdDrive")
.AddParameter("VMDvdDrive", i.PsObject))).Map(x => x.Lefts().HeadOrNone())
.MatchAsync(
Some: l => LeftAsync <PowershellFailure, Unit>(l).ToEither(),
None: () => RightAsync <PowershellFailure, Unit>(Unit.Default)
.ToEither()));
}
public static Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > CreateVirtualMachine(
IPowershellEngine engine,
string vmName,
string storageIdentifier,
string vmPath,
int startupMemory)
{
var memoryStartupBytes = startupMemory * 1024L * 1024;
return(engine.GetObjectsAsync <VirtualMachineInfo>(PsCommandBuilder.Create()
.AddCommand("New-VM")
.AddParameter("Name", storageIdentifier)
.AddParameter("Path", vmPath)
.AddParameter("MemoryStartupBytes", memoryStartupBytes)
.AddParameter("Generation", 2))
.MapAsync(x => x.Head).MapAsync(
result =>
{
engine.RunAsync(PsCommandBuilder.Create().AddCommand("Get-VMNetworkAdapter")
.AddParameter("VM", result.PsObject).AddCommand("Remove-VMNetworkAdapter"));
return result;
})
.BindAsync(info => RenameVirtualMachine(engine, info, vmName)));
}
public async Task <Either <PowershellFailure, Unit> > RunAsync(PsCommandBuilder builder, Func <int, Task> reportProgress = null)
{
using (var ps = CreateShell())
{
builder.Build(ps);
InitializeAsyncProgressReporting(ps, reportProgress);
return(await ps.RunAsync().ConfigureAwait(false));
}
}
public Either <PowershellFailure, Unit> Run(PsCommandBuilder builder, Action <int> reportProgress = null)
{
using (var ps = CreateShell())
{
builder.Build(ps);
InitializeProgressReporting(ps, reportProgress);
return(ps.Run());
}
}
public Either <PowershellFailure, Seq <TypedPsObject <T> > > GetObjects <T>(PsCommandBuilder builder, Action <int> reportProgress = null)
{
using (var ps = CreateShell())
{
builder.Build(ps);
InitializeProgressReporting(ps, reportProgress);
return(ps.GetObjects <T>());
}
}
public static Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > RenameVirtualMachine(
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
string newName)
{
return(engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Rename-VM")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("NewName", newName)
).BindAsync(u => vmInfo.RecreateOrReload(engine)));
}
private static Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > RemoveConfigDriveDisk(
TypedPsObject <VirtualMachineInfo> vmInfo,
IPowershellEngine engine)
{
return(FindAndApply(vmInfo, l => l.DVDDrives,
drive => drive.ControllerLocation == 63 && drive.ControllerNumber == 0,
drive => engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Remove-VMDvdDrive")
.AddParameter("VMDvdDrive", drive.PsObject)))
.Map(list => list.Lefts().HeadOrNone()).MatchAsync(
None: () => vmInfo.RecreateOrReload(engine),
Some: l => LeftAsync <PowershellFailure, TypedPsObject <VirtualMachineInfo> >(l).ToEither()));
}
public async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > GetInfoAsync(Guid id)
{
if (_buffer.TryGetValue(id, out var vmInfo))
{
return(vmInfo);
}
return((await _engine.GetObjectsAsync <VirtualMachineInfo>(
PsCommandBuilder.Create()
.AddCommand("get-vm").AddParameter("Id", id)).ConfigureAwait(false)).Map(newVmInfo =>
{
SetInfo(newVmInfo.Head);
return newVmInfo.Head;
}));
}
public static async Task Definition(
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
MachineConfig vmConfig,
Func <string, Task> reportProgress)
{
if (vmInfo.Value.Generation >= 2)
{
await engine.GetObjectsAsync <VMFirmwareInfo>(PsCommandBuilder.Create()
.AddCommand("Get-VMFirmware")
.AddParameter("VM", vmInfo.PsObject)).MapAsync(async(firmwareInfo) =>
{
if (firmwareInfo.Head.Value.SecureBoot != OnOffState.Off)
{
await reportProgress($"Configure VM Firmware - Secure Boot: {OnOffState.Off}")
.ConfigureAwait(false);
await engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMFirmware")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("EnableSecureBoot", OnOffState.Off)).ConfigureAwait(false);
}
}).ConfigureAwait(false);
}
if (vmInfo.Value.ProcessorCount != vmConfig.VM.Cpu.Count)
{
await reportProgress($"Configure VM Processor: Count: {vmConfig.VM.Cpu.Count}").ConfigureAwait(false);
await engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMProcessor")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("Count", vmConfig.VM.Cpu.Count)).ConfigureAwait(false);
}
var memoryStartupBytes = vmConfig.VM.Memory.Startup * 1024L * 1024;
if (vmInfo.Value.MemoryStartup != memoryStartupBytes)
{
await reportProgress($"Configure VM Memory: Startup: {vmConfig.VM.Memory.Startup} MB").ConfigureAwait(false);
await engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMMemory")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("StartupBytes", memoryStartupBytes)).ConfigureAwait(false);
}
}
private static Task <Either <PowershellFailure, Unit> > InsertConfigDriveDisk(
string configDriveIsoPath,
TypedPsObject <VirtualMachineInfo> vmInfo,
IPowershellEngine engine)
{
return(vmInfo.GetList(l => l.DVDDrives, drive => string.IsNullOrWhiteSpace(drive.Path))
.HeadOrNone().MatchAsync(
None: () => engine.RunAsync(
PsCommandBuilder.Create().AddCommand("Add-VMDvdDrive")
.AddParameter("VM", vmInfo.PsObject).AddParameter("Path", configDriveIsoPath)),
Some: dvdDriveInfo => engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMDvdDrive")
.AddParameter("VMDvdDrive", dvdDriveInfo.PsObject)
.AddParameter("Path", configDriveIsoPath)
)));
}
public static async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > Firmware(TypedPsObject <VirtualMachineInfo> vmInfo,
MachineConfig config, IPowershellEngine engine, Func <string, Task> reportProgress)
{
if (vmInfo.Value.Generation < 2)
{
return(vmInfo);
}
return(await engine.GetObjectsAsync <VMFirmwareInfo>(PsCommandBuilder.Create()
.AddCommand("Get-VMFirmware")
.AddParameter("VM", vmInfo.PsObject))
.BindAsync(firmwareInfoSeq =>
firmwareInfoSeq.HeadOrNone()
.Match <Either <PowershellFailure, TypedPsObject <VMFirmwareInfo> > >(
None: () => new PowershellFailure {
Message = "Failed to get VM Firmware"
},
Some: s => s
))
.BindAsync(async firmwareInfo =>
{
if (firmwareInfo.Value.SecureBoot != OnOffState.Off)
{
await reportProgress($"Configure VM Firmware - Secure Boot: {OnOffState.Off}")
.ConfigureAwait(false);
var res = await engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMFirmware")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("EnableSecureBoot", OnOffState.Off)).MapAsync(
r => new TypedPsObject <VirtualMachineInfo>(vmInfo.PsObject)
).ConfigureAwait(false);
return res;
}
return new TypedPsObject <VirtualMachineInfo>(vmInfo.PsObject);
}
).ConfigureAwait(false));
}
public static async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > Cpu(TypedPsObject <VirtualMachineInfo> vmInfo,
Option <VirtualMachineCpuConfig> cpuConfig, IPowershellEngine engine, Func <string, Task> reportProgress)
{
return(await cpuConfig.MatchAsync(
None : () => vmInfo,
Some : async config =>
{
if (vmInfo.Value.ProcessorCount != config.Count)
{
await reportProgress($"Configure VM Processor: Count: {config.Count}").ConfigureAwait(false);
await engine.RunAsync(PsCommandBuilder.Create()
.AddCommand("Set-VMProcessor")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("Count", config.Count)).ConfigureAwait(false);
return await vmInfo.RecreateOrReload(engine).ConfigureAwait(false);
}
return vmInfo;
}).ConfigureAwait(false));
}
public static async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > NetworkAdapter(
VirtualMachineNetworkAdapterConfig networkAdapterConfig,
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
MachineConfig machineConfig,
Func <string, Task> reportProgress)
{
var optionalAdapter = await GetOrCreateInfoAsync(vmInfo,
i => i.NetworkAdapters,
adapter => networkAdapterConfig.Name.Equals(adapter.Name, StringComparison.OrdinalIgnoreCase),
async() =>
{
await reportProgress($"Add Network Adapter: {networkAdapterConfig.Name}").ConfigureAwait(false);
return(await engine.GetObjectsAsync <VMNetworkAdapter>(PsCommandBuilder.Create()
.AddCommand("Add-VmNetworkAdapter")
.AddParameter("Passthru")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("Name", networkAdapterConfig.Name)
.AddParameter("StaticMacAddress", UseOrGenerateMacAddress(networkAdapterConfig, vmInfo))
.AddParameter("SwitchName", networkAdapterConfig.SwitchName)).ConfigureAwait(false));
}).ConfigureAwait(false);
return(optionalAdapter.Map(_ => vmInfo.Recreate()));
//optionalAdapter.Map(async (adapter) =>
//{
// if (!adapter.Value.Connected || adapter.Value.SwitchName != networkConfig.SwitchName)
// {
// await reportProgress($"Connect Network Adapter {adapter.Value.Name} to switch {networkConfig.SwitchName}").ConfigureAwait(false);
// await engine.RunAsync(
// PsCommandBuilder.Create().AddCommand("Connect-VmNetworkAdapter")
// .AddParameter("VMNetworkAdapter", adapter.PsObject)
// .AddParameter("SwitchName", networkConfig.SwitchName)).ConfigureAwait(false);
// }
//});
return(vmInfo.Recreate());
}
public async Task <Either <PowershellFailure, Seq <TypedPsObject <T> > > > GetObjectsAsync <T>(PsCommandBuilder builder, Func <int, Task> reportProgress = null)
{
using (var ps = CreateShell())
{
builder.Build(ps);
InitializeAsyncProgressReporting(ps, reportProgress);
return(await ps.GetObjectsAsync <T>().ConfigureAwait(false));
}
}
public static async Task <Either <PowershellFailure, TypedPsObject <VirtualMachineInfo> > > VirtualDisk(
VMDiskStorageSettings diskSettings,
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
Seq <CurrentVMDiskStorageSettings> currentStorageSettings,
Func <string, Task> reportProgress)
{
var currentSettings =
currentStorageSettings.Find(x => diskSettings.Path.Equals(x.Path, StringComparison.OrdinalIgnoreCase) &&
diskSettings.Name.Equals(x.Name, StringComparison.InvariantCultureIgnoreCase));
var frozenOptional = currentSettings.Map(x => x.Frozen);
if (frozenOptional.IsSome && frozenOptional.ValueUnsafe())
{
await reportProgress($"Skipping disk '{diskSettings.Name}': storage management is disabled for this disk.").ConfigureAwait(false);
return(vmInfo);
}
return(await diskSettings.AttachPath.Map(async (vhdPath) =>
{
if (!File.Exists(vhdPath))
{
await reportProgress($"Create VHD: {diskSettings.Name}").ConfigureAwait(false);
var createDiskResult = await diskSettings.ParentPath.Match(Some: parentPath =>
{
return engine.RunAsync(PsCommandBuilder.Create().Script(
$"New-VHD -Path \"{vhdPath}\" -ParentPath \"{parentPath}\" -Differencing"));
},
None: () =>
{
return engine.RunAsync(PsCommandBuilder.Create().Script(
$"New-VHD -Path \"{vhdPath}\" -Dynamic -SizeBytes {diskSettings.SizeBytes}"));
});
if (createDiskResult.IsLeft)
{
return Prelude.Left(createDiskResult.LeftAsEnumerable().FirstOrDefault());
}
}
var sizeResult = await engine
.GetObjectsAsync <VhdInfo>(new PsCommandBuilder().AddCommand("get-vhd").AddArgument(vhdPath))
.BindAsync(x => x.HeadOrLeft(new PowershellFailure())).BindAsync(async(vhdInfo) =>
{
if (vhdInfo.Value.Size != diskSettings.SizeBytes && diskSettings.SizeBytes > 0)
{
var gb = Math.Round(diskSettings.SizeBytes / 1024d / 1024 / 1024, 1);
await reportProgress(
$"Resizing disk {diskSettings.Name} to {gb} GB");
return await engine.RunAsync(PsCommandBuilder.Create().AddCommand("Resize-VHD")
.AddArgument(vhdPath)
.AddParameter("Size", diskSettings.SizeBytes));
}
return Unit.Default;
});
if (sizeResult.IsLeft)
{
return Prelude.Left(sizeResult.LeftAsEnumerable().FirstOrDefault());
}
return await GetOrCreateInfoAsync(vmInfo,
i => i.HardDrives,
disk => currentSettings.Map(x => x.AttachedVMId) == disk.Id,
async() =>
{
await reportProgress($"Attaching disk {diskSettings.Name} to controller: {diskSettings.ControllerNumber}, Location: {diskSettings.ControllerLocation}").ConfigureAwait(false);
return (await engine.GetObjectsAsync <HardDiskDriveInfo>(PsCommandBuilder.Create()
.AddCommand("Add-VMHardDiskDrive")
.AddParameter("VM", vmInfo.PsObject)
.AddParameter("Path", vhdPath)
.AddParameter("ControllerNumber", diskSettings.ControllerNumber)
.AddParameter("ControllerLocation", diskSettings.ControllerLocation)
.AddParameter("PassThru")
).ConfigureAwait(false));
}).BindAsync(_ => vmInfo.RecreateOrReload(engine))
.ConfigureAwait(false);
}).IfNone(vmInfo.RecreateOrReload(engine)));
}
public static Task <Either <PowershellFailure, Unit> > DetachUndefinedHardDrives(
IPowershellEngine engine,
TypedPsObject <VirtualMachineInfo> vmInfo,
Seq <VMDriveStorageSettings> plannedStorageSettings,
Seq <CurrentVMDiskStorageSettings> currentDiskStorageSettings,
Func <string, Task> reportProgress)
{
var planedDiskSettings = plannedStorageSettings.Where(x =>
x.Type == VirtualMachineDriveType.VHD || x.Type == VirtualMachineDriveType.SharedVHD)
.Cast <VMDiskStorageSettings>().ToSeq();
var attachedPaths = planedDiskSettings.Map(s => s.AttachPath).Map(x => x.IfNone(""))
.Where(x => !string.IsNullOrWhiteSpace(x));
var frozenDiskIds = currentDiskStorageSettings.Where(x => x.Frozen).Map(x => x.AttachedVMId);
return(FindAndApply(vmInfo,
i => i.HardDrives,
hd =>
{
var plannedDiskAtControllerPos = planedDiskSettings
.FirstOrDefault(x =>
x.ControllerLocation == hd.ControllerLocation && x.ControllerNumber == hd.ControllerNumber);
var detach = plannedDiskAtControllerPos == null;
if (!detach && plannedDiskAtControllerPos.AttachPath.IsSome)
{
var plannedAttachPath = plannedDiskAtControllerPos.AttachPath.IfNone("");
if (hd.Path == null || !hd.Path.Equals(plannedAttachPath, StringComparison.InvariantCultureIgnoreCase))
{
detach = true;
}
}
if (detach && frozenDiskIds.Contains(hd.Id))
{
reportProgress(hd.Path != null
? $"Skipping detach of frozen disk {Path.GetFileNameWithoutExtension(hd.Path)}"
: $"Skipping detach of unknown frozen disk at controller {hd.ControllerNumber}, Location: {hd.ControllerLocation}");
return false;
}
if (detach)
{
reportProgress(hd.Path != null
? $"Detaching disk {Path.GetFileNameWithoutExtension(hd.Path)} from controller: {hd.ControllerNumber}, Location: {hd.ControllerLocation}"
: $"Detaching unknown disk at controller: {hd.ControllerNumber}, Location: {hd.ControllerLocation}");
}
return detach;
},
i => engine.RunAsync(PsCommandBuilder.Create().AddCommand("Remove-VMHardDiskDrive")
.AddParameter("VMHardDiskDrive", i.PsObject))).Map(x => x.Lefts().HeadOrNone())
.MatchAsync(
Some: l => LeftAsync <PowershellFailure, Unit>(l).ToEither(),
None: () => RightAsync <PowershellFailure, Unit>(Unit.Default)
.ToEither()));
}