import SwiftUI
import UniformTypeIdentifiers

enum FilePreviewType: Equatable {
  case unknown
  case image
  case text
}

/// State for a file preview download
@MainActor
@Observable
final class FilePreviewState {
  enum LoadState: Equatable {
    case unloaded
    case loading
    case loaded
    case failed
  }
  
  let info: PreviewFileInfo

  var state: LoadState = .unloaded
  var progress: Double = 0.0

  var fileURL: URL? = nil

  #if os(iOS)
  var image: UIImage? = nil
  #elseif os(macOS)
  var image: NSImage? = nil
  #endif

  var text: String? = nil
  var styledText: NSAttributedString? = nil
  
  @ObservationIgnored private var previewClient: HotlineFilePreviewClient?
  @ObservationIgnored private var previewTask: Task<Void, Never>?

  var previewType: FilePreviewType {
    self.info.previewType
  }

  init(info: PreviewFileInfo) {
    self.info = info
  }

  // MARK: - API

  func download() {
    // Cancel any existing download
    self.previewTask?.cancel()
    self.previewClient?.cleanup()

    let task = Task { @MainActor in
      do {
        let client = HotlineFilePreviewClient(
          fileName: self.info.name,
          address: self.info.address,
          port: UInt16(self.info.port),
          reference: self.info.id,
          size: UInt32(self.info.size),
          fileType: self.info.type,
          fileCreator: self.info.creator
        )
        self.previewClient = client

        self.state = .loading
        self.progress = 0.0

        let url = try await client.preview { [weak self] progress in
          guard let self else { return }

          Task { @MainActor in
            switch progress {
            case .preparing:
              self.state = .loading
              self.progress = 0.0
              
            case .connecting:
              self.state = .loading
              self.progress = 0.0

            case .connected:
              self.state = .loading
              self.progress = 0.0

            case .transfer(name: _, size: _, total: _, progress: let p, speed: _, estimate: _):
              self.state = .loading
              self.progress = p

            case .completed(url: let url):
              self.state = .loaded
              self.progress = 1.0
              self.fileURL = url
//              self.loadPreview(from: url)

            case .error(let error):
              self.state = .failed
              self.progress = 0.0
              print("FilePreviewState: Download failed: \(error)")

            case .unconnected:
              break
            }
          }
        }

        // Final load if not already loaded
        if self.state != .loaded {
          self.state = .loaded
          self.progress = 1.0
          self.fileURL = url
//          self.loadPreview(from: url)
        }

      } catch is CancellationError {
        // Cancelled, do nothing
        return
      } catch {
        self.state = .failed
        self.progress = 0.0
        print("FilePreviewState: Download error: \(error)")
      }
    }

    self.previewTask = task
  }

  func cancel() {
    self.previewTask?.cancel()
    self.previewTask = nil
    self.previewClient?.cancel()
  }

  func cleanup() {
    self.previewClient?.cleanup()
    self.previewClient = nil
    self.fileURL = nil
    self.image = nil
    self.text = nil
    self.styledText = nil
  }

  // MARK: - Utility

//  private func loadPreview(from url: URL) {
//    guard let data = try? Data(contentsOf: url) else {
//      self.state = .failed
//      print("FilePreviewState: Failed to read preview data from \(url.path)")
//      return
//    }
//
//    switch self.previewType {
//    case .image:
//      #if os(iOS)
//      self.image = UIImage(data: data)
//      #elseif os(macOS)
//      self.image = NSImage(data: data)
//      #endif
//
//      if self.image == nil {
//        self.state = .failed
//        print("FilePreviewState: Failed to create image from data")
//      }
//
//    case .text:
//      let encoding: UInt = NSString.stringEncoding(for: data, convertedString: nil, usedLossyConversion: nil)
//      if encoding != 0 {
//        self.text = String(data: data, encoding: String.Encoding(rawValue: encoding))
//      } else {
//        self.text = String(data: data, encoding: .utf8)
//      }
//
//      if self.text == nil {
//        self.state = .failed
//        print("FilePreviewState: Failed to decode text data")
//      }
//
//    case .unknown:
//      print("FilePreviewState: Unknown preview type for \(info.name)")
//      break
//    }
//  }
}