ios-projects/AIStudyApp/AIStudyApp/Features/Library/LibraryViewModel.swift

import Combine
import Foundation

@MainActor
class LibraryViewModel: ObservableObject {
    @Published var knowledgeBases: [KnowledgeBase] = []
    @Published var isLoading = false
    @Published var errorMessage: String?

    func loadKnowledgeBases() async {
        isLoading = true
        errorMessage = nil
        do {
            knowledgeBases = try await KnowledgeBaseService.shared.list()
        } catch {
            errorMessage = "加载知识库失败"
        }
        isLoading = false
    }

    func createKnowledgeBase(title: String, description: String?) async -> KnowledgeBase? {
        do {
            let kb = try await KnowledgeBaseService.shared.create(title: title, description: description)
            knowledgeBases.insert(kb, at: 0)
            return kb
        } catch {
            errorMessage = "创建知识库失败"
            return nil
        }
    }

    func deleteKnowledgeBase(id: String) async {
        do {
            _ = try await KnowledgeBaseService.shared.delete(id: id)
            knowledgeBases.removeAll { $0.id == id }
        } catch {
            errorMessage = "删除知识库失败"
        }
    }
}

@MainActor
class LibraryDetailViewModel: ObservableObject {
    @Published var items: [KnowledgeItem] = []
    @Published var knowledgeBase: KnowledgeBase?
    @Published var isLoading = false
    @Published var errorMessage: String?

    func loadItems(knowledgeBaseId: String) async {
        isLoading = true
        errorMessage = nil
        do {
            items = try await KnowledgeItemService.shared.list(knowledgeBaseId: knowledgeBaseId)
        } catch {
            errorMessage = "加载知识点失败"
        }
        isLoading = false
    }

    func loadKnowledgeBase(id: String) async {
        do {
            knowledgeBase = try await KnowledgeBaseService.shared.detail(id: id)
        } catch {
            errorMessage = "加载知识库详情失败"
        }
    }

    func addItem(knowledgeBaseId: String, title: String, content: String?) async -> KnowledgeItem? {
        do {
            let item = try await KnowledgeItemService.shared.create(
                knowledgeBaseId: knowledgeBaseId, title: title, content: content
            )
            items.append(item)
            return item
        } catch {
            errorMessage = "添加知识点失败"
            return nil
        }
    }
}