package apemanager

import (
	"context"
	"sync/atomic"

	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/apemanager"
	ape_grpc "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/apemanager/grpc"
	"git.frostfs.info/TrueCloudLab/frostfs-node/internal/audit"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
)

var _ Server = (*auditService)(nil)

type auditService struct {
	next    Server
	log     *logger.Logger
	enabled *atomic.Bool
}

func NewAuditService(next Server, log *logger.Logger, enabled *atomic.Bool) Server {
	return &auditService{
		next:    next,
		log:     log,
		enabled: enabled,
	}
}

// AddChain implements Server.
func (a *auditService) AddChain(ctx context.Context, req *apemanager.AddChainRequest) (*apemanager.AddChainResponse, error) {
	res, err := a.next.AddChain(ctx, req)
	if !a.enabled.Load() {
		return res, err
	}

	var respChainID []byte
	if respBody := res.GetBody(); respBody != nil {
		respChainID = respBody.GetChainID()
	}

	audit.LogRequest(a.log, ape_grpc.APEManagerService_AddChain_FullMethodName, req,
		audit.TargetFromChainID(req.GetBody().GetTarget().GetTargetType().String(),
			req.GetBody().GetTarget().GetName(),
			respChainID),
		err == nil)

	return res, err
}

// ListChains implements Server.
func (a *auditService) ListChains(ctx context.Context, req *apemanager.ListChainsRequest) (*apemanager.ListChainsResponse, error) {
	res, err := a.next.ListChains(ctx, req)
	if !a.enabled.Load() {
		return res, err
	}

	audit.LogRequest(a.log, ape_grpc.APEManagerService_ListChains_FullMethodName, req,
		audit.TargetFromChainID(req.GetBody().GetTarget().GetTargetType().String(),
			req.GetBody().GetTarget().GetName(),
			nil),
		err == nil)

	return res, err
}

// RemoveChain implements Server.
func (a *auditService) RemoveChain(ctx context.Context, req *apemanager.RemoveChainRequest) (*apemanager.RemoveChainResponse, error) {
	res, err := a.next.RemoveChain(ctx, req)
	if !a.enabled.Load() {
		return res, err
	}

	audit.LogRequest(a.log, ape_grpc.APEManagerService_RemoveChain_FullMethodName, req,
		audit.TargetFromChainID(req.GetBody().GetTarget().GetTargetType().String(),
			req.GetBody().GetTarget().GetName(),
			req.GetBody().GetChainID()),
		err == nil)

	return res, err
}