package control

import (
	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/grpc"
	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/message"
)

type requestWrapper struct {
	message.Message
	m grpc.Message
}

func (w *requestWrapper) ToGRPCMessage() grpc.Message {
	return w.m
}

type healthCheckResponseWrapper struct {
	m *HealthCheckResponse
}

func (w *healthCheckResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.m
}

func (w *healthCheckResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	var ok bool

	w.m, ok = m.(*HealthCheckResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, w.m)
	}

	return nil
}

type setNetmapStatusResponseWrapper struct {
	message.Message
	m *SetNetmapStatusResponse
}

func (w *setNetmapStatusResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.m
}

func (w *setNetmapStatusResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	var ok bool

	w.m, ok = m.(*SetNetmapStatusResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, w.m)
	}

	return nil
}

type dropObjectsResponseWrapper struct {
	message.Message
	m *DropObjectsResponse
}

func (w *dropObjectsResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.m
}

func (w *dropObjectsResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	var ok bool

	w.m, ok = m.(*DropObjectsResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, w.m)
	}

	return nil
}

type listShardsResponseWrapper struct {
	m *ListShardsResponse
}

func (w *listShardsResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.m
}

func (w *listShardsResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	var ok bool

	w.m, ok = m.(*ListShardsResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, w.m)
	}

	return nil
}

type setShardModeResponseWrapper struct {
	m *SetShardModeResponse
}

func (w *setShardModeResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.m
}

func (w *setShardModeResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	var ok bool

	w.m, ok = m.(*SetShardModeResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, w.m)
	}

	return nil
}

type synchronizeTreeResponseWrapper struct {
	*SynchronizeTreeResponse
}

func (w *synchronizeTreeResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.SynchronizeTreeResponse
}

func (w *synchronizeTreeResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	r, ok := m.(*SynchronizeTreeResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, (*SynchronizeTreeResponse)(nil))
	}

	w.SynchronizeTreeResponse = r
	return nil
}

type evacuateShardResponseWrapper struct {
	*EvacuateShardResponse
}

func (w *evacuateShardResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.EvacuateShardResponse
}

func (w *evacuateShardResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	r, ok := m.(*EvacuateShardResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, (*EvacuateShardResponse)(nil))
	}

	w.EvacuateShardResponse = r
	return nil
}

type flushCacheResponseWrapper struct {
	*FlushCacheResponse
}

func (w *flushCacheResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.FlushCacheResponse
}

func (w *flushCacheResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	r, ok := m.(*FlushCacheResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, (*FlushCacheResponse)(nil))
	}

	w.FlushCacheResponse = r
	return nil
}

type doctorResponseWrapper struct {
	*DoctorResponse
}

func (w *doctorResponseWrapper) ToGRPCMessage() grpc.Message {
	return w.DoctorResponse
}

func (w *doctorResponseWrapper) FromGRPCMessage(m grpc.Message) error {
	r, ok := m.(*DoctorResponse)
	if !ok {
		return message.NewUnexpectedMessageType(m, (*DoctorResponse)(nil))
	}

	w.DoctorResponse = r
	return nil
}