Merge pull request #15 from Nexters/feature/chat-messaging

[FEAT] 채팅 기능 추가

Author: cchanmi

GAMSS.xcodeproj/project.pbxproj

@@ -23,7 +23,7 @@ struct GAMSSApp: App {
 
     var body: some Scene {
         WindowGroup {
-            LoginView(viewModel: LoginViewModel(loginUseCase: DefaultLoginUseCase(authRepository: DefaultAuthRepository(networkManager: NetworkManager.shared, tokenStorage: TokenStorage.shared))))
+            MainTabView()
         }
     }
 }

GAMSS/Sources/App/GAMSSApp.swift

@@ -34,11 +34,18 @@ final class NetworkManager: NetworkRequesting {
     ) async throws -> T {
 
         let request = try endpoint.asURLRequest()
-        
+
+        let bodyString = request.httpBody.flatMap { String(data: $0, encoding: .utf8) } ?? "-"
+        Log.debug("📤 \(endpoint.method.rawValue) \(endpoint.path) body: \(bodyString)")
+
         let (data, response) = try await session.data(
             for: request
         )
-        
+
+        if let jsonString = String(data: data, encoding: .utf8) {
+            Log.debug("📦 \(endpoint.path) response: \(jsonString)")
+        }
+
         guard let response = response as? HTTPURLResponse else {
             throw NetworkError.invalidResponse
         }

GAMSS/Sources/Core/Network/NetworkManager.swift

@@ -8,9 +8,9 @@
 import Foundation
 
 struct CreateMessageRequestDTO: Encodable {
-    let conversationId: Int
+    let conversationId: Int?
     let content: String
-    let repliesMessageId: Int
-    let currentConversationSummary: String
+    let repliesToMessageId: Int?
+    let currentConversationSummary: String?
     let excludeCharacters: [String]
 }

GAMSS/Sources/Data/DTO/Chat/Request/CreateMessageRequestDTO.swift

@@ -0,0 +1,46 @@
+//
+//  ConversationMessageDTO.swift
+//  GAMSS
+//
+//  Created by cchanmi on 8/7/26.
+//
+
+import Foundation
+
+/// 서버 필드명과 1:1 (Android ConversationMessage.kt 기준). senderType이 "CHARACTER"일 때만
+/// emotionType이 채워진다. senderType이 알 수 없거나 emotionType을 못 알아보면 toDomain()이
+/// nil을 반환 — 잘못된 주체로 그리는 것보다 목록에서 제외하는 쪽을 택한다.
+struct ConversationMessageDTO: Decodable {
+    let id: Int
+    let conversationId: Int
+    let senderType: String
+    let emotionType: String?
+    let content: String
+    let repliesToMessageId: Int?
+    let rootMessageId: Int?
+    let createdAt: String
+
+    private static let serverTypeToCharacter: [String: EmotionCharacter] = [
+        "JOY": .joy, "ANGER": .anger, "ANXIETY": .anxiety,
+        "GRUMPY": .prickly, "WARM": .sadness, "QUIRKY": .quirky,
+    ]
+
+    func toDomain() -> Message? {
+        let sender: MessageSender
+        switch senderType {
+        case "USER":
+            sender = .user
+        case "CHARACTER":
+            guard let character = Self.serverTypeToCharacter[emotionType ?? ""] else { return nil }
+            sender = .character(character)
+        default:
+            return nil
+        }
+        return Message(id: id, conversationId: conversationId, sender: sender, content: content, repliesToMessageId: repliesToMessageId)
+    }
+
+    /// 방금 보낸 사용자 메시지 응답 전용 — 항상 .user로 간주한다.
+    func toSentUserMessage() -> Message {
+        Message(id: id, conversationId: conversationId, sender: .user, content: content, repliesToMessageId: repliesToMessageId)
+    }
+}

GAMSS/Sources/Data/DTO/ConversationMessageDTO.swift

@@ -0,0 +1,20 @@
+//
+//  ConversationSummaryDTO.swift
+//  GAMSS
+//
+//  Created by cchanmi on 8/8/26.
+//
+
+import Foundation
+
+struct ConversationSummaryDTO: Decodable {
+    let id: Int
+    let title: String?
+    let status: String
+    let createdAt: String
+    let updatedAt: String
+
+    func toDomain() -> ConversationSummary {
+        ConversationSummary(id: id, title: title, status: status, createdAt: createdAt)
+    }
+}

GAMSS/Sources/Data/DTO/ConversationSummaryDTO.swift

@@ -0,0 +1,32 @@
+//
+//  SaveMessageResponseDTO.swift
+//  GAMSS
+//
+//  Created by cchanmi on 8/7/26.
+//
+
+import Foundation
+
+struct SaveMessageResponseDTO: Decodable {
+    let message: ConversationMessageDTO
+    let commentStatus: String
+    let comments: [ConversationMessageDTO]
+
+    func toDomain() -> SentMessage {
+        SentMessage(
+            message: message.toSentUserMessage(),
+            commentStatus: commentStatus.toCommentGenerationStatus(),
+            comments: comments.compactMap { $0.toDomain() }
+        )
+    }
+}
+
+private extension String {
+    func toCommentGenerationStatus() -> CommentGenerationStatus {
+        switch self {
+        case "DONE": return .done
+        case "LIMIT_EXCEEDED": return .limitExceeded
+        default: return .failed
+        }
+    }
+}

GAMSS/Sources/Data/DTO/SaveMessageResponseDTO.swift

@@ -0,0 +1,45 @@
+//
+//  DefaultConversationRepository.swift
+//  GAMSS
+//
+//  Created by cchanmi on 8/7/26.
+//
+
+final class DefaultConversationRepository: ConversationRepository {
+    private let networkManager: NetworkRequesting
+
+    init(networkManager: NetworkRequesting) {
+        self.networkManager = networkManager
+    }
+
+    func sendMessage(conversationId: Int?, content: String, repliesToMessageId: Int?, contextSummary: String?) async throws -> SentMessage {
+        let request = CreateMessageRequestDTO(
+            conversationId: conversationId,
+            content: content,
+            repliesToMessageId: repliesToMessageId,
+            currentConversationSummary: contextSummary,
+            excludeCharacters: []
+        )
+        let response = try await networkManager.request(
+            ChatEndpoint.createMessage(request),
+            responseType: APIResponse<SaveMessageResponseDTO>.self
+        )
+        return response.data.toDomain()
+    }
+
+    func getMessages(conversationId: Int) async throws -> [Message] {
+        let response = try await networkManager.request(
+            ChatEndpoint.fetchMessages(chatId: String(conversationId)),
+            responseType: APIResponse<[ConversationMessageDTO]>.self
+        )
+        return response.data.compactMap { $0.toDomain() }
+    }
+
+    func getConversations(date: String) async throws -> [ConversationSummary] {
+        let response = try await networkManager.request(
+            ChatEndpoint.fetchChats(date: date),
+            responseType: APIResponse<[ConversationSummaryDTO]>.self
+        )
+        return response.data.map { $0.toDomain() }
+    }
+}

GAMSS/Sources/Data/Repository/DefaultConversationRepository.swift

@@ -1,8 +1,8 @@
 //
-//  DefaultDiarySummaryRepository.swift
+//  DefaultSummaryRepository.swift
 //  GAMSS
 //
-//  Created by cchanmi on 7/31/26.
+//  Created by cchanmi on 8/7/26.
 //
 
 import Foundation
@@ -12,7 +12,7 @@ import Tokenizers
 // actor로 선언해 encoder/decoder ORTSession과 tokenizer에 대한 동시 접근을 직렬화한다.
 // ONNX Runtime 세션은 스레드 세이프하지 않아서, summarize가 여러 곳에서 동시에
 // 호출되면 run() 호출들이 서로 레이스할 수 있다.
-actor DefaultDiarySummaryRepository: DiarySummaryRepository {
+actor DefaultSummaryRepository: SummaryRepository {
     // 모델이 이 값들에 맞춰 학습/export되어 있으므로 임의 변경 금지.
     private static let decoderStartToken = 1
     private static let eosToken = 1
@@ -40,12 +40,12 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
         self.tokenizer = tokenizer
     }
 
-    static func make() async throws -> DefaultDiarySummaryRepository {
+    static func make() async throws -> DefaultSummaryRepository {
         guard
             let encoderPath = Bundle.main.path(forResource: "kobart_encoder_int8", ofType: "onnx"),
             let decoderPath = Bundle.main.path(forResource: "kobart_decoder_int8", ofType: "onnx")
         else {
-            throw DiarySummaryError.modelLoadFailed()
+            throw SummaryError.modelLoadFailed()
         }
 
         let encoder: ORTSession
@@ -55,11 +55,11 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
             encoder = try ORTSession(env: env, modelPath: encoderPath, sessionOptions: nil)
             decoder = try ORTSession(env: env, modelPath: decoderPath, sessionOptions: nil)
         } catch {
-            throw DiarySummaryError.modelLoadFailed(underlying: error)
+            throw SummaryError.modelLoadFailed(underlying: error)
         }
 
         let tokenizer = try await Self.loadTokenizer()
-        return DefaultDiarySummaryRepository(encoder: encoder, decoder: decoder, tokenizer: tokenizer)
+        return DefaultSummaryRepository(encoder: encoder, decoder: decoder, tokenizer: tokenizer)
     }
 
     // AutoTokenizer.from(modelFolder:)는 폴더 안에서 표준 파일명("tokenizer.json"/"tokenizer_config.json")을
@@ -68,7 +68,7 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
     // 이를 피하기 위해 표준 파일명으로만 구성된 격리된 임시 폴더를 만들어 그 안에서 로드한다.
     private static func loadTokenizer() async throws -> Tokenizer {
         guard let tokenizerURL = Bundle.main.url(forResource: "kobart_tokenizer", withExtension: "json") else {
-            throw DiarySummaryError.modelLoadFailed()
+            throw SummaryError.modelLoadFailed()
         }
 
         let isolatedFolder = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString, isDirectory: true)
@@ -87,7 +87,7 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
 
             return try await AutoTokenizer.from(modelFolder: isolatedFolder)
         } catch {
-            throw DiarySummaryError.modelLoadFailed(underlying: error)
+            throw SummaryError.modelLoadFailed(underlying: error)
         }
     }
 
