package wrapper

import (
	"fmt"

	"github.com/nspcc-dev/neofs-api-go/pkg/netmap"
	netmap2 "github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap"
)

// Fetch returns current netmap node infos.
// Consider using pkg/morph/client/netmap for this.
func (w *Wrapper) Snapshot() (*netmap.Netmap, error) {
	res, err := w.client.Snapshot(netmap2.GetSnapshotArgs{})
	if err != nil {
		return nil, err
	}

	peers := res.Peers()
	result := make([]netmap.NodeInfo, len(peers))

	for i := range peers {
		if err := result[i].Unmarshal(peers[i]); err != nil {
			return nil, fmt.Errorf("can't unmarshal node info: %w", err)
		}
	}

	return netmap.NewNetmap(netmap.NodesFromInfo(result))
}