kangkang/康康/Models/Models.swift

import Foundation
import SwiftData

enum IndicatorStatus: String, Codable, CaseIterable {
    case high, low, normal
}

enum ReportType: String, Codable, CaseIterable {
    case checkup, lab, imaging, prescription, other

    var label: String {
        switch self {
        case .checkup:      return String(appLoc: "体检报告")
        case .lab:          return String(appLoc: "化验单")
        case .imaging:      return String(appLoc: "影像报告")
        case .prescription: return String(appLoc: "处方")
        case .other:        return String(appLoc: "其他")
        }
    }
}

@Model
final class Indicator {
    var name: String
    var value: String
    var unit: String
    var range: String
    var statusRaw: String
    var note: String?
    var capturedAt: Date

    var report: Report?
    var asset: Asset?
    var pinned: Bool = false

    /// 长期指标系列 key,如 "bp.systolic" / "glucose.fasting" / "weight"。
    /// 来源:IndicatorRecordSheet 选预设时填;VL/Report/自由输入留 nil。
    /// 用途:Trends 按 seriesKey 分组;Timeline 配对(如 bp.systolic + bp.diastolic 合并)。
    var seriesKey: String?

    init(name: String,
         value: String,
         unit: String,
         range: String,
         status: IndicatorStatus,
         note: String? = nil,
         capturedAt: Date = .now,
         report: Report? = nil,
         asset: Asset? = nil,
         pinned: Bool = false,
         seriesKey: String? = nil) {
        self.name = name
        self.value = value
        self.unit = unit
        self.range = range
        self.statusRaw = status.rawValue
        self.note = note
        self.capturedAt = capturedAt
        self.report = report
        self.asset = asset
        self.pinned = pinned
        self.seriesKey = seriesKey
    }

    var status: IndicatorStatus {
        IndicatorStatus(rawValue: statusRaw) ?? .normal
    }
}

@Model
final class Report {
    var title: String
    var typeRaw: String
    var reportDate: Date
    var institution: String?
    var note: String?
    var summary: String?
    var pageCount: Int
    var createdAt: Date

    @Relationship(deleteRule: .cascade, inverse: \Indicator.report)
    var indicators: [Indicator] = []

    @Relationship(deleteRule: .cascade)
    var assets: [Asset] = []

    init(title: String,
         type: ReportType,
         reportDate: Date,
         institution: String? = nil,
         note: String? = nil,
         summary: String? = nil,
         pageCount: Int = 1,
         createdAt: Date = .now) {
        self.title = title
        self.typeRaw = type.rawValue
        self.reportDate = reportDate
        self.institution = institution
        self.note = note
        self.summary = summary
        self.pageCount = pageCount
        self.createdAt = createdAt
    }

    var type: ReportType {
        ReportType(rawValue: typeRaw) ?? .other
    }
}

@Model
final class DiaryEntry {
    var content: String
    var createdAt: Date
    var tags: [String]

    init(content: String, createdAt: Date = .now, tags: [String] = []) {
        self.content = content
        self.createdAt = createdAt
        self.tags = tags
    }
}

@Model
final class Asset {
    var relativePath: String
    var mimeType: String
    var bytes: Int
    var createdAt: Date

    init(relativePath: String,
         mimeType: String = "image/jpeg",
         bytes: Int = 0,
         createdAt: Date = .now) {
        self.relativePath = relativePath
        self.mimeType = mimeType
        self.bytes = bytes
        self.createdAt = createdAt
    }
}

@Model
final class Symptom {
    var name: String
    var startedAt: Date
    var endedAt: Date?
    var note: String?
    var severity: Int
    var tags: [String]
    var createdAt: Date

    init(name: String,
         startedAt: Date = .now,
         endedAt: Date? = nil,
         note: String? = nil,
         severity: Int = 3,
         tags: [String] = [],
         createdAt: Date = .now) {
        self.name = name
        self.startedAt = startedAt
        self.endedAt = endedAt
        self.note = note
        self.severity = max(1, min(5, severity))
        self.tags = tags
        self.createdAt = createdAt
    }

    var isOngoing: Bool { endedAt == nil }

    var duration: TimeInterval {
        (endedAt ?? .now).timeIntervalSince(startedAt)
    }
}

