ui/lib/shared/addon/components/form-pod-affinity-k8s/component.js

import Component from '@ember/component';
import layout from './template';
import {
  computed,
  get,
  set,
  observer
} from '@ember/object';
import { randomStr } from '../../utils/util';

/**  podAffinity: {
  *    preferredDuringSchedulingIgnoredDuringExecution:
  *      - {
  *          weight: int,
  *          podaffinityterm: podaffinityterm
  *        }
  *    requiredDuringSchedulingIgnoredDuringExecution:
  *      - podaffinityterm
  *  }
  * podAntiAffinity: same shape
  */

export const TERM_PRIORITY = {
  PREFERRED: 'preferred',
  REQUIRED:  'required'
}

export default Component.extend({
  layout,

  TERM_PRIORITY,
  podAffinity:     null,
  podAntiAffinity: null,
  value:           null,
  mode:            'new',
  anti:            false,
  allTerms:        [],
  /**
 * this component renders one list for required & preferred arrays of terms in affinity & antiAffinity
 * each podaffinitytermk8s component can change between affinity and antiaffinity and between required and perferred
 * the overall list shouldn't re-order when a term is moved to a different underlying array so rather than computing this off the arrays in spec
 * this list will track which array a term should belong to and the arrays in spec will be computed off this
 * list of all terms
 * - {
 *    priority: preferred/required
 *    anti: bool
 *    term:preferred or required term
 *   }
 *
 */
  init(){
    this._super(...arguments);
    const allTerms = []
    const addTerms = function(terms = [], priority, isAnti){
      terms.forEach((term) => {
        allTerms.push({
          priority,
          anti: isAnti,
          term
        })
      })
    }

    addTerms(this.podAffinity?.preferredDuringSchedulingIgnoredDuringExecution, TERM_PRIORITY.PREFERRED, false)
    addTerms(this.podAffinity?.requiredDuringSchedulingIgnoredDuringExecution, TERM_PRIORITY.REQUIRED, false)
    addTerms(this.podAntiAffinity?.preferredDuringSchedulingIgnoredDuringExecution, TERM_PRIORITY.PREFERRED, true)
    addTerms(this.podAntiAffinity?.requiredDuringSchedulingIgnoredDuringExecution, TERM_PRIORITY.REQUIRED, true)

    set(this, 'allTerms', allTerms)
  },

  actions: {
    addRequired(){
      this.requiredDuringSchedulingIgnoredDuringExecution.pushObject({})
    },
    addPreferred(){
      this.preferredDuringSchedulingIgnoredDuringExecution.pushObject({})
    },

    addTerm(){
      const neu = {
        priority: TERM_PRIORITY.REQUIRED,
        anti:     false,
        term:     { _id: randomStr() }
      }

      this.allTerms.push(neu)
      get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution').addObject(neu.term)
      this.notifyPropertyChange('allTerms')
      this.notifyPropertyChange('podAffinity')

      // TODO nb update relevent spec array
    },

    removeTerm(term){
      // TODO nb update relevent spec array
      get(this, 'allTerms').removeObject(term)
      if (term.anti){
        if (term.priority === TERM_PRIORITY.REQUIRED){
          const removeFrom = get(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')

          removeFrom.removeObject(term.term)
        } else {
          const removeFrom = get(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')

          removeFrom.removeObject(term.term)
        }
        this.notifyPropertyChange('podAntiAffinity')
      } else {
        if (term.priority === TERM_PRIORITY.REQUIRED){
          const removeFrom = get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')

          removeFrom.removeObject(term.term)
        } else {
          const removeFrom = get(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')

          removeFrom.removeObject(term.term)
        }
        this.notifyPropertyChange('podAffinity')
      }
    },

    affinityChanged(){
      // TODO nb this.update()
      // needed?
    },



    typeChanged(term, old, neu){
      if (term.anti){
        if (old === TERM_PRIORITY.REQUIRED){
          const removeFrom = get(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        } else {
          const removeFrom = get(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        }
        this.notifyPropertyChange('podAntiAffinity')
      } else {
        if (old === TERM_PRIORITY.REQUIRED){
          const removeFrom = get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        } else {
          const removeFrom = get(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))
          addTo.pushObject(term.term)
        }
        this.notifyPropertyChange('podAffinity')
      }
    },

    antiChanged(term, old, neu){
      if (old){
        if (term.priority === TERM_PRIORITY.REQUIRED){
          const removeFrom = get(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        } else {
          const removeFrom = get(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        }
      } else {
        if (term.priority === TERM_PRIORITY.REQUIRED){
          const removeFrom = get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        } else {
          const removeFrom = get(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')
          const addTo = get(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution')

          // removeFrom.removeObject(term.term)
          set(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution', this.removeFromSpec(term.term, removeFrom))

          addTo.pushObject(term.term)
        }
      }
      this.notifyPropertyChange('podAffinity')
      this.notifyPropertyChange('podAntiAffinity')
    }
  },

  editing: computed('mode', function() {
    const mode = get(this, 'mode')

    return mode === 'new' || mode === 'edit'
  }),

  removeFromSpec: (term, array) => {
    return array.filter((each) => each._id !== term._id)
  },

  // affinityRequiredDuringSchedulingIgnoredDuringExecution: computed('podAffinity.requiredDuringSchedulingIgnoredDuringExecution', {
  //   get(){
  //     return get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution') || []
  //   },
  //   set(key, val){
  //     debugger
  //     if (get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution')){
  //       get(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution').addObject(val)
  //     } else {
  //       set(this.podAffinity, 'requiredDuringSchedulingIgnoredDuringExecution', [val])
  //     }
  //     this.notifyPropertyChange('podAffinity')

  //     return val
  //   }
  // }),

  // affinityPreferredDuringSchedulingIgnoredDuringExecution: computed('podAffinity.preferredDuringSchedulingIngnoredDuringExecution', function(){
  //   return get(this.podAffinity, 'preferredDuringSchedulingIgnoredDuringExecution') || []
  // }),

  // antiAffinityRequiredDuringSchedulingIgnoredDuringExecution: computed('podAntiAffinity.requiredDuringSchedulingIgnoredDuringExecution', function(){
  //   return get(this.podAntiAffinity, 'requiredDuringSchedulingIgnoredDuringExecution') || []
  // }),

  // antiAffinityPreferredDuringSchedulingIgnoredDuringExecution: computed('podAntiAffinity.preferredDuringSchedulingIngnoredDuringExecution', function(){
  //   return get(this.podAntiAffinity, 'preferredDuringSchedulingIgnoredDuringExecution') || []
  // }),


})