Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Модель секции
// Сделаем нашей модели read only протокол, только с необходимыми полями
protocol SectionCompoundItemProtocol {
var header: String? {get}
var footer: String? {get}
}
class SectionCompoundItem: CompoundItemProtocol {
private let decoratedComposite: CompoundItemProtocol
// основные поля которые мы добавили с помощью декоратора
var header: String?
var footer: String?
var identifier: String {
return self.decoratedComposite.identifier
}
// из нашего декарируемого объекта будем рекурсивно
// запрашивать элементы, вернет только листья
var children: [CompoundItemProtocol] {
return self.decoratedComposite.children
}
var items: [CompoundItemProtocol] {
return self.decoratedComposite.children.flatMap {$0.items}
}
var level: CompoundItemLevel {
return self.decoratedComposite.level
}
init(identifier: String, header: String?, footer: String?) {
// Для упрощения мы не будем передавать декарируемый
// объект, а просто созданим его с нужным уровнем вложенности
self.decoratedComposite = BaseCompoundItem(identifier: identifier, level: .section)
self.header = header
self.footer = footer
}
// В секцию можно добавить все кроме секций и корневых элементов
func add(_ model: CompoundItemProtocol...) {
for item in model {
if item.level != .section && item.level != .root {
self.decoratedComposite.add(item)
} else {
print("You can`t add section in other section")
}
}
}
func remove(_ model: CompoundItemProtocol) {
self.decoratedComposite.remove(model)
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.