mirror of
git://jb55.com/damus
synced 2024-09-18 19:23:49 +00:00
63 lines
1.3 KiB
Swift
63 lines
1.3 KiB
Swift
//
|
|
// NdbTagsIterator.swift
|
|
// damus
|
|
//
|
|
// Created by William Casarin on 2023-07-21.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
struct TagsIterator: IteratorProtocol {
|
|
typealias Element = TagSequence
|
|
|
|
var done: Bool
|
|
var iter: ndb_iterator
|
|
var note: NdbNote
|
|
|
|
mutating func next() -> TagSequence? {
|
|
guard ndb_tags_iterate_next(&self.iter) == 1 else {
|
|
done = true
|
|
return nil
|
|
}
|
|
|
|
return TagSequence(note: note, tag: self.iter.tag)
|
|
}
|
|
|
|
var count: UInt16 {
|
|
return note.note.pointee.tags.count
|
|
}
|
|
|
|
init(note: NdbNote) {
|
|
self.iter = ndb_iterator()
|
|
ndb_tags_iterate_start(note.note, &self.iter)
|
|
self.done = false
|
|
self.note = note
|
|
}
|
|
}
|
|
|
|
struct TagsSequence: Sequence {
|
|
let note: NdbNote
|
|
|
|
var count: UInt16 {
|
|
note.note.pointee.tags.count
|
|
}
|
|
|
|
// no O(1) indexing on top-level tag lists unfortunately :(
|
|
// bit it's very fast to iterate over each tag since the number of tags
|
|
// are stored and the elements are fixed size.
|
|
subscript(index: Int) -> Iterator.Element? {
|
|
var i = 0
|
|
for element in self {
|
|
if i == index {
|
|
return element
|
|
}
|
|
i += 1
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func makeIterator() -> TagsIterator {
|
|
return .init(note: note)
|
|
}
|
|
}
|