audit_log.go

package database

import (
	"context"

	"github.com/google/uuid"
	"github.com/nais/api/internal/auditlogger/audittype"
	"github.com/nais/api/internal/database/gensql"
	"github.com/nais/api/internal/slug"
)

type AuditLogsRepo interface {
	CreateAuditLogEntry(ctx context.Context, correlationID uuid.UUID, actor *string, targetType audittype.AuditLogsTargetType, targetIdentifier string, action audittype.AuditAction, message string) error
	GetAuditLogsForCorrelationID(ctx context.Context, correlationID uuid.UUID, p Page) ([]*AuditLog, int, error)
	GetAuditLogsForReconciler(ctx context.Context, reconcilerName string, p Page) ([]*AuditLog, int, error)
	GetAuditLogsForTeam(ctx context.Context, teamSlug slug.Slug, p Page) ([]*AuditLog, int, error)
}

var _ AuditLogsRepo = (*database)(nil)

type AuditLog struct {
	*gensql.AuditLog
}

func (d *database) GetAuditLogsForTeam(ctx context.Context, teamSlug slug.Slug, p Page) ([]*AuditLog, int, error) {
	rows, err := d.querier.GetAuditLogsForTeam(ctx, gensql.GetAuditLogsForTeamParams{
		TargetIdentifier: string(teamSlug),
		Offset:           int32(p.Offset),
		Limit:            int32(p.Limit),
	})
	if err != nil {
		return nil, 0, err
	}

	entries := make([]*AuditLog, len(rows))
	for i, row := range rows {
		entries[i] = &AuditLog{AuditLog: row}
	}

	total, err := d.querier.GetAuditLogsForTeamCount(ctx, string(teamSlug))
	if err != nil {
		return nil, 0, err
	}

	return entries, int(total), nil
}

func (d *database) GetAuditLogsForReconciler(ctx context.Context, reconcilerName string, p Page) ([]*AuditLog, int, error) {
	rows, err := d.querier.GetAuditLogsForReconciler(ctx, gensql.GetAuditLogsForReconcilerParams{
		TargetIdentifier: reconcilerName,
		Offset:           int32(p.Offset),
		Limit:            int32(p.Limit),
	})
	if err != nil {
		return nil, 0, err
	}

	entries := make([]*AuditLog, len(rows))
	for i, row := range rows {
		entries[i] = &AuditLog{AuditLog: row}
	}

	total, err := d.querier.GetAuditLogsForReconcilerCount(ctx, reconcilerName)
	if err != nil {
		return nil, 0, err
	}

	return entries, int(total), nil
}

func (d *database) CreateAuditLogEntry(ctx context.Context, correlationID uuid.UUID, actor *string, targetType audittype.AuditLogsTargetType, targetIdentifier string, action audittype.AuditAction, message string) error {
	return d.querier.CreateAuditLog(ctx, gensql.CreateAuditLogParams{
		CorrelationID:    correlationID,
		Actor:            actor,
		TargetType:       string(targetType),
		TargetIdentifier: targetIdentifier,
		Action:           string(action),
		Message:          message,
	})
}

func (d *database) GetAuditLogsForCorrelationID(ctx context.Context, correlationID uuid.UUID, p Page) ([]*AuditLog, int, error) {
	rows, err := d.querier.GetAuditLogsForCorrelationID(ctx, gensql.GetAuditLogsForCorrelationIDParams{
		CorrelationID: correlationID,
		Offset:        int32(p.Offset),
		Limit:         int32(p.Limit),
	})
	if err != nil {
		return nil, 0, err
	}

	entries := make([]*AuditLog, len(rows))
	for i, row := range rows {
		entries[i] = &AuditLog{AuditLog: row}
	}
	total, err := d.querier.GetAuditLogsForCorrelationIDCount(ctx, correlationID)
	if err != nil {
		return nil, 0, err
	}

	return entries, int(total), nil
}