IceCubesApp/Packages/Status/Sources/Status/Editor/StatusEditorViewModel.swift

import SwiftUI
import DesignSystem
import Models
import Network

@MainActor
class StatusEditorViewModel: ObservableObject {
  @Published var statusText = NSAttributedString(string: "") {
    didSet {
      guard !internalUpdate else { return }
      highlightMeta()
    }
  }
  
  var client: Client?
  private var internalUpdate: Bool = false
  private var inReplyTo: Status?
  
  init(inReplyTo: Status?) {
    self.inReplyTo = inReplyTo
  }
  
  func postStatus() async -> Status? {
    guard let client else { return nil }
    do {
      let status: Status = try await client.post(endpoint: Statuses.postStatus(status: statusText.string,
                                                                               inReplyTo: inReplyTo?.id,
                                                                               mediaIds: nil,
                                                                               spoilerText: nil))
      return status
    } catch {
      return nil
    }
  }
  
  func insertReplyTo() {
    if let inReplyTo {
      statusText = .init(string: "@\(inReplyTo.account.acct) ")
    }
  }
  
  func highlightMeta() {
    let mutableString = NSMutableAttributedString(attributedString: statusText)
    let hashtagPattern = "(#+[a-zA-Z0-9(_)]{1,})"
    let mentionPattern = "(@+[a-zA-Z0-9(_).]{1,})"
    var ranges: [NSRange] = [NSRange]()

    do {
      let hashtagRegex = try NSRegularExpression(pattern: hashtagPattern, options: [])
      let mentionRegex = try NSRegularExpression(pattern: mentionPattern, options: [])
      
      ranges = hashtagRegex.matches(in: mutableString.string,
                                    options: [],
                                    range: NSMakeRange(0, mutableString.string.utf16.count)).map { $0.range }
      ranges.append(contentsOf: mentionRegex.matches(in: mutableString.string,
                                                     options: [],
                                                     range: NSMakeRange(0, mutableString.string.utf16.count)).map {$0.range})

      for range in ranges {
        mutableString.addAttributes([.foregroundColor: UIColor(Color.brand)],
                                   range: NSRange(location: range.location, length: range.length))
      }
      internalUpdate = true
      statusText = mutableString
      internalUpdate = false
    } catch {
      
    }
  }
   
}