package disks

import (
	"context"
	"errors"
	"net/http"
	"strconv"
)

type LimitIORequest struct {
	DiskID           uint64 `url:"diskId"`
	IOps             uint64 `url:"iops"`
	TotalBytesSec    uint64 `url:"total_bytes_sec"`
	ReadBytesSec     uint64 `url:"read_bytes_sec"`
	WriteBytesSec    uint64 `url:"write_bytes_sec"`
	TotalIOpsSec     uint64 `url:"total_iops_sec"`
	ReadIOpsSec      uint64 `url:"read_iops_sec"`
	WriteIOpsSec     uint64 `url:"write_iops_sec"`
	TotalBytesSecMax uint64 `url:"total_bytes_sec_max"`
	ReadBytesSecMax  uint64 `url:"read_bytes_sec_max"`
	WriteBytesSecMax uint64 `url:"write_bytes_sec_max"`
	TotalIOpsSecMax  uint64 `url:"total_iops_sec_max"`
	ReadIOpsSecMax   uint64 `url:"read_iops_sec_max"`
	WriteIOpsSecMax  uint64 `url:"write_iops_sec_max"`
	SizeIOpsSec      uint64 `url:"size_iops_sec"`
}

func (drq LimitIORequest) Validate() error {
	if drq.DiskID == 0 {
		return errors.New("validation-error: field DiskID can not be empty or equal to 0")
	}

	return nil
}

func (d Disks) LimitIO(ctx context.Context, req LimitIORequest) (bool, error) {
	err := req.Validate()
	if err != nil {
		return false, err
	}

	url := "/cloudapi/disks/limitIO"

	res, err := d.client.DecortApiCall(ctx, http.MethodPost, url, req)
	if err != nil {
		return false, err
	}

	result, err := strconv.ParseBool(string(res))
	if err != nil {
		return false, err
	}

	return result, nil

}