Add reaction

Author: dena-sohrabi

apple/InlineIOS/Chat/UIMessageView.swift

@@ -2,6 +2,12 @@ import InlineKit
 import SwiftUI
 import UIKit
 
+struct Reaction2: Equatable {
+  let emoji: String
+  var count: Int
+  var hasReacted: Bool
+}
+
 class UIMessageView: UIView {
   // MARK: - Properties
 
@@ -40,19 +46,26 @@ class UIMessageView: UIView {
     return stack
   }()
 
+  private var reactions: [Reaction2] = []
+  private let reactionHeight: CGFloat = 24
+
+  private lazy var reactionsContainer: UIStackView = {
+    let stack = UIStackView()
+    stack.axis = .horizontal
+    stack.spacing = 4
+    stack.alignment = .center
+    stack.distribution = .fillProportionally
+    stack.translatesAutoresizingMaskIntoConstraints = false
+    return stack
+  }()
+
   private var leadingConstraint: NSLayoutConstraint?
   private var trailingConstraint: NSLayoutConstraint?
   private var fullMessage: FullMessage
 
   private let horizontalPadding: CGFloat = 12
   private let verticalPadding: CGFloat = 8
 
-//  private let embedView: MessageEmbedView = {
-//    let view = MessageEmbedView(repliedToMessage: nil)
-//    view.translatesAutoresizingMaskIntoConstraints = false
-//    return view
-//  }()
-
   // MARK: - Initialization
 
   init(fullMessage: FullMessage) {
@@ -76,12 +89,6 @@ class UIMessageView: UIView {
 
     bubbleView.addSubview(contentStack)
 
-    // Add embed view if there's a reply
-//    if let replyMessage = fullMessage.repliedToMessage {
-//      contentStack.addArrangedSubview(embedView)
-//      embedView.repliedToMessage = replyMessage
-//    }
-
     leadingConstraint = bubbleView.leadingAnchor.constraint(equalTo: leadingAnchor, constant: 8)
     trailingConstraint = bubbleView.trailingAnchor.constraint(equalTo: trailingAnchor, constant: -8)
 
@@ -112,12 +119,6 @@ class UIMessageView: UIView {
     contentStack.arrangedSubviews.forEach { $0.removeFromSuperview() }
     shortMessageStack.arrangedSubviews.forEach { $0.removeFromSuperview() }
 
-    // Add embed view first if there's a reply
-//    if let replyMessage = fullMessage.repliedToMessage {
-//      contentStack.addArrangedSubview(embedView)
-//      embedView.repliedToMessage = replyMessage
-//    }
-
     let messageLength = fullMessage.message.text?.count ?? 0
     let messageText = fullMessage.message.text ?? ""
     let hasLineBreak = messageText.contains("\n")
@@ -147,10 +148,16 @@ class UIMessageView: UIView {
       shortMessageStack.addArrangedSubview(metadataView)
       contentStack.addArrangedSubview(shortMessageStack)
     }
+
+    // Add reactions container if there are reactions
+    if !reactions.isEmpty {
+      contentStack.addArrangedSubview(reactionsContainer)
+    }
   }
 
   private func configureForMessage() {
-    messageLabel.text = fullMessage.message.text
+    messageLabel.text =
+      "\(fullMessage.message.text) \(fullMessage.message.messageId) \(fullMessage.message.chatId)"
 
     if fullMessage.message.out == true {
       bubbleView.backgroundColor = ColorManager.shared.selectedColor
@@ -188,15 +195,77 @@ class UIMessageView: UIView {
     super.layoutSubviews()
   }
 
-  //  override func traitCollectionDidChange(_ previousTraitCollection: UITraitCollection?) {
-  //    super.traitCollectionDidChange(previousTraitCollection)
-  //
-  //    if previousTraitCollection?.preferredContentSizeCategory
-  //      != traitCollection.preferredContentSizeCategory
-  //    {
-  //      setNeedsLayout()
-  //    }
-  //  }
+  // MARK: - Reaction2 Handling
+
+  private func handleReaction2(_ emoji: String) {
+    if let index = reactions.firstIndex(where: { $0.emoji == emoji }) {
+      // Toggle existing reaction
+      var reaction = reactions[index]
+      reaction.hasReacted.toggle()
+      reaction.count += reaction.hasReacted ? 1 : -1
+
+      if reaction.count > 0 {
+        reactions[index] = reaction
+      } else {
+        reactions.remove(at: index)
+      }
+    } else {
+      // Add new reaction
+      reactions.append(Reaction2(emoji: emoji, count: 1, hasReacted: true))
+    }
+
+    updateReaction2Views()
+  }
+
+  private func updateReaction2Views() {
+    // Clear existing reaction views
+    reactionsContainer.arrangedSubviews.forEach { $0.removeFromSuperview() }
+
+    // Add reaction bubbles
+    for reaction in reactions {
+      let reactionView = createReaction2Bubble(for: reaction)
+      reactionsContainer.addArrangedSubview(reactionView)
+    }
+
+    updateMetadataLayout()
+  }
+
+  private func createReaction2Bubble(for reaction: Reaction2) -> UIView {
+    let container = UIView()
+    container.backgroundColor =
+      reaction.hasReacted ? ColorManager.shared.selectedColor.withAlphaComponent(0.1) : .systemGray6
+    container.layer.cornerRadius = reactionHeight / 2
+
+    let label = UILabel()
+    label.text = "\(reaction.emoji) \(reaction.count)"
+    label.font = .systemFont(ofSize: 12)
+    label.textAlignment = .center
+
+    container.addSubview(label)
+    label.translatesAutoresizingMaskIntoConstraints = false
+
+    NSLayoutConstraint.activate([
+      container.heightAnchor.constraint(equalToConstant: reactionHeight),
+      label.leadingAnchor.constraint(equalTo: container.leadingAnchor, constant: 8),
+      label.trailingAnchor.constraint(equalTo: container.trailingAnchor, constant: -8),
+      label.centerYAnchor.constraint(equalTo: container.centerYAnchor),
+    ])
+
+    // Add tap gesture
+    let tapGesture = UITapGestureRecognizer(target: self, action: #selector(reactionTapped(_:)))
+    container.addGestureRecognizer(tapGesture)
+    container.tag = reactions.firstIndex(where: { $0.emoji == reaction.emoji }) ?? 0
+
+    return container
+  }
+
+  @objc private func reactionTapped(_ gesture: UITapGestureRecognizer) {
+    guard let view = gesture.view,
+          reactions.indices.contains(view.tag)
+    else { return }
+
+    handleReaction2(reactions[view.tag].emoji)
+  }
 }
 
 // MARK: - Context Menu
@@ -206,18 +275,35 @@ extension UIMessageView: UIContextMenuInteractionDelegate {
     _ interaction: UIContextMenuInteraction,
     configurationForMenuAtLocation location: CGPoint
   ) -> UIContextMenuConfiguration? {
-    return UIContextMenuConfiguration(identifier: nil, previewProvider: nil) { _ in
+    return UIContextMenuConfiguration(identifier: nil, previewProvider: nil) { [weak self] _ in
       let copyAction = UIAction(title: "Copy") { [weak self] _ in
         UIPasteboard.general.string = self?.fullMessage.message.text
       }
 
       let replyAction = UIAction(title: "Reply") { [weak self] _ in
         ChatState.shared.setReplyingMessageId(
-          chatId: self?.fullMessage.message.chatId ?? 0, id: self?.fullMessage.message.id ?? 0
+          chatId: self?.fullMessage.message.chatId ?? 0,
+          id: self?.fullMessage.message.id ?? 0
         )
       }
 
-      return UIMenu(children: [copyAction, replyAction])
+      // Create reaction submenu
+      let reactionActions = self?.createReaction2Actions() ?? []
+      let reactMenu = UIMenu(
+        title: "React", image: UIImage(systemName: "face.smiling"), children: reactionActions
+      )
+
+      return UIMenu(children: [copyAction, replyAction, reactMenu])
+    }
+  }
+
+  private func createReaction2Actions() -> [UIAction] {
+    let commonEmojis = ["👍", "❤️", "😂", "🎉", "🤔", "👀"]
+
+    return commonEmojis.map { emoji in
+      UIAction(title: emoji) { [weak self] _ in
+        self?.handleReaction2(emoji)
+      }
     }
   }
 }

apple/InlineKit/Sources/InlineKit/ApiClient.swift

@@ -40,6 +40,7 @@ public enum Path: String {
   case savePushNotification
   case updateStatus
   case sendComposeAction
+  case addReaction
 }
 
 public final class ApiClient: ObservableObject, @unchecked Sendable {
@@ -92,7 +93,7 @@ public final class ApiClient: ObservableObject, @unchecked Sendable {
       }
 
       switch httpResponse.statusCode {
-      case 200 ... 299:
+      case 200...299:
         let apiResponse = try decoder.decode(APIResponse<T>.self, from: data)
         switch apiResponse {
         case let .success(data):
@@ -254,7 +255,7 @@ public final class ApiClient: ObservableObject, @unchecked Sendable {
     repliedToMessageId: Int64?
   ) async throws -> SendMessage {
     var queryItems: [URLQueryItem] = [
-      URLQueryItem(name: "text", value: text),
+      URLQueryItem(name: "text", value: text)
     ]
 
     if let peerUserId = peerUserId {
@@ -299,7 +300,7 @@ public final class ApiClient: ObservableObject, @unchecked Sendable {
     try await request(
       .savePushNotification,
       queryItems: [
-        URLQueryItem(name: "applePushToken", value: pushToken),
+        URLQueryItem(name: "applePushToken", value: pushToken)
 
       ],
       includeToken: true
@@ -310,7 +311,7 @@ public final class ApiClient: ObservableObject, @unchecked Sendable {
     try await request(
       .updateStatus,
       queryItems: [
-        URLQueryItem(name: "online", value: online ? "true" : "false"),
+        URLQueryItem(name: "online", value: online ? "true" : "false")
       ],
       includeToken: true
     )
@@ -332,6 +333,20 @@ public final class ApiClient: ObservableObject, @unchecked Sendable {
       includeToken: true
     )
   }
+
+  public func addReaction(messageId: Int64, chatId: Int64, emoji: String) async throws
+    -> AddReaction
+  {
+    try await request(
+      .addReaction,
+      queryItems: [
+        URLQueryItem(name: "messageId", value: "\(messageId)"),
+        URLQueryItem(name: "chatId", value: "\(chatId)"),
+        URLQueryItem(name: "emoji", value: emoji),
+      ],
+      includeToken: true
+    )
+  }
 }
 
 /// Example
@@ -448,6 +463,10 @@ public struct SendMessage: Codable, Sendable {
   public let message: ApiMessage
 }
 
+public struct AddReaction: Codable, Sendable {
+  public let reaction: ApiReaction
+}
+
 public struct GetDialogs: Codable, Sendable {
   // Threads
   public let chats: [ApiChat]

apple/InlineKit/Sources/InlineKit/DataManager.swift

@@ -296,7 +296,7 @@ public class DataManager: ObservableObject {
 
           var chat = Chat(from: chat)
           // to avoid foriegn key constraint
-          chat.lastMsgId = nil // TODO: fix
+          chat.lastMsgId = nil  // TODO: fix
 
           return chat
         }
@@ -463,6 +463,17 @@ public class DataManager: ObservableObject {
     }
   }
 
+  public func addReaction(messageId: Int64, chatId: Int64, emoji: String) async throws {
+    let result = try await ApiClient.shared.addReaction(
+      messageId: messageId, chatId: chatId, emoji: emoji)
+
+    try await database.dbWriter.write { db in
+      let reaction = Reaction(from: result.reaction)
+      try reaction.save(db, onConflict: .replace)
+      print("saved reaction: \(reaction)")
+    }
+  }
+
   public func updateStatus(online: Bool) async throws {
     log.debug("updateStatus")
     let _ = try await ApiClient.shared.updateStatus(online: online)

apple/InlineKit/Sources/InlineKit/Database.swift

@@ -98,6 +98,27 @@ public extension AppDatabase {
         t.column("readOutboxMaxId", .integer)
         t.column("pinned", .boolean)
       }
+
+      try db.create(table: "reaction") { t in
+        t.primaryKey("id", .integer).notNull().unique()
+        t.column("messageId", .integer)
+          .references(
+            "message", column: "messageId", onDelete: .cascade)
+          .notNull()
+          .unique()
+        t.column("userId", .integer)
+          .references("user", column: "id", onDelete: .cascade)
+          .notNull()
+          .unique()
+        t.column("chatId", .integer)
+          .references("chat", column: "id", onDelete: .cascade)
+          .notNull()
+          .unique()
+        t.column("emoji", .text)
+          .notNull()
+
+        t.column("date", .datetime).notNull()
+      }
     }
 
     migrator.registerMigration("v2") { db in

apple/InlineKit/Sources/InlineKit/Models/Message.swift

@@ -77,11 +77,17 @@ public struct Message: FetchableRecord, Identifiable, Codable, Hashable, Persist
     request(for: Message.from)
   }
 
-  public static let repliedToMessage = belongsTo(Message.self, key: "repliedToMessage", using: ForeignKey(["messageId"]))
+  public static let repliedToMessage = belongsTo(
+    Message.self, key: "repliedToMessage", using: ForeignKey(["messageId"]))
   public var repliedToMessage: QueryInterfaceRequest<Message> {
     request(for: Message.repliedToMessage)
   }
 
+  public static let reactions = hasMany(Reaction.self)
+  public var reactions: QueryInterfaceRequest<Reaction> {
+    request(for: Message.reactions)
+  }
+
   public init(
     messageId: Int64,
     randomId: Int64? = nil,
@@ -148,8 +154,8 @@ public struct Message: FetchableRecord, Identifiable, Codable, Hashable, Persist
 
 // MARK: Helpers
 
-public extension Message {
-  mutating func saveMessage(_ db: Database, onConflict: Database.ConflictResolution = .abort)
+extension Message {
+  public mutating func saveMessage(_ db: Database, onConflict: Database.ConflictResolution = .abort)
     throws
   {
     if self.globalId == nil {
@@ -165,7 +171,7 @@ public extension Message {
 
       if let existing =
         try? Message
-          .fetchOne(db, key: ["messageId": self.messageId, "chatId": self.chatId])
+        .fetchOne(db, key: ["messageId": self.messageId, "chatId": self.chatId])
       {
         self.globalId = existing.globalId
       }