package contracts

import (
	"google.golang.org/protobuf/proto"

	altv1 "git.toki-labs.com/toki/alt/packages/contracts/gen/go/alt/v1"
	protoSocket "git.toki-labs.com/toki/proto-socket/go"
)

// ParserMap returns a fresh ParserMap populated with all ALT API messages.
func ParserMap() protoSocket.ParserMap {
	return protoSocket.ParserMap{
		protoSocket.TypeNameOf(&altv1.HelloRequest{}):                 parserFor(func() proto.Message { return &altv1.HelloRequest{} }),
		protoSocket.TypeNameOf(&altv1.HelloResponse{}):                parserFor(func() proto.Message { return &altv1.HelloResponse{} }),
		protoSocket.TypeNameOf(&altv1.ListInstrumentsRequest{}):       parserFor(func() proto.Message { return &altv1.ListInstrumentsRequest{} }),
		protoSocket.TypeNameOf(&altv1.ListInstrumentsResponse{}):      parserFor(func() proto.Message { return &altv1.ListInstrumentsResponse{} }),
		protoSocket.TypeNameOf(&altv1.ListBarsRequest{}):              parserFor(func() proto.Message { return &altv1.ListBarsRequest{} }),
		protoSocket.TypeNameOf(&altv1.ListBarsResponse{}):             parserFor(func() proto.Message { return &altv1.ListBarsResponse{} }),
		protoSocket.TypeNameOf(&altv1.StartBacktestRequest{}):         parserFor(func() proto.Message { return &altv1.StartBacktestRequest{} }),
		protoSocket.TypeNameOf(&altv1.StartBacktestResponse{}):        parserFor(func() proto.Message { return &altv1.StartBacktestResponse{} }),
		protoSocket.TypeNameOf(&altv1.GetBacktestRunRequest{}):        parserFor(func() proto.Message { return &altv1.GetBacktestRunRequest{} }),
		protoSocket.TypeNameOf(&altv1.GetBacktestRunResponse{}):       parserFor(func() proto.Message { return &altv1.GetBacktestRunResponse{} }),
		protoSocket.TypeNameOf(&altv1.GetBacktestResultRequest{}):     parserFor(func() proto.Message { return &altv1.GetBacktestResultRequest{} }),
		protoSocket.TypeNameOf(&altv1.GetBacktestResultResponse{}):    parserFor(func() proto.Message { return &altv1.GetBacktestResultResponse{} }),
		protoSocket.TypeNameOf(&altv1.BacktestResult{}):               parserFor(func() proto.Message { return &altv1.BacktestResult{} }),
		protoSocket.TypeNameOf(&altv1.ListBacktestRunsRequest{}):      parserFor(func() proto.Message { return &altv1.ListBacktestRunsRequest{} }),
		protoSocket.TypeNameOf(&altv1.ListBacktestRunsResponse{}):     parserFor(func() proto.Message { return &altv1.ListBacktestRunsResponse{} }),
		protoSocket.TypeNameOf(&altv1.GetBacktestRunDetailRequest{}):  parserFor(func() proto.Message { return &altv1.GetBacktestRunDetailRequest{} }),
		protoSocket.TypeNameOf(&altv1.GetBacktestRunDetailResponse{}): parserFor(func() proto.Message { return &altv1.GetBacktestRunDetailResponse{} }),
		protoSocket.TypeNameOf(&altv1.CompareBacktestRunsRequest{}):   parserFor(func() proto.Message { return &altv1.CompareBacktestRunsRequest{} }),
		protoSocket.TypeNameOf(&altv1.CompareBacktestRunsResponse{}):  parserFor(func() proto.Message { return &altv1.CompareBacktestRunsResponse{} }),
	}
}

func parserFor(factory func() proto.Message) func([]byte) (proto.Message, error) {
	return func(data []byte) (proto.Message, error) {
		msg := factory()
		if err := proto.Unmarshal(data, msg); err != nil {
			return nil, err
		}
		return msg, nil
	}
}