source: branches/HeuristicLab.Hive.Azure/HeuristicLab.Clients.Hive.CloudManager/3.3/Views/AddAzureServiceDialog.cs @ 7366

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Windows.Forms;
using HeuristicLab.Clients.Hive.CloudManager.Model;
using HeuristicLab.Core;

namespace HeuristicLab.Clients.Hive.CloudManager.Views {
  public partial class AddAzureServiceDialog : Form {
    private BindingList<Subscription> subscriptions;
    private BindingList<HostedService> hostedServices;
    private BindingList<string> locations;
    private BindingList<AffinityGroup> affinityGroups;

    private BackgroundWorker workerUpdate;
    private BackgroundWorker workerCreate;

    private bool isInitializing;
    private bool bwCompleted;
    private bool closePending;

    private string certificateFile;
    private string certificatePassword;

    public AddAzureServiceDialog(IItemList<Subscription> subs) {
      InitializeComponent();
      isInitializing = true;
      bwCompleted = true;

      this.subscriptions = new BindingList<Subscription>(subs);
      this.hostedServices = new BindingList<HostedService>();
      this.locations = new BindingList<string>();
      this.affinityGroups = new BindingList<AffinityGroup>();

      SetLookupBinding(cmbLocation, locations, "", "");
      SetLookupBinding(cmbAffinityGroup, affinityGroups, "Label", "Name");
      SetLookupBinding(cmbChooseHostedService, hostedServices, "ServiceName", "ServiceName");
      SetLookupBinding(cmbChooseSubscription, subscriptions, "SubscriptionName", "SubscriptionID");

      if (!cbNewHostedService.Checked) {
        SetGroupBoxControlsEnabled(gbNewHostedService, cbNewHostedService.Checked);
      }

      workerUpdate = new BackgroundWorker();
      workerUpdate.DoWork += new DoWorkEventHandler(UpdateTask);
      workerUpdate.RunWorkerCompleted += new RunWorkerCompletedEventHandler(WorkerCompleted);

      workerCreate = new BackgroundWorker();
      workerCreate.DoWork += new DoWorkEventHandler(CreateDoploymentTask);
      workerCreate.RunWorkerCompleted += new RunWorkerCompletedEventHandler(CreateDeploymentCompleted);

      isInitializing = false;
      cmbChooseSubscription_SelectedValueChanged(this, EventArgs.Empty);
    }

    protected override void OnFormClosing(FormClosingEventArgs e) {
      if (!bwCompleted) {
        SetAllControlsEnabled(this, false);
        e.Cancel = true;
        closePending = true;
      }
    }

    private void SetLookupBinding(ListControl toBind, object dataSource, string displayMember, string valueMember) {
      toBind.DisplayMember = displayMember;
      toBind.ValueMember = valueMember;
      toBind.DataSource = dataSource;
    }

    private void SetGroupBoxControlsEnabled(GroupBox gb, bool isEnabled) {
      foreach (Control ctrl in gb.Controls) {
        ctrl.Enabled = isEnabled;
      }
    }

    private void SetAllControlsEnabled(Form frm, bool isEnabled) {
      foreach (Control ctrl in frm.Controls) {
        if (!(ctrl is ProgressBar)) {
          ctrl.Enabled = isEnabled;
        }
      }
    }

    private void SetBindingList<T>(ref BindingList<T> bindinglist, List<T> list) {
      if (list != null) {
        BindingList<T> newList = new BindingList<T>(list);
        bindinglist = newList;
      }
    }

    private void UpdateBindingList<T>(BindingList<T> bindinglist, List<T> list) {
      if (list != null) {
        bindinglist.Clear();
        foreach (T item in list)
          bindinglist.Add(item);
      }
    }

    private void UpdateSubscriptionItem(BindingList<Subscription> bindinglist, Subscription item) {
      if (item != null) {
        foreach (Subscription blItem in bindinglist)
          if (blItem.Equals(item)) {
            blItem.Merge(item);
          }
      }
    }

    private void cbNewHostedService_CheckedChanged(object sender, EventArgs e) {
      if (cbNewHostedService.Checked) {
        Subscription subscription = (Subscription)cmbChooseSubscription.SelectedItem;
        if (subscription.CurrentHostedServices == subscription.MaxHostedServices) {
          MessageBox.Show("You have already created the maximum number of hosted services", "Limit Reached", MessageBoxButtons.OK, MessageBoxIcon.Error);
          cbNewHostedService.Checked = false;
        } else {
          SetGroupBoxControlsEnabled(gbNewHostedService, cbNewHostedService.Checked);
          cmbChooseHostedService.Enabled = false;
        }
      } else {
        // checked == false
        SetGroupBoxControlsEnabled(gbNewHostedService, false);
        if (hostedServices.Count == 0) {
          cbNewHostedService.Checked = true;
        }
      }
    }

    private void cmbChooseSubscription_SelectedValueChanged(object sender, EventArgs e) {
      if (!isInitializing) {
        Subscription item = (Subscription)cmbChooseSubscription.SelectedItem;
        if (item != null) {
          bwCompleted = false;
          progressBar.Visible = true;
          SetAllControlsEnabled(this, false);
          workerUpdate.RunWorkerAsync(item);
        }
      }
    }

