searchreplacefuncs.go

// Copyright (c) 2021-2025 Dell Inc. or its subsidiaries.  All rights reserved.
// Licensed to You under the Apache License, Version 2.0.   See the LICENSE file for more details.
package jsonstream

import (
	"sync"
)

type dropperType int

const (
	dropperTypeFn dropperType = iota
	dropperTypePropertyPathEqual
)

// Dropper is a SearchReplacer that searches for things and doesnt replace them with anything
type Dropper struct {
	dType      dropperType
	searchFunc func(JSONPath) JSONPath
	prefix     JSONPath
}

//nolint:gochecknoglobals
var dropperPool = sync.Pool{
	New: func() interface{} {
		return &Dropper{}
	},
}

func DropPropertyByMatchFunc(drop func(JSONPath) JSONPath) func(TokenIterator) TokenIterator {
	dropper := dropperPool.Get().(*Dropper)
	dropper.searchFunc = drop
	dropper.dType = dropperTypeFn
	return SearchReplace(dropper, true)
}

func DropPropertyByPathMatch(prefix JSONPath, iter TokenIterator) TokenIterator {
	dropper := dropperPool.Get().(*Dropper)
	dropper.dType = dropperTypePropertyPathEqual
	dropper.prefix = prefix
	return SearchReplaceWithIterator(dropper, true, iter)
}

func (bm *Dropper) Search(p JSONPath, _ []byte) JSONPath {
	switch bm.dType {
	case dropperTypePropertyPathEqual:
		return bm.propertyPathEqual(p)
	default:
		return bm.searchFunc(p)
	}
}

func (bm *Dropper) propertyPathEqual(tokPath JSONPath) JSONPath {
	if ret := MatchPathPrefix(tokPath, bm.prefix); len(ret) == len(tokPath) {
		return ret
	}
	return nil
}

func (bm *Dropper) Save(JSONPath, []byte, int) {}

func (bm *Dropper) GetReplaceIter() TokenIterator { return nil }

func (bm *Dropper) Close() error {
	defer dropperPool.Put(bm)
	return nil
}