/// 用户自定义的长期监测指标。
/// 与 hardcoded `MonitorMetric` 并列出现在 IndicatorQuickSheet 的 grid 里;
/// `seriesKey` 自动生成成 `"custom.<uuid>"`,以此和 Indicator 双向关联。
@Model
final class CustomMonitorMetric {
    @Attribute(.unique) var seriesKey: String
    var name: String
    var unit: String
    var lowerBound: Double?
    var upperBound: Double?
    var icon: String
    var createdAt: Date

    init(name: String,
         unit: String,
         lowerBound: Double? = nil,
         upperBound: Double? = nil,
         icon: String = "circle.fill",
         createdAt: Date = .now) {
        self.seriesKey = "custom.\(UUID().uuidString)"
        self.name = name
        self.unit = unit
        self.lowerBound = lowerBound
        self.upperBound = upperBound
        self.icon = icon
        self.createdAt = createdAt
    }

    var referenceRange: ClosedRange<Double>? {
        guard let lo = lowerBound, let hi = upperBound, lo <= hi else { return nil }
        return lo...hi
    }

    var rangeText: String {
        guard let r = referenceRange else { return "" }
        return "\(Self.format(r.lowerBound)) - \(Self.format(r.upperBound))"
    }

    private static func format(_ v: Double) -> String {
        v.truncatingRemainder(dividingBy: 1) == 0
            ? String(format: "%.0f", v)
            : String(format: "%.1f", v)
    }
}

/// 长期监测指标的周期性记录提醒。
/// 一个 metric 一条(`metricId` = `MonitorMetric.rawValue`)。
/// 关闭通过 `enabled=false`(保留时间设置),删除走 `ctx.delete`。
@Model
final class MetricReminder {
    @Attribute(.unique) var metricId: String
    var displayName: String
    var enabled: Bool
    var hour: Int            // 0...23
    var minute: Int          // 0...59
    var weekdays: [Int]      // iOS Calendar 约定:1=日, 2=一, ..., 7=六。全 7 个 = 每天
    var createdAt: Date
    var updatedAt: Date

    init(metricId: String,
         displayName: String,
         hour: Int = 8,
         minute: Int = 0,
         weekdays: [Int] = [1, 2, 3, 4, 5, 6, 7],
         enabled: Bool = true,
         createdAt: Date = .now) {
        self.metricId = metricId
        self.displayName = displayName
        self.enabled = enabled
        self.hour = max(0, min(23, hour))
        self.minute = max(0, min(59, minute))
        self.weekdays = weekdays
        self.createdAt = createdAt
        self.updatedAt = createdAt
    }

    var isEveryDay: Bool { Set(weekdays) == Set(1...7) }

    var frequencyLabel: String {
        if !enabled { return String(appLoc: "已关闭") }
        if isEveryDay { return String(appLoc: "每天") }
        if weekdays.isEmpty { return String(appLoc: "未选日") }
        let names = [String(appLoc: "日"), String(appLoc: "一"), String(appLoc: "二"), String(appLoc: "三"), String(appLoc: "四"), String(appLoc: "五"), String(appLoc: "六")]
        let sorted = weekdays.sorted()
        return String(appLoc: "每周 ") + sorted.map { names[$0 - 1] }.joined()
    }

    var timeLabel: String {
        String(format: "%02d:%02d", hour, minute)
    }

    /// 这条指标提醒在给定日期「这天」是否会触发(weekday 制,全 7 = 每天);关闭则恒为 false。
    /// 供主页「今日提醒」筛选。
    func occurs(on date: Date, calendar: Calendar = .current) -> Bool {
        guard enabled else { return false }
        return weekdays.contains(calendar.component(.weekday, from: date))
    }
}

/// 自由文案的周期性提醒(如「每天 20:00 跑步 5 公里」「每天 12:30 吃 2 片护肝片」)。
/// 与 `MetricReminder`(去记录某指标)语义独立:这里是用户自定义的动作提醒,
/// 量词(5 公里 / 2 片)直接写在 `title` 自由文本里。
/// 周期粒度沿用 weekday 约定(全 7 = 每天);本地通知调度见 `ReminderService`。
@Model
final class CustomReminder {
    /// 周期粒度。每日只看时间;每周看 weekdays;每月看 dayOfMonth;每年看 month + dayOfMonth。
    enum Frequency: String, CaseIterable, Sendable {
        case daily, weekly, monthly, yearly
    }