@@ -108,20 +108,26 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
                 runOptions: nil
             )
             guard let encoderHidden = encoderOutputs[Self.encoderOutputKey] else {
-                throw DiarySummaryError.inferenceFailed()
+                throw SummaryError.inferenceFailed()
             }
 
             let generatedTokens = try greedyDecode(encoderHidden: encoderHidden, encoderAttentionMask: maskTensor)
             let summary = tokenizer.decode(tokens: generatedTokens, skipSpecialTokens: true)
             return summary.trimmingCharacters(in: .whitespacesAndNewlines)
-        } catch let error as DiarySummaryError {
+        } catch let error as SummaryError {
             throw error
         } catch {
             Log.error("요약 추론 실패: \(error)")
-            throw DiarySummaryError.inferenceFailed(underlying: error)
+            throw SummaryError.inferenceFailed(underlying: error)
         }
     }
 
+    /// 절단 없는 실제 토큰 수. swift-transformers의 encode()는 자체적으로 truncation을 하지 않으므로
+    /// (그 truncation은 summarize()의 buildEncoderInputs에서만 수동으로 함) 그대로 개수를 세면 된다.
+    func countTokens(text: String) async throws -> Int {
+        tokenizer.encode(text: text).count
+    }
+
     // 인코더 1회 실행 결과(encoderHidden)를 매 스텝 재사용하며, 지금까지 생성된 전체 시퀀스를
     // 다시 디코더에 통째로 넣는 cache-free 그리디 디코딩(디코더가 KV 캐시를 안 쓰므로).
     private func greedyDecode(encoderHidden: ORTValue, encoderAttentionMask: ORTValue) throws -> [Int] {
@@ -140,7 +146,7 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
                 runOptions: nil
             )
             guard let logitsValue = decoderOutputs[Self.decoderOutputKey] else {
-                throw DiarySummaryError.inferenceFailed()
+                throw SummaryError.inferenceFailed()
             }
 
             let nextToken = try Self.argmaxLastPosition(
@@ -189,7 +195,7 @@ actor DefaultDiarySummaryRepository: DiarySummaryRepository {
         let shapeInfo = try logits.tensorTypeAndShapeInfo()
         let shape = shapeInfo.shape.map(\.intValue)
         guard shape.count == 3, shape[1] == sequenceLength, shape[2] == vocabSize else {
-            throw DiarySummaryError.inferenceFailed()
+            throw SummaryError.inferenceFailed()
         }
 
         let data = try logits.tensorData() as Data

…tory/DefaultDiarySummaryRepository.swift …epository/DefaultSummaryRepository.swiftGAMSS/Sources/Data/Repository/DefaultDiarySummaryRepository.swift renamed to GAMSS/Sources/Data/Repository/DefaultSummaryRepository.swift GAMSS/Sources/Data/Repository/DefaultDiarySummaryRepository.swift renamed to GAMSS/Sources/Data/Repository/DefaultSummaryRepository.swift

@@ -0,0 +1,48 @@
+//
+//  LazyConversationSummaryStore.swift
+//  GAMSS
+//
+//  Created by cchanmi on 8/8/26.
+//
+
+import Foundation
+
+/// DefaultSummaryRepository.make()가 async throws라 화면 생성 시점에 바로 만들 수 없다.
+/// 실제로 처음 쓰일 때(add/current/restore 최초 호출) 한 번만 만들어 캐싱한다.
+/// 생성에 실패해도(예: 모델 로드 실패) 압축은 부가 기능이므로 조용히 무시한다.
+actor LazyConversationSummaryStore: ConversationSummaryStore {
+    private let makeRepository: () async throws -> SummaryRepository
+    private var underlying: DefaultConversationSummaryStore?
+    private var resolutionFailed = false
+
+    init(makeRepository: @escaping () async throws -> SummaryRepository = { try await DefaultSummaryRepository.make() }) {
+        self.makeRepository = makeRepository
+    }
+
+    func add(_ utterance: String) async {
+        await resolve()?.add(utterance)
+    }
+
+    func current() async -> String? {
+        await resolve()?.current()
+    }
+
+    func reset() async {
+        await resolve()?.reset()
+    }
+
+    func restore(historicalUtterances: [String]) async {
+        await resolve()?.restore(historicalUtterances: historicalUtterances)
+    }
+
+    private func resolve() async -> DefaultConversationSummaryStore? {
+        if let underlying { return underlying }
+        guard !resolutionFailed, let repository = try? await makeRepository() else {
+            resolutionFailed = true
+            return nil
+        }
+        let store = DefaultConversationSummaryStore(summaryRepository: repository)
+        underlying = store
+        return store
+    }
+}