package sarama

import (
	"time"
)

type DeleteGroupsResponse struct {
	Version         int16
	ThrottleTime    time.Duration
	GroupErrorCodes map[string]KError
}

func (r *DeleteGroupsResponse) encode(pe packetEncoder) error {
	pe.putInt32(int32(r.ThrottleTime / time.Millisecond))

	if err := pe.putArrayLength(len(r.GroupErrorCodes)); err != nil {
		return err
	}
	for groupID, errorCode := range r.GroupErrorCodes {
		if err := pe.putString(groupID); err != nil {
			return err
		}
		pe.putInt16(int16(errorCode))
	}

	return nil
}

func (r *DeleteGroupsResponse) decode(pd packetDecoder, version int16) error {
	throttleTime, err := pd.getInt32()
	if err != nil {
		return err
	}
	r.ThrottleTime = time.Duration(throttleTime) * time.Millisecond

	n, err := pd.getArrayLength()
	if err != nil {
		return err
	}
	if n == 0 {
		return nil
	}

	r.GroupErrorCodes = make(map[string]KError, n)
	for i := 0; i < n; i++ {
		groupID, err := pd.getString()
		if err != nil {
			return err
		}
		errorCode, err := pd.getInt16()
		if err != nil {
			return err
		}

		r.GroupErrorCodes[groupID] = KError(errorCode)
	}

	return nil
}

func (r *DeleteGroupsResponse) key() int16 {
	return 42
}

func (r *DeleteGroupsResponse) version() int16 {
	return r.Version
}

func (r *DeleteGroupsResponse) headerVersion() int16 {
	return 0
}

func (r *DeleteGroupsResponse) isValidVersion() bool {
	return r.Version >= 0 && r.Version <= 1
}

func (r *DeleteGroupsResponse) requiredVersion() KafkaVersion {
	switch r.Version {
	case 1:
		return V2_0_0_0
	case 0:
		return V1_1_0_0
	default:
		return V2_0_0_0
	}
}

func (r *DeleteGroupsResponse) throttleTime() time.Duration {
	return r.ThrottleTime
}