package grpcapi

import (
	"context"
	"encoding/json"

	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"

	"git.5iot.com/iotlcm-public/iotlcm-sms/api/pb"
	"git.5iot.com/iotlcm-public/iotlcm-sms/internal/sms"
)

type Sender interface {
	Send(context.Context, sms.Config, sms.Message) (sms.Result, error)
}

type Server struct {
	pb.UnimplementedNoticeServer
	sender Sender
}

func NewServer(sender Sender) *Server {
	return &Server{sender: sender}
}

func (s *Server) Send(ctx context.Context, request *pb.SendReq) (*pb.SendRes, error) {
	if request == nil {
		return nil, status.Error(codes.InvalidArgument, "请求不能为空")
	}

	var config sms.Config
	if err := json.Unmarshal(request.GetConfig(), &config); err != nil {
		return nil, status.Errorf(codes.InvalidArgument, "config JSON 无效: %v", err)
	}
	var message sms.Message
	if err := json.Unmarshal(request.GetParams(), &message); err != nil {
		return nil, status.Errorf(codes.InvalidArgument, "params JSON 无效: %v", err)
	}

	result, _ := s.sender.Send(ctx, config, message)
	responseHead := &pb.HeadRes{
		Code:    int32(result.Code),
		Message: result.Message,
	}
	if request.Head != nil {
		responseHead.RouterId = request.Head.RouterId
		responseHead.TraceId = request.Head.TraceId
	}
	return &pb.SendRes{Head: responseHead}, nil
}