package strategy

import (
	"sort"

	"github.com/docker/swarm/scheduler/node"
	"github.com/samalba/dockerclient"
)

// SpreadPlacementStrategy is exported
type SpreadPlacementStrategy struct {
}

// Initialize is exported
func (p *SpreadPlacementStrategy) Initialize() error {
	return nil
}

// Name returns the name of the strategy
func (p *SpreadPlacementStrategy) Name() string {
	return "spread"
}

// PlaceContainer is exported
func (p *SpreadPlacementStrategy) PlaceContainer(config *dockerclient.ContainerConfig, nodes []*node.Node) (*node.Node, error) {
	weightedNodes, err := weighNodes(config, nodes)
	if err != nil {
		return nil, err
	}

	// sort by lowest weight
	sort.Sort(weightedNodes)

	bottomNode := weightedNodes[0]
	for _, node := range weightedNodes {
		if node.Weight != bottomNode.Weight {
			break
		}
		if len(node.Node.Containers) < len(bottomNode.Node.Containers) {
			bottomNode = node
		}
	}

	return bottomNode.Node, nil
}