module model

pub type NodeGetter = fn (NodeFilter) ![]Node

pub struct NodeIterator {
pub mut:
	filter NodeFilter
pub:
	get_func NodeGetter @[required]
}

pub fn (mut i NodeIterator) next() ?[]Node {
	if v := i.filter.page {
		i.filter.page = v + 1
	} else {
		i.filter.page = u64(1)
	}

	nodes := i.get_func(i.filter) or { return none }
	if nodes.len == 0 {
		return none
	}
	return nodes
}

pub type FarmGetter = fn (FarmFilter) ![]Farm

pub struct FarmIterator {
pub mut:
	filter FarmFilter
pub:
	get_func FarmGetter @[required]
}

pub fn (mut i FarmIterator) next() ?[]Farm {
	if v := i.filter.page {
		i.filter.page = v + 1
	} else {
		i.filter.page = u64(1)
	}
	farms := i.get_func(i.filter) or { return none }
	if farms.len == 0 {
		return none
	}
	return farms
}

pub type ContractGetter = fn (ContractFilter) ![]Contract

pub struct ContractIterator {
pub mut:
	filter ContractFilter
pub:
	get_func ContractGetter @[required]
}

pub fn (mut i ContractIterator) next() ?[]Contract {
	if v := i.filter.page {
		i.filter.page = v + 1
	} else {
		i.filter.page = u64(1)
	}
	contracts := i.get_func(i.filter) or { return none }
	if contracts.len == 0 {
		return none
	}
	return contracts
}

pub type TwinGetter = fn (TwinFilter) ![]Twin

pub struct TwinIterator {
pub mut:
	filter TwinFilter
pub:
	get_func TwinGetter @[required]
}

pub fn (mut i TwinIterator) next() ?[]Twin {
	if v := i.filter.page {
		i.filter.page = v + 1
	} else {
		i.filter.page = u64(1)
	}
	twins := i.get_func(i.filter) or { return none }
	if twins.len == 0 {
		return none
	}
	return twins
}