    @Attribute(.unique) var id: UUID
    var title: String        // 用户文案,如 "跑步5公里"
    var note: String         // 可选备注 → 通知正文
    var hour: Int            // 0...23
    var minute: Int          // 0...59
    var weekdays: [Int]      // iOS Calendar 约定:1=日, 2=一, ..., 7=六。全 7 个 = 每天
    var frequencyRaw: String = "daily"   // CustomReminder.Frequency 原始值
    var dayOfMonth: Int = 1              // monthly / yearly 用,1...31
    var month: Int = 1                   // yearly 用,1...12
    var enabled: Bool
    var createdAt: Date
    var updatedAt: Date

    init(id: UUID = UUID(),
         title: String,
         note: String = "",
         hour: Int = 8,
         minute: Int = 0,
         weekdays: [Int] = [1, 2, 3, 4, 5, 6, 7],
         frequency: Frequency = .daily,
         dayOfMonth: Int = 1,
         month: Int = 1,
         enabled: Bool = true,
         createdAt: Date = .now) {
        self.id = id
        self.title = title
        self.note = note
        self.hour = max(0, min(23, hour))
        self.minute = max(0, min(59, minute))
        self.weekdays = weekdays
        self.frequencyRaw = frequency.rawValue
        self.dayOfMonth = max(1, min(31, dayOfMonth))
        self.month = max(1, min(12, month))
        self.enabled = enabled
        self.createdAt = createdAt
        self.updatedAt = createdAt
    }

    var isEveryDay: Bool { Set(weekdays) == Set(1...7) }

    var frequency: Frequency {
        get { Frequency(rawValue: frequencyRaw) ?? .daily }
        set { frequencyRaw = newValue.rawValue }
    }

    /// 列表行副标题:按频率展示「每天 / 每周 一三五 / 每月15日 / 每年3月15日」。
    var frequencyLabel: String {
        if !enabled { return String(appLoc: "已关闭") }
        switch frequency {
        case .daily:
            return String(appLoc: "每天")
        case .weekly:
            if isEveryDay { return String(appLoc: "每天") }
            if weekdays.isEmpty { return String(appLoc: "未选日") }
            let names = [String(appLoc: "日"), String(appLoc: "一"), String(appLoc: "二"), String(appLoc: "三"), String(appLoc: "四"), String(appLoc: "五"), String(appLoc: "六")]
            return String(appLoc: "每周 ") + weekdays.sorted().map { names[$0 - 1] }.joined()
        case .monthly:
            return String(appLoc: "每月\(dayOfMonth)日")
        case .yearly:
            return String(appLoc: "每年\(month)月\(dayOfMonth)日")
        }
    }

    var timeLabel: String {
        String(format: "%02d:%02d", hour, minute)
    }

    /// 这条提醒在给定日期「这天」是否会触发(只看哪天,不看时分);关闭则恒为 false。
    /// 供主页「今日提醒」筛选。monthly/yearly 选了无此日的月份(如 31 日)自然返回 false,
    /// 与 iOS「该月跳过、不顺延」的行为一致。
    func occurs(on date: Date, calendar: Calendar = .current) -> Bool {
        guard enabled else { return false }
        let c = calendar.dateComponents([.weekday, .day, .month], from: date)
        switch frequency {
        case .daily:   return true
        case .weekly:  return weekdays.contains(c.weekday ?? -1)
        case .monthly: return dayOfMonth == (c.day ?? -1)
        case .yearly:  return month == (c.month ?? -1) && dayOfMonth == (c.day ?? -1)
        }
    }
}

@Model
final class ChatTurn {
    var question: String
    var answer: String
    var referencedIndicatorIDs: [String]
    var referencedReportIDs: [String]
    var createdAt: Date
    var decodeRate: Double

    init(question: String,
         answer: String,
         referencedIndicatorIDs: [String] = [],
         referencedReportIDs: [String] = [],
         createdAt: Date = .now,
         decodeRate: Double = 0) {
        self.question = question
        self.answer = answer
        self.referencedIndicatorIDs = referencedIndicatorIDs
        self.referencedReportIDs = referencedReportIDs
        self.createdAt = createdAt
        self.decodeRate = decodeRate
    }
}