/*
 * This file is part of the KubeVirt project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * Copyright The KubeVirt Authors.
 *
 */

package generic

import (
	v1 "kubevirt.io/api/core/v1"

	"kubevirt.io/kubevirt/pkg/virt-launcher/virtwrap/device/hostdevice"
)

// NewPCIAddressPool creates a PCI address pool based on the provided list of host-devices and
// the environment variables that describe the resource.
func NewPCIAddressPool(hostDevices []v1.HostDevice) *hostdevice.AddressPool {
	return hostdevice.NewAddressPool(v1.PCIResourcePrefix, extractResources(hostDevices))
}

// NewMDEVAddressPool creates a MDEV address pool based on the provided list of host-devices and
// the environment variables that describe the resource.
func NewMDEVAddressPool(hostDevices []v1.HostDevice) *hostdevice.AddressPool {
	return hostdevice.NewAddressPool(v1.MDevResourcePrefix, extractResources(hostDevices))
}

// NewUSBAddressPool creates a USB address pool based on the environment variable that describes
// the resource
func NewUSBAddressPool(hostDevices []v1.HostDevice) *hostdevice.AddressPool {
	return hostdevice.NewAddressPool(v1.USBResourcePrefix, extractResources(hostDevices))
}

func extractResources(hostDevices []v1.HostDevice) []string {
	var resourceSet = make(map[string]struct{})
	for _, hostDevice := range hostDevices {
		resourceSet[hostDevice.DeviceName] = struct{}{}
	}

	var resources []string
	for resource := range resourceSet {
		resources = append(resources, resource)
	}
	return resources
}
