291 lines
9.6 KiB
Swift
291 lines
9.6 KiB
Swift
import Foundation
|
||
import SwiftData
|
||
|
||
/// SwiftData 持久化模型 - 对应 Android 版 Room 数据库
|
||
|
||
/// 收藏/历史的业务唯一键（source + vodId）。
|
||
private func makeVodBusinessKey(vodId: String, sourceKey: String) -> String {
|
||
let normalizedVodId = vodId.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
let normalizedSourceKey = sourceKey.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
return "\(normalizedSourceKey)::\(normalizedVodId)"
|
||
}
|
||
|
||
/// 单部剧的续播状态
|
||
struct VodPlaybackState: Codable {
|
||
/// 当前播放线路标识。
|
||
var flag: String
|
||
/// 剧集索引。
|
||
var episodeIndex: Int
|
||
/// 播放进度（秒）。
|
||
var progressSeconds: Double
|
||
}
|
||
|
||
/// 视频收藏
|
||
@Model
|
||
final class VodCollect {
|
||
/// 业务唯一键（sourceKey + vodId）。
|
||
var bizKey: String = ""
|
||
/// 视频 ID（与 sourceKey 组成唯一语义键）。
|
||
var vodId: String = ""
|
||
/// 片名。
|
||
var vodName: String = ""
|
||
/// 海报地址。
|
||
var vodPic: String = ""
|
||
/// 来源站点 key。
|
||
var sourceKey: String = ""
|
||
/// 最近更新时间（收藏创建/刷新时间）。
|
||
var updateTime: Date = Date()
|
||
|
||
init(vodId: String, vodName: String, vodPic: String, sourceKey: String) {
|
||
self.bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
self.vodId = vodId
|
||
self.vodName = vodName
|
||
self.vodPic = vodPic
|
||
self.sourceKey = sourceKey
|
||
self.updateTime = Date()
|
||
}
|
||
}
|
||
|
||
/// 播放历史记录
|
||
@Model
|
||
final class VodRecord {
|
||
/// 业务唯一键（sourceKey + vodId）。
|
||
var bizKey: String = ""
|
||
/// 视频 ID。
|
||
var vodId: String = ""
|
||
/// 片名。
|
||
var vodName: String = ""
|
||
/// 海报地址。
|
||
var vodPic: String = ""
|
||
/// 来源站点 key。
|
||
var sourceKey: String = ""
|
||
/// 播放标记，如“第5集 03:45”。
|
||
var playNote: String = ""
|
||
/// 续播状态 JSON（`VodPlaybackState` 编码结果）。
|
||
var dataJson: String = ""
|
||
/// 最近播放时间。
|
||
var updateTime: Date = Date()
|
||
|
||
init(vodId: String, vodName: String, vodPic: String, sourceKey: String, playNote: String = "") {
|
||
self.bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
self.vodId = vodId
|
||
self.vodName = vodName
|
||
self.vodPic = vodPic
|
||
self.sourceKey = sourceKey
|
||
self.playNote = playNote
|
||
self.updateTime = Date()
|
||
}
|
||
}
|
||
|
||
/// 通用缓存
|
||
@Model
|
||
final class CacheItem {
|
||
/// 唯一缓存键。
|
||
@Attribute(.unique) var key: String = ""
|
||
/// 缓存值（字符串形式）。
|
||
var value: String = ""
|
||
/// 更新时间。
|
||
var updateTime: Date = Date()
|
||
|
||
init(key: String, value: String) {
|
||
self.key = key
|
||
self.value = value
|
||
self.updateTime = Date()
|
||
}
|
||
}
|
||
|
||
/// 缓存管理器
|
||
actor CacheStore {
|
||
static let shared = CacheStore()
|
||
|
||
private init() {}
|
||
|
||
@MainActor
|
||
func addCollect(_ video: Movie.Video, context: ModelContext) {
|
||
let vodId = video.id
|
||
let sourceKey = video.sourceKey
|
||
let bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
|
||
do {
|
||
let matched = try fetchCollects(vodId: vodId, sourceKey: sourceKey, context: context)
|
||
if let first = matched.first {
|
||
first.bizKey = bizKey
|
||
first.vodName = video.name
|
||
first.vodPic = video.pic
|
||
first.updateTime = Date()
|
||
for duplicate in matched.dropFirst() {
|
||
context.delete(duplicate)
|
||
}
|
||
} else {
|
||
let collect = VodCollect(
|
||
vodId: vodId,
|
||
vodName: video.name,
|
||
vodPic: video.pic,
|
||
sourceKey: sourceKey
|
||
)
|
||
context.insert(collect)
|
||
}
|
||
try context.save()
|
||
} catch {
|
||
print("写入收藏失败: \(error)")
|
||
}
|
||
}
|
||
|
||
@MainActor
|
||
func removeCollect(vodId: String, sourceKey: String, context: ModelContext) {
|
||
do {
|
||
let items = try fetchCollects(vodId: vodId, sourceKey: sourceKey, context: context)
|
||
guard !items.isEmpty else { return }
|
||
for item in items {
|
||
context.delete(item)
|
||
}
|
||
try context.save()
|
||
} catch {
|
||
print("删除收藏失败: \(error)")
|
||
}
|
||
}
|
||
|
||
@MainActor
|
||
func isCollected(vodId: String, sourceKey: String, context: ModelContext) -> Bool {
|
||
let bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
let predicate = #Predicate<VodCollect> { item in
|
||
item.bizKey == bizKey || (item.bizKey == "" && item.vodId == vodId && item.sourceKey == sourceKey)
|
||
}
|
||
let descriptor = FetchDescriptor<VodCollect>(predicate: predicate)
|
||
|
||
do {
|
||
let count = try context.fetchCount(descriptor)
|
||
return count > 0
|
||
} catch {
|
||
print("查询收藏状态失败: \(error)")
|
||
return false
|
||
}
|
||
}
|
||
|
||
@MainActor
|
||
func addRecord(
|
||
_ video: Movie.Video,
|
||
playNote: String,
|
||
playbackState: VodPlaybackState? = nil,
|
||
context: ModelContext
|
||
) {
|
||
let vodId = video.id
|
||
let sourceKey = video.sourceKey
|
||
let encodedState = Self.encodePlaybackState(playbackState)
|
||
let bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
|
||
do {
|
||
let matched = try fetchRecords(vodId: vodId, sourceKey: sourceKey, context: context)
|
||
|
||
// 更新或插入
|
||
if let record = matched.first {
|
||
record.bizKey = bizKey
|
||
record.playNote = playNote
|
||
if let encodedState {
|
||
record.dataJson = encodedState
|
||
}
|
||
record.updateTime = Date()
|
||
for duplicate in matched.dropFirst() {
|
||
context.delete(duplicate)
|
||
}
|
||
} else {
|
||
let record = VodRecord(
|
||
vodId: vodId,
|
||
vodName: video.name,
|
||
vodPic: video.pic,
|
||
sourceKey: sourceKey,
|
||
playNote: playNote
|
||
)
|
||
if let encodedState {
|
||
record.dataJson = encodedState
|
||
}
|
||
context.insert(record)
|
||
}
|
||
|
||
try context.save()
|
||
} catch {
|
||
print("写入播放记录失败: \(error)")
|
||
}
|
||
}
|
||
|
||
/// 读取续播状态（若无记录或 JSON 无法解码则返回 `nil`）。
|
||
@MainActor
|
||
func getPlaybackState(vodId: String, sourceKey: String, context: ModelContext) -> VodPlaybackState? {
|
||
do {
|
||
guard let record = try fetchRecords(vodId: vodId, sourceKey: sourceKey, context: context).first else {
|
||
return nil
|
||
}
|
||
return Self.decodePlaybackState(record.dataJson)
|
||
} catch {
|
||
print("读取续播状态失败: \(error)")
|
||
return nil
|
||
}
|
||
}
|
||
|
||
@MainActor
|
||
func clearHistory(context: ModelContext) {
|
||
do {
|
||
try context.delete(model: VodRecord.self)
|
||
try context.save()
|
||
} catch {
|
||
print("清空历史记录失败: \(error)")
|
||
}
|
||
}
|
||
|
||
@MainActor
|
||
private func fetchRecords(vodId: String, sourceKey: String, context: ModelContext) throws -> [VodRecord] {
|
||
let bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
let predicate = #Predicate<VodRecord> { item in
|
||
item.bizKey == bizKey || (item.bizKey == "" && item.vodId == vodId && item.sourceKey == sourceKey)
|
||
}
|
||
let descriptor = FetchDescriptor<VodRecord>(predicate: predicate)
|
||
let records = try context.fetch(descriptor)
|
||
|
||
// 兼容旧数据：命中 legacy 记录时补写业务键。
|
||
var needsSave = false
|
||
for record in records where record.bizKey.isEmpty {
|
||
record.bizKey = bizKey
|
||
needsSave = true
|
||
}
|
||
if needsSave {
|
||
try context.save()
|
||
}
|
||
|
||
return records.sorted(by: { $0.updateTime > $1.updateTime })
|
||
}
|
||
|
||
@MainActor
|
||
private func fetchCollects(vodId: String, sourceKey: String, context: ModelContext) throws -> [VodCollect] {
|
||
let bizKey = makeVodBusinessKey(vodId: vodId, sourceKey: sourceKey)
|
||
let predicate = #Predicate<VodCollect> { item in
|
||
item.bizKey == bizKey || (item.bizKey == "" && item.vodId == vodId && item.sourceKey == sourceKey)
|
||
}
|
||
let descriptor = FetchDescriptor<VodCollect>(predicate: predicate)
|
||
let collects = try context.fetch(descriptor)
|
||
|
||
// 兼容旧数据：命中 legacy 记录时补写业务键。
|
||
var needsSave = false
|
||
for collect in collects where collect.bizKey.isEmpty {
|
||
collect.bizKey = bizKey
|
||
needsSave = true
|
||
}
|
||
if needsSave {
|
||
try context.save()
|
||
}
|
||
|
||
return collects.sorted(by: { $0.updateTime > $1.updateTime })
|
||
}
|
||
|
||
private nonisolated static func encodePlaybackState(_ state: VodPlaybackState?) -> String? {
|
||
guard let state else { return nil }
|
||
guard let data = try? JSONEncoder().encode(state) else { return nil }
|
||
return String(data: data, encoding: .utf8)
|
||
}
|
||
|
||
/// 从 JSON 字符串反序列化续播状态。
|
||
private nonisolated static func decodePlaybackState(_ json: String) -> VodPlaybackState? {
|
||
guard let data = json.data(using: .utf8) else { return nil }
|
||
return try? JSONDecoder().decode(VodPlaybackState.self, from: data)
|
||
}
|
||
}
|