package iot

import (
	"context"
	"strconv"

	"github.com/gogf/gf/v2/errors/gcode"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"

	v1 "git.5iot.com/iotlcm-public/iotlcm-admin-svr/api/iot/v1"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/contexts"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service/fullsyncretry"
)

type FullSyncReplayController struct{ client fullsyncretry.Client }

func NewFullSyncReplay() *FullSyncReplayController {
	return &FullSyncReplayController{client: fullsyncretry.FromEnvironment()}
}

func (c *FullSyncReplayController) Replay(ctx context.Context, req *v1.FullSyncReplayReq) (*v1.FullSyncReplayRes, error) {
	userID := contexts.GetUserId(ctx)
	if userID == 0 {
		return nil, gerror.NewCode(gcode.CodeNotAuthorized, "Login required")
	}
	receipt, err := c.client.Replay(ctx, req.SourceRunID, strconv.FormatInt(userID, 10), req.MaxRounds)
	if err != nil {
		g.Log().Warningf(ctx, "full_sync_replay rejected actor=%d sourceRunId=%s maxRounds=%d", userID, req.SourceRunID, req.MaxRounds)
		return nil, err
	}
	g.Log().Infof(ctx, "full_sync_replay accepted actor=%d sourceRunId=%s maxRounds=%d replayRunId=%s", userID, req.SourceRunID, req.MaxRounds, receipt.ReplayRunID)
	return &v1.FullSyncReplayRes{ReplayRunID: receipt.ReplayRunID, Status: receipt.Status}, nil
}

func (c *FullSyncReplayController) Status(ctx context.Context, req *v1.FullSyncReplayStatusReq) (*v1.FullSyncReplayStatusRes, error) {
	if contexts.GetUserId(ctx) == 0 {
		return nil, gerror.NewCode(gcode.CodeNotAuthorized, "Login required")
	}
	status, err := c.client.Get(ctx, req.ReplayRunID)
	if err != nil {
		return nil, err
	}
	return &v1.FullSyncReplayStatusRes{ReplayRunID: status.ReplayRunID, SourceRunID: status.SourceRunID, Status: status.Status, Enqueued: status.Enqueued, Remaining: status.Remaining, RunStatus: status.RunStatus, Error: status.Error}, nil
}