From 018ff22032a2e9dbfbe8d3f5fd8e62fea20bc405 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B8=B8=E8=96=AA=E6=B8=9D=28=E6=8F=BD=E6=B8=85=29?=
 <azure.yxy@alibaba-inc.com>
Date: Fri, 20 Jun 2025 17:52:14 +0800
Subject: [PATCH] [feat] pin an unpin models

---
 .../ModelList/Models/ModelListViewModel.swift | 24 ++++++------
 .../ModelList/Views/ModelListView.swift       | 24 +-----------
 .../ModelList/Views/SwipeActionsView.swift    | 39 +++++++++++++++++++
 3 files changed, 53 insertions(+), 34 deletions(-)
 create mode 100644 apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/SwipeActionsView.swift

diff --git a/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Models/ModelListViewModel.swift b/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Models/ModelListViewModel.swift
index cd909fa0..3b4675e8 100644
--- a/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Models/ModelListViewModel.swift
+++ b/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Models/ModelListViewModel.swift
@@ -19,10 +19,10 @@ class ModelListViewModel: ObservableObject {
     @Published var selectedModel: ModelInfo?
     
     private let modelClient = ModelClient()
-    private let pinnedModelKey = "com.mnnllm.pinnedModelId"
+    private let pinnedModelKey = "com.mnnllm.pinnedModelIds"
     
-    private var pinnedModelId: String? {
-        get { UserDefaults.standard.string(forKey: pinnedModelKey) }
+    public var pinnedModelIds: [String] {
+        get { UserDefaults.standard.stringArray(forKey: pinnedModelKey) ?? [] }
         set { UserDefaults.standard.setValue(newValue, forKey: pinnedModelKey) }
     }
     
@@ -70,12 +70,12 @@ class ModelListViewModel: ObservableObject {
             for i in 0..<fetchedModels.count {
                 fetchedModels[i].isDownloaded = ModelStorageManager.shared.isModelDownloaded(fetchedModels[i].modelId)
             }
-            // 保持置顶顺序
-            if let pinnedId = pinnedModelId, let idx = fetchedModels.firstIndex(where: { $0.modelId == pinnedId }) {
-                let pinned = fetchedModels.remove(at: idx)
-                fetchedModels.insert(pinned, at: 0)
+            // 多模型置顶顺序，最新置顶的在最顶部
+            let pinned = pinnedModelIds.reversed().compactMap { id in
+                fetchedModels.first(where: { $0.modelId == id })
             }
-            models = fetchedModels
+            let rest = fetchedModels.filter { !pinnedModelIds.contains($0.modelId) }
+            models = pinned + rest
         } catch {
             showError = true
             errorMessage = "Error: \(error.localizedDescription)"
@@ -126,15 +126,17 @@ class ModelListViewModel: ObservableObject {
         guard let index = models.firstIndex(where: { $0.modelId == model.modelId }) else { return }
         let pinned = models.remove(at: index)
         models.insert(pinned, at: 0)
-        pinnedModelId = model.modelId
+        var ids = pinnedModelIds.filter { $0 != model.modelId }
+        ids.append(model.modelId)
+        pinnedModelIds = ids
     }
     
     func unpinModel(_ model: ModelInfo) {
         guard let index = models.firstIndex(where: { $0.modelId == model.modelId }) else { return }
         let unpinned = models.remove(at: index)
-        let insertIndex = models.firstIndex(where: { !$0.isDownloaded || $0.modelId != model.modelId }) ?? models.endIndex
+        let insertIndex = models.count // 取消置顶后放到未置顶最后
         models.insert(unpinned, at: insertIndex)
-        pinnedModelId = nil
+        pinnedModelIds = pinnedModelIds.filter { $0 != model.modelId }
     }
     
     func deleteModel(_ model: ModelInfo) async {
diff --git a/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/ModelListView.swift b/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/ModelListView.swift
index f1a81ab6..ddac5ca0 100644
--- a/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/ModelListView.swift
+++ b/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/ModelListView.swift
@@ -44,29 +44,7 @@ struct ModelListView: View {
                             }
                         }
                         .swipeActions(edge: .trailing, allowsFullSwipe: false) {
-                            if let firstModel = viewModel.filteredModels.first, firstModel.modelId == model.modelId {
-                                Button {
-                                    viewModel.unpinModel(model)
-                                } label: {
-                                    Label("取消置顶", systemImage: "pin.slash")
-                                }.tint(.gray)
-                            } else {
-                                Button {
-                                    viewModel.pinModel(model)
-                                } label: {
-                                    Label("置顶", systemImage: "pin")
-                                }.tint(.yellow)
-                            }
-                            
-                            if model.isDownloaded {
-                                Button(role: .destructive) {
-                                    Task {
-                                        await viewModel.deleteModel(model)
-                                    }
-                                } label: {
-                                    Label("Delete", systemImage: "trash")
-                                }
-                            }
+                            SwipeActionsView(model: model, viewModel: viewModel)
                         }
                     }
                 }
diff --git a/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/SwipeActionsView.swift b/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/SwipeActionsView.swift
new file mode 100644
index 00000000..1a46eea0
--- /dev/null
+++ b/apps/iOS/MNNLLMChat/MNNLLMiOS/MainTab/ModelList/Views/SwipeActionsView.swift
@@ -0,0 +1,39 @@
+//
+//  SwipeActionsView.swift
+//  MNNLLMiOS
+//
+//  Created by 游薪渝(揽清) on 2025/1/3.
+//
+
+
+import SwiftUI
+
+struct SwipeActionsView: View {
+    let model: ModelInfo
+    @ObservedObject var viewModel: ModelListViewModel
+    
+    var body: some View {
+        if viewModel.pinnedModelIds.contains(model.modelId) {
+            Button {
+                viewModel.unpinModel(model)
+            } label: {
+                Label("取消置顶", systemImage: "pin.slash")
+            }.tint(.gray)
+        } else {
+            Button {
+                viewModel.pinModel(model)
+            } label: {
+                Label("置顶", systemImage: "pin")
+            }.tint(.yellow)
+        }
+        if model.isDownloaded {
+            Button(role: .destructive) {
+                Task {
+                    await viewModel.deleteModel(model)
+                }
+            } label: {
+                Label("Delete", systemImage: "trash")
+            }
+        }
+    }
+}