vnidrop/apple/VniDrop/Core/TransferProgress.swift

import Foundation

/// Window size classes, ported from `AppUiModels.kt`. Thresholds are in points,
/// matching the Compose dp thresholds.
enum WindowClass {
	case phone
	case tablet
	case desktop
}

func windowClassFor(width: Double) -> WindowClass {
	if width >= 920 { return .desktop }
	if width >= 600 { return .tablet }
	return .phone
}

/// A progress snapshot derived from core events. `label` is a localization key
/// resolved at the view layer.
struct TransferProgress: Equatable {
	let transferId: UInt64?
	let phase: EventPhase
	let kind: EventKind
	let labelKey: String.LocalizationValue
	let progress: Double?
	var detail: String? = nil
	/// Pre-resolved label that overrides `labelKey` when set (e.g. "Sending to 2",
	/// which needs a runtime count).
	var label: String? = nil
}

func statusLabelKey(_ status: TransferStatus) -> String.LocalizationValue {
	switch status {
	case .importing: return L10n.Status.preparing
	case .sharing: return L10n.Status.available
	case .receiving: return L10n.Status.receiving
	case .done: return L10n.Status.completed
	case .cancelled: return L10n.Status.cancelled
	case .stopped: return L10n.Status.stopped
	case .failed: return L10n.Status.failed
	}
}

/// Latest progress snapshot for a transfer. Events are newest-first. Only events
/// whose `phase` and `kind` map to known cases participate.
func progressForTransfer(events: [CoreEventModel], transferId: UInt64) -> TransferProgress? {
	let relevant = events.filter { event in
		event.transferId == transferId && event.eventPhase != nil && event.eventKind != nil
	}
	guard let latest = relevant.first, let phase = latest.eventPhase, let kind = latest.eventKind else { return nil }
	let sizeHint = findKnownSize(events: events, transferId: transferId)
	return TransferProgress(
		transferId: transferId,
		phase: phase,
		kind: kind,
		labelKey: humanProgressLabel(phase: phase, kind: kind),
		progress: parseProgress(latest.dataJson, sizeHint: sizeHint),
		detail: progressDetail(latest)
	)
}

/// Live byte progress for one receiver on an outgoing share.
func progressForReceiver(
	events: [CoreEventModel],
	transferId: UInt64,
	remoteEndpointId: String,
	totalSizeHint: UInt64? = nil
) -> TransferProgress? {
	if remoteEndpointId.isEmpty { return nil }
	let connectionIds = connectionIdsForEndpoint(events: events, remoteEndpointId: remoteEndpointId)
	let receiverKinds: Set<EventKind> = [.started, .progress, .completed, .aborted]
	let transferEvents = events.filter { event in
		event.transferId == transferId
			&& event.eventDirection == .send
			&& event.eventPhase == .transfer
			&& (event.eventKind.map(receiverKinds.contains) ?? false)
			&& eventBelongsToReceiver(event, remoteEndpointId: remoteEndpointId, connectionIds: connectionIds)
	}
	guard let latest = transferEvents.first, let latestKind = latest.eventKind else { return nil }
	if latestKind == .aborted {
		return TransferProgress(
			transferId: transferId, phase: .transfer, kind: .aborted,
			labelKey: L10n.Progress.interrupted, progress: nil, detail: nil
		)
	}
	if latestKind == .completed && !transferEvents.contains(where: { $0.eventKind == .progress || $0.eventKind == .started }) {
		return TransferProgress(
			transferId: transferId, phase: .transfer, kind: .completed,
			labelKey: L10n.Progress.completed, progress: 1, detail: nil
		)
	}
	let progress = aggregateReceiverProgress(events: transferEvents, totalSizeHint: totalSizeHint)

	return TransferProgress(
		transferId: transferId, phase: .transfer, kind: latestKind,
		labelKey: L10n.Progress.sending, progress: progress, detail: progressDetail(latest)
	)
}

