// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.

package storage

// ListKeys returns keys starting from first and upto limit
func ListKeys(store KeyValueStore, first Key, limit Limit) (Keys, error) {
	const unlimited = Limit(1 << 31)

	keys := make(Keys, 0, limit)
	if limit == 0 {
		// TODO: this shouldn't be probably the case
		limit = unlimited
	}

	err := store.Iterate(IterateOptions{
		First:   first,
		Recurse: true,
	}, func(it Iterator) error {
		var item ListItem
		for ; limit > 0 && it.Next(&item); limit-- {
			if item.Key == nil {
				panic("nil key")
			}
			keys = append(keys, CloneKey(item.Key))
		}
		return nil
	})

	return keys, err
}

// ReverseListKeys returns keys starting from first and upto limit in reverse order
func ReverseListKeys(store KeyValueStore, first Key, limit Limit) (Keys, error) {
	const unlimited = Limit(1 << 31)

	keys := make(Keys, 0, limit)
	if limit == 0 {
		// TODO: this shouldn't be probably the case
		limit = unlimited
	}

	err := store.Iterate(IterateOptions{
		First:   first,
		Recurse: true,
		Reverse: true,
	}, func(it Iterator) error {
		var item ListItem
		for ; limit > 0 && it.Next(&item); limit-- {
			if item.Key == nil {
				panic("nil key")
			}
			keys = append(keys, CloneKey(item.Key))
		}
		return nil
	})

	return keys, err
}