/**
 * Copyright (c) HashiCorp, Inc.
 * SPDX-License-Identifier: MPL-2.0
 */

import { computed } from '@ember/object';
import { alias, and } from '@ember/object/computed';
import Fragment from 'ember-data-model-fragments/fragment';
import { attr } from '@ember-data/model';
import {
  fragment,
  fragmentOwner,
  fragmentArray,
} from 'ember-data-model-fragments/attributes';
import classic from 'ember-classic-decorator';

@classic
export default class TaskState extends Fragment {
  @fragmentOwner() allocation;

  @attr('string') name;
  @attr('string') state;
  @attr('date') startedAt;
  @attr('date') finishedAt;
  @attr('boolean') failed;

  @and('isActive', 'allocation.isRunning') isRunning;

  @computed('task.kind')
  get isConnectProxy() {
    return (this.get('task.kind') || '').startsWith('connect-proxy:');
  }

  @computed('name', 'allocation.taskGroup.tasks.[]')
  get task() {
    const tasks = this.get('allocation.taskGroup.tasks');
    return tasks && tasks.findBy('name', this.name);
  }

  @alias('task.driver') driver;

  // TaskState represents a task running on a node, so in addition to knowing the
  // driver via the task, the health of the driver is also known via the node
  @computed('task.driver', 'allocation.node.drivers.[]')
  get driverStatus() {
    const nodeDrivers = this.get('allocation.node.drivers') || [];
    return nodeDrivers.findBy('name', this.get('task.driver'));
  }

  @fragment('resources') resources;
  @fragmentArray('task-event') events;

  @computed('state')
  get stateClass() {
    const classMap = {
      pending: 'is-pending',
      running: 'is-primary',
      finished: 'is-complete',
      failed: 'is-error',
    };

    return classMap[this.state] || 'is-dark';
  }

  @computed('state')
  get isActive() {
    return this.state === 'running';
  }

  restart() {
    return this.allocation.restart(this.name);
  }
}
