ui/app/models/pod.js

import C from 'ui/utils/constants';
import Resource from '@rancher/ember-api-store/models/resource';
import { alias } from '@ember/object/computed';
import { reference } from '@rancher/ember-api-store/utils/denormalize';
import { get, computed, set } from '@ember/object';
import { inject as service } from '@ember/service';
import { strPad } from 'ui/utils/util';
import { formatSi } from 'shared/utils/parse-unit';
import { later } from '@ember/runloop';
import { gt } from '@ember/object/computed';
import Grafana from 'shared/mixins/grafana';
import DisplayImage from 'shared/mixins/display-image';

var Pod = Resource.extend(Grafana, DisplayImage, {
  router:       service(),
  modalService:  service('modal'),
  globalStore:  service(),
  clusterStore:  service(),
  scope:        service(),

  canHaveLabels: true,
  escToClose:    true,

  canEdit:      false,
  canClone: false,

  grafanaDashboardName: 'Pods',
  grafanaResourceId:    alias('name'),

  namespace:    reference('namespaceId', 'namespace', 'clusterStore'),
  node:          reference('nodeId', 'node', 'globalStore'),
  workload:      reference('workloadId'),
  hasSidekicks:  gt('containers.length', 1),
  canEditYaml:  computed('links.update', 'links.yaml', function() {
    return !!get(this, 'links.update') && !!get(this, 'links.yaml');
  }),

  canShell: computed('containers', function() {
    return !!this.containers.findBy('canShell', true);
  }),

  availableActions: computed('canShell', function() {
    const canShell = this.canShell;

    var choices = [
      {
        label:     'action.execute',
        icon:      'icon icon-terminal',
        action:    'shell',
        enabled:   canShell,
        altAction: 'popoutShell'
      },
      {
        label:     'action.logs',
        icon:      'icon icon-file',
        action:    'logs',
        enabled:   true,
        altAction: 'popoutLogs'
      },
    ];

    return choices;
  }),

  memoryReservationBlurb: computed('memoryReservation', function() {
    if ( this.memoryReservation ) {
      return formatSi(this.memoryReservation, 1024, 'iB', 'B');
    }

    return;
  }),

  combinedState: computed('node.state', 'workload.state', 'state', 'healthState', 'healthCheck', function() {
    var node = get(this, 'node.state');
    var resource = this.state;
    // var workload = get(this,'workload.state');
    var health = this.healthState;
    var hasCheck = !!this.healthCheck;

    if ( !hasCheck && C.DISCONNECTED_STATES.includes(node) ) {
      return 'unknown';
    } else if ( C.ACTIVEISH_STATES.includes(resource) && health ) {
      return health;
    } else {
      return resource;
    }
  }),

  isOn: computed('state', function() {
    return ['running', 'migrating', 'restarting'].indexOf(this.state) >= 0;
  }),

  displayState: computed('_displayState', 'exitCode', 'state', function() {
    let out = this._displayState;
    let code = this.exitCode;

    if ( this.state === 'stopped' && this.exitCode > 0) {
      out += ` (${  code  })`;
    }

    return out;
  }),

  displayEnvironmentVars: computed('environment', function() {
    var envs = [];
    var environment = this.environment || {};

    Object.keys(environment).forEach((key) => {
      envs.pushObject({
        key,
        value: environment[key]
      })
    });

    return envs;
  }),

  displayIp: computed('status.podIp', function() {
    return get(this, 'status.podIp') || null;
  }),

  dislayContainerMessage: computed('containers.@each.showTransitioningMessage', function() {
    return !!this.containers.findBy('showTransitioningMessage', true);
  }),

  restarts: computed('status.containerStatuses.@each.restartCount', function() {
    let out = 0;

    (get(this, 'status.containerStatuses') || []).forEach((state) => {
      out += get(state, 'restartCount');
    });

    return out;
  }),

  nodeIp: computed('status.nodeIp', function() {
    return get(this, 'status.nodeIp') || null;
  }),

  sortIp: computed('primaryIpAddress', 'primaryAssociatedIpAddress', function() {
    var ip = this.primaryAssociatedIpAddress || this.primaryIpAddress;

    if ( !ip ) {
      return '';
    }
    var match = ip.match(/^(\d+)\.(\d+)\.(\d+)\.(\d+)$/);

    if ( match ) {
      return match.slice(1).map((octet) => strPad(octet, 3, '0', false))
        .join('.');
    }

    return '';
  }),

  isGlobalScale: computed('labels', function() {
    return `${ (this.labels || {})[C.LABEL.SCHED_GLOBAL]  }` === 'true';
  }),

  actions: {
    clone() {
      this.router.transitionTo('containers.run', { queryParams: { podId: this.id, } });
    },

    shell() {
      this.modalService.toggleModal('modal-shell', { model: this, });
    },

    popoutShell() {
      const projectId = get(this, 'scope.currentProject.id');
      const podId = this.id;
      const route = this.router.urlFor('authenticated.project.console', projectId);

      const system = get(this, 'node.info.os.operatingSystem') || ''
      let windows = false;

      if (system.startsWith('Windows')) {
        windows = true;
      }

      later(() => {
        window.open(`//${ window.location.host }${ route }?podId=${ podId }&windows=${ windows }&isPopup=true`, '_blank', 'toolbars=0,width=900,height=700,left=200,top=200');
      });
    },

    logs() {
      const dataToModal = { model: this };

      if (this.containers && this.containers.firstObject) {
        set(dataToModal, 'containerName', this.containers.firstObject.name);
      }

      this.modalService.toggleModal('modal-container-logs', dataToModal);
    },

    popoutLogs() {
      const projectId = get(this, 'scope.currentProject.id');
      const podId = this.id;
      const route = this.router.urlFor('authenticated.project.container-log', projectId);

      later(() => {
        window.open(`//${ window.location.host }${ route }?podId=${ podId }&isPopup=true`, '_blank', 'toolbars=0,width=900,height=700,left=200,top=200');
      });
    },
  },

  hasLabel(key, desiredValue) {
    const labels = this.labels || {};
    const value = get(labels, key);

    if ( value === undefined ) {
      return false;
    }

    if ( desiredValue === undefined ) {
      return true;
    }

    return ( value === desiredValue );
  }
});

export function stoppedIcon(inst) {
  if ( inst.get('restartPolicy.name') === 'no' && inst.get('exitCode') === 0 ) {
    return 'icon icon-dot-circlefill';
  }

  return 'icon icon-circle';
}

export function stoppedColor(inst) {
  if ( inst.get('restartPolicy.name') === 'no' && inst.get('exitCode') === 0 ) {
    return 'text-success';
  }

  return 'text-error';
}

Pod.reopenClass({
  stateMap: {
    'stopped': {
      icon:  stoppedIcon,
      color: stoppedColor
    },
  },

  mangleIn(data) {
    if ( data && data.containers ) {
      data.containers.forEach((container) => {
        container.type = 'container';
        container.podId = data.id;
      })
    }

    return data;
  }
});

export default Pod;