xenadmin/XenAdmin/Commands/Controls/AssignVMGroupToolStripMenuItem.cs

314 lines
13 KiB
C#
Raw Normal View History

/* Copyright (c) Citrix Systems, Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms,
* with or without modification, are permitted provided
* that the following conditions are met:
*
* * Redistributions of source code must retain the above
* copyright notice, this list of conditions and the
* following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the
* following disclaimer in the documentation and/or other
* materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Windows.Forms;
using XenAdmin.Core;
using XenAPI;
using XenAdmin.Dialogs;
using System.Drawing;
namespace XenAdmin.Commands
{
/// <summary>
/// Assign VMs to a group of VMs (e.g., to a VMPP or a vApp)
/// </summary>
internal class AssignGroupToolStripMenuItem<T> : CommandToolStripMenuItem where T : XenObject<T>
{
public AssignGroupToolStripMenuItem()
: base(new AssignVMsToGroup(), false)
{
base.DropDownItems.Add(new ToolStripMenuItem());
}
public AssignGroupToolStripMenuItem(IMainWindow mainWindow, SelectedItemCollection selection, bool inContextMenu)
: base(new AssignVMsToGroup(mainWindow, selection), inContextMenu)
{
base.DropDownItems.Add(new ToolStripMenuItem());
}
protected override void OnDropDownOpening(EventArgs e)
{
base.DropDownItems.Clear();
var cmd = new NewGroupCommand(Command.MainWindowCommandInterface, Command.GetSelection());
var item = new CommandToolStripMenuItem(cmd);
base.DropDownItems.Add(item);
T[] groups = VMGroup<T>.GroupsInCache(Command.GetSelection()[0].Connection.Cache);
if (groups.Length > 0)
base.DropDownItems.Add(new ToolStripSeparator());
Array.Sort(groups);
for (int index = 0, offset = 0; index < groups.Length; index++)
{
T group = groups[index];
/* do not add unsupported policies to the drop down for VMSS */
XenAPI.VMSS policy = group as VMSS;
if (policy != null && policy.type == vmss_type.snapshot_with_quiesce)
{
List<VM> vms = Command.GetSelection().AsXenObjects<VM>();
bool doNotInclude = vms.Any(vm => !vm.allowed_operations.Contains(vm_operations.snapshot_with_quiesce));
if (doNotInclude)
{
offset--;
continue;
}
}
var menuText = (index + offset) < 9
? String.Format(Messages.DYNAMIC_MENUITEM_WITH_ACCESS_KEY, (index + offset) + 1, group.Name)
: String.Format(Messages.DYNAMIC_MENUITEM_WITHOUT_ACCESS_KEY, group.Name);
var cmdGroup = new AssignGroupToVMCommand(Command.MainWindowCommandInterface, Command.GetSelection(), group, menuText);
var itemGroup = new CommandToolStripMenuItem(cmdGroup);
if (Command.GetSelection().Count == 1 &&
VMGroup<T>.VmToGroup((VM)Command.GetSelection()[0].XenObject).opaque_ref == group.opaque_ref)
itemGroup.Checked = true;
base.DropDownItems.Add(itemGroup);
}
}
[Browsable(false)]
[DesignerSerializationVisibility(DesignerSerializationVisibility.Hidden)]
public new Command Command
{
get
{
return base.Command;
}
}
[Browsable(false)]
[DesignerSerializationVisibility(DesignerSerializationVisibility.Hidden)]
public new event EventHandler DropDownOpening
{
add
{
throw new InvalidOperationException();
}
remove
{
throw new InvalidOperationException();
}
}
[Browsable(false)]
[DesignerSerializationVisibility(DesignerSerializationVisibility.Hidden)]
public new ToolStripItemCollection DropDownItems
{
get
{
throw new InvalidOperationException();
}
}
public class AssignGroupToVMCommand : Command
{
private readonly T _group;
private readonly string _menuText;
public AssignGroupToVMCommand(IMainWindow mainWindow, SelectedItemCollection selection, T group, string menuText)
: base(mainWindow, selection)
{
_group = group;
_menuText = menuText;
}
public override string MenuText
{
get { return (String.IsNullOrEmpty(_menuText) ? _group.Name : _menuText).Ellipsise(30); }
}
protected override string EnabledToolTipText
{
get { return _group.Name; }
}
protected override bool CanExecuteCore(SelectedItemCollection selection)
{
return true;
}
/// <summary>
/// Find out if any VMs are already assigned to a different group, and if so, check they can be moved
/// </summary>
/// <param name="vms">All the VMs to be assigned to the group</param>
/// <param name="group">The group to assign the VMs to (null for a new group)</param>
/// <param name="groupName">The name of the group to assign the VMs to</param>
/// <returns>Whether the user is happy to proceed</returns>
public static bool ChangesOK(List<VM> vms, T group, string groupName)
{
var vmsWithExistingGroup = vms.FindAll(vm =>
{
T oldGroup = vm.Connection.Resolve(VMGroup<T>.VmToGroup(vm));
return oldGroup != null && (group == null || oldGroup.opaque_ref != group.opaque_ref);
});
if (vmsWithExistingGroup.Count == 0)
return true;
string text;
if (vmsWithExistingGroup.Count == 1)
{
VM vm = vmsWithExistingGroup[0];
T oldGroup = vm.Connection.Resolve(VMGroup<T>.VmToGroup(vm));
text = string.Format(VMGroup<T>.ChangeOneWarningString,
vm.Name.Ellipsise(250), oldGroup.Name.Ellipsise(250), groupName.Ellipsise(250));
}
else
{
text = string.Format(VMGroup<T>.ChangeMultipleWarningString, groupName.Ellipsise(250));
}
DialogResult dialogResult;
using (var dlg = new ThreeButtonDialog(
new ThreeButtonDialog.Details(SystemIcons.Warning, text, VMGroup<T>.ChangeVMsGroupString),
ThreeButtonDialog.ButtonYes,
ThreeButtonDialog.ButtonNo))
{
dialogResult = dlg.ShowDialog();
}
return dialogResult == DialogResult.Yes;
}
protected override void ExecuteCore(SelectedItemCollection selection)
{
// remove single VM from group
if (selection.Count == 1)
{
XenRef<VM> vmRefInGroup = VMGroup<T>.GroupToVMs(_group).FirstOrDefault(vmRef => vmRef.opaque_ref == selection[0].XenObject.opaque_ref);
if (vmRefInGroup != null)
{
var vmRefs = new List<XenRef<VM>> { vmRefInGroup };
VMGroup<T>.RemoveVMsFromGroupAction(_group, vmRefs).RunAsync();
return;
}
}
if (!ChangesOK(selection.AsXenObjects<VM>(), _group, _group.Name))
return;
var selectedRefVMs = selection.AsXenObjects().ConvertAll<XenRef<VM>>(converterVMRefs);
selectedRefVMs.AddRange(VMGroup<T>.GroupToVMs(_group));
VMGroup<T>.AssignVMsToGroupAction(_group, selectedRefVMs, false).RunAsync();
}
private XenRef<VM> converterVMRefs(IXenObject input)
{
var vm = input as VM;
if (vm == null)
return null;
return new XenRef<VM>(vm.opaque_ref);
}
}
public class NewGroupCommand : Command
{
public NewGroupCommand(IMainWindow mainWindowCommandInterface, SelectedItemCollection getSelection)
: base(mainWindowCommandInterface, getSelection)
{
}
protected override void ExecuteCore(SelectedItemCollection selection)
{
MainWindowCommandInterface.ShowPerConnectionWizard(selection[0].Connection,
VMGroup<T>.NewGroupWizard(Helpers.GetPoolOfOne(selection[0].Connection), selection.AsXenObjects<VM>()));
}
public override string MenuText
{
get { return VMGroup<T>.NewGroupString; }
}
protected override bool CanExecuteCore(SelectedItemCollection selection)
{
return !Helpers.FeatureForbidden(selection.FirstAsXenObject.Connection, VMGroup<T>.FeatureRestricted)
&& (selection.PoolAncestor != null || selection.HostAncestor != null); //CA-61207: this check ensures there's no cross-pool selection
}
}
private class AssignVMsToGroup : Command
{
/// <summary>
/// Initializes a new instance of this Command. The parameter-less constructor is required if
/// this Command is to be attached to a ToolStrip menu item or button. It should not be used in any other scenario.
/// </summary>
public AssignVMsToGroup()
{
}
public AssignVMsToGroup(IMainWindow mainWindow, IEnumerable<SelectedItem> selection)
: base(mainWindow, selection)
{
}
public bool CanExecute(VM vm)
{
return vm != null && vm.is_a_real_vm && !vm.Locked && VMGroup<T>.FeaturePossible(vm.Connection) &&
!Helpers.FeatureForbidden(vm.Connection, VMGroup<T>.FeatureRestricted);
}
protected override bool CanExecuteCore(SelectedItemCollection selection)
{
return selection.AllItemsAre<VM>() && selection.AtLeastOneXenObjectCan<VM>(CanExecute)
&& (selection.PoolAncestor != null || selection.HostAncestor != null); //CA-61207: this check ensures there's no cross-pool selection
}
public override string MenuText
{
get { return VMGroup<T>.AssignMainMenuString; }
}
public override string ContextMenuText
{
get { return VMGroup<T>.AssignContextMenuString; }
}
}
}
/// <summary>
/// Class used for the benefit of visual studio's form designer which has trouble with generic controls
/// </summary>
internal sealed class AssignGroupToolStripMenuItemVMSS : AssignGroupToolStripMenuItem<VMSS>
{ }
/// <summary>
/// Class used for the benefit of visual studio's form designer which has trouble with generic controls
/// </summary>
internal sealed class AssignGroupToolStripMenuItemVM_appliance : AssignGroupToolStripMenuItem<VM_appliance>
{ }
}