func formatBytes(_ size: UInt64) -> String {
	var scaled = Double(size)
	let units = ["B", "KB", "MB", "GB", "TB"]
	var unitIndex = 0
	while scaled >= 1024 && unitIndex < units.count - 1 {
		scaled /= 1024
		unitIndex += 1
	}
	if unitIndex == 0 {
		return "\(size) \(units[unitIndex])"
	}
	let rounded = (scaled * 10).rounded() / 10
	return "\(rounded) \(units[unitIndex])"
}

// MARK: - Internals (ported literally from AppUiModels.kt)

private func humanProgressLabel(phase: EventPhase, kind: EventKind) -> String.LocalizationValue {
	switch (phase, kind) {
	case (.importing, .copyProgress), (.importing, .outboardProgress), (.importing, .started):
		return L10n.Progress.preparing
	case (.importing, .done): return L10n.Progress.ready
	case (.ticket, .created): return L10n.Progress.shareReady
	case (.network, .connecting): return L10n.Progress.connecting
	case (.network, .connected): return L10n.Progress.connected
	case (.download, .foundCollection): return L10n.Progress.gettingReady
	case (.download, .progress): return L10n.Progress.downloading
	case (.export, .progress): return L10n.Progress.saving
	case (.transfer, .progress): return L10n.Progress.sending
	case (.transfer, .started): return L10n.Progress.connected
	case (.transfer, .completed): return L10n.Progress.completed
	case (.lifecycle, .done): return L10n.Progress.completed
	case (.lifecycle, .cancelled): return L10n.Progress.cancelled
	default:
		if phase == .handshake { return L10n.Progress.requestingAccess }
		if kind == .failed { return L10n.Progress.failed }
		return L10n.Progress.working
	}
}

private func progressDetail(_ event: CoreEventModel) -> String? {
	let fileName = findString(event.dataJson, key: "file_name")
	let current = findNumber(event.dataJson, key: "current_file_index").map { Int64($0) }
	let totalFiles = findNumber(event.dataJson, key: "total_files").map { Int64($0) }
	if let fileName, let current, let totalFiles, totalFiles > 0 {
		return "\(fileName) (\(current + 1)/\(totalFiles))"
	}
	return fileName
}

func parseProgress(_ json: String, sizeHint: Double? = nil) -> Double? {
	let transferred = findNumber(json, key: "exported")
		?? findNumber(json, key: "downloaded")
		?? findNumber(json, key: "offset")
		?? findNumber(json, key: "end_offset")
		?? findNumber(json, key: "transferred")
		?? findNumber(json, key: "written")
	let total = findNumber(json, key: "file_size")
		?? findNumber(json, key: "total_size")
		?? findNumber(json, key: "size")
		?? findNumber(json, key: "total")
		?? sizeHint
	guard let transferred, let total, total > 0 else { return nil }
	return min(1, max(0, transferred / total))
}

private func findKnownSize(events: [CoreEventModel], transferId: UInt64) -> Double? {
	for event in events where event.transferId == transferId {
		if let s = findNumber(event.dataJson, key: "size"), s > 0 { return s }
		if let s = findNumber(event.dataJson, key: "total_size"), s > 0 { return s }
		if let s = findNumber(event.dataJson, key: "file_size"), s > 0 { return s }
	}
	return nil
}

private final class BlobState {
	var size: Double?
	var offset: Double = 0
	var completed = false
	var aborted = false
}

