package blobovniczatree

import (
	"context"
	"encoding/hex"
	"time"

	"git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobovnicza"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/common"
	tracingPkg "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/tracing"
	"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client"
	"go.opentelemetry.io/otel/attribute"
	"go.opentelemetry.io/otel/trace"
	"go.uber.org/zap"
)

// Exists implements common.Storage.
func (b *Blobovniczas) Exists(ctx context.Context, prm common.ExistsPrm) (common.ExistsRes, error) {
	var (
		startedAt = time.Now()
		success   = false
		found     = false
	)
	defer func() {
		b.metrics.Exists(time.Since(startedAt), success, prm.StorageID != nil)
	}()

	ctx, span := tracing.StartSpanFromContext(ctx, "Blobovniczas.Exists",
		trace.WithAttributes(
			attribute.String("path", b.rootPath),
			attribute.String("address", prm.Address.EncodeToString()),
			attribute.String("storage_id", hex.EncodeToString(prm.StorageID)),
		))
	defer span.End()

	if prm.StorageID != nil {
		id := NewIDFromBytes(prm.StorageID)
		shBlz := b.getBlobovnicza(ctx, id.Path())
		blz, err := shBlz.Open(ctx)
		if err != nil {
			return common.ExistsRes{}, err
		}
		defer shBlz.Close(ctx)

		exists, err := blz.Exists(ctx, prm.Address)
		return common.ExistsRes{Exists: exists}, err
	}

	var gPrm blobovnicza.GetPrm
	gPrm.SetAddress(prm.Address)

	err := b.iterateSortedDBPaths(ctx, prm.Address, func(p string) (bool, error) {
		_, err := b.getObjectFromLevel(ctx, gPrm, p)
		if err != nil {
			if !client.IsErrObjectNotFound(err) {
				b.log.Debug(ctx, logs.BlobovniczatreeCouldNotGetObjectFromLevel,
					zap.String("level", p),
					zap.Error(err),
					zap.String("trace_id", tracingPkg.GetTraceID(ctx)))
			}
		}

		found = err == nil
		return found, nil
	})

	success = err == nil
	return common.ExistsRes{Exists: found}, err
}