open-nomad/ui/app/components/drain-popover.js

import Component from '@ember/component';
import { computed } from '@ember/object';
import { equal } from '@ember/object/computed';
import { computed as overridable } from 'ember-overridable-computed';
import { task } from 'ember-concurrency';
import Duration from 'duration-js';
import { tagName } from '@ember-decorators/component';
import classic from 'ember-classic-decorator';
import localStorageProperty from 'nomad-ui/utils/properties/local-storage';

@classic
@tagName('')
export default class DrainPopover extends Component {
  client = null;
  isDisabled = false;

  onError() {}
  onDrain() {}

  parseError = '';

  deadlineEnabled = false;
  forceDrain = false;
  drainSystemJobs = true;

  @localStorageProperty('nomadDrainOptions', {}) drainOptions;

  didReceiveAttrs() {
    super.didReceiveAttrs();
    // Load drain config values from local storage if availabe.
    [
      'deadlineEnabled',
      'customDuration',
      'forceDrain',
      'drainSystemJobs',
      'selectedDurationQuickOption',
    ].forEach((k) => {
      if (k in this.drainOptions) {
        this[k] = this.drainOptions[k];
      }
    });
  }

  @overridable(function () {
    return this.durationQuickOptions[0];
  })
  selectedDurationQuickOption;

  @equal('selectedDurationQuickOption.value', 'custom') durationIsCustom;
  customDuration = '';

  @computed
  get durationQuickOptions() {
    return [
      { label: '1 Hour', value: '1h' },
      { label: '4 Hours', value: '4h' },
      { label: '8 Hours', value: '8h' },
      { label: '12 Hours', value: '12h' },
      { label: '1 Day', value: '1d' },
      { label: 'Custom', value: 'custom' },
    ];
  }

  @computed(
    'deadlineEnabled',
    'durationIsCustom',
    'customDuration',
    'selectedDurationQuickOption.value'
  )
  get deadline() {
    if (!this.deadlineEnabled) return 0;
    if (this.durationIsCustom) return this.customDuration;
    return this.selectedDurationQuickOption.value;
  }

  @task(function* (close) {
    if (!this.client) return;
    const isUpdating = this.client.isDraining;

    let deadline;
    try {
      deadline = new Duration(this.deadline).nanoseconds();
    } catch (err) {
      this.set('parseError', err.message);
      return;
    }

    const spec = {
      Deadline: deadline,
      IgnoreSystemJobs: !this.drainSystemJobs,
    };

    this.drainOptions = {
      deadlineEnabled: this.deadlineEnabled,
      customDuration: this.deadline,
      selectedDurationQuickOption: this.selectedDurationQuickOption,
      drainSystemJobs: this.drainSystemJobs,
      forceDrain: this.forceDrain,
    };

    close();

    try {
      if (this.forceDrain) {
        yield this.client.forceDrain(spec);
      } else {
        yield this.client.drain(spec);
      }
      this.onDrain(isUpdating);
    } catch (err) {
      this.onError(err);
    }
  })
  drain;

  preventDefault(e) {
    e.preventDefault();
  }
}