    private void btnAddCertificate_Click(object sender, EventArgs e) {
      using (var form = new AddCertificate()) {
        form.ShowDialog();
        if (!form.ErrorOccured) {
          certificateFile = form.CertificateFile;
          certificatePassword = form.CertificatePassword;
          lblCertificateFile.Text = certificateFile;
        }
      }
    }

    private void btnCancel_Click(object sender, EventArgs e) {
      this.Close();
    }

    private void btnOk_Click(object sender, EventArgs e) {
      if (cmbChooseSubscription.SelectedItem != null) {
        Subscription sub = (Subscription)cmbChooseSubscription.SelectedItem;
        bool newHostedServiceChecked = cbNewHostedService.Checked;
        bool regionChecked = rbRegion.Checked;
        string certFile = certificateFile;
        string certPw = certificatePassword;
        int instanceCount = Convert.ToInt32(tbInstanceCount.Text);
        HostedService hostedService = new HostedService();
        hostedService.ServiceName = tbServiceName.Text;
        HostedServiceProperties properties = new HostedServiceProperties();
        if (regionChecked) {
          properties.AffinityGroup = string.Empty;
          properties.Location = (string)cmbLocation.SelectedItem;
        } else {
          properties.AffinityGroup = ((AffinityGroup)cmbAffinityGroup.SelectedItem).Name;
          properties.Location = string.Empty;
        }
        properties.AffinityGroup = ((AffinityGroup)cmbAffinityGroup.SelectedItem).Name;
        properties.Label = tbLabel.Text;
        properties.Location = (string)cmbLocation.SelectedItem;
        hostedService.HostedServiceProperties = properties;

        var parameters = Tuple.Create<Subscription, bool, bool, string, string, HostedService, int>(sub, newHostedServiceChecked, regionChecked, certificateFile, certificatePassword, hostedService, instanceCount);

        bwCompleted = false;
        progressBar.Visible = true;
        SetAllControlsEnabled(this, false);
        workerCreate.RunWorkerAsync(parameters);
      }
    }

    private void UpdateTask(object sender, DoWorkEventArgs e) {
      try {
        Subscription item = (Subscription)e.Argument;
        item = CloudManagerClient.Instance.AzureProvider.GetSubscriptionInfo(item.SubscriptionID, item.CertificateThumbprint);
        this.Invoke((MethodInvoker)delegate { lblCores.Text = item.CurrentCoreCount + " / " + item.MaxCoreCount; });

        List<HostedService> services = CloudManagerClient.Instance.AzureProvider.ListHostedServices(item);
        List<string> locs = CloudManagerClient.Instance.AzureProvider.ListLocations(item);
        List<AffinityGroup> groups = CloudManagerClient.Instance.AzureProvider.ListAffinityGroups(item);

        this.Invoke((MethodInvoker)delegate { UpdateSubscriptionItem(subscriptions, item); });
        this.Invoke((MethodInvoker)delegate { UpdateBindingList<HostedService>(hostedServices, services); });
        this.Invoke((MethodInvoker)delegate { UpdateBindingList<string>(locations, locs); });
        this.Invoke((MethodInvoker)delegate { UpdateBindingList<AffinityGroup>(affinityGroups, groups); });
      }
      catch (Exception ex) {
        MessageBox.Show(ex.Message, "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
      }
    }

    private void WorkerCompleted(object sender, RunWorkerCompletedEventArgs e) {
      SetAllControlsEnabled(this, true);
      progressBar.Visible = false;
      bwCompleted = true;
      if (closePending) {
        this.Close();
      }
    }

    private void CreateDoploymentTask(object sender, DoWorkEventArgs e) {
      Tuple<Subscription, bool, bool, string, string, HostedService, int> parameters = (Tuple<Subscription, bool, bool, string, string, HostedService, int>)e.Argument;

      Subscription sub = parameters.Item1;
      bool newHostedServiceChecked = parameters.Item2;
      bool regionChecked = parameters.Item3;
      string certFile = parameters.Item4;
      string certPw = parameters.Item5;
      HostedService hostedService = parameters.Item6;
      int instanceCount = parameters.Item7;

      //STEP 1 - Create Hosted Service
      if (newHostedServiceChecked) {
        if (regionChecked) {
          CloudManagerClient.Instance.AzureProvider.CreateHostedService(sub, hostedService.ServiceName, hostedService.HostedServiceProperties.Label, hostedService.HostedServiceProperties.Description, hostedService.HostedServiceProperties.Location);
        } else {
          CloudManagerClient.Instance.AzureProvider.CreateHostedService(sub, hostedService.ServiceName, hostedService.HostedServiceProperties.Label, hostedService.HostedServiceProperties.Description, new AffinityGroup() { Name = hostedService.HostedServiceProperties.AffinityGroup });
        }
      }

    }

    private void CreateDeploymentCompleted(object sender, RunWorkerCompletedEventArgs e) {
      SetAllControlsEnabled(this, true);
      progressBar.Visible = false;
      bwCompleted = true;
      if (closePending) {
        this.Close();
      }
    }
  }
}