import _ from 'lodash-es';
import { PorImageRegistryModel } from 'Docker/models/porImageRegistry';
import { ResourceControlType } from '@/react/portainer/access-control/types';

angular.module('portainer.docker').controller('ContainerController', ContainerController);

/* @ngInject */
function ContainerController($q, $scope, $state, $transition$, $filter, $async, ContainerService, ImageHelper, Notifications, HttpRequestHelper, Authentication, endpoint) {
  $scope.resourceType = ResourceControlType.Container;
  $scope.endpoint = endpoint;
  $scope.activityTime = 0;
  $scope.portBindings = [];

  $scope.onSuccessAction = update;

  $scope.config = {
    RegistryModel: new PorImageRegistryModel(),
  };

  $scope.state = {
    pullImageValidity: false,
  };

  $scope.setPullImageValidity = setPullImageValidity;
  function setPullImageValidity(validity) {
    $scope.state.pullImageValidity = validity;
  }

  $scope.onUpdateRestartPolicySuccess = onUpdateRestartPolicySuccess;
  function onUpdateRestartPolicySuccess(policy) {
    $scope.container.HostConfig.RestartPolicy = {
      Name: policy.name,
      MaximumRetryCount: policy.maximumRetryCount,
    };
  }

  $scope.onUpdateResourceControlSuccess = function () {
    $state.reload();
  };

  $scope.computeDockerGPUCommand = () => {
    const gpuOptions = _.find($scope.container.HostConfig.DeviceRequests, function (o) {
      return o.Driver === 'nvidia' || (o.Capabilities && o.Capabilities.length > 0 && o.Capabilities[0] > 0 && o.Capabilities[0][0] === 'gpu');
    });
    if (!gpuOptions) {
      return 'No GPU config found';
    }
    let gpuStr = 'all';
    if (gpuOptions.Count !== -1) {
      gpuStr = `"device=${_.join(gpuOptions.DeviceIDs, ',')}"`;
    }
    // we only support a single set of capabilities for now
    // creation UI needs to be reworked in order to support OR combinations of AND capabilities
    const capStr = `"capabilities=${_.join(gpuOptions.Capabilities[0], ',')}"`;
    return `${gpuStr},${capStr}`;
  };

  function update() {
    var nodeName = $transition$.params().nodeName;
    HttpRequestHelper.setPortainerAgentTargetHeader(nodeName);
    $scope.nodeName = nodeName;

    ContainerService.container(endpoint.Id, $transition$.params().id)
      .then(function success(data) {
        var container = data;
        $scope.container = container;

        $scope.portBindings = [];
        if (container.NetworkSettings.Ports) {
          _.forEach(Object.keys(container.NetworkSettings.Ports), function (key) {
            if (container.NetworkSettings.Ports[key]) {
              _.forEach(container.NetworkSettings.Ports[key], (portMapping) => {
                const mapping = {};
                mapping.container = key;
                mapping.host = `${portMapping.HostIp}:${portMapping.HostPort}`;
                $scope.portBindings.push(mapping);
              });
            }
          });
        }

        $scope.container.Config.Env = _.sortBy($scope.container.Config.Env, _.toLower);
      })
      .catch(function error(err) {
        Notifications.error('Failure', err, 'Unable to retrieve container info');
      });
  }

  // TODO - need to fix this
  $scope.getRegistryId = function () {
    return _.get($scope.config.RegistryModel, 'Registry.Id', 0);
  };

  update();
}