policy-engine/pkg/engine/inmemory/morph_storage.go

package inmemory

import (
	"git.frostfs.info/TrueCloudLab/policy-engine/pkg/chain"
	"git.frostfs.info/TrueCloudLab/policy-engine/pkg/engine"
	"github.com/google/uuid"
	"github.com/nspcc-dev/neo-go/pkg/neorpc/result"
	"github.com/nspcc-dev/neo-go/pkg/util"
	"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
)

type inmemoryMorphRuleChainStorage struct {
	nameToNamespaceChains engine.LocalOverrideStorage
	nameToContainerChains engine.LocalOverrideStorage
}

func NewInmemoryMorphRuleChainStorage() engine.MorphRuleChainStorage {
	return &inmemoryMorphRuleChainStorage{
		nameToNamespaceChains: NewInmemoryLocalStorage(),
		nameToContainerChains: NewInmemoryLocalStorage(),
	}
}

func (s *inmemoryMorphRuleChainStorage) AddMorphRuleChain(name chain.Name, target engine.Target, c *chain.Chain) (_ util.Uint256, _ uint32, err error) {
	switch target.Type {
	case engine.Namespace:
		_, err = s.nameToNamespaceChains.AddOverride(name, target, c)
	case engine.Container:
		_, err = s.nameToContainerChains.AddOverride(name, target, c)
	default:
		err = engine.ErrUnknownTarget
	}
	return
}

func (s *inmemoryMorphRuleChainStorage) RemoveMorphRuleChain(name chain.Name, target engine.Target, chainID chain.ID) (_ util.Uint256, _ uint32, err error) {
	switch target.Type {
	case engine.Namespace:
		err = s.nameToNamespaceChains.RemoveOverride(name, target, chainID)
	case engine.Container:
		err = s.nameToContainerChains.RemoveOverride(name, target, chainID)
	default:
		err = engine.ErrUnknownTarget
	}
	return
}

func (s *inmemoryMorphRuleChainStorage) ListMorphRuleChains(name chain.Name, target engine.Target) ([]*chain.Chain, error) {
	switch target.Type {
	case engine.Namespace:
		return s.nameToNamespaceChains.ListOverrides(name, target)
	case engine.Container:
		return s.nameToContainerChains.ListOverrides(name, target)
	default:
	}
	return nil, engine.ErrUnknownTarget
}

func (s *inmemoryMorphRuleChainStorage) GetAdmin() (util.Uint160, error) {
	panic("not implemented")
}

func (s *inmemoryMorphRuleChainStorage) SetAdmin(_ util.Uint160) (util.Uint256, uint32, error) {
	panic("not implemented")
}

func (s *inmemoryMorphRuleChainStorage) ListTargetsIterator(targetType engine.TargetType) (uuid uuid.UUID, it result.Iterator, err error) {
	it.Values = make([]stackitem.Item, 0)

	switch targetType {
	case engine.Namespace:
		// Listing targets may look bizarre, because inmemory rule chain storage use inmemory local overrides where
		// targets are listed by chain names.
		var targets []engine.Target
		targets, err = s.nameToNamespaceChains.ListOverrideDefinedTargets(chain.Ingress)
		if err != nil {
			return
		}
		for _, t := range targets {
			it.Values = append(it.Values, stackitem.NewByteArray([]byte(t.Name)))
		}

		targets, err = s.nameToNamespaceChains.ListOverrideDefinedTargets(chain.S3)
		if err != nil {
			return
		}
		for _, t := range targets {
			it.Values = append(it.Values, stackitem.NewByteArray([]byte(t.Name)))
		}
	case engine.Container:
		var targets []engine.Target
		targets, err = s.nameToContainerChains.ListOverrideDefinedTargets(chain.Ingress)
		if err != nil {
			return
		}
		for _, t := range targets {
			it.Values = append(it.Values, stackitem.NewByteArray([]byte(t.Name)))
		}

		targets, err = s.nameToContainerChains.ListOverrideDefinedTargets(chain.S3)
		if err != nil {
			return
		}
		for _, t := range targets {
			it.Values = append(it.Values, stackitem.NewByteArray([]byte(t.Name)))
		}
	default:
	}
	return
}