private func aggregateReceiverProgress(events: [CoreEventModel], totalSizeHint: UInt64?) -> Double? {
	let chronological = events.reversed()
	var byRequest = [String: BlobState]()
	var order = [String]()
	var connectionScopedOffset: Double?
	var connectionScopedSize: Double?

	for event in chronological {
		let requestKey = findNumber(event.dataJson, key: "request_id").map { String(Int64($0)) }
			?? findString(event.dataJson, key: "request_id")
		let size = findNumber(event.dataJson, key: "size")
		let endOffset = findNumber(event.dataJson, key: "end_offset")
			?? findNumber(event.dataJson, key: "offset")
			?? findNumber(event.dataJson, key: "transferred")

		if let requestKey {
			let state: BlobState
			if let existing = byRequest[requestKey] {
				state = existing
			} else {
				state = BlobState()
				byRequest[requestKey] = state
				order.append(requestKey)
			}
			if let size, size > 0 { state.size = size }
			switch event.eventKind {
			case .progress, .started:
				if let endOffset { state.offset = max(state.offset, endOffset) }
				state.aborted = false
			case .completed:
				state.completed = true
				if let s = state.size { state.offset = s }
			case .aborted:
				state.aborted = true
			default:
				break
			}
		} else {
			if let size, size > 0 { connectionScopedSize = size }
			if let endOffset { connectionScopedOffset = endOffset }
		}
	}

	if !byRequest.isEmpty {
		let active = order.compactMap { byRequest[$0] }.filter { !$0.aborted }
		if active.isEmpty { return nil }
		let transferred = active.reduce(0.0) { acc, state in
			acc + (state.completed ? (state.size ?? state.offset) : state.offset)
		}
		let observedSize = active.compactMap { $0.size }.reduce(0, +)
		let total = totalSizeHint.map { Double($0) }.flatMap { $0 > 0 ? $0 : nil }
			?? (observedSize > 0 ? observedSize : nil)
		guard let total, total > 0 else { return nil }
		return min(1, max(0, transferred / total))
	}

	let total = totalSizeHint.map { Double($0) }.flatMap { $0 > 0 ? $0 : nil }
		?? connectionScopedSize.flatMap { $0 > 0 ? $0 : nil }
	guard let transferred = connectionScopedOffset, let total, total > 0 else { return nil }
	return min(1, max(0, transferred / total))
}

private func connectionIdsForEndpoint(events: [CoreEventModel], remoteEndpointId: String) -> Set<String> {
	var ids = Set<String>()
	for event in events {
		guard let endpoint = findString(event.dataJson, key: "endpoint_id"), endpoint == remoteEndpointId else { continue }
		if let n = findNumber(event.dataJson, key: "connection_id") { ids.insert(String(Int64(n))) }
		if let s = findString(event.dataJson, key: "connection_id") { ids.insert(s) }
	}
	return ids
}

private func eventBelongsToReceiver(_ event: CoreEventModel, remoteEndpointId: String, connectionIds: Set<String>) -> Bool {
	if let endpoint = findString(event.dataJson, key: "endpoint_id") {
		return endpoint == remoteEndpointId
	}
	let connectionId = findNumber(event.dataJson, key: "connection_id").map { String(Int64($0)) }
		?? findString(event.dataJson, key: "connection_id")
	guard let connectionId else { return false }
	return connectionIds.contains(connectionId)
}

// Lightweight JSON scraping, ported from AppUiModels.kt (matches core event shapes).
func findNumber(_ json: String, key: String) -> Double? {
	let marker = "\"\(key)\":"
	guard let range = json.range(of: marker) else { return nil }
	let after = json[range.upperBound...].drop { $0 == " " }
	if after.hasPrefix("null") { return nil }
	let terminators: Set<Character> = [",", "}", "]"]
	var value = ""
	for ch in json[range.upperBound...] {
		if terminators.contains(ch) { break }
		value.append(ch)
	}
	let trimmed = value.trimmingCharacters(in: .whitespaces).trimmingCharacters(in: CharacterSet(charactersIn: "\""))
	return Double(trimmed)
}

func findString(_ json: String, key: String) -> String? {
	let marker = "\"\(key)\":"
	guard let range = json.range(of: marker) else { return nil }
	let after = json[range.upperBound...].drop { $0 == " " }
	if after.hasPrefix("null") { return nil }
	guard after.hasPrefix("\"") else { return nil }
	let content = after.dropFirst()
	guard let endIdx = content.firstIndex(of: "\"") else { return nil }
	if content.startIndex == endIdx { return nil }
	return String(content[content.startIndex..<endIdx])
}