DataSource

datasource.swift

protocol DataSourceSection {
    associatedtype Element
    var elements: [Element] { get }
}

protocol DataSource {
    associatedtype Section: DataSourceSection
    var sections: [Section] { get }
}

extension DataSource {
    func numberOfSections() -> Int {
        return sections.count
    }

    func numberOfElements(in section: Int) -> Int {
        return self[section].elements.count
    }

    subscript (section: Int) -> Section {
        return sections[section]
    }

    subscript (indexPath: IndexPath) -> Section.Element {
        return self[indexPath.section].elements[indexPath.row]
    }
}

final class SettingDataSource: DataSource {
    struct Section: DataSourceSection {
        enum SectionType {
            case others
        }
        let type: SectionType
        let title: String
        let elements: [SettingDataSource.Row]
    }

    struct Row {
        enum RowType {
            case logout
        }
        let type: RowType
        let title: String
    }

    let sections: [Section]
    init(sections: [Section]) {
        self.sections = sections
    }
}