package gkeheterogeneous

import (
	regionaldedicatedv1 "github.com/tidbcloud/infra-provider/api/tidbcloud/regional-server/v1"
)

func newCreateComputeClassRequest(
	clusterID string,
	componentType regionaldedicatedv1.ComponentType,
	bindings *NodePoolBindingSet,
) *regionaldedicatedv1.CreateGKEHeterogeneousComputeClassRequest {
	return &regionaldedicatedv1.CreateGKEHeterogeneousComputeClassRequest{
		ClusterId:     clusterID,
		ComponentType: componentType,
		NodePoolNames: bindings.NodePoolNames,
		InstanceTypes: bindings.InstanceTypes,
	}
}

func newGetComputeClassRequest(
	clusterID string,
	baseInstanceType string,
	heterogeneousInstanceType string,
) *regionaldedicatedv1.GetGKEHeterogeneousComputeClassRequest {
	return &regionaldedicatedv1.GetGKEHeterogeneousComputeClassRequest{
		ClusterId:                 clusterID,
		BaseInstanceType:          baseInstanceType,
		HeterogeneousInstanceType: heterogeneousInstanceType,
	}
}