import AVFAudio import CoreMedia import Defaults import Foundation import SwiftUI extension PlayerModel { func handleSegments(at time: CMTime) { if let segment = lastSkipped { if time > .secondsInDefaultTimescale(segment.end + 5) { resetLastSegment() } } guard let firstSegment = sponsorBlock.segments.first(where: { $0.timeInSegment(time) }) else { return } // find last segment in case they are 2 sec or less after each other // to avoid multiple skips in a row var nextSegments = [firstSegment] while let segment = sponsorBlock.segments.first(where: { !nextSegments.contains($0) && $0.timeInSegment(.secondsInDefaultTimescale(nextSegments.last!.end + 2)) }) { nextSegments.append(segment) } if let segmentToSkip = nextSegments.last, shouldSkip(segmentToSkip, at: time) { skip(segmentToSkip, at: time) } } private func skip(_ segment: Segment, at time: CMTime) { if let duration = playerItemDuration, segment.endTime.seconds >= duration.seconds - 3 { logger.error("segment end time is: \(segment.end) when player item duration is: \(duration.seconds)") DispatchQueue.main.async { [weak self] in guard let self else { return } self.pause() self.backend.eofPlaybackModeAction() } return } backend.seek(to: segment.endTime, seekType: .segmentSkip(segment.category)) DispatchQueue.main.async { [weak self] in withAnimation { self?.lastSkipped = segment } self?.segmentRestorationTime = time } logger.info("SponsorBlock skipping to: \(segment.end)") } private func shouldSkip(_ segment: Segment, at time: CMTime) -> Bool { guard isPlaying, !restoredSegments.contains(segment), Defaults[.sponsorBlockCategories].contains(segment.category) else { return false } return time.seconds - segment.start < 2 && segment.end - time.seconds > 2 } func restoreLastSkippedSegment() { guard let segment = lastSkipped, let time = segmentRestorationTime else { return } restoredSegments.append(segment) backend.seek(to: time, seekType: .segmentRestore) resetLastSegment() } private func resetLastSegment() { DispatchQueue.main.async { [weak self] in withAnimation { self?.lastSkipped = nil self?.controls.objectWillChange.send() } self?.segmentRestorationTime = nil } } func resetSegments() { resetLastSegment() DispatchQueue.main.async { [weak self] in self?.restoredSegments = [] } } }