Chaim Lev-Ari
96 lines
3.2 KiB
JavaScript
96 lines
3.2 KiB
JavaScript
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();
|
|
}
|