wolfplus
1 year ago
59 changed files with 6578 additions and 804 deletions
@ -1,17 +1,18 @@ |
|||||
package com.yunqiinnovation.ble_service |
package com.yunqiinnovation.agent_service |
||||
|
|
||||
import android.content.Context |
import android.content.Context |
||||
import com.yunqiinnovation.agent_service.AgentService |
import com.yunqiinnovation.agent_service.AgentService |
||||
import com.yunqiinnovation.agent_service.AgentServiceListener |
import com.yunqiinnovation.agent_service.AgentServiceListener |
||||
import com.yunqiinnovation.deepvoice_utils.FileLogger |
import com.yunqiinnovation.deepvoice_utils.FileLogger |
||||
|
import com.yunqiinnovation.ble_service.BleService |
||||
|
import com.yunqiinnovation.ble_service.BleConst |
||||
|
|
||||
/** |
/** |
||||
* 蓝牙语音交互代理 |
* 蓝牙语音交互代理 |
||||
* 连接BleService和AgentService，响应蓝牙设备唤醒信号，启动语音识别和交互 |
* 连接BleService和AgentService，响应蓝牙设备唤醒信号，启动语音识别和交互 |
||||
*/ |
*/ |
||||
object ChatAgent : BleService.Callback, AgentServiceListener { |
object BleAgent : BleService.Callback, AgentServiceListener { |
||||
private const val TAG = "ChatAgent" |
private const val TAG = "BleAgent" |
||||
|
|
||||
// 是否已初始化 |
// 是否已初始化 |
||||
private var isInitialized = false |
private var isInitialized = false |
||||
@ -1,4 +1,4 @@ |
|||||
package com.yunqiinnovation.ble_service |
package com.yunqiinnovation.agent_service |
||||
|
|
||||
import android.content.* |
import android.content.* |
||||
import android.os.Build |
import android.os.Build |
||||
@ -1,4 +1,4 @@ |
|||||
package com.yunqiinnovation.ble_service |
package com.yunqiinnovation.agent_service |
||||
|
|
||||
import android.content.Context |
import android.content.Context |
||||
import androidx.security.crypto.EncryptedSharedPreferences |
import androidx.security.crypto.EncryptedSharedPreferences |
||||
@ -0,0 +1,994 @@ |
|||||
|
import Foundation |
||||
|
import MicrosoftCognitiveServicesSpeech |
||||
|
import open_ai_service |
||||
|
import chat_storage |
||||
|
|
||||
|
/// 代理服务 - 处理语音识别、TTS和AI对话相关逻辑 |
||||
|
/// |
||||
|
/// 负责集成Azure语音服务、OpenAI服务和本地存储服务， |
||||
|
/// 提供语音识别、语音合成、AI对话等功能 |
||||
|
class AgentService { |
||||
|
// 常量定义 |
||||
|
private let TAG = "AgentService" |
||||
|
|
||||
|
// 上下文和监听器 |
||||
|
private var listeners = NSHashTable<AnyObject>.weakObjects() |
||||
|
|
||||
|
// 配置参数 |
||||
|
private var azureSpeechKey: String = "" |
||||
|
private var azureSpeechRegion: String = "" |
||||
|
private var openaiApiKey: String = "" |
||||
|
private var openaiBaseUrl: String = "" |
||||
|
private var openaiModel: String = "gpt-3.5-turbo" |
||||
|
private var mcpServer: String = "" |
||||
|
|
||||
|
// Azure服务 |
||||
|
private var azureAsrHelper: AzureAsrHelper? |
||||
|
private var azureTtsHelper: AzureTtsHelper? |
||||
|
|
||||
|
// OpenAI服务 |
||||
|
private var openAIService: OpenAIServiceBridge? |
||||
|
|
||||
|
// 聊天存储服务 |
||||
|
private var chatStorageHelper: ChatStorageHelperBridge? |
||||
|
|
||||
|
// 会话ID，用于区分不同聊天上下文 |
||||
|
private let sessionId = "default_agent" |
||||
|
|
||||
|
// 历史聊天消息缓存 |
||||
|
private var historyMessages = [[String: Any]]() |
||||
|
|
||||
|
// 系统提示词 |
||||
|
private var systemPrompt = "" |
||||
|
|
||||
|
// 状态 |
||||
|
private var isInitialized = false |
||||
|
private var isRecognitionActive = false |
||||
|
private var isTtsSpeaking = false |
||||
|
private var hasSpeechDetected = false |
||||
|
private var isAiStreaming = false |
||||
|
|
||||
|
// AI流生成相关 |
||||
|
private var currentAiTask: DispatchWorkItem? |
||||
|
|
||||
|
// 空闲检测相关 |
||||
|
private var idleCheckTimer: Timer? |
||||
|
private let maxIdleSeconds: TimeInterval = 10 // 最大空闲秒数 |
||||
|
|
||||
|
/// 初始化 |
||||
|
init() { |
||||
|
// 创建OpenAI服务桥接器 |
||||
|
openAIService = OpenAIServiceBridge() |
||||
|
|
||||
|
// 创建聊天存储桥接器 |
||||
|
chatStorageHelper = ChatStorageHelperBridge() |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 初始化系统提示词 |
||||
|
*/ |
||||
|
private func initSystemPrompt() { |
||||
|
systemPrompt = """ |
||||
|
你是一个友好、专业的语音助手，名叫"小语"。你的目标是通过对话为用户提供帮助、解答问题和完成任务。 |
||||
|
遵循以下指导原则： |
||||
|
1. 保持简短精炼的回答，因为用户是通过语音与你交流 |
||||
|
2. 优先使用中文回复，除非用户明确要求使用其他语言 |
||||
|
3. 当用户问题不明确时，礼貌地请求更多信息 |
||||
|
4. 避免过长的列表，尽量将信息分成小段 |
||||
|
5. 不要使用需要视觉展示的元素（如表格、图表或代码块） |
||||
|
6. 记住用户之前的对话内容，保持对话连贯 |
||||
|
7. 如果用户发送了图片，请根据图片内容和文字要求回答问题 |
||||
|
你不仅可以回答知识性问题，还可以帮助用户设置提醒、提供建议，或进行轻松愉快的对话。 |
||||
|
无论遇到什么问题，都要尽力以温暖、贴心的语气提供最佳帮助。 |
||||
|
""" |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 初始化 |
||||
|
* |
||||
|
* - Parameter config: 配置参数，包含所需的所有API密钥和设置 |
||||
|
* - Returns: 是否初始化成功 |
||||
|
*/ |
||||
|
func initialize(config: [String: Any]) -> Bool { |
||||
|
if isInitialized { return true } |
||||
|
|
||||
|
do { |
||||
|
// 从配置中获取参数 |
||||
|
if let azureKey = config["azureSpeechKey"] as? String { |
||||
|
azureSpeechKey = azureKey |
||||
|
} |
||||
|
|
||||
|
if let azureRegion = config["azureSpeechRegion"] as? String { |
||||
|
azureSpeechRegion = azureRegion |
||||
|
} |
||||
|
|
||||
|
if let openaiKey = config["openaiApiKey"] as? String { |
||||
|
openaiApiKey = openaiKey |
||||
|
} |
||||
|
|
||||
|
if let baseUrl = config["openaiBaseUrl"] as? String { |
||||
|
openaiBaseUrl = baseUrl |
||||
|
} |
||||
|
|
||||
|
if let model = config["openaiModel"] as? String { |
||||
|
openaiModel = model |
||||
|
} |
||||
|
|
||||
|
if let server = config["mcpServer"] as? String { |
||||
|
mcpServer = server |
||||
|
} |
||||
|
|
||||
|
// 自定义系统提示词 |
||||
|
if let customSystemPrompt = config["systemPrompt"] as? String, !customSystemPrompt.isEmpty { |
||||
|
systemPrompt = customSystemPrompt |
||||
|
} else { |
||||
|
// 使用默认系统提示词 |
||||
|
initSystemPrompt() |
||||
|
} |
||||
|
|
||||
|
// 检查必要参数 |
||||
|
if azureSpeechKey.isEmpty || azureSpeechRegion.isEmpty || openaiApiKey.isEmpty { |
||||
|
print("\(TAG): 初始化失败：关键配置参数缺失") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 初始化OpenAI服务 |
||||
|
openAIService?.initialize(apiKey: openaiApiKey, baseUrl: openaiBaseUrl, model: openaiModel, mcpServer: mcpServer) |
||||
|
|
||||
|
// 初始化Azure ASR |
||||
|
azureAsrHelper = AzureAsrHelper(self) |
||||
|
let asrInitResult = azureAsrHelper?.initialize( |
||||
|
subscriptionKey: azureSpeechKey, |
||||
|
region: azureSpeechRegion, |
||||
|
audioSourceType: .microphone |
||||
|
) ?? false |
||||
|
|
||||
|
if !asrInitResult { |
||||
|
print("\(TAG): Azure ASR初始化失败") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 初始化Azure TTS |
||||
|
azureTtsHelper = AzureTtsHelper(self) |
||||
|
let ttsInitResult = azureTtsHelper?.initialize( |
||||
|
subscriptionKey: azureSpeechKey, |
||||
|
region: azureSpeechRegion, |
||||
|
callback: self |
||||
|
) ?? false |
||||
|
|
||||
|
if !ttsInitResult { |
||||
|
print("\(TAG): Azure TTS初始化失败") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 加载最近的聊天记录 |
||||
|
loadChatHistory() |
||||
|
|
||||
|
isInitialized = true |
||||
|
print("\(TAG): 代理服务初始化成功") |
||||
|
return true |
||||
|
} catch { |
||||
|
print("\(TAG): 初始化失败：\(error.localizedDescription)") |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 添加事件监听器 |
||||
|
* |
||||
|
* - Parameter listener: 要添加的监听器 |
||||
|
*/ |
||||
|
func addListener(_ listener: AgentServiceListener) { |
||||
|
listeners.add(listener as AnyObject) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 设置事件监听器（替换现有监听器） |
||||
|
* |
||||
|
* - Parameter listener: 要设置的监听器 |
||||
|
*/ |
||||
|
func setListener(_ listener: AgentServiceListener) { |
||||
|
listeners.removeAllObjects() |
||||
|
listeners.add(listener as AnyObject) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 移除事件监听器 |
||||
|
* |
||||
|
* - Parameter listener: 要移除的监听器 |
||||
|
*/ |
||||
|
func removeListener(_ listener: AgentServiceListener) { |
||||
|
listeners.remove(listener as AnyObject) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 移除所有事件监听器 |
||||
|
*/ |
||||
|
func clearListeners() { |
||||
|
listeners.removeAllObjects() |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 启动空闲检测 |
||||
|
*/ |
||||
|
private func startIdleCheck() { |
||||
|
stopIdleCheck() // 先停止现有的检查 |
||||
|
|
||||
|
if !isRecognitionActive { return } |
||||
|
|
||||
|
// 创建定时器 |
||||
|
idleCheckTimer = Timer.scheduledTimer(withTimeInterval: maxIdleSeconds, repeats: false) { [weak self] _ in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
// 如果状态仍然是空闲的，则停止识别 |
||||
|
if self.isRecognitionActive && !self.hasSpeechDetected && !self.isTtsSpeaking && !self.isAiStreaming { |
||||
|
self.stopRecognition() |
||||
|
self.sendEvent("auto_stop", ["reason": "idle_timeout", "seconds": self.maxIdleSeconds]) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 停止空闲检测 |
||||
|
*/ |
||||
|
private func stopIdleCheck() { |
||||
|
idleCheckTimer?.invalidate() |
||||
|
idleCheckTimer = nil |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 重启空闲检测 |
||||
|
* 当状态发生变化时调用 |
||||
|
*/ |
||||
|
private func restartIdleCheck() { |
||||
|
if isRecognitionActive { |
||||
|
startIdleCheck() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 开始语音识别 |
||||
|
* |
||||
|
* - Returns: 是否成功开始识别 |
||||
|
*/ |
||||
|
func startRecognition() -> Bool { |
||||
|
if !isInitialized { |
||||
|
print("\(TAG): 服务未初始化") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if isRecognitionActive { return true } |
||||
|
|
||||
|
isRecognitionActive = true |
||||
|
hasSpeechDetected = false |
||||
|
|
||||
|
do { |
||||
|
// 创建连续识别回调 |
||||
|
class ContinuousRecognizeCallbackImpl: AzureAsrHelper.ContinuousRecognizeCallback { |
||||
|
private weak var service: AgentService? |
||||
|
|
||||
|
init(_ service: AgentService) { |
||||
|
self.service = service |
||||
|
} |
||||
|
|
||||
|
func onRecognizing(recognizing: String, detectedLanguage: String) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
if !recognizing.isEmpty { |
||||
|
// 检测到语音，更新状态 |
||||
|
let previousHasSpeech = service.hasSpeechDetected |
||||
|
service.hasSpeechDetected = true |
||||
|
|
||||
|
// 状态发生变化时重启空闲检测 |
||||
|
if !previousHasSpeech { |
||||
|
service.restartIdleCheck() |
||||
|
} |
||||
|
|
||||
|
service.sendEvent("recognizing", [ |
||||
|
"text": recognizing, |
||||
|
"language": detectedLanguage |
||||
|
]) |
||||
|
|
||||
|
// 如果TTS正在播放或AI正在生成，则触发打断 |
||||
|
if service.isTtsSpeaking || service.isAiStreaming { |
||||
|
service.interruptCurrentResponse() |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func onResult(text: String, detectedLanguage: String) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
if !text.isEmpty { |
||||
|
service.sendEvent("recognition_result", [ |
||||
|
"text": text, |
||||
|
"language": detectedLanguage |
||||
|
]) |
||||
|
|
||||
|
service.processWithOpenAI(text: text) |
||||
|
} |
||||
|
|
||||
|
// 重置状态，继续识别 |
||||
|
let previousHasSpeech = service.hasSpeechDetected |
||||
|
service.hasSpeechDetected = false |
||||
|
|
||||
|
// 状态发生变化时重启空闲检测 |
||||
|
if previousHasSpeech { |
||||
|
service.restartIdleCheck() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func onSessionStarted() { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
service.sendEvent("recognition_started", ["status": "started"]) |
||||
|
// 启动空闲检测 |
||||
|
service.startIdleCheck() |
||||
|
} |
||||
|
|
||||
|
func onSessionStopped() { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
service.isRecognitionActive = false |
||||
|
service.stopIdleCheck() |
||||
|
service.sendEvent("recognition_stopped", ["status": "stopped"]) |
||||
|
} |
||||
|
|
||||
|
func onCanceled(reason: String, errorDetails: String) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
service.isRecognitionActive = false |
||||
|
service.stopIdleCheck() |
||||
|
service.sendEvent("recognition_canceled", [ |
||||
|
"reason": reason, |
||||
|
"details": errorDetails |
||||
|
]) |
||||
|
} |
||||
|
|
||||
|
func onError(error: String) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
service.isRecognitionActive = false |
||||
|
service.stopIdleCheck() |
||||
|
print("\(service.TAG): 语音识别出错: \(error)") |
||||
|
service.sendEvent("error", [ |
||||
|
"code": "RECOGNITION_ERROR", |
||||
|
"message": error |
||||
|
]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
let callback = ContinuousRecognizeCallbackImpl(self) |
||||
|
let result = azureAsrHelper?.startContinuousRecognition(callback) ?? false |
||||
|
|
||||
|
if !result { |
||||
|
isRecognitionActive = false |
||||
|
print("\(TAG): 启动语音识别失败") |
||||
|
sendEvent("error", [ |
||||
|
"code": "RECOGNITION_START_ERROR", |
||||
|
"message": "启动语音识别失败" |
||||
|
]) |
||||
|
} |
||||
|
|
||||
|
return result |
||||
|
} catch { |
||||
|
isRecognitionActive = false |
||||
|
print("\(TAG): 启动语音识别失败: \(error.localizedDescription)") |
||||
|
sendEvent("error", [ |
||||
|
"code": "RECOGNITION_START_ERROR", |
||||
|
"message": error.localizedDescription |
||||
|
]) |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 停止语音识别 |
||||
|
*/ |
||||
|
func stopRecognition() { |
||||
|
if !isRecognitionActive { return } |
||||
|
|
||||
|
print("\(TAG): 停止语音识别") |
||||
|
|
||||
|
// 停止识别 |
||||
|
let _ = azureAsrHelper?.stopContinuousRecognition() |
||||
|
isRecognitionActive = false |
||||
|
stopIdleCheck() |
||||
|
print("\(TAG): 语音识别已停止") |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 打断当前响应 |
||||
|
* 停止TTS播放和AI流输出 |
||||
|
*/ |
||||
|
func interruptCurrentResponse() { |
||||
|
if isAiStreaming || isTtsSpeaking { |
||||
|
// 停止TTS播放 |
||||
|
stopTts() |
||||
|
|
||||
|
// 停止AI流输出 |
||||
|
stopAiStream() |
||||
|
|
||||
|
// 发送打断事件 |
||||
|
sendEvent("response_interrupted", ["status": "interrupted"]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 停止AI流输出 |
||||
|
*/ |
||||
|
private func stopAiStream() { |
||||
|
if isAiStreaming { |
||||
|
// 取消当前AI生成任务 |
||||
|
currentAiTask?.cancel() |
||||
|
currentAiTask = nil |
||||
|
|
||||
|
// 通知OpenAI服务终止当前流式请求 |
||||
|
openAIService?.cancelCurrentStream() |
||||
|
|
||||
|
// 更新状态 |
||||
|
isAiStreaming = false |
||||
|
|
||||
|
// 记录日志 |
||||
|
print("\(TAG): AI流输出已停止") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 处理文本输入 |
||||
|
* 作为语音输入的补充，直接处理文本并通过事件返回结果 |
||||
|
* |
||||
|
* - Parameters: |
||||
|
* - text: 用户输入文本 |
||||
|
* - speakResponse: 是否朗读回复，默认为false |
||||
|
* - Returns: 是否成功开始处理 |
||||
|
*/ |
||||
|
func processTextInput(text: String, speakResponse: Bool = false) -> Bool { |
||||
|
if !isInitialized { |
||||
|
print("\(TAG): 服务未初始化") |
||||
|
sendEvent("error", ["code": "NOT_INITIALIZED", "message": "服务未初始化"]) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if text.isEmpty { |
||||
|
print("\(TAG): 文本输入不能为空") |
||||
|
sendEvent("error", ["code": "EMPTY_TEXT", "message": "文本输入不能为空"]) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 使用OpenAI处理文本 |
||||
|
processWithOpenAI(text: text, speakResponse: speakResponse) |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 使用OpenAI处理语音识别结果 |
||||
|
* |
||||
|
* - Parameter text: 用户输入文本 |
||||
|
*/ |
||||
|
private func processWithOpenAI(text: String) { |
||||
|
processWithOpenAI(text: text, speakResponse: true) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 使用OpenAI处理文本消息 |
||||
|
* |
||||
|
* - Parameters: |
||||
|
* - text: 用户输入文本 |
||||
|
* - speakResponse: 是否使用TTS朗读回复 |
||||
|
*/ |
||||
|
private func processWithOpenAI(text: String, speakResponse: Bool = true) { |
||||
|
print("\(TAG): 用户问题: \(text)") |
||||
|
|
||||
|
// 创建用户文本消息并处理 |
||||
|
guard let userMessage = openAIService?.createUserMessage(text: text) else { |
||||
|
print("\(TAG): 创建用户消息失败") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
processWithOpenAIInternal(userMessage: userMessage, displayText: text, speakResponse: speakResponse) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 使用OpenAI处理图片 |
||||
|
* |
||||
|
* - Parameters: |
||||
|
* - imagePath: 图片文件路径 |
||||
|
* - text: 可选的文本描述或问题 |
||||
|
* - speakResponse: 是否朗读回复 |
||||
|
* - Returns: 是否成功开始处理 |
||||
|
*/ |
||||
|
func processImageInput(imagePath: String, text: String = "", speakResponse: Bool = false) -> Bool { |
||||
|
if !isInitialized { |
||||
|
print("\(TAG): 服务未初始化") |
||||
|
sendEvent("error", ["code": "NOT_INITIALIZED", "message": "服务未初始化"]) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if imagePath.isEmpty { |
||||
|
print("\(TAG): 图片路径不能为空") |
||||
|
sendEvent("error", ["code": "EMPTY_IMAGE_PATH", "message": "图片路径不能为空"]) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 通知开始处理图片 |
||||
|
sendEvent("image_processing", [ |
||||
|
"status": "processing", |
||||
|
"imagePath": imagePath |
||||
|
]) |
||||
|
|
||||
|
// 异步处理图片 |
||||
|
DispatchQueue.global(qos: .userInitiated).async { [weak self] in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
// 将图片转换为Base64格式 |
||||
|
guard let imageBase64 = self.openAIService?.fileToBase64(filePath: imagePath) else { |
||||
|
DispatchQueue.main.async { |
||||
|
print("\(self.TAG): 图片转换失败: \(imagePath)") |
||||
|
self.sendEvent("error", [ |
||||
|
"code": "IMAGE_CONVERSION_FAILED", |
||||
|
"message": "图片转换失败" |
||||
|
]) |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 通知图片准备完成 |
||||
|
DispatchQueue.main.async { |
||||
|
self.sendEvent("image_ready", [ |
||||
|
"status": "ready", |
||||
|
"imagePath": imagePath |
||||
|
]) |
||||
|
|
||||
|
// 处理包含图片的消息 |
||||
|
self.processImageWithOpenAI(imageBase64: imageBase64, text: text, speakResponse: speakResponse) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 使用OpenAI处理图片 |
||||
|
* |
||||
|
* - Parameters: |
||||
|
* - imageBase64: Base64编码的图片数据 |
||||
|
* - text: 可选的文本描述或问题 |
||||
|
* - speakResponse: 是否朗读回复 |
||||
|
*/ |
||||
|
private func processImageWithOpenAI(imageBase64: String, text: String = "", speakResponse: Bool = false) { |
||||
|
print("\(TAG): 处理图片输入: \(text.isEmpty ? "无附加文本" : "附带文本: \(text)")") |
||||
|
|
||||
|
// 创建带图片的用户消息并处理 |
||||
|
guard let userMessage = openAIService?.createUserMessageWithImage(text: text, imageBase64: imageBase64) else { |
||||
|
print("\(TAG): 创建带图片的用户消息失败") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 图片描述用于存储 |
||||
|
let displayText = text.isEmpty ? "[图片]" : text |
||||
|
processWithOpenAIInternal(userMessage: userMessage, displayText: displayText, speakResponse: speakResponse, hasImage: true) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 内部方法：通用的OpenAI处理逻辑 |
||||
|
* |
||||
|
* - Parameters: |
||||
|
* - userMessage: 用户消息（可以是文本或图片格式） |
||||
|
* - displayText: 用于显示和存储的文本 |
||||
|
* - speakResponse: 是否朗读回复 |
||||
|
* - hasImage: 是否包含图片 |
||||
|
*/ |
||||
|
private func processWithOpenAIInternal(userMessage: [String: Any], displayText: String, speakResponse: Bool = true, hasImage: Bool = false) { |
||||
|
// 如果有正在进行的AI流式输出，先停止它 |
||||
|
stopAiStream() |
||||
|
|
||||
|
// 创建AI任务 |
||||
|
let workItem = DispatchWorkItem { [weak self] in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
// 设置状态为正在流式输出 |
||||
|
self.isAiStreaming = true |
||||
|
|
||||
|
// 使用历史记录作为上下文发送到OpenAI |
||||
|
var responseBuilder = "" |
||||
|
|
||||
|
// 添加系统提示词到历史消息的副本中 |
||||
|
var messagesWithSystemPrompt: [[String: Any]] = [] |
||||
|
|
||||
|
// 先添加系统提示词 |
||||
|
if !self.systemPrompt.isEmpty { |
||||
|
if let systemMessage = self.openAIService?.createSystemMessage(text: self.systemPrompt) { |
||||
|
messagesWithSystemPrompt.append(systemMessage) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 再添加历史消息 |
||||
|
messagesWithSystemPrompt.append(contentsOf: self.historyMessages) |
||||
|
messagesWithSystemPrompt.append(userMessage) |
||||
|
|
||||
|
// 将用户消息添加到历史记录 |
||||
|
self.addToHistoryMessages(userMessage) |
||||
|
|
||||
|
// 流式回调 |
||||
|
class StreamCallbackBridge: NSObject, OpenAIStreamCallback { |
||||
|
private weak var service: AgentService? |
||||
|
private var responseBuilder: String |
||||
|
private let speakResponse: Bool |
||||
|
private let displayText: String |
||||
|
private let hasImage: Bool |
||||
|
|
||||
|
init(_ service: AgentService, responseBuilder: String = "", speakResponse: Bool, displayText: String, hasImage: Bool) { |
||||
|
self.service = service |
||||
|
self.responseBuilder = responseBuilder |
||||
|
self.speakResponse = speakResponse |
||||
|
self.displayText = displayText |
||||
|
self.hasImage = hasImage |
||||
|
super.init() |
||||
|
} |
||||
|
|
||||
|
func onToken(token: String) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
responseBuilder.append(token) |
||||
|
if speakResponse { |
||||
|
_ = service.azureTtsHelper?.speakStream(token) |
||||
|
} |
||||
|
// 发送流式回复token |
||||
|
service.sendEvent("assistant_token", ["token": token]) |
||||
|
} |
||||
|
|
||||
|
func onComplete() { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
// 视情况决定是否朗读回复 |
||||
|
if speakResponse { |
||||
|
_ = service.azureTtsHelper?.flushStream() |
||||
|
} |
||||
|
|
||||
|
if !responseBuilder.isEmpty { |
||||
|
// 发送完整回复，包含是否有图片的标记 |
||||
|
var responseData: [String: Any] = [ |
||||
|
"text": responseBuilder, |
||||
|
"userInput": displayText |
||||
|
] |
||||
|
|
||||
|
if hasImage { |
||||
|
responseData["hasImage"] = true |
||||
|
} |
||||
|
|
||||
|
service.sendEvent("assistant_response", responseData) |
||||
|
|
||||
|
// 添加AI回复到历史记录 |
||||
|
if let assistantMessage = service.openAIService?.createAssistantMessage(text: responseBuilder) { |
||||
|
service.addToHistoryMessages(assistantMessage) |
||||
|
} |
||||
|
|
||||
|
// 保存聊天记录 |
||||
|
service.saveChatMessage(userMessage: displayText, assistantMessage: responseBuilder) |
||||
|
} |
||||
|
|
||||
|
// 标记AI流式输出已完成 |
||||
|
service.isAiStreaming = false |
||||
|
service.currentAiTask = nil |
||||
|
} |
||||
|
|
||||
|
func onError(error: Error) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
print("\(service.TAG): AI处理出错: \(error.localizedDescription)") |
||||
|
service.sendEvent("error", [ |
||||
|
"code": "AI_ERROR", |
||||
|
"message": error.localizedDescription |
||||
|
]) |
||||
|
|
||||
|
// 标记AI流式输出已完成 |
||||
|
service.isAiStreaming = false |
||||
|
service.currentAiTask = nil |
||||
|
} |
||||
|
|
||||
|
func onFunctionCall(call: [String: Any]) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
if let name = call["name"] as? String { |
||||
|
service.sendEvent("function_call", [ |
||||
|
"name": name, |
||||
|
"arguments": call |
||||
|
]) |
||||
|
|
||||
|
if name == "exit_interaction" { |
||||
|
service.stopRecognition() |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func onFunctionCallResult(functionCall: [String: Any], functionCallResult: [String: Any]) { |
||||
|
guard let service = service else { return } |
||||
|
|
||||
|
service.sendEvent("function_call_result", [ |
||||
|
"function_call": functionCall, |
||||
|
"result": functionCallResult, |
||||
|
]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
let callback = StreamCallbackBridge(self, responseBuilder: responseBuilder, speakResponse: speakResponse, displayText: displayText, hasImage: hasImage) |
||||
|
|
||||
|
self.openAIService?.sendMessageStream(messages: messagesWithSystemPrompt, callback: callback) |
||||
|
} |
||||
|
|
||||
|
// 保存任务引用并在全局队列中执行 |
||||
|
currentAiTask = workItem |
||||
|
DispatchQueue.global(qos: .userInitiated).async(execute: workItem) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 加载最近的聊天历史记录 |
||||
|
*/ |
||||
|
private func loadChatHistory() { |
||||
|
// 清空当前历史记录 |
||||
|
historyMessages.removeAll() |
||||
|
|
||||
|
// 使用ChatStorageHelper直接获取最近消息 |
||||
|
guard let recentMessages = chatStorageHelper?.getRecentMessages(sessionId: sessionId, count: 10) else { |
||||
|
print("\(TAG): 没有找到历史记录") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 将消息添加到历史记录 |
||||
|
for message in recentMessages { |
||||
|
if let sender = message["sender"] as? String, |
||||
|
let content = message["message"] as? String { |
||||
|
|
||||
|
if sender == "user" { |
||||
|
if let userMessage = openAIService?.createUserMessage(text: content) { |
||||
|
addToHistoryMessages(userMessage) |
||||
|
} |
||||
|
} else if sender == "assistant" { |
||||
|
if let assistantMessage = openAIService?.createAssistantMessage(text: content) { |
||||
|
addToHistoryMessages(assistantMessage) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
print("\(TAG): 已加载\(recentMessages.count)条历史记录") |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 添加消息到历史记录，保持最近10条 |
||||
|
*/ |
||||
|
private func addToHistoryMessages(_ message: [String: Any]) { |
||||
|
// 添加新消息 |
||||
|
historyMessages.append(message) |
||||
|
|
||||
|
// 如果超过10条，删除最早的消息 |
||||
|
while historyMessages.count > 10 { |
||||
|
historyMessages.removeFirst() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* TTS播放函数 |
||||
|
* |
||||
|
* - Parameter text: 要播放的文本 |
||||
|
*/ |
||||
|
func speakText(text: String) { |
||||
|
if text.isEmpty { return } |
||||
|
|
||||
|
// 更新状态 |
||||
|
isTtsSpeaking = true |
||||
|
restartIdleCheck() // 状态变化，重启检测 |
||||
|
|
||||
|
// 播放文本 |
||||
|
_ = azureTtsHelper?.speakText(text) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 停止TTS播放 |
||||
|
*/ |
||||
|
func stopTts() { |
||||
|
if isTtsSpeaking { |
||||
|
_ = azureTtsHelper?.stopSpeaking() |
||||
|
isTtsSpeaking = false |
||||
|
restartIdleCheck() // 状态变化，重启检测 |
||||
|
sendEvent("tts_stopped", ["status": "stopped"]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 保存聊天消息 |
||||
|
*/ |
||||
|
private func saveChatMessage(userMessage: String, assistantMessage: String) { |
||||
|
DispatchQueue.global(qos: .background).async { [weak self] in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
// 保存用户消息 |
||||
|
let userMessageId = self.chatStorageHelper?.saveMessage( |
||||
|
sessionId: self.sessionId, |
||||
|
message: userMessage, |
||||
|
sender: "user" |
||||
|
) ?? -1 |
||||
|
|
||||
|
if userMessageId != -1 { |
||||
|
// 保存AI回复 |
||||
|
let assistantMessageId = self.chatStorageHelper?.saveMessage( |
||||
|
sessionId: self.sessionId, |
||||
|
message: assistantMessage, |
||||
|
sender: "assistant" |
||||
|
) ?? -1 |
||||
|
|
||||
|
if assistantMessageId == -1 { |
||||
|
print("\(self.TAG): 保存助手消息失败") |
||||
|
} |
||||
|
} else { |
||||
|
print("\(self.TAG): 保存用户消息失败") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 清除聊天历史 |
||||
|
*/ |
||||
|
func clearChatHistory(completion: @escaping (Bool) -> Void) { |
||||
|
DispatchQueue.global(qos: .background).async { [weak self] in |
||||
|
guard let self = self else { |
||||
|
DispatchQueue.main.async { |
||||
|
completion(false) |
||||
|
} |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 清除指定会话的所有消息 |
||||
|
let success = self.chatStorageHelper?.deleteMessages(sessionId: self.sessionId) ?? false |
||||
|
|
||||
|
if success { |
||||
|
// 清空内存中的历史记录 |
||||
|
self.historyMessages.removeAll() |
||||
|
print("\(self.TAG): 聊天历史已清除") |
||||
|
} else { |
||||
|
print("\(self.TAG): 清除聊天历史失败") |
||||
|
} |
||||
|
|
||||
|
DispatchQueue.main.async { |
||||
|
completion(success) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 发送事件 |
||||
|
*/ |
||||
|
private func sendEvent(_ eventName: String, _ data: [String: Any]) { |
||||
|
// 向所有监听器发送事件 |
||||
|
let allListeners = listeners.allObjects |
||||
|
for case let listener as AgentServiceListener in allListeners { |
||||
|
listener.onEvent(eventName: eventName, data: data) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 释放资源 |
||||
|
*/ |
||||
|
func dispose() { |
||||
|
stopRecognition() |
||||
|
stopTts() |
||||
|
stopAiStream() |
||||
|
stopIdleCheck() |
||||
|
|
||||
|
azureAsrHelper?.dispose() |
||||
|
azureTtsHelper?.dispose() |
||||
|
|
||||
|
// 清除所有监听器 |
||||
|
clearListeners() |
||||
|
|
||||
|
// 重置单例状态，以便下次使用时可以重新初始化 |
||||
|
isInitialized = false |
||||
|
|
||||
|
print("\(TAG): 代理服务资源已释放") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - AzureTtsHelper.TtsCallback |
||||
|
extension AgentService: AzureTtsHelper.TtsCallback { |
||||
|
func onSynthesisStarted() { |
||||
|
isTtsSpeaking = true |
||||
|
// 状态变化，重置空闲检测 |
||||
|
restartIdleCheck() |
||||
|
sendEvent("tts_started", ["status": "started"]) |
||||
|
} |
||||
|
|
||||
|
func onSynthesizing() { |
||||
|
// 可以在此添加TTS合成中的处理逻辑 |
||||
|
} |
||||
|
|
||||
|
func onSynthesisCompleted() { |
||||
|
isTtsSpeaking = false |
||||
|
// 状态变化，重启空闲检测 |
||||
|
restartIdleCheck() |
||||
|
sendEvent("tts_completed", ["status": "completed"]) |
||||
|
} |
||||
|
|
||||
|
func onSynthesisCanceled() { |
||||
|
isTtsSpeaking = false |
||||
|
// 状态变化，重启空闲检测 |
||||
|
restartIdleCheck() |
||||
|
sendEvent("tts_canceled", ["status": "canceled"]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 外部插件桥接器 |
||||
|
|
||||
|
/// 桥接OpenAIService插件 |
||||
|
class OpenAIServiceBridge { |
||||
|
private let service = OpenAIService() |
||||
|
|
||||
|
/// 初始化服务 |
||||
|
func initialize(apiKey: String, baseUrl: String, model: String, mcpServer: String) { |
||||
|
service.initialize(apiKey: apiKey, baseUrl: baseUrl, model: model, mcpServer: mcpServer) |
||||
|
} |
||||
|
|
||||
|
/// 创建系统消息 |
||||
|
func createSystemMessage(text: String) -> [String: Any] { |
||||
|
return service.createSystemMessage(text) |
||||
|
} |
||||
|
|
||||
|
/// 创建用户消息 |
||||
|
func createUserMessage(text: String) -> [String: Any] { |
||||
|
return service.createUserMessage(text) |
||||
|
} |
||||
|
|
||||
|
/// 创建助手消息 |
||||
|
func createAssistantMessage(text: String) -> [String: Any] { |
||||
|
return service.createAssistantMessage(text) |
||||
|
} |
||||
|
|
||||
|
/// 创建带图片的用户消息 |
||||
|
func createUserMessageWithImage(text: String, imageBase64: String) -> [String: Any]? { |
||||
|
return service.createUserMessageWithImage(text: text, imageBase64: imageBase64) |
||||
|
} |
||||
|
|
||||
|
/// 文件转Base64 |
||||
|
func fileToBase64(filePath: String) -> String? { |
||||
|
return service.fileToBase64(filePath: filePath) |
||||
|
} |
||||
|
|
||||
|
/// 发送流式消息 |
||||
|
func sendMessageStream(messages: [[String: Any]], callback: OpenAIStreamCallback) { |
||||
|
service.sendMessageStream(messages: messages, callback: callback) |
||||
|
} |
||||
|
|
||||
|
/// 取消当前流 |
||||
|
func cancelCurrentStream() { |
||||
|
service.cancelCurrentStream() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// OpenAI流式回调协议 |
||||
|
@objc protocol OpenAIStreamCallback { |
||||
|
func onToken(token: String) |
||||
|
func onComplete() |
||||
|
func onError(error: Error) |
||||
|
func onFunctionCall(call: [String: Any]) |
||||
|
func onFunctionCallResult(functionCall: [String: Any], functionCallResult: [String: Any]) |
||||
|
} |
||||
|
|
||||
|
/// 桥接ChatStorageHelper插件 |
||||
|
class ChatStorageHelperBridge { |
||||
|
private let helper = ChatStorageHelper() |
||||
|
|
||||
|
/// 保存消息 |
||||
|
func saveMessage(sessionId: String, message: String, sender: String) -> Int64 { |
||||
|
return helper.saveMessage(sessionId: sessionId, message: message, sender: sender) |
||||
|
} |
||||
|
|
||||
|
/// 获取最近消息 |
||||
|
func getRecentMessages(sessionId: String, count: Int) -> [[String: Any]]? { |
||||
|
return helper.getRecentMessages(sessionId: sessionId, count: count) |
||||
|
} |
||||
|
|
||||
|
/// 删除消息 |
||||
|
func deleteMessages(sessionId: String) -> Bool { |
||||
|
return helper.deleteMessages(sessionId: sessionId) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,11 @@ |
|||||
|
import Foundation |
||||
|
|
||||
|
/// 代理服务事件监听器协议 |
||||
|
/// 用于接收代理服务发送的各类事件 |
||||
|
protocol AgentServiceListener: AnyObject { |
||||
|
/// 事件回调 |
||||
|
/// - Parameters: |
||||
|
/// - eventName: 事件名称 |
||||
|
/// - data: 事件数据 |
||||
|
func onEvent(eventName: String, data: [String: Any]) |
||||
|
} |
||||
@ -0,0 +1,4 @@ |
|||||
|
#import <Flutter/Flutter.h> |
||||
|
|
||||
|
@interface AgentServicePlugin : NSObject<FlutterPlugin> |
||||
|
@end |
||||
@ -0,0 +1,12 @@ |
|||||
|
#import "AgentServicePlugin.h" |
||||
|
#if __has_include(<agent_service/agent_service-Swift.h>) |
||||
|
#import <agent_service/agent_service-Swift.h> |
||||
|
#else |
||||
|
#import "agent_service-Swift.h" |
||||
|
#endif |
||||
|
|
||||
|
@implementation AgentServicePlugin |
||||
|
+ (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar>*)registrar { |
||||
|
[SwiftAgentServicePlugin registerWithRegistrar:registrar]; |
||||
|
} |
||||
|
@end |
||||
@ -0,0 +1,259 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
import open_ai_service |
||||
|
import chat_storage |
||||
|
|
||||
|
/// Swift版Agent Service插件 |
||||
|
/// 处理Flutter与原生层的通信 |
||||
|
public class SwiftAgentServicePlugin: NSObject, FlutterPlugin { |
||||
|
// Flutter方法通道 |
||||
|
private var methodChannel: FlutterMethodChannel? |
||||
|
|
||||
|
// Flutter事件通道 |
||||
|
private var eventChannel: FlutterEventChannel? |
||||
|
|
||||
|
// 事件sink，用于向Flutter发送事件 |
||||
|
private var eventSink: FlutterEventSink? |
||||
|
|
||||
|
// 核心服务实例 |
||||
|
private var agentService: AgentService? |
||||
|
|
||||
|
/// 注册插件 |
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
let instance = SwiftAgentServicePlugin() |
||||
|
|
||||
|
// 初始化方法通道 |
||||
|
let methodChannel = FlutterMethodChannel(name: "com.yunqiinnovation.agent_service", binaryMessenger: registrar.messenger()) |
||||
|
registrar.addMethodCallDelegate(instance, channel: methodChannel) |
||||
|
instance.methodChannel = methodChannel |
||||
|
|
||||
|
// 初始化事件通道 |
||||
|
let eventChannel = FlutterEventChannel(name: "com.yunqiinnovation.agent_service/events", binaryMessenger: registrar.messenger()) |
||||
|
eventChannel.setStreamHandler(instance) |
||||
|
instance.eventChannel = eventChannel |
||||
|
} |
||||
|
|
||||
|
/// 处理方法调用 |
||||
|
public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) { |
||||
|
switch call.method { |
||||
|
case "initialize": |
||||
|
handleInitialize(call, result) |
||||
|
|
||||
|
case "startConversation": |
||||
|
handleStartConversation(result) |
||||
|
|
||||
|
case "stopConversation": |
||||
|
handleStopConversation(result) |
||||
|
|
||||
|
case "processTextInput": |
||||
|
handleProcessTextInput(call, result) |
||||
|
|
||||
|
case "speakText": |
||||
|
handleSpeakText(call, result) |
||||
|
|
||||
|
case "stopTts": |
||||
|
handleStopTts(result) |
||||
|
|
||||
|
case "clearChatHistory": |
||||
|
handleClearChatHistory(result) |
||||
|
|
||||
|
case "interruptCurrentResponse": |
||||
|
handleInterruptCurrentResponse(result) |
||||
|
|
||||
|
case "processImageInput": |
||||
|
handleProcessImageInput(call, result) |
||||
|
|
||||
|
case "dispose": |
||||
|
handleDispose(result) |
||||
|
|
||||
|
default: |
||||
|
result(FlutterMethodNotImplemented) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 方法处理 |
||||
|
|
||||
|
/// 初始化 |
||||
|
private func handleInitialize(_ call: FlutterMethodCall, _ result: @escaping FlutterResult) { |
||||
|
guard let args = call.arguments as? [String: Any], |
||||
|
let config = args["config"] as? [String: Any] else { |
||||
|
result(FlutterError(code: "INVALID_ARGUMENTS", message: "缺少配置参数", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 初始化代理服务 |
||||
|
if agentService == nil { |
||||
|
agentService = AgentService() |
||||
|
|
||||
|
// 设置事件监听器 |
||||
|
agentService?.setListener(self) |
||||
|
} |
||||
|
|
||||
|
// 执行初始化 |
||||
|
let success = agentService?.initialize(config: config) ?? false |
||||
|
result(success) |
||||
|
} |
||||
|
|
||||
|
/// 开始对话 |
||||
|
private func handleStartConversation(_ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let success = service.startRecognition() |
||||
|
result(success) |
||||
|
} |
||||
|
|
||||
|
/// 停止对话 |
||||
|
private func handleStopConversation(_ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
service.stopRecognition() |
||||
|
result(true) |
||||
|
} |
||||
|
|
||||
|
/// 处理文本输入 |
||||
|
private func handleProcessTextInput(_ call: FlutterMethodCall, _ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let args = call.arguments as? [String: Any], |
||||
|
let text = args["text"] as? String else { |
||||
|
result(FlutterError(code: "INVALID_ARGUMENTS", message: "无效的文本参数", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let speakResponse = (args["speakResponse"] as? Bool) ?? false |
||||
|
let success = service.processTextInput(text: text, speakResponse: speakResponse) |
||||
|
result(success) |
||||
|
} |
||||
|
|
||||
|
/// 朗读文本 |
||||
|
private func handleSpeakText(_ call: FlutterMethodCall, _ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let args = call.arguments as? [String: Any], |
||||
|
let text = args["text"] as? String else { |
||||
|
result(FlutterError(code: "INVALID_ARGUMENTS", message: "无效的文本参数", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
service.speakText(text: text) |
||||
|
result(true) |
||||
|
} |
||||
|
|
||||
|
/// 停止语音合成 |
||||
|
private func handleStopTts(_ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
service.stopTts() |
||||
|
result(true) |
||||
|
} |
||||
|
|
||||
|
/// 清除聊天历史 |
||||
|
private func handleClearChatHistory(_ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
service.clearChatHistory { success in |
||||
|
result(success) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 中断当前响应 |
||||
|
private func handleInterruptCurrentResponse(_ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
service.interruptCurrentResponse() |
||||
|
result(true) |
||||
|
} |
||||
|
|
||||
|
/// 处理图片输入 |
||||
|
private func handleProcessImageInput(_ call: FlutterMethodCall, _ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "AgentService尚未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let args = call.arguments as? [String: Any], |
||||
|
let imagePath = args["imagePath"] as? String else { |
||||
|
result(FlutterError(code: "INVALID_ARGUMENTS", message: "无效的图片路径参数", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let text = (args["text"] as? String) ?? "" |
||||
|
let speakResponse = (args["speakResponse"] as? Bool) ?? false |
||||
|
|
||||
|
let success = service.processImageInput(imagePath: imagePath, text: text, speakResponse: speakResponse) |
||||
|
result(success) |
||||
|
} |
||||
|
|
||||
|
/// 释放资源 |
||||
|
private func handleDispose(_ result: @escaping FlutterResult) { |
||||
|
guard let service = agentService else { |
||||
|
result(true) // 已经是nil，直接返回成功 |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
service.dispose() |
||||
|
agentService = nil |
||||
|
|
||||
|
result(true) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - FlutterStreamHandler |
||||
|
extension SwiftAgentServicePlugin: FlutterStreamHandler { |
||||
|
public func onListen(withArguments arguments: Any?, eventSink events: @escaping FlutterEventSink) -> FlutterError? { |
||||
|
self.eventSink = events |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
public func onCancel(withArguments arguments: Any?) -> FlutterError? { |
||||
|
self.eventSink = nil |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - AgentServiceListener |
||||
|
extension SwiftAgentServicePlugin: AgentServiceListener { |
||||
|
func onEvent(eventName: String, data: [String: Any]) { |
||||
|
guard let sink = eventSink else { return } |
||||
|
|
||||
|
do { |
||||
|
// 构建事件JSON |
||||
|
let eventMap: [String: Any] = [ |
||||
|
"event": eventName, |
||||
|
"data": data |
||||
|
] |
||||
|
|
||||
|
// 将事件转换为JSON字符串 |
||||
|
let jsonData = try JSONSerialization.data(withJSONObject: eventMap, options: []) |
||||
|
if let jsonString = String(data: jsonData, encoding: .utf8) { |
||||
|
// 发送到Flutter |
||||
|
DispatchQueue.main.async { |
||||
|
sink(jsonString) |
||||
|
} |
||||
|
} |
||||
|
} catch { |
||||
|
print("事件序列化失败: \(error.localizedDescription)") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
File diff suppressed because it is too large
@ -0,0 +1,15 @@ |
|||||
|
<?xml version="1.0" encoding="UTF-8"?> |
||||
|
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd"> |
||||
|
<plist version="1.0"> |
||||
|
<dict> |
||||
|
<key>NSBluetoothAlwaysUsageDescription</key> |
||||
|
<string>需要蓝牙权限与设备进行通信和传输音频数据</string> |
||||
|
<key>NSBluetoothPeripheralUsageDescription</key> |
||||
|
<string>需要蓝牙权限与设备进行通信和传输音频数据</string> |
||||
|
<key>UIBackgroundModes</key> |
||||
|
<array> |
||||
|
<string>bluetooth-central</string> |
||||
|
<string>audio</string> |
||||
|
</array> |
||||
|
</dict> |
||||
|
</plist> |
||||
@ -0,0 +1,90 @@ |
|||||
|
import Foundation |
||||
|
import CoreBluetooth |
||||
|
|
||||
|
/// BLE服务常量 |
||||
|
struct BleConst { |
||||
|
// ================== 协议相关UUID ================== |
||||
|
/// 主服务UUID - 0000ABC0-0000-1111-2222-123456789ABC |
||||
|
static let PRIMARY_SERVICE_UUID = CBUUID(string: "0000abc0-0000-1111-2222-123456789abc") |
||||
|
|
||||
|
/// 音频服务UUID - 00001801-0000-1000-8000-00805f9b34fb |
||||
|
static let AUDIO_SERVICE_UUID = CBUUID(string: "0000ae00-0000-1000-8000-00805f9b34fb") |
||||
|
|
||||
|
/// 接收音频特征UUID - 0000ABC1-0000-1111-2222-123456789ABC |
||||
|
static let RECEIVE_AUDIO_CHAR_UUID = CBUUID(string: "0000ae02-0000-1000-8000-00805f9b34fb") |
||||
|
|
||||
|
/// 写入特征UUID - 0000ABC1-0000-1111-2222-123456789ABC |
||||
|
static let WRITE_CHAR_UUID = CBUUID(string: "0000abc1-0000-1111-2222-123456789abc") |
||||
|
|
||||
|
/// 通知特征UUID - 0000ABC2-0000-1111-2222-123456789ABC |
||||
|
static let NOTIFY_CHAR_UUID = CBUUID(string: "0000abc2-0000-1111-2222-123456789abc") |
||||
|
|
||||
|
/// 客户端特征配置描述符UUID |
||||
|
static let CLIENT_CHAR_CONFIG_UUID = CBUUID(string: "0000fd7b-0000-1000-8000-00805f9b34fb") |
||||
|
|
||||
|
/// BLE扫描Action |
||||
|
static let ACTION_SCAN_RESULT = "com.yunqiinnovation.ble_service.ACTION_BLE_SCAN" |
||||
|
|
||||
|
// ================== 连接状态 ================== |
||||
|
static let STATE_DISCONNECTED = 0 |
||||
|
static let STATE_CONNECTING = 1 |
||||
|
static let STATE_CONNECTED = 2 |
||||
|
static let STATE_DISCONNECTING = 3 |
||||
|
static let STATE_ERROR = 4 |
||||
|
|
||||
|
// ================== 广播厂商信息 ================== |
||||
|
/// 厂商ID - 0xABEF |
||||
|
static let MANUFACTURER_ID: UInt16 = 0xABEF |
||||
|
/// 客户ID - 0xA001 |
||||
|
static let CUSTOMER_ID: UInt16 = 0xA001 |
||||
|
/// 产品ID - 0xA001 |
||||
|
static let PRODUCT_ID: UInt16 = 0xA001 |
||||
|
/// 蓝牙状态-未连接 |
||||
|
static let BT_STATE_DISCONNECTED: UInt8 = 0x00 |
||||
|
/// 蓝牙状态-已连接 |
||||
|
static let BT_STATE_CONNECTED: UInt8 = 0x0C |
||||
|
|
||||
|
// ================== 协议格式 ================== |
||||
|
/// APP请求帧头 |
||||
|
static let APP_REQUEST_HEADER: UInt8 = 0xAA |
||||
|
/// 设备响应帧头 |
||||
|
static let DEVICE_RESPONSE_HEADER: UInt8 = 0xBB |
||||
|
/// 设备主动上报帧头 |
||||
|
static let DEVICE_NOTIFY_HEADER: UInt8 = 0xCC |
||||
|
|
||||
|
// ================== 协议指令 (CMD) ================== |
||||
|
/// 获取耳机版本号 |
||||
|
static let CMD_GET_VERSION: UInt8 = 0x01 |
||||
|
/// 获取耳机产品ID |
||||
|
static let CMD_GET_PRODUCT_ID: UInt8 = 0x02 |
||||
|
/// 获取耳机颜色ID |
||||
|
static let CMD_GET_COLOR_ID: UInt8 = 0x03 |
||||
|
/// 获取耳机电量ID |
||||
|
static let CMD_GET_BATTERY_INFO: UInt8 = 0x04 |
||||
|
/// APP控制编解码指令 |
||||
|
static let CMD_CONTROL_CODEC: UInt8 = 0x05 |
||||
|
/// 音量增大 |
||||
|
static let CMD_VOLUME_UP: UInt8 = 0x06 |
||||
|
/// 音量减小 |
||||
|
static let CMD_VOLUME_DOWN: UInt8 = 0x07 |
||||
|
/// A2DP播放 |
||||
|
static let CMD_PLAY_A2DP: UInt8 = 0x08 |
||||
|
/// 唤醒指令 |
||||
|
static let CMD_WAKE_UP: UInt8 = 0x11 |
||||
|
|
||||
|
// ================== 编解码控制值 ================== |
||||
|
/// 关闭编解码指令 |
||||
|
static let CODEC_CONTROL_CLOSE: UInt8 = 0x00 |
||||
|
/// 打开解码指令 |
||||
|
static let CODEC_CONTROL_DECODE_ON: UInt8 = 0xA1 |
||||
|
/// 通过A2DP播放 |
||||
|
static let CODEC_CONTROL_A2DP_PLAY: UInt8 = 0xA2 |
||||
|
/// 打开编码指令 |
||||
|
static let CODEC_CONTROL_ENCODE_ON: UInt8 = 0xB1 |
||||
|
/// 左声道 |
||||
|
static let AUDIO_CHANNEL_LEFT: UInt8 = 0x01 |
||||
|
/// 右声道 |
||||
|
static let AUDIO_CHANNEL_RIGHT: UInt8 = 0x02 |
||||
|
/// 立体声 |
||||
|
static let AUDIO_CHANNEL_STEREO: UInt8 = 0x03 |
||||
|
} |
||||
@ -0,0 +1,25 @@ |
|||||
|
import Foundation |
||||
|
|
||||
|
/// BLE服务回调代理协议 |
||||
|
protocol BleDelegate: AnyObject { |
||||
|
/// 扫描结果回调 |
||||
|
/// - Parameter devices: 设备列表，每个设备包含id、name、rssi和uuids等信息 |
||||
|
func onScanResult(devices: [[String: Any]]) |
||||
|
|
||||
|
/// 连接状态改变回调 |
||||
|
/// - Parameter state: 连接状态，参考BleConst中的STATE_*常量 |
||||
|
func onConnectionStateChanged(state: Int) |
||||
|
|
||||
|
/// 音频数据接收回调 |
||||
|
/// - Parameter data: 接收到的音频数据 |
||||
|
func onAudioDataReceived(data: Data) |
||||
|
|
||||
|
/// 唤醒信号接收回调 |
||||
|
func onWakeupSignalReceived() |
||||
|
|
||||
|
/// 设备信息接收回调 |
||||
|
/// - Parameters: |
||||
|
/// - infoType: 信息类型 |
||||
|
/// - infoData: 信息数据 |
||||
|
func onDeviceInfoReceived(infoType: Int, infoData: [String: Any]) |
||||
|
} |
||||
@ -0,0 +1,963 @@ |
|||||
|
import Foundation |
||||
|
import CoreBluetooth |
||||
|
import os.log |
||||
|
|
||||
|
/// BLE服务类 - 单例实现 |
||||
|
actor BleService { |
||||
|
// MARK: - 单例实现 |
||||
|
static let shared = BleService() |
||||
|
|
||||
|
// MARK: - 常量 |
||||
|
|
||||
|
/// 用于状态恢复的唯一标识符 |
||||
|
private let restoreIdentifier = "com.yunqiinnovation.ble_service.restore" |
||||
|
|
||||
|
private init() { |
||||
|
// 初始化连接状态 |
||||
|
connectionState = BleConst.STATE_DISCONNECTED |
||||
|
|
||||
|
// 将通知处理移到非actor方法中 |
||||
|
setupNotifications() |
||||
|
} |
||||
|
|
||||
|
deinit { |
||||
|
// 移除通知观察者 |
||||
|
for observer in notificationObservers { |
||||
|
NotificationCenter.default.removeObserver(observer) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 属性 |
||||
|
|
||||
|
/// 中央管理器 |
||||
|
private var centralManager: CBCentralManager? |
||||
|
|
||||
|
/// 当前连接的外设 |
||||
|
private var peripheral: CBPeripheral? |
||||
|
|
||||
|
/// 主服务特征 |
||||
|
private var writeCharacteristic: CBCharacteristic? |
||||
|
private var notifyCharacteristic: CBCharacteristic? |
||||
|
private var audioCharacteristic: CBCharacteristic? |
||||
|
|
||||
|
/// 连接状态 |
||||
|
private(set) var connectionState: Int { |
||||
|
didSet { |
||||
|
if oldValue != connectionState { |
||||
|
Task { await notifyConnectionStateChanged(state: connectionState) } |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 是否正在扫描 |
||||
|
private(set) var isScanning = false |
||||
|
|
||||
|
/// 扫描结果缓存 |
||||
|
private var scanResults = [CBPeripheral: [String: Any]]() |
||||
|
|
||||
|
/// 扫描超时任务 |
||||
|
private var scanTimeoutTask: Task<Void, Never>? |
||||
|
|
||||
|
/// 是否已初始化 |
||||
|
private var isInitialized = false |
||||
|
|
||||
|
/// 回调代理列表 - 使用actor保证线程安全 |
||||
|
private var delegates = [BleDelegate]() |
||||
|
|
||||
|
/// 日志 |
||||
|
private let logger = OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService") |
||||
|
|
||||
|
/// 通知观察者 |
||||
|
private var notificationObservers = [NSObjectProtocol]() |
||||
|
|
||||
|
// MARK: - 通知设置 |
||||
|
|
||||
|
/// 设置应用生命周期通知 |
||||
|
private nonisolated func setupNotifications() { |
||||
|
// 使用nonisolated函数以避免actor隔离问题 |
||||
|
let becomeActiveObserver = NotificationCenter.default.addObserver( |
||||
|
forName: UIApplication.didBecomeActiveNotification, |
||||
|
object: nil, |
||||
|
queue: .main |
||||
|
) { [weak self] _ in |
||||
|
guard let self = self else { return } |
||||
|
Task { |
||||
|
await self.handleAppDidBecomeActive() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
let enterBackgroundObserver = NotificationCenter.default.addObserver( |
||||
|
forName: UIApplication.didEnterBackgroundNotification, |
||||
|
object: nil, |
||||
|
queue: .main |
||||
|
) { [weak self] _ in |
||||
|
guard let self = self else { return } |
||||
|
Task { |
||||
|
await self.handleAppDidEnterBackground() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 存储观察者以便后续移除 |
||||
|
notificationObservers = [becomeActiveObserver, enterBackgroundObserver] |
||||
|
} |
||||
|
|
||||
|
// MARK: - 公共方法 |
||||
|
|
||||
|
/// 初始化BLE服务 |
||||
|
/// - Returns: 是否初始化成功 |
||||
|
func initialize() -> Bool { |
||||
|
if isInitialized { |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
do { |
||||
|
// 初始化中央管理器 - 配置状态恢复选项 |
||||
|
let options: [String: Any] = [ |
||||
|
CBCentralManagerOptionRestoreIdentifierKey: restoreIdentifier, |
||||
|
CBCentralManagerOptionShowPowerAlertKey: true |
||||
|
] |
||||
|
|
||||
|
centralManager = CBCentralManager( |
||||
|
delegate: CBManagerHandler.shared, |
||||
|
queue: nil, |
||||
|
options: options |
||||
|
) |
||||
|
|
||||
|
CBManagerHandler.shared.bleService = self |
||||
|
|
||||
|
isInitialized = true |
||||
|
os_log("BLE服务初始化成功", log: logger, type: .info) |
||||
|
return true |
||||
|
} catch { |
||||
|
os_log("BLE服务初始化失败: %{public}@", log: logger, type: .error, error.localizedDescription) |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 添加回调代理 |
||||
|
/// - Parameter delegate: 要添加的代理 |
||||
|
/// - Returns: 是否添加成功 |
||||
|
func addDelegate(_ delegate: BleDelegate) -> Bool { |
||||
|
if !delegates.contains(where: { $0 === delegate }) { |
||||
|
delegates.append(delegate) |
||||
|
os_log("已添加BLE回调, 当前回调数: %d", log: logger, type: .debug, delegates.count) |
||||
|
return true |
||||
|
} |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
/// 移除回调代理 |
||||
|
/// - Parameter delegate: 要移除的代理 |
||||
|
/// - Returns: 是否移除成功 |
||||
|
func removeDelegate(_ delegate: BleDelegate) -> Bool { |
||||
|
if let index = delegates.firstIndex(where: { $0 === delegate }) { |
||||
|
delegates.remove(at: index) |
||||
|
os_log("已移除BLE回调, 当前回调数: %d", log: logger, type: .debug, delegates.count) |
||||
|
return true |
||||
|
} |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
/// 清除所有回调代理 |
||||
|
func clearDelegates() { |
||||
|
delegates.removeAll() |
||||
|
os_log("已清除所有BLE回调", log: logger, type: .debug) |
||||
|
} |
||||
|
|
||||
|
/// 检查是否已连接 |
||||
|
/// - Returns: 是否已连接 |
||||
|
func isConnected() -> Bool { |
||||
|
return connectionState == BleConst.STATE_CONNECTED |
||||
|
} |
||||
|
|
||||
|
// MARK: - 扫描方法 |
||||
|
|
||||
|
/// 开始主动扫描BLE设备 |
||||
|
/// - Parameter timeout: 扫描超时时间(秒) |
||||
|
/// - Returns: 是否成功开始扫描 |
||||
|
func startScan(timeout: TimeInterval = 15.0) -> Bool { |
||||
|
if !isInitialized { |
||||
|
os_log("BLE服务未初始化", log: logger, type: .error) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
if isScanning { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
guard let manager = centralManager, manager.state == .poweredOn else { |
||||
|
os_log("蓝牙未开启", log: logger, type: .error) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
os_log("开始主动扫描BLE设备...", log: logger, type: .info) |
||||
|
unregisterBackgroundScan() // 避免双重扫描 |
||||
|
scanResults.removeAll() |
||||
|
|
||||
|
// 扫描选项 - 允许重复结果，以获取最新的RSSI值 |
||||
|
let options: [String: Any] = [ |
||||
|
CBCentralManagerScanOptionAllowDuplicatesKey: true |
||||
|
] |
||||
|
|
||||
|
// 设置过滤服务UUID |
||||
|
let scanFilters = [ |
||||
|
// 如果需要过滤，添加CBUUID |
||||
|
// CBUUID(string: BleConst.PRIMARY_SERVICE_UUID.uuidString) |
||||
|
] |
||||
|
|
||||
|
// 使用过滤器开始扫描 |
||||
|
manager.scanForPeripherals( |
||||
|
withServices: scanFilters.isEmpty ? nil : scanFilters, |
||||
|
options: options |
||||
|
) |
||||
|
|
||||
|
isScanning = true |
||||
|
|
||||
|
// 设置扫描超时 |
||||
|
scanTimeoutTask?.cancel() |
||||
|
scanTimeoutTask = Task { [weak self] in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
do { |
||||
|
try await Task.sleep(nanoseconds: UInt64(timeout * 1_000_000_000)) |
||||
|
if !Task.isCancelled { |
||||
|
await self.stopScan() |
||||
|
} |
||||
|
} catch { |
||||
|
// 任务被取消，忽略错误 |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
/// 停止主动扫描 |
||||
|
/// - Returns: 是否成功停止扫描 |
||||
|
func stopScan() -> Bool { |
||||
|
if !isScanning { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
os_log("停止主动扫描", log: logger, type: .info) |
||||
|
|
||||
|
centralManager?.stopScan() |
||||
|
scanTimeoutTask?.cancel() |
||||
|
scanTimeoutTask = nil |
||||
|
isScanning = false |
||||
|
|
||||
|
// 重新注册后台扫描，保证唤醒能力 |
||||
|
registerBackgroundScan() |
||||
|
|
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
/// 注册后台扫描 |
||||
|
/// - Returns: 是否成功注册 |
||||
|
func registerBackgroundScan() -> Bool { |
||||
|
// iOS中无法直接实现与Android相同的后台扫描唤醒功能 |
||||
|
os_log("iOS不支持直接的后台扫描唤醒，将使用状态保存和恢复功能", log: logger, type: .info) |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
/// 注销后台扫描 |
||||
|
/// - Returns: 是否成功注销 |
||||
|
func unregisterBackgroundScan() -> Bool { |
||||
|
// iOS没有直接的后台扫描注销机制 |
||||
|
os_log("iOS不支持直接的后台扫描注销", log: logger, type: .info) |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
// MARK: - 连接方法 |
||||
|
|
||||
|
/// 连接到指定地址的BLE设备 |
||||
|
/// - Parameter address: 设备UUID字符串 |
||||
|
func connect(address: String) async { |
||||
|
if connectionState == BleConst.STATE_CONNECTED { |
||||
|
os_log("已经连接中，忽略连接请求", log: logger, type: .warning) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let central = centralManager, central.state == .poweredOn else { |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 关闭现有连接 |
||||
|
disconnectGatt() |
||||
|
|
||||
|
// 查找对应的外设 |
||||
|
if let uuid = UUID(uuidString: address), |
||||
|
let targetPeripheral = scanResults.keys.first(where: { $0.identifier == uuid }) { |
||||
|
// 如果在扫描结果中找到，直接连接 |
||||
|
os_log("从扫描结果中找到目标设备，开始连接...", log: logger, type: .debug) |
||||
|
await connectToPeripheral(targetPeripheral) |
||||
|
} else { |
||||
|
// 如果没有找到，尝试从已知设备中恢复 |
||||
|
os_log("尝试使用UUID恢复设备连接: %{public}@", log: logger, type: .debug, address) |
||||
|
if let uuid = UUID(uuidString: address) { |
||||
|
let peripherals = central.retrievePeripherals(withIdentifiers: [uuid]) |
||||
|
if let targetPeripheral = peripherals.first { |
||||
|
await connectToPeripheral(targetPeripheral) |
||||
|
} else { |
||||
|
os_log("未找到指定UUID的设备: %{public}@", log: logger, type: .error, address) |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
} |
||||
|
} else { |
||||
|
os_log("无效的UUID格式: %{public}@", log: logger, type: .error, address) |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 连接到指定外设 |
||||
|
/// - Parameter peripheral: 要连接的外设 |
||||
|
private func connectToPeripheral(_ peripheral: CBPeripheral) async { |
||||
|
self.peripheral = peripheral |
||||
|
peripheral.delegate = CBPeripheralHandler.shared |
||||
|
CBPeripheralHandler.shared.bleService = self |
||||
|
updateConnectionState(BleConst.STATE_CONNECTING) |
||||
|
centralManager?.connect(peripheral, options: nil) |
||||
|
} |
||||
|
|
||||
|
/// 断开当前连接 |
||||
|
func disconnect() async { |
||||
|
guard let peripheral = peripheral, let central = centralManager else { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
updateConnectionState(BleConst.STATE_DISCONNECTING) |
||||
|
central.cancelPeripheralConnection(peripheral) |
||||
|
} |
||||
|
|
||||
|
/// 安全关闭GATT连接 |
||||
|
private func disconnectGatt() { |
||||
|
if let peripheral = peripheral, let central = centralManager { |
||||
|
central.cancelPeripheralConnection(peripheral) |
||||
|
} |
||||
|
|
||||
|
peripheral = nil |
||||
|
writeCharacteristic = nil |
||||
|
notifyCharacteristic = nil |
||||
|
audioCharacteristic = nil |
||||
|
} |
||||
|
|
||||
|
/// 更新连接状态并通知回调 |
||||
|
private func updateConnectionState(_ state: Int) { |
||||
|
connectionState = state |
||||
|
} |
||||
|
|
||||
|
// MARK: - 设备通信方法 |
||||
|
|
||||
|
/// 获取设备版本号 |
||||
|
/// - Returns: 是否成功发送请求 |
||||
|
func getDeviceVersion() -> Bool { |
||||
|
return sendCommand(BleConst.CMD_GET_VERSION) |
||||
|
} |
||||
|
|
||||
|
/// 获取电池信息 |
||||
|
/// - Returns: 是否成功发送请求 |
||||
|
func getBatteryInfo() -> Bool { |
||||
|
return sendCommand(BleConst.CMD_GET_BATTERY_INFO) |
||||
|
} |
||||
|
|
||||
|
/// 打开编码器 |
||||
|
/// - Returns: 是否成功发送请求 |
||||
|
func openEncoder() -> Bool { |
||||
|
return sendCodecControl(BleConst.CODEC_CONTROL_ENCODE_ON) |
||||
|
} |
||||
|
|
||||
|
/// 打开解码器 |
||||
|
/// - Returns: 是否成功发送请求 |
||||
|
func openDecoder() -> Bool { |
||||
|
return sendCodecControl(BleConst.CODEC_CONTROL_DECODE_ON) |
||||
|
} |
||||
|
|
||||
|
/// 关闭编解码器 |
||||
|
/// - Returns: 是否成功发送请求 |
||||
|
func closeCodec() -> Bool { |
||||
|
return sendCodecControl(BleConst.CODEC_CONTROL_CLOSE) |
||||
|
} |
||||
|
|
||||
|
/// 播放A2DP |
||||
|
/// - Returns: 是否成功发送请求 |
||||
|
func playA2DP() -> Bool { |
||||
|
return sendCodecControl(BleConst.CODEC_CONTROL_A2DP_PLAY) |
||||
|
} |
||||
|
|
||||
|
/// 发送编解码控制指令 |
||||
|
/// - Parameter control: 控制值 |
||||
|
/// - Returns: 是否成功发送 |
||||
|
private func sendCodecControl(_ control: UInt8) -> Bool { |
||||
|
let data = Data([BleConst.APP_REQUEST_HEADER, BleConst.CMD_CONTROL_CODEC, control]) |
||||
|
return writeData(data) |
||||
|
} |
||||
|
|
||||
|
/// 发送命令 |
||||
|
/// - Parameter command: 命令字 |
||||
|
/// - Returns: 是否成功发送 |
||||
|
private func sendCommand(_ command: UInt8) -> Bool { |
||||
|
let data = Data([BleConst.APP_REQUEST_HEADER, command]) |
||||
|
return writeData(data) |
||||
|
} |
||||
|
|
||||
|
/// 写入数据 |
||||
|
/// - Parameter data: 要写入的数据 |
||||
|
/// - Returns: 是否成功写入 |
||||
|
private func writeData(_ data: Data) -> Bool { |
||||
|
guard let peripheral = peripheral, |
||||
|
let characteristic = writeCharacteristic, |
||||
|
connectionState == BleConst.STATE_CONNECTED else { |
||||
|
os_log("无法写入数据，设备未连接或特征未找到", log: logger, type: .error) |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
let hexString = data.map { String(format: "0x%02X", $0) }.joined(separator: " ") |
||||
|
os_log("写入数据: %{public}@", log: logger, type: .debug, hexString) |
||||
|
|
||||
|
peripheral.writeValue(data, for: characteristic, type: .withoutResponse) |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
// MARK: - 数据处理方法 |
||||
|
|
||||
|
/// 处理音频数据 |
||||
|
/// - Parameter data: 音频数据 |
||||
|
func processAudioData(_ data: Data) async { |
||||
|
os_log("收到音频数据: %d字节", log: logger, type: .debug, data.count) |
||||
|
await notifyAudioDataReceived(data: data) |
||||
|
} |
||||
|
|
||||
|
/// 处理设备响应 |
||||
|
/// - Parameter data: 设备响应数据 |
||||
|
func processDeviceResponse(_ data: Data) async { |
||||
|
guard data.count >= 2 else { return } |
||||
|
|
||||
|
let cmd = data[1] |
||||
|
os_log("收到设备响应: CMD=0x%02X", log: logger, type: .debug, cmd) |
||||
|
|
||||
|
switch cmd { |
||||
|
case BleConst.CMD_GET_VERSION: |
||||
|
if data.count >= 3 { |
||||
|
// 解析版本号 |
||||
|
let version = String(format: "%d.%d.%d", data[2], data.count > 3 ? data[3] : 0, data.count > 4 ? data[4] : 0) |
||||
|
let infoData: [String: Any] = ["version": version] |
||||
|
await notifyDeviceInfoReceived(infoType: Int(cmd), infoData: infoData) |
||||
|
} |
||||
|
case BleConst.CMD_GET_BATTERY_INFO: |
||||
|
if data.count >= 3 { |
||||
|
// 解析电池电量 |
||||
|
let batteryLevel = Int(data[2]) |
||||
|
let infoData: [String: Any] = ["level": batteryLevel] |
||||
|
await notifyDeviceInfoReceived(infoType: Int(cmd), infoData: infoData) |
||||
|
} |
||||
|
default: |
||||
|
os_log("未处理的响应命令: 0x%02X", log: logger, type: .debug, cmd) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理设备通知 |
||||
|
/// - Parameter data: 设备通知数据 |
||||
|
func processDeviceNotification(_ data: Data) async { |
||||
|
guard data.count >= 2 else { return } |
||||
|
|
||||
|
let notifyType = data[1] |
||||
|
os_log("收到设备通知: Type=0x%02X", log: logger, type: .debug, notifyType) |
||||
|
|
||||
|
switch notifyType { |
||||
|
case BleConst.CMD_WAKE_UP: |
||||
|
// 设备唤醒信号 |
||||
|
await notifyWakeupSignalReceived() |
||||
|
default: |
||||
|
os_log("未处理的通知类型: 0x%02X", log: logger, type: .debug, notifyType) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 应用状态处理 |
||||
|
|
||||
|
/// 处理应用进入活跃状态 |
||||
|
private func handleAppDidBecomeActive() async { |
||||
|
os_log("应用进入活跃状态", log: logger, type: .debug) |
||||
|
// 重新初始化蓝牙管理器，如果需要 |
||||
|
if centralManager == nil { |
||||
|
_ = initialize() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理应用进入后台 |
||||
|
private func handleAppDidEnterBackground() async { |
||||
|
os_log("应用进入后台", log: logger, type: .debug) |
||||
|
// 在iOS中，后台蓝牙操作有一定限制，这里可以做一些清理工作 |
||||
|
} |
||||
|
|
||||
|
// MARK: - 回调通知方法 |
||||
|
|
||||
|
/// 通知扫描结果 |
||||
|
/// - Parameter devices: 设备列表 |
||||
|
func notifyScanResult(devices: [[String: Any]]) async { |
||||
|
let currentDelegates = delegates |
||||
|
|
||||
|
for delegate in currentDelegates { |
||||
|
await MainActor.run { |
||||
|
delegate.onScanResult(devices: devices) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 通知连接状态变化 |
||||
|
/// - Parameter state: 连接状态 |
||||
|
func notifyConnectionStateChanged(state: Int) async { |
||||
|
let currentDelegates = delegates |
||||
|
|
||||
|
for delegate in currentDelegates { |
||||
|
await MainActor.run { |
||||
|
delegate.onConnectionStateChanged(state: state) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 通知音频数据接收 |
||||
|
/// - Parameter data: 音频数据 |
||||
|
func notifyAudioDataReceived(data: Data) async { |
||||
|
let currentDelegates = delegates |
||||
|
|
||||
|
for delegate in currentDelegates { |
||||
|
await MainActor.run { |
||||
|
delegate.onAudioDataReceived(data: data) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 通知唤醒信号接收 |
||||
|
func notifyWakeupSignalReceived() async { |
||||
|
let currentDelegates = delegates |
||||
|
|
||||
|
for delegate in currentDelegates { |
||||
|
await MainActor.run { |
||||
|
delegate.onWakeupSignalReceived() |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 通知设备信息接收 |
||||
|
/// - Parameters: |
||||
|
/// - infoType: 信息类型 |
||||
|
/// - infoData: 信息数据 |
||||
|
func notifyDeviceInfoReceived(infoType: Int, infoData: [String: Any]) async { |
||||
|
let currentDelegates = delegates |
||||
|
|
||||
|
for delegate in currentDelegates { |
||||
|
await MainActor.run { |
||||
|
delegate.onDeviceInfoReceived(infoType: infoType, infoData: infoData) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理扫描到的设备 |
||||
|
func handleDiscoveredPeripheral(_ peripheral: CBPeripheral, advertisementData: [String: Any], rssi: NSNumber) async { |
||||
|
// 打印外设信息 |
||||
|
os_log("扫描到设备: 名称=%{public}@, ID=%{public}@, RSSI=%d", log: logger, type: .debug, |
||||
|
peripheral.name ?? "未知", peripheral.identifier.uuidString, rssi.intValue) |
||||
|
|
||||
|
// 提取服务UUID |
||||
|
var uuids: [String] = [] |
||||
|
if let serviceUUIDs = advertisementData[CBAdvertisementDataServiceUUIDsKey] as? [CBUUID] { |
||||
|
uuids = serviceUUIDs.map { $0.uuidString } |
||||
|
} |
||||
|
|
||||
|
// 创建设备信息字典 |
||||
|
let deviceInfo: [String: Any] = [ |
||||
|
"id": peripheral.identifier.uuidString, |
||||
|
"name": peripheral.name ?? "未知", |
||||
|
"rssi": rssi.intValue, |
||||
|
"uuids": uuids |
||||
|
] |
||||
|
|
||||
|
// 更新扫描缓存 |
||||
|
scanResults[peripheral] = deviceInfo |
||||
|
|
||||
|
// 通知扫描结果 |
||||
|
await notifyScanResult(devices: [deviceInfo]) |
||||
|
} |
||||
|
|
||||
|
/// 处理连接结果 |
||||
|
func handleConnectionResult(peripheral: CBPeripheral, success: Bool, error: Error?) async { |
||||
|
if success { |
||||
|
os_log("已连接到设备: %{public}@", log: logger, type: .info, peripheral.identifier.uuidString) |
||||
|
updateConnectionState(BleConst.STATE_CONNECTED) |
||||
|
|
||||
|
// 开始服务发现 |
||||
|
peripheral.discoverServices([BleConst.PRIMARY_SERVICE_UUID, BleConst.AUDIO_SERVICE_UUID]) |
||||
|
} else { |
||||
|
os_log("连接设备失败: %{public}@", log: logger, type: .error, error?.localizedDescription ?? "未知错误") |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
|
||||
|
// 清理资源 |
||||
|
self.peripheral = nil |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理断开连接 |
||||
|
func handleDisconnection(peripheral: CBPeripheral, error: Error?) async { |
||||
|
if let error = error { |
||||
|
os_log("设备连接断开(错误): %{public}@", log: logger, type: .error, error.localizedDescription) |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
} else { |
||||
|
os_log("设备连接断开(正常)", log: logger, type: .info) |
||||
|
updateConnectionState(BleConst.STATE_DISCONNECTED) |
||||
|
} |
||||
|
|
||||
|
// 清理资源 |
||||
|
if self.peripheral?.identifier == peripheral.identifier { |
||||
|
self.peripheral = nil |
||||
|
self.writeCharacteristic = nil |
||||
|
self.notifyCharacteristic = nil |
||||
|
self.audioCharacteristic = nil |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理发现服务 |
||||
|
func handleDiscoveredServices(peripheral: CBPeripheral, error: Error?) async { |
||||
|
if let error = error { |
||||
|
os_log("发现服务失败: %{public}@", log: logger, type: .error, error.localizedDescription) |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let services = peripheral.services else { |
||||
|
os_log("没有发现任何服务", log: logger, type: .error) |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 发现所有服务的特征 |
||||
|
for service in services { |
||||
|
os_log("发现服务: %{public}@", log: logger, type: .debug, service.uuid.uuidString) |
||||
|
peripheral.discoverCharacteristics(nil, for: service) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理发现的特征 |
||||
|
func handleDiscoveredCharacteristics(for service: CBService, peripheral: CBPeripheral, error: Error?) async { |
||||
|
if let error = error { |
||||
|
os_log("发现特征失败: %{public}@", log: logger, type: .error, error.localizedDescription) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let characteristics = service.characteristics else { |
||||
|
os_log("服务没有任何特征: %{public}@", log: logger, type: .warning, service.uuid.uuidString) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 根据UUID识别和保存特征 |
||||
|
for characteristic in characteristics { |
||||
|
os_log("发现特征: %{public}@", log: logger, type: .debug, characteristic.uuid.uuidString) |
||||
|
|
||||
|
// 主服务特征 |
||||
|
if service.uuid == BleConst.PRIMARY_SERVICE_UUID { |
||||
|
if characteristic.uuid == BleConst.WRITE_CHAR_UUID { |
||||
|
writeCharacteristic = characteristic |
||||
|
os_log("找到写入特征", log: logger, type: .debug) |
||||
|
} else if characteristic.uuid == BleConst.NOTIFY_CHAR_UUID { |
||||
|
notifyCharacteristic = characteristic |
||||
|
os_log("找到通知特征", log: logger, type: .debug) |
||||
|
// 设置通知 |
||||
|
await setupNotification(for: characteristic, peripheral: peripheral) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 音频服务特征 |
||||
|
if service.uuid == BleConst.AUDIO_SERVICE_UUID { |
||||
|
if characteristic.uuid == BleConst.RECEIVE_AUDIO_CHAR_UUID { |
||||
|
audioCharacteristic = characteristic |
||||
|
os_log("找到音频特征", log: logger, type: .debug) |
||||
|
// 设置通知 |
||||
|
await setupNotification(for: characteristic, peripheral: peripheral) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 检查是否找到所有必要的特征 |
||||
|
if writeCharacteristic == nil || notifyCharacteristic == nil { |
||||
|
os_log("未找到所有必要的特征", log: logger, type: .error) |
||||
|
updateConnectionState(BleConst.STATE_ERROR) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 设置特征的通知 |
||||
|
/// - Parameters: |
||||
|
/// - characteristic: 特征 |
||||
|
/// - peripheral: 外设 |
||||
|
private func setupNotification(for characteristic: CBCharacteristic, peripheral: CBPeripheral) async { |
||||
|
if characteristic.properties.contains(.notify) { |
||||
|
peripheral.setNotifyValue(true, for: characteristic) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 处理特征值更新 |
||||
|
func handleValueUpdate(for characteristic: CBCharacteristic, peripheral: CBPeripheral, error: Error?) async { |
||||
|
if let error = error { |
||||
|
os_log("特征值更新失败: %{public}@", log: logger, type: .error, error.localizedDescription) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let data = characteristic.value else { |
||||
|
os_log("特征没有值", log: logger, type: .warning) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let hexString = data.map { String(format: "%02X", $0) }.joined(separator: " ") |
||||
|
os_log("收到特征数据: %{public}@ - %{public}@", log: logger, type: .debug, characteristic.uuid.uuidString, hexString) |
||||
|
|
||||
|
// 根据特征UUID处理不同类型的数据 |
||||
|
if characteristic.uuid == BleConst.RECEIVE_AUDIO_CHAR_UUID { |
||||
|
// 音频数据处理 |
||||
|
await processAudioData(data) |
||||
|
} else if characteristic.uuid == BleConst.NOTIFY_CHAR_UUID { |
||||
|
// 通知特征处理 |
||||
|
if !data.isEmpty { |
||||
|
switch data[0] { |
||||
|
case BleConst.DEVICE_RESPONSE_HEADER: |
||||
|
// 设备响应 |
||||
|
await processDeviceResponse(data) |
||||
|
case BleConst.DEVICE_NOTIFY_HEADER: |
||||
|
// 设备通知 |
||||
|
await processDeviceNotification(data) |
||||
|
default: |
||||
|
os_log("未知数据类型: %d", log: logger, type: .debug, data[0]) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 状态恢复处理 |
||||
|
|
||||
|
/// 处理蓝牙状态恢复 |
||||
|
func handleStateRestoration(dict: [String: Any]) async { |
||||
|
os_log("正在恢复蓝牙状态...", log: logger, type: .info) |
||||
|
|
||||
|
// 1. 恢复之前连接的外设 |
||||
|
if let peripherals = dict[CBCentralManagerRestoredStatePeripheralsKey] as? [CBPeripheral] { |
||||
|
os_log("恢复 %d 个已连接的外设", log: logger, type: .debug, peripherals.count) |
||||
|
|
||||
|
for peripheral in peripherals { |
||||
|
self.peripheral = peripheral |
||||
|
peripheral.delegate = CBPeripheralHandler.shared |
||||
|
CBPeripheralHandler.shared.bleService = self |
||||
|
|
||||
|
// 根据外设当前状态采取不同操作 |
||||
|
if peripheral.state == .connected { |
||||
|
os_log("外设 %{public}@ 已连接", log: logger, type: .debug, peripheral.identifier.uuidString) |
||||
|
updateConnectionState(BleConst.STATE_CONNECTED) |
||||
|
|
||||
|
// 重新发现服务 |
||||
|
peripheral.discoverServices([BleConst.PRIMARY_SERVICE_UUID, BleConst.AUDIO_SERVICE_UUID]) |
||||
|
} else if peripheral.state == .connecting { |
||||
|
os_log("外设 %{public}@ 正在连接中", log: logger, type: .debug, peripheral.identifier.uuidString) |
||||
|
updateConnectionState(BleConst.STATE_CONNECTING) |
||||
|
} else { |
||||
|
os_log("外设 %{public}@ 状态异常: %d", log: logger, type: .debug, peripheral.identifier.uuidString, peripheral.state.rawValue) |
||||
|
// 尝试重新连接 |
||||
|
updateConnectionState(BleConst.STATE_CONNECTING) |
||||
|
centralManager?.connect(peripheral, options: nil) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 2. 恢复扫描状态 |
||||
|
if let scanServices = dict[CBCentralManagerRestoredStateScanServicesKey] as? [CBUUID] { |
||||
|
os_log("恢复扫描任务，服务UUID数: %d", log: logger, type: .debug, scanServices.count) |
||||
|
|
||||
|
let scanOptions = dict[CBCentralManagerRestoredStateScanOptionsKey] as? [String: Any] ?? [ |
||||
|
CBCentralManagerScanOptionAllowDuplicatesKey: true |
||||
|
] |
||||
|
|
||||
|
// 重新启动扫描任务 |
||||
|
centralManager?.scanForPeripherals( |
||||
|
withServices: scanServices.isEmpty ? nil : scanServices, |
||||
|
options: scanOptions |
||||
|
) |
||||
|
|
||||
|
isScanning = true |
||||
|
|
||||
|
// 设置新的扫描超时（默认15秒） |
||||
|
scanTimeoutTask?.cancel() |
||||
|
scanTimeoutTask = Task { [weak self] in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
do { |
||||
|
try await Task.sleep(nanoseconds: UInt64(15.0 * 1_000_000_000)) |
||||
|
if !Task.isCancelled { |
||||
|
await self.stopScan() |
||||
|
} |
||||
|
} catch { |
||||
|
// 任务被取消，忽略错误 |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 通知外部：BLE状态已恢复 |
||||
|
await notifyStateRestored() |
||||
|
} |
||||
|
|
||||
|
/// 通知状态恢复 |
||||
|
private func notifyStateRestored() async { |
||||
|
let currentDelegates = delegates |
||||
|
|
||||
|
for delegate in currentDelegates { |
||||
|
if let stateDelegate = delegate as? BleStateRestorationDelegate { |
||||
|
await MainActor.run { |
||||
|
stateDelegate.onBleStateRestored() |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 中央管理器委托处理器 |
||||
|
|
||||
|
class CBManagerHandler: NSObject, CBCentralManagerDelegate { |
||||
|
static let shared = CBManagerHandler() |
||||
|
weak var bleService: BleService? |
||||
|
|
||||
|
func centralManagerDidUpdateState(_ central: CBCentralManager) { |
||||
|
Task { |
||||
|
guard let service = bleService else { return } |
||||
|
|
||||
|
switch central.state { |
||||
|
case .poweredOn: |
||||
|
await MainActor.run { |
||||
|
os_log("蓝牙已开启", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .info) |
||||
|
} |
||||
|
case .poweredOff: |
||||
|
await MainActor.run { |
||||
|
os_log("蓝牙已关闭", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .warning) |
||||
|
} |
||||
|
await service.updateConnectionState(BleConst.STATE_DISCONNECTED) |
||||
|
case .unauthorized, .unsupported: |
||||
|
await MainActor.run { |
||||
|
os_log("蓝牙不可用: %d", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .error, central.state.rawValue) |
||||
|
} |
||||
|
await service.updateConnectionState(BleConst.STATE_ERROR) |
||||
|
default: |
||||
|
await MainActor.run { |
||||
|
os_log("蓝牙状态变化: %d", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .debug, central.state.rawValue) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func centralManager(_ central: CBCentralManager, didDiscover peripheral: CBPeripheral, advertisementData: [String : Any], rssi RSSI: NSNumber) { |
||||
|
Task { |
||||
|
await bleService?.handleDiscoveredPeripheral(peripheral, advertisementData: advertisementData, rssi: RSSI) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func centralManager(_ central: CBCentralManager, didConnect peripheral: CBPeripheral) { |
||||
|
Task { |
||||
|
await bleService?.handleConnectionResult(peripheral: peripheral, success: true, error: nil) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func centralManager(_ central: CBCentralManager, didFailToConnect peripheral: CBPeripheral, error: Error?) { |
||||
|
Task { |
||||
|
await bleService?.handleConnectionResult(peripheral: peripheral, success: false, error: error) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func centralManager(_ central: CBCentralManager, didDisconnectPeripheral peripheral: CBPeripheral, error: Error?) { |
||||
|
Task { |
||||
|
await bleService?.handleDisconnection(peripheral: peripheral, error: error) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 状态恢复处理 |
||||
|
func centralManager(_ central: CBCentralManager, willRestoreState dict: [String : Any]) { |
||||
|
os_log("蓝牙状态恢复被触发", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .info) |
||||
|
|
||||
|
// 打印恢复信息 |
||||
|
if let peripherals = dict[CBCentralManagerRestoredStatePeripheralsKey] as? [CBPeripheral] { |
||||
|
os_log("恢复外设数量: %d", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .debug, peripherals.count) |
||||
|
for (i, peripheral) in peripherals.enumerated() { |
||||
|
os_log("恢复外设[%d]: %{public}@", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .debug, i, peripheral.identifier.uuidString) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if let services = dict[CBCentralManagerRestoredStateScanServicesKey] as? [CBUUID] { |
||||
|
os_log("恢复扫描服务数量: %d", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .debug, services.count) |
||||
|
} |
||||
|
|
||||
|
// 异步处理状态恢复 |
||||
|
Task { |
||||
|
if let service = bleService { |
||||
|
await service.handleStateRestoration(dict: dict) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 外设委托处理器 |
||||
|
|
||||
|
class CBPeripheralHandler: NSObject, CBPeripheralDelegate { |
||||
|
static let shared = CBPeripheralHandler() |
||||
|
weak var bleService: BleService? |
||||
|
|
||||
|
func peripheral(_ peripheral: CBPeripheral, didDiscoverServices error: Error?) { |
||||
|
Task { |
||||
|
await bleService?.handleDiscoveredServices(peripheral: peripheral, error: error) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func peripheral(_ peripheral: CBPeripheral, didDiscoverCharacteristicsFor service: CBService, error: Error?) { |
||||
|
Task { |
||||
|
await bleService?.handleDiscoveredCharacteristics(for: service, peripheral: peripheral, error: error) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func peripheral(_ peripheral: CBPeripheral, didUpdateValueFor characteristic: CBCharacteristic, error: Error?) { |
||||
|
Task { |
||||
|
await bleService?.handleValueUpdate(for: characteristic, peripheral: peripheral, error: error) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func peripheral(_ peripheral: CBPeripheral, didWriteValueFor characteristic: CBCharacteristic, error: Error?) { |
||||
|
if let error = error { |
||||
|
os_log("写入特征值失败: %{public}@", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .error, error.localizedDescription) |
||||
|
} else { |
||||
|
os_log("写入特征值成功", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .debug) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func peripheral(_ peripheral: CBPeripheral, didUpdateNotificationStateFor characteristic: CBCharacteristic, error: Error?) { |
||||
|
if let error = error { |
||||
|
os_log("设置通知状态失败: %{public}@", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .error, error.localizedDescription) |
||||
|
} else { |
||||
|
os_log("设置通知状态成功: %{public}@", log: OSLog(subsystem: "com.yunqiinnovation.ble_service", category: "BleService"), type: .debug, characteristic.uuid.uuidString) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - BLE状态恢复代理 |
||||
|
/// 蓝牙状态恢复代理 |
||||
|
protocol BleStateRestorationDelegate: AnyObject { |
||||
|
/// 当蓝牙状态被系统恢复时调用 |
||||
|
func onBleStateRestored() |
||||
|
} |
||||
|
|
||||
|
// MARK: - BLE代理 |
||||
|
/// 蓝牙事件代理 |
||||
|
protocol BleDelegate: AnyObject { |
||||
|
/// 扫描结果回调 |
||||
|
func onScanResult(devices: [[String: Any]]) |
||||
|
|
||||
|
/// 连接状态变化回调 |
||||
|
func onConnectionStateChanged(state: Int) |
||||
|
|
||||
|
/// 音频数据接收回调 |
||||
|
func onAudioDataReceived(data: Data) |
||||
|
|
||||
|
/// 唤醒信号接收回调 |
||||
|
func onWakeupSignalReceived() |
||||
|
|
||||
|
/// 设备信息接收回调 |
||||
|
func onDeviceInfoReceived(infoType: Int, infoData: [String: Any]) |
||||
|
} |
||||
@ -0,0 +1,4 @@ |
|||||
|
#import <Flutter/Flutter.h> |
||||
|
|
||||
|
@interface BleServicePlugin : NSObject<FlutterPlugin> |
||||
|
@end |
||||
@ -0,0 +1,12 @@ |
|||||
|
#import "BleServicePlugin.h" |
||||
|
#if __has_include(<ble_service/ble_service-Swift.h>) |
||||
|
#import <ble_service/ble_service-Swift.h> |
||||
|
#else |
||||
|
#import "ble_service-Swift.h" |
||||
|
#endif |
||||
|
|
||||
|
@implementation BleServicePlugin |
||||
|
+ (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar>*)registrar { |
||||
|
[SwiftBleServicePlugin registerWithRegistrar:registrar]; |
||||
|
} |
||||
|
@end |
||||
@ -0,0 +1,203 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
|
||||
|
public class SwiftBleServicePlugin: NSObject, FlutterPlugin, BleDelegate { |
||||
|
private let TAG = "BleServicePlugin" |
||||
|
|
||||
|
// 方法通道 |
||||
|
private var methodChannel: FlutterMethodChannel? |
||||
|
|
||||
|
// 事件通道 |
||||
|
private var dataEventSink: FlutterEventSink? |
||||
|
private var statusEventSink: FlutterEventSink? |
||||
|
|
||||
|
// BLE服务实例 |
||||
|
private let bleService = BleService.shared |
||||
|
|
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
let methodChannel = FlutterMethodChannel(name: "com.yunqiinnovation.ble_service", binaryMessenger: registrar.messenger()) |
||||
|
|
||||
|
// 创建插件实例 |
||||
|
let instance = SwiftBleServicePlugin() |
||||
|
instance.methodChannel = methodChannel |
||||
|
|
||||
|
// 设置方法通道处理器 |
||||
|
registrar.addMethodCallDelegate(instance, channel: methodChannel) |
||||
|
|
||||
|
// 设置事件通道 |
||||
|
instance.setupEventChannels(with: registrar) |
||||
|
|
||||
|
// 初始化BLE服务 |
||||
|
_ = instance.bleService.initialize() |
||||
|
|
||||
|
// 添加自身作为BLE回调 |
||||
|
_ = instance.bleService.addDelegate(instance) |
||||
|
} |
||||
|
|
||||
|
/// 设置事件通道 |
||||
|
private func setupEventChannels(with registrar: FlutterPluginRegistrar) { |
||||
|
// 数据接收事件通道 |
||||
|
let dataEventChannel = FlutterEventChannel(name: "com.yunqiinnovation.ble_service/data_events", binaryMessenger: registrar.messenger()) |
||||
|
dataEventChannel.setStreamHandler(StreamHandler { [weak self] sink in |
||||
|
self?.dataEventSink = sink |
||||
|
} onCancel: { [weak self] in |
||||
|
self?.dataEventSink = nil |
||||
|
}) |
||||
|
|
||||
|
// 状态更新事件通道 |
||||
|
let statusEventChannel = FlutterEventChannel(name: "com.yunqiinnovation.ble_service/status_events", binaryMessenger: registrar.messenger()) |
||||
|
statusEventChannel.setStreamHandler(StreamHandler { [weak self] sink in |
||||
|
self?.statusEventSink = sink |
||||
|
} onCancel: { [weak self] in |
||||
|
self?.statusEventSink = nil |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
/// 发送事件到Flutter |
||||
|
private func sendEvent(_ eventSink: FlutterEventSink?, _ data: Any) { |
||||
|
DispatchQueue.main.async { |
||||
|
eventSink?(data) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) { |
||||
|
NSLog("Method call: \(call.method)") |
||||
|
|
||||
|
switch call.method { |
||||
|
case "isConnected": |
||||
|
result(bleService.isConnected()) |
||||
|
|
||||
|
case "getBatteryInfo": |
||||
|
result(bleService.getBatteryInfo()) |
||||
|
|
||||
|
case "getDeviceVersion": |
||||
|
result(bleService.getDeviceVersion()) |
||||
|
|
||||
|
case "openEncoder": |
||||
|
result(bleService.openEncoder()) |
||||
|
|
||||
|
case "openDecoder": |
||||
|
result(bleService.openDecoder()) |
||||
|
|
||||
|
case "closeCodec": |
||||
|
result(bleService.closeCodec()) |
||||
|
|
||||
|
case "playA2DP": |
||||
|
result(bleService.playA2DP()) |
||||
|
|
||||
|
case "registerBackgroundScan": |
||||
|
result(bleService.registerBackgroundScan()) |
||||
|
|
||||
|
case "unregisterBackgroundScan": |
||||
|
result(bleService.unregisterBackgroundScan()) |
||||
|
|
||||
|
case "startBleService": |
||||
|
guard let config = call.arguments as? [String: Any] else { |
||||
|
result(FlutterError(code: "INVALID_ARGUMENTS", message: "配置参数无效", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 处理配置参数 |
||||
|
if let address = config["mac"] as? String { |
||||
|
// 如果提供了设备地址，尝试连接 |
||||
|
bleService.connect(address: address) |
||||
|
} |
||||
|
|
||||
|
// 开始扫描，寻找设备 |
||||
|
let scanSuccess = bleService.startScan() |
||||
|
result(scanSuccess) |
||||
|
|
||||
|
case "stopBleService": |
||||
|
// 停止扫描和断开连接 |
||||
|
_ = bleService.stopScan() |
||||
|
if bleService.isConnected() { |
||||
|
bleService.disconnect() |
||||
|
} |
||||
|
result(true) |
||||
|
|
||||
|
default: |
||||
|
result(FlutterMethodNotImplemented) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - BleDelegate Methods |
||||
|
|
||||
|
func onScanResult(devices: [[String: Any]]) { |
||||
|
// 目前Flutter层没有直接使用此事件，可以选择不转发 |
||||
|
} |
||||
|
|
||||
|
func onConnectionStateChanged(state: Int) { |
||||
|
// 将连接状态变化通过状态通道发送 |
||||
|
let stateStr: String |
||||
|
switch state { |
||||
|
case BleConst.STATE_DISCONNECTED: |
||||
|
stateStr = "Disconnected" |
||||
|
case BleConst.STATE_CONNECTING: |
||||
|
stateStr = "Connecting" |
||||
|
case BleConst.STATE_CONNECTED: |
||||
|
stateStr = "Connected" |
||||
|
case BleConst.STATE_DISCONNECTING: |
||||
|
stateStr = "Disconnecting" |
||||
|
case BleConst.STATE_ERROR: |
||||
|
stateStr = "Connection Error" |
||||
|
default: |
||||
|
stateStr = "Unknown State(\(state))" |
||||
|
} |
||||
|
|
||||
|
let stateMap: [String: Any] = [ |
||||
|
"type": "connectionStateChanged", |
||||
|
"state": state, |
||||
|
"stateStr": stateStr |
||||
|
] |
||||
|
|
||||
|
sendEvent(statusEventSink, stateMap) |
||||
|
} |
||||
|
|
||||
|
func onAudioDataReceived(data: Data) { |
||||
|
// 将音频数据通过数据通道发送 |
||||
|
let dataMap: [String: Any] = [ |
||||
|
"type": "audioData", |
||||
|
"data": FlutterStandardTypedData(bytes: data) |
||||
|
] |
||||
|
|
||||
|
sendEvent(dataEventSink, dataMap) |
||||
|
} |
||||
|
|
||||
|
func onWakeupSignalReceived() { |
||||
|
// 将唤醒事件通过状态通道发送 |
||||
|
sendEvent(statusEventSink, ["type": "wakeup"]) |
||||
|
} |
||||
|
|
||||
|
func onDeviceInfoReceived(infoType: Int, infoData: [String: Any]) { |
||||
|
// 将设备信息通过状态通道发送 |
||||
|
var resultMap = infoData |
||||
|
resultMap["type"] = "deviceInfo" |
||||
|
resultMap["result"] = infoType |
||||
|
|
||||
|
sendEvent(statusEventSink, resultMap) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 事件处理器类 |
||||
|
|
||||
|
/// 简化的事件处理器类 |
||||
|
class StreamHandler: NSObject, FlutterStreamHandler { |
||||
|
private let onListen: (FlutterEventSink) -> Void |
||||
|
private let onCancel: () -> Void |
||||
|
|
||||
|
init(onListen: @escaping (FlutterEventSink) -> Void, onCancel: @escaping () -> Void) { |
||||
|
self.onListen = onListen |
||||
|
self.onCancel = onCancel |
||||
|
super.init() |
||||
|
} |
||||
|
|
||||
|
func onListen(withArguments arguments: Any?, eventSink events: @escaping FlutterEventSink) -> FlutterError? { |
||||
|
onListen(events) |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func onCancel(withArguments arguments: Any?) -> FlutterError? { |
||||
|
onCancel() |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,23 @@ |
|||||
|
# |
||||
|
# To learn more about a Podspec see http://guides.cocoapods.org/syntax/podspec.html. |
||||
|
# Run `pod lib lint ble_service.podspec` to validate before publishing. |
||||
|
# |
||||
|
Pod::Spec.new do |s| |
||||
|
s.name = 'ble_service' |
||||
|
s.version = '0.0.1' |
||||
|
s.summary = 'BLE服务Flutter插件' |
||||
|
s.description = <<-DESC |
||||
|
BLE服务Flutter插件，提供蓝牙设备扫描、连接和通信功能。 |
||||
|
DESC |
||||
|
s.homepage = 'http://example.com' |
||||
|
s.license = { :file => '../LICENSE' } |
||||
|
s.author = { 'Your Company' => 'email@example.com' } |
||||
|
s.source = { :path => '.' } |
||||
|
s.source_files = 'Classes/**/*' |
||||
|
s.dependency 'Flutter' |
||||
|
s.platform = :ios, '11.0' |
||||
|
|
||||
|
# Flutter.framework does not contain a i386 slice. |
||||
|
s.pod_target_xcconfig = { 'DEFINES_MODULE' => 'YES', 'EXCLUDED_ARCHS[sdk=iphonesimulator*]' => 'i386' } |
||||
|
s.swift_version = '5.0' |
||||
|
end |
||||
@ -0,0 +1,410 @@ |
|||||
|
import Foundation |
||||
|
import SQLite3 |
||||
|
|
||||
|
/// ChatStorageHelper - iOS版本的聊天存储助手 |
||||
|
/// 负责SQLite数据库操作和业务逻辑 |
||||
|
class ChatStorageHelper { |
||||
|
// MARK: - Constants |
||||
|
private let TAG = "ChatStorageHelper" |
||||
|
|
||||
|
// 数据库常量 |
||||
|
private let DATABASE_NAME = "chat_messages.db" |
||||
|
private let DATABASE_VERSION = 1 |
||||
|
|
||||
|
// 表和列名 |
||||
|
private let TABLE_MESSAGES = "messages" |
||||
|
private let COLUMN_ID = "id" |
||||
|
private let COLUMN_SESSION_ID = "session_id" |
||||
|
private let COLUMN_MESSAGE = "message" |
||||
|
private let COLUMN_SENDER = "sender" |
||||
|
private let COLUMN_TIMESTAMP = "timestamp" |
||||
|
private let COLUMN_METADATA = "metadata" |
||||
|
|
||||
|
// MARK: - Properties |
||||
|
private var db: OpaquePointer? |
||||
|
private let dbPath: String |
||||
|
|
||||
|
// MARK: - Initialization |
||||
|
init(_ context: Any) { |
||||
|
// 获取数据库文件路径 |
||||
|
let fileManager = FileManager.default |
||||
|
if let documentsDirectory = fileManager.urls(for: .documentDirectory, in: .userDomainMask).first { |
||||
|
let databaseDirectory = documentsDirectory.appendingPathComponent("Databases") |
||||
|
|
||||
|
// 确保目录存在 |
||||
|
try? fileManager.createDirectory(at: databaseDirectory, withIntermediateDirectories: true) |
||||
|
|
||||
|
dbPath = databaseDirectory.appendingPathComponent(DATABASE_NAME).path |
||||
|
|
||||
|
// 打开数据库连接 |
||||
|
openDatabase() |
||||
|
} else { |
||||
|
dbPath = "" |
||||
|
print("\(TAG) 无法获取文档目录路径") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
deinit { |
||||
|
closeDatabase() |
||||
|
} |
||||
|
|
||||
|
// MARK: - Database Operations |
||||
|
|
||||
|
/// 打开数据库连接 |
||||
|
private func openDatabase() { |
||||
|
if sqlite3_open(dbPath, &db) == SQLITE_OK { |
||||
|
createTables() |
||||
|
print("\(TAG) 数据库已成功打开") |
||||
|
} else { |
||||
|
print("\(TAG) 无法打开数据库: \(String(describing: errorMessage))") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 关闭数据库连接 |
||||
|
private func closeDatabase() { |
||||
|
if db != nil { |
||||
|
sqlite3_close(db) |
||||
|
db = nil |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 创建必要的数据库表 |
||||
|
private func createTables() { |
||||
|
// 创建消息表 |
||||
|
let createTableSQL = """ |
||||
|
CREATE TABLE IF NOT EXISTS \(TABLE_MESSAGES) ( |
||||
|
\(COLUMN_ID) INTEGER PRIMARY KEY AUTOINCREMENT, |
||||
|
\(COLUMN_SESSION_ID) TEXT NOT NULL, |
||||
|
\(COLUMN_MESSAGE) TEXT NOT NULL, |
||||
|
\(COLUMN_SENDER) TEXT NOT NULL, |
||||
|
\(COLUMN_TIMESTAMP) TEXT NOT NULL, |
||||
|
\(COLUMN_METADATA) TEXT |
||||
|
); |
||||
|
""" |
||||
|
|
||||
|
if executeUpdate(sql: createTableSQL) { |
||||
|
// 创建索引以加快查询速度 |
||||
|
executeUpdate(sql: "CREATE INDEX IF NOT EXISTS idx_session_id ON \(TABLE_MESSAGES) (\(COLUMN_SESSION_ID))") |
||||
|
executeUpdate(sql: "CREATE INDEX IF NOT EXISTS idx_timestamp ON \(TABLE_MESSAGES) (\(COLUMN_TIMESTAMP))") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 获取SQLite错误消息 |
||||
|
private var errorMessage: String { |
||||
|
if let errorPointer = sqlite3_errmsg(db) { |
||||
|
let errorMessage = String(cString: errorPointer) |
||||
|
return errorMessage |
||||
|
} else { |
||||
|
return "未知错误" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 执行SQL更新操作（INSERT, UPDATE, DELETE, CREATE等） |
||||
|
private func executeUpdate(sql: String, parameters: [Any]? = nil) -> Bool { |
||||
|
var statement: OpaquePointer? |
||||
|
|
||||
|
if sqlite3_prepare_v2(db, sql, -1, &statement, nil) == SQLITE_OK { |
||||
|
// 绑定参数 |
||||
|
if let params = parameters { |
||||
|
for (index, param) in params.enumerated() { |
||||
|
bindParameter(statement: statement, index: index + 1, value: param) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 执行语句 |
||||
|
let result = sqlite3_step(statement) |
||||
|
sqlite3_finalize(statement) |
||||
|
|
||||
|
return result == SQLITE_DONE |
||||
|
} else { |
||||
|
print("\(TAG) 准备SQL语句失败: \(errorMessage)") |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 绑定SQL参数 |
||||
|
private func bindParameter(statement: OpaquePointer?, index: Int, value: Any) { |
||||
|
if let value = value as? Int { |
||||
|
sqlite3_bind_int(statement, Int32(index), Int32(value)) |
||||
|
} else if let value = value as? Double { |
||||
|
sqlite3_bind_double(statement, Int32(index), value) |
||||
|
} else if let value = value as? String { |
||||
|
sqlite3_bind_text(statement, Int32(index), (value as NSString).utf8String, -1, nil) |
||||
|
} else if value is NSNull { |
||||
|
sqlite3_bind_null(statement, Int32(index)) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 查询数据并处理结果 |
||||
|
private func executeQuery(sql: String, parameters: [Any]? = nil, handler: (OpaquePointer) -> Void) { |
||||
|
var statement: OpaquePointer? |
||||
|
|
||||
|
if sqlite3_prepare_v2(db, sql, -1, &statement, nil) == SQLITE_OK { |
||||
|
// 绑定参数 |
||||
|
if let params = parameters { |
||||
|
for (index, param) in params.enumerated() { |
||||
|
bindParameter(statement: statement, index: index + 1, value: param) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 处理结果 |
||||
|
while sqlite3_step(statement) == SQLITE_ROW { |
||||
|
handler(statement!) |
||||
|
} |
||||
|
} else { |
||||
|
print("\(TAG) 执行查询失败: \(errorMessage)") |
||||
|
} |
||||
|
|
||||
|
sqlite3_finalize(statement) |
||||
|
} |
||||
|
|
||||
|
// MARK: - Public Methods |
||||
|
|
||||
|
/// 保存消息 |
||||
|
/// - Parameters: |
||||
|
/// - sessionId: 会话ID |
||||
|
/// - message: 消息内容 |
||||
|
/// - sender: 发送者 |
||||
|
/// - metadata: 元数据 |
||||
|
/// - Returns: 成功返回消息ID，失败返回-1 |
||||
|
func saveMessage(sessionId: String, message: String, sender: String, metadata: String? = nil) -> Int64 { |
||||
|
let dateFormatter = DateFormatter() |
||||
|
dateFormatter.dateFormat = "yyyy-MM-dd HH:mm:ss" |
||||
|
let timestamp = dateFormatter.string(from: Date()) |
||||
|
|
||||
|
let sql = """ |
||||
|
INSERT INTO \(TABLE_MESSAGES) (\(COLUMN_SESSION_ID), \(COLUMN_MESSAGE), \(COLUMN_SENDER), \(COLUMN_TIMESTAMP), \(COLUMN_METADATA)) |
||||
|
VALUES (?, ?, ?, ?, ?) |
||||
|
""" |
||||
|
|
||||
|
var parameters: [Any] = [sessionId, message, sender, timestamp] |
||||
|
if let metadata = metadata { |
||||
|
parameters.append(metadata) |
||||
|
} else { |
||||
|
parameters.append(NSNull()) |
||||
|
} |
||||
|
|
||||
|
if executeUpdate(sql: sql, parameters: parameters) { |
||||
|
return sqlite3_last_insert_rowid(db) |
||||
|
} else { |
||||
|
print("\(TAG) 保存消息失败") |
||||
|
return -1 |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 获取指定会话的消息 |
||||
|
/// - Parameters: |
||||
|
/// - sessionId: 会话ID |
||||
|
/// - page: 页码 |
||||
|
/// - pageSize: 每页大小 |
||||
|
/// - Returns: JSON字符串，包含消息列表和分页信息 |
||||
|
func getMessages(sessionId: String, page: Int = 1, pageSize: Int = 20) -> [String: Any] { |
||||
|
var messages: [[String: Any]] = [] |
||||
|
let offset = (page - 1) * pageSize |
||||
|
|
||||
|
// 查询消息 |
||||
|
let sql = """ |
||||
|
SELECT * FROM \(TABLE_MESSAGES) |
||||
|
WHERE \(COLUMN_SESSION_ID) = ? |
||||
|
ORDER BY \(COLUMN_TIMESTAMP) DESC |
||||
|
LIMIT ? OFFSET ? |
||||
|
""" |
||||
|
|
||||
|
executeQuery(sql: sql, parameters: [sessionId, pageSize, offset]) { statement in |
||||
|
var message: [String: Any] = [:] |
||||
|
|
||||
|
// 获取ID |
||||
|
message["id"] = Int(sqlite3_column_int(statement, 0)) |
||||
|
|
||||
|
// 获取会话ID |
||||
|
if let sessionId = sqlite3_column_text(statement, 1) { |
||||
|
message["sessionId"] = String(cString: sessionId) |
||||
|
} |
||||
|
|
||||
|
// 获取消息内容 |
||||
|
if let messageText = sqlite3_column_text(statement, 2) { |
||||
|
message["message"] = String(cString: messageText) |
||||
|
} |
||||
|
|
||||
|
// 获取发送者 |
||||
|
if let sender = sqlite3_column_text(statement, 3) { |
||||
|
message["sender"] = String(cString: sender) |
||||
|
} |
||||
|
|
||||
|
// 获取时间戳 |
||||
|
if let timestamp = sqlite3_column_text(statement, 4) { |
||||
|
message["timestamp"] = String(cString: timestamp) |
||||
|
} |
||||
|
|
||||
|
// 获取元数据（如果有） |
||||
|
if sqlite3_column_type(statement, 5) != SQLITE_NULL { |
||||
|
if let metadata = sqlite3_column_text(statement, 5) { |
||||
|
message["metadata"] = String(cString: metadata) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
messages.append(message) |
||||
|
} |
||||
|
|
||||
|
// 获取总记录数 |
||||
|
var totalCount = 0 |
||||
|
let countSql = "SELECT COUNT(*) FROM \(TABLE_MESSAGES) WHERE \(COLUMN_SESSION_ID) = ?" |
||||
|
|
||||
|
executeQuery(sql: countSql, parameters: [sessionId]) { statement in |
||||
|
totalCount = Int(sqlite3_column_int(statement, 0)) |
||||
|
} |
||||
|
|
||||
|
// 计算总页数 |
||||
|
let totalPages = (totalCount + pageSize - 1) / pageSize |
||||
|
|
||||
|
return [ |
||||
|
"messages": messages, |
||||
|
"page": page, |
||||
|
"pageSize": pageSize, |
||||
|
"totalCount": totalCount, |
||||
|
"totalPages": totalPages |
||||
|
] |
||||
|
} |
||||
|
|
||||
|
/// 获取会话列表 |
||||
|
/// - Returns: 会话列表数据 |
||||
|
func getSessions() -> [[String: Any]] { |
||||
|
var sessions: [[String: Any]] = [] |
||||
|
|
||||
|
// 查询不同的会话ID及其最新消息 |
||||
|
let sql = """ |
||||
|
SELECT s1.\(COLUMN_SESSION_ID), s1.\(COLUMN_MESSAGE), s1.\(COLUMN_SENDER), s1.\(COLUMN_TIMESTAMP) |
||||
|
FROM \(TABLE_MESSAGES) s1 |
||||
|
INNER JOIN ( |
||||
|
SELECT \(COLUMN_SESSION_ID), MAX(\(COLUMN_TIMESTAMP)) as max_timestamp |
||||
|
FROM \(TABLE_MESSAGES) |
||||
|
GROUP BY \(COLUMN_SESSION_ID) |
||||
|
) s2 ON s1.\(COLUMN_SESSION_ID) = s2.\(COLUMN_SESSION_ID) AND s1.\(COLUMN_TIMESTAMP) = s2.max_timestamp |
||||
|
ORDER BY s1.\(COLUMN_TIMESTAMP) DESC |
||||
|
""" |
||||
|
|
||||
|
executeQuery(sql: sql) { statement in |
||||
|
var session: [String: Any] = [:] |
||||
|
|
||||
|
// 获取会话ID |
||||
|
if let sessionId = sqlite3_column_text(statement, 0) { |
||||
|
let sessionIdStr = String(cString: sessionId) |
||||
|
session["sessionId"] = sessionIdStr |
||||
|
|
||||
|
// 获取该会话的消息数量 |
||||
|
var messageCount = 0 |
||||
|
let countSql = "SELECT COUNT(*) FROM \(TABLE_MESSAGES) WHERE \(COLUMN_SESSION_ID) = ?" |
||||
|
|
||||
|
executeQuery(sql: countSql, parameters: [sessionIdStr]) { countStmt in |
||||
|
messageCount = Int(sqlite3_column_int(countStmt, 0)) |
||||
|
} |
||||
|
|
||||
|
session["messageCount"] = messageCount |
||||
|
} |
||||
|
|
||||
|
// 获取最后一条消息 |
||||
|
if let message = sqlite3_column_text(statement, 1) { |
||||
|
session["lastMessage"] = String(cString: message) |
||||
|
} |
||||
|
|
||||
|
// 获取最后发送者 |
||||
|
if let sender = sqlite3_column_text(statement, 2) { |
||||
|
session["lastSender"] = String(cString: sender) |
||||
|
} |
||||
|
|
||||
|
// 获取时间戳 |
||||
|
if let timestamp = sqlite3_column_text(statement, 3) { |
||||
|
session["timestamp"] = String(cString: timestamp) |
||||
|
} |
||||
|
|
||||
|
sessions.append(session) |
||||
|
} |
||||
|
|
||||
|
return sessions |
||||
|
} |
||||
|
|
||||
|
/// 删除消息 |
||||
|
/// - Parameters: |
||||
|
/// - sessionId: 会话ID，如果提供则删除整个会话 |
||||
|
/// - messageIds: 消息ID列表，如果提供则删除指定消息 |
||||
|
/// - Returns: 成功返回true，失败返回false |
||||
|
func deleteMessages(sessionId: String? = nil, messageIds: [Int]? = nil) -> Bool { |
||||
|
if let messageIds = messageIds, !messageIds.isEmpty { |
||||
|
// 删除指定ID的消息 |
||||
|
let placeholders = Array(repeating: "?", count: messageIds.count).joined(separator: ",") |
||||
|
let sql = "DELETE FROM \(TABLE_MESSAGES) WHERE \(COLUMN_ID) IN (\(placeholders))" |
||||
|
|
||||
|
return executeUpdate(sql: sql, parameters: messageIds.map { $0 }) |
||||
|
} else if let sessionId = sessionId, !sessionId.isEmpty { |
||||
|
// 删除会话中的所有消息 |
||||
|
let sql = "DELETE FROM \(TABLE_MESSAGES) WHERE \(COLUMN_SESSION_ID) = ?" |
||||
|
return executeUpdate(sql: sql, parameters: [sessionId]) |
||||
|
} |
||||
|
|
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
/// 清空数据库 |
||||
|
/// - Returns: 成功返回true，失败返回false |
||||
|
func clearDatabase() -> Bool { |
||||
|
let sql = "DELETE FROM \(TABLE_MESSAGES)" |
||||
|
return executeUpdate(sql: sql) |
||||
|
} |
||||
|
|
||||
|
/// 获取指定会话的最近消息 |
||||
|
/// - Parameters: |
||||
|
/// - sessionId: 会话ID |
||||
|
/// - limit: 限制数量 |
||||
|
/// - Returns: 消息列表，按时间正序排列 |
||||
|
func getRecentMessages(sessionId: String, limit: Int) -> [[String: Any]] { |
||||
|
var messages: [[String: Any]] = [] |
||||
|
|
||||
|
// 查询语句 - 按时间倒序获取最近的N条，然后在结果中再按时间正序 |
||||
|
let sql = """ |
||||
|
SELECT * FROM ( |
||||
|
SELECT * FROM \(TABLE_MESSAGES) |
||||
|
WHERE \(COLUMN_SESSION_ID) = ? |
||||
|
ORDER BY \(COLUMN_TIMESTAMP) DESC |
||||
|
LIMIT ? |
||||
|
) tmp ORDER BY \(COLUMN_TIMESTAMP) ASC |
||||
|
""" |
||||
|
|
||||
|
executeQuery(sql: sql, parameters: [sessionId, limit]) { statement in |
||||
|
var message: [String: Any] = [:] |
||||
|
|
||||
|
// 获取ID |
||||
|
message["id"] = Int(sqlite3_column_int(statement, 0)) |
||||
|
|
||||
|
// 获取会话ID |
||||
|
if let sessionId = sqlite3_column_text(statement, 1) { |
||||
|
message["sessionId"] = String(cString: sessionId) |
||||
|
} |
||||
|
|
||||
|
// 获取消息内容 |
||||
|
if let messageText = sqlite3_column_text(statement, 2) { |
||||
|
message["message"] = String(cString: messageText) |
||||
|
} |
||||
|
|
||||
|
// 获取发送者 |
||||
|
if let sender = sqlite3_column_text(statement, 3) { |
||||
|
message["sender"] = String(cString: sender) |
||||
|
} |
||||
|
|
||||
|
// 获取时间戳 |
||||
|
if let timestamp = sqlite3_column_text(statement, 4) { |
||||
|
message["timestamp"] = String(cString: timestamp) |
||||
|
} |
||||
|
|
||||
|
// 获取元数据（如果有） |
||||
|
if sqlite3_column_type(statement, 5) != SQLITE_NULL { |
||||
|
if let metadata = sqlite3_column_text(statement, 5) { |
||||
|
message["metadata"] = String(cString: metadata) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
messages.append(message) |
||||
|
} |
||||
|
|
||||
|
return messages |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,4 @@ |
|||||
|
#import <Flutter/Flutter.h> |
||||
|
|
||||
|
@interface ChatStoragePlugin : NSObject<FlutterPlugin> |
||||
|
@end |
||||
@ -0,0 +1,12 @@ |
|||||
|
#import "ChatStoragePlugin.h" |
||||
|
#if __has_include(<chat_storage/chat_storage-Swift.h>) |
||||
|
#import <chat_storage/chat_storage-Swift.h> |
||||
|
#else |
||||
|
#import "chat_storage-Swift.h" |
||||
|
#endif |
||||
|
|
||||
|
@implementation ChatStoragePlugin |
||||
|
+ (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar>*)registrar { |
||||
|
[SwiftChatStoragePlugin registerWithRegistrar:registrar]; |
||||
|
} |
||||
|
@end |
||||
@ -0,0 +1,182 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
|
||||
|
public class ChatStoragePlugin: NSObject, FlutterPlugin { |
||||
|
private let TAG = "ChatStoragePlugin" |
||||
|
private var chatStorageHelper: ChatStorageHelper? |
||||
|
|
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
let channel = FlutterMethodChannel(name: "chat_storage", binaryMessenger: registrar.messenger()) |
||||
|
let instance = ChatStoragePlugin() |
||||
|
registrar.addMethodCallDelegate(instance, channel: channel) |
||||
|
} |
||||
|
|
||||
|
override init() { |
||||
|
super.init() |
||||
|
chatStorageHelper = ChatStorageHelper(()) |
||||
|
print("\(TAG) 聊天存储插件已加载") |
||||
|
} |
||||
|
|
||||
|
public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) { |
||||
|
guard let helper = chatStorageHelper else { |
||||
|
result(FlutterError(code: "NOT_INITIALIZED", message: "存储助手未初始化", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
switch call.method { |
||||
|
case "saveMessage": |
||||
|
handleSaveMessage(call, result, helper) |
||||
|
|
||||
|
case "getMessages": |
||||
|
handleGetMessages(call, result, helper) |
||||
|
|
||||
|
case "getSessions": |
||||
|
handleGetSessions(call, result, helper) |
||||
|
|
||||
|
case "deleteMessages": |
||||
|
handleDeleteMessages(call, result, helper) |
||||
|
|
||||
|
case "clearDatabase": |
||||
|
handleClearDatabase(call, result, helper) |
||||
|
|
||||
|
case "getRecentMessages": |
||||
|
handleGetRecentMessages(call, result, helper) |
||||
|
|
||||
|
default: |
||||
|
result(FlutterMethodNotImplemented) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - Method handlers |
||||
|
|
||||
|
private func handleSaveMessage(_ call: FlutterMethodCall, _ result: @escaping FlutterResult, _ helper: ChatStorageHelper) { |
||||
|
guard let arguments = call.arguments as? [String: Any] else { |
||||
|
result(FlutterError(code: "INVALID_ARGS", message: "参数无效", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let sessionId = arguments["sessionId"] as? String, |
||||
|
let message = arguments["message"] as? String, |
||||
|
let sender = arguments["sender"] as? String else { |
||||
|
result(FlutterError(code: "MISSING_ARGS", message: "缺少必要参数", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let metadata = arguments["metadata"] as? String |
||||
|
|
||||
|
do { |
||||
|
let id = helper.saveMessage(sessionId: sessionId, message: message, sender: sender, metadata: metadata) |
||||
|
|
||||
|
if id != -1 { |
||||
|
result(["success": true, "id": id]) |
||||
|
} else { |
||||
|
result(FlutterError(code: "DB_ERROR", message: "插入消息失败", details: nil)) |
||||
|
} |
||||
|
} catch { |
||||
|
print("\(TAG) 保存消息时发生错误: \(error.localizedDescription)") |
||||
|
result(FlutterError(code: "EXCEPTION", message: "保存消息时发生错误: \(error.localizedDescription)", details: nil)) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func handleGetMessages(_ call: FlutterMethodCall, _ result: @escaping FlutterResult, _ helper: ChatStorageHelper) { |
||||
|
guard let arguments = call.arguments as? [String: Any] else { |
||||
|
result(FlutterError(code: "INVALID_ARGS", message: "参数无效", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let sessionId = arguments["sessionId"] as? String else { |
||||
|
result(FlutterError(code: "MISSING_ARGS", message: "缺少会话ID", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let page = arguments["page"] as? Int ?? 1 |
||||
|
let pageSize = arguments["pageSize"] as? Int ?? 20 |
||||
|
|
||||
|
do { |
||||
|
let messagesData = helper.getMessages(sessionId: sessionId, page: page, pageSize: pageSize) |
||||
|
|
||||
|
// 将结果转换为JSON字符串 |
||||
|
if let jsonData = try? JSONSerialization.data(withJSONObject: messagesData), |
||||
|
let jsonString = String(data: jsonData, encoding: .utf8) { |
||||
|
result(jsonString) |
||||
|
} else { |
||||
|
result(FlutterError(code: "JSON_ERROR", message: "无法序列化结果", details: nil)) |
||||
|
} |
||||
|
} catch { |
||||
|
print("\(TAG) 获取消息时发生错误: \(error.localizedDescription)") |
||||
|
result(FlutterError(code: "EXCEPTION", message: "获取消息时发生错误: \(error.localizedDescription)", details: nil)) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func handleGetSessions(_ call: FlutterMethodCall, _ result: @escaping FlutterResult, _ helper: ChatStorageHelper) { |
||||
|
do { |
||||
|
let sessions = helper.getSessions() |
||||
|
|
||||
|
// 将结果转换为JSON字符串 |
||||
|
if let jsonData = try? JSONSerialization.data(withJSONObject: sessions), |
||||
|
let jsonString = String(data: jsonData, encoding: .utf8) { |
||||
|
result(jsonString) |
||||
|
} else { |
||||
|
result(FlutterError(code: "JSON_ERROR", message: "无法序列化结果", details: nil)) |
||||
|
} |
||||
|
} catch { |
||||
|
print("\(TAG) 获取会话列表时发生错误: \(error.localizedDescription)") |
||||
|
result(FlutterError(code: "EXCEPTION", message: "获取会话列表时发生错误: \(error.localizedDescription)", details: nil)) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func handleDeleteMessages(_ call: FlutterMethodCall, _ result: @escaping FlutterResult, _ helper: ChatStorageHelper) { |
||||
|
guard let arguments = call.arguments as? [String: Any] else { |
||||
|
result(FlutterError(code: "INVALID_ARGS", message: "参数无效", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let sessionId = arguments["sessionId"] as? String |
||||
|
let messageIds = arguments["messageIds"] as? [Int] |
||||
|
|
||||
|
do { |
||||
|
let success = helper.deleteMessages(sessionId: sessionId, messageIds: messageIds) |
||||
|
|
||||
|
if success { |
||||
|
result(["success": true]) |
||||
|
} else { |
||||
|
result(FlutterError(code: "INVALID_ARGS", message: "必须提供sessionId或messageIds", details: nil)) |
||||
|
} |
||||
|
} catch { |
||||
|
print("\(TAG) 删除消息时发生错误: \(error.localizedDescription)") |
||||
|
result(FlutterError(code: "EXCEPTION", message: "删除消息时发生错误: \(error.localizedDescription)", details: nil)) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func handleClearDatabase(_ call: FlutterMethodCall, _ result: @escaping FlutterResult, _ helper: ChatStorageHelper) { |
||||
|
do { |
||||
|
let success = helper.clearDatabase() |
||||
|
result(["success": success]) |
||||
|
} catch { |
||||
|
print("\(TAG) 清空数据库时发生错误: \(error.localizedDescription)") |
||||
|
result(FlutterError(code: "EXCEPTION", message: "清空数据库时发生错误: \(error.localizedDescription)", details: nil)) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func handleGetRecentMessages(_ call: FlutterMethodCall, _ result: @escaping FlutterResult, _ helper: ChatStorageHelper) { |
||||
|
guard let arguments = call.arguments as? [String: Any] else { |
||||
|
result(FlutterError(code: "INVALID_ARGS", message: "参数无效", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
guard let sessionId = arguments["sessionId"] as? String else { |
||||
|
result(FlutterError(code: "MISSING_ARGS", message: "缺少会话ID", details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let limit = arguments["limit"] as? Int ?? 10 |
||||
|
|
||||
|
do { |
||||
|
let messages = helper.getRecentMessages(sessionId: sessionId, limit: limit) |
||||
|
result(messages) |
||||
|
} catch { |
||||
|
print("\(TAG) 获取最近消息时发生错误: \(error.localizedDescription)") |
||||
|
result(FlutterError(code: "EXCEPTION", message: "获取最近消息时发生错误: \(error.localizedDescription)", details: nil)) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,11 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
|
||||
|
// 这个类是为了满足Flutter插件的命名约定 |
||||
|
// 实际实现在ChatStoragePlugin.swift中 |
||||
|
public class SwiftChatStoragePlugin: NSObject, FlutterPlugin { |
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
// 调用实际实现类的注册方法 |
||||
|
ChatStoragePlugin.register(with: registrar) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,64 @@ |
|||||
|
# Chat Storage 插件 - iOS 实现 |
||||
|
|
||||
|
## 概述 |
||||
|
|
||||
|
Chat Storage 插件的 iOS 实现，为 Flutter 应用提供本地 SQLite 数据库支持，用于存储和管理聊天记录。 |
||||
|
|
||||
|
## 功能特性 |
||||
|
|
||||
|
- 基于 SQLite 的聊天记录存储 |
||||
|
- 支持按会话分组消息 |
||||
|
- 提供分页查询功能 |
||||
|
- 支持元数据存储 |
||||
|
- 完整的增删改查 API |
||||
|
|
||||
|
## 技术实现 |
||||
|
|
||||
|
本插件使用原生 Swift 实现，通过 SQLite C API 直接操作数据库，不依赖第三方 SQLite 封装库，以保持最小的依赖要求和最佳性能。 |
||||
|
|
||||
|
### 核心组件 |
||||
|
|
||||
|
- `ChatStorageHelper`: 负责 SQLite 数据库操作，提供完整的数据操作 API |
||||
|
- `ChatStoragePlugin`: 插件主类，处理与 Flutter 的通信 |
||||
|
|
||||
|
### 数据库结构 |
||||
|
|
||||
|
数据库包含一个 `messages` 表，结构如下： |
||||
|
|
||||
|
```sql |
||||
|
CREATE TABLE messages ( |
||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT, |
||||
|
session_id TEXT NOT NULL, |
||||
|
message TEXT NOT NULL, |
||||
|
sender TEXT NOT NULL, |
||||
|
timestamp TEXT NOT NULL, |
||||
|
metadata TEXT |
||||
|
); |
||||
|
|
||||
|
-- 索引 |
||||
|
CREATE INDEX idx_session_id ON messages (session_id); |
||||
|
CREATE INDEX idx_timestamp ON messages (timestamp); |
||||
|
``` |
||||
|
|
||||
|
## 使用注意事项 |
||||
|
|
||||
|
### SQLite 多线程访问 |
||||
|
|
||||
|
本插件所有数据库操作均在主线程执行，以避免 SQLite 多线程访问的复杂性。在处理大量数据时，建议在 Flutter 层面使用 `compute` 函数或 `Isolate` 进行异步处理。 |
||||
|
|
||||
|
### 数据格式 |
||||
|
|
||||
|
- 时间戳使用 `yyyy-MM-dd HH:mm:ss` 格式的字符串 |
||||
|
- JSON 转换遵循 Swift/Objective-C 的类型映射规则 |
||||
|
|
||||
|
## 调试与故障排除 |
||||
|
|
||||
|
- 日志前缀为 `ChatStoragePlugin` 和 `ChatStorageHelper` |
||||
|
- 可通过 Console.app 查看日志输出 |
||||
|
- 数据库文件存储在应用的 Documents/Databases 目录下 |
||||
|
|
||||
|
## 开发者须知 |
||||
|
|
||||
|
如果需要扩展此插件的功能，比如添加新的表或字段，应同时更新 Android 和 iOS 端的实现，并确保在版本升级时提供适当的数据库迁移策略。 |
||||
|
|
||||
|
数据库迁移逻辑目前实现为重建表，在生产环境中应该改为保留用户数据的方式。 |
||||
@ -0,0 +1,197 @@ |
|||||
|
import Foundation |
||||
|
import ExternalAccessory |
||||
|
import CoreBluetooth |
||||
|
|
||||
|
/// 蓝牙事件类型 |
||||
|
enum BluetoothEventType { |
||||
|
case stateChanged |
||||
|
case deviceConnected |
||||
|
case deviceDisconnected |
||||
|
} |
||||
|
|
||||
|
/// 蓝牙状态枚举 |
||||
|
enum BluetoothState: String { |
||||
|
case unknown = "unknown" |
||||
|
case off = "off" |
||||
|
case on = "on" |
||||
|
case turningOn = "turningOn" |
||||
|
case turningOff = "turningOff" |
||||
|
} |
||||
|
|
||||
|
/// 经典蓝牙助手类，负责iOS上的蓝牙设备管理 |
||||
|
class ClassicBluetoothHelper: NSObject { |
||||
|
private let TAG = "ClassicBluetoothHelper" |
||||
|
|
||||
|
// 状态监控相关 |
||||
|
private var isMonitoring = false |
||||
|
private var stateCallback: ((BluetoothEventType, [String: Any]) -> Void)? |
||||
|
|
||||
|
// 用于监控蓝牙系统状态 |
||||
|
private var centralManager: CBCentralManager? |
||||
|
|
||||
|
// 通知观察者 |
||||
|
private var accessoryConnectionObserver: NSObjectProtocol? |
||||
|
private var accessoryDisconnectionObserver: NSObjectProtocol? |
||||
|
|
||||
|
override init() { |
||||
|
super.init() |
||||
|
centralManager = CBCentralManager(delegate: nil, queue: nil, options: [CBCentralManagerOptionShowPowerAlertKey: false]) |
||||
|
} |
||||
|
|
||||
|
deinit { |
||||
|
stopBluetoothStateMonitoring() |
||||
|
} |
||||
|
|
||||
|
/// 检查蓝牙是否启用 |
||||
|
func isBluetoothEnabled() -> Bool { |
||||
|
guard let centralManager = centralManager else { return false } |
||||
|
return centralManager.state == .poweredOn |
||||
|
} |
||||
|
|
||||
|
/// 获取已连接的蓝牙设备 |
||||
|
/// - Parameter completion: 完成回调 |
||||
|
func getConnectedDevices(completion: @escaping ([[String: String]], Error?) -> Void) { |
||||
|
// 获取所有已连接的外部配件 |
||||
|
let accessoryManager = EAAccessoryManager.shared() |
||||
|
let connectedAccessories = accessoryManager.connectedAccessories |
||||
|
|
||||
|
// 将配件转换为设备信息数组 |
||||
|
let devices = connectedAccessories.map { accessory -> [String: String] in |
||||
|
return [ |
||||
|
"name": accessory.name, |
||||
|
"address": accessory.serialNumber // iOS使用序列号作为唯一标识 |
||||
|
] |
||||
|
} |
||||
|
|
||||
|
// 返回结果 |
||||
|
completion(devices, nil) |
||||
|
} |
||||
|
|
||||
|
/// 开始监听蓝牙状态变化 |
||||
|
/// - Parameter callback: 状态变化回调 |
||||
|
func startBluetoothStateMonitoring(callback: @escaping (BluetoothEventType, [String: Any]) -> Void) { |
||||
|
self.stateCallback = callback |
||||
|
|
||||
|
if !isMonitoring { |
||||
|
isMonitoring = true |
||||
|
|
||||
|
// 设置蓝牙中央管理器代理 |
||||
|
centralManager?.delegate = self |
||||
|
|
||||
|
// 注册蓝牙配件连接/断开通知 |
||||
|
let notificationCenter = NotificationCenter.default |
||||
|
|
||||
|
// 观察配件连接 |
||||
|
accessoryConnectionObserver = notificationCenter.addObserver( |
||||
|
forName: NSNotification.Name.EAAccessoryDidConnect, |
||||
|
object: nil, |
||||
|
queue: nil |
||||
|
) { [weak self] notification in |
||||
|
guard let self = self, |
||||
|
let accessory = notification.userInfo?[EAAccessoryKey] as? EAAccessory else { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let device: [String: String] = [ |
||||
|
"name": accessory.name, |
||||
|
"address": accessory.serialNumber |
||||
|
] |
||||
|
|
||||
|
self.stateCallback?(.deviceConnected, ["device": device]) |
||||
|
} |
||||
|
|
||||
|
// 观察配件断开 |
||||
|
accessoryDisconnectionObserver = notificationCenter.addObserver( |
||||
|
forName: NSNotification.Name.EAAccessoryDidDisconnect, |
||||
|
object: nil, |
||||
|
queue: nil |
||||
|
) { [weak self] notification in |
||||
|
guard let self = self, |
||||
|
let accessory = notification.userInfo?[EAAccessoryKey] as? EAAccessory else { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let device: [String: String] = [ |
||||
|
"name": accessory.name, |
||||
|
"address": accessory.serialNumber |
||||
|
] |
||||
|
|
||||
|
self.stateCallback?(.deviceDisconnected, ["device": device]) |
||||
|
} |
||||
|
|
||||
|
// 开始监视配件变化 |
||||
|
EAAccessoryManager.shared().registerForLocalNotifications() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 停止监听蓝牙状态变化 |
||||
|
func stopBluetoothStateMonitoring() { |
||||
|
if isMonitoring { |
||||
|
isMonitoring = false |
||||
|
|
||||
|
// 移除蓝牙中央管理器代理 |
||||
|
centralManager?.delegate = nil |
||||
|
|
||||
|
// 注销配件通知观察者 |
||||
|
if let observer = accessoryConnectionObserver { |
||||
|
NotificationCenter.default.removeObserver(observer) |
||||
|
accessoryConnectionObserver = nil |
||||
|
} |
||||
|
|
||||
|
if let observer = accessoryDisconnectionObserver { |
||||
|
NotificationCenter.default.removeObserver(observer) |
||||
|
accessoryDisconnectionObserver = nil |
||||
|
} |
||||
|
|
||||
|
// 停止监视配件变化 |
||||
|
EAAccessoryManager.shared().unregisterForLocalNotifications() |
||||
|
} |
||||
|
|
||||
|
// 清除回调 |
||||
|
stateCallback = nil |
||||
|
} |
||||
|
|
||||
|
/// 报告当前蓝牙状态 |
||||
|
func reportCurrentBluetoothState() { |
||||
|
guard let centralManager = centralManager else { return } |
||||
|
|
||||
|
let state: BluetoothState |
||||
|
|
||||
|
switch centralManager.state { |
||||
|
case .poweredOn: |
||||
|
state = .on |
||||
|
case .poweredOff: |
||||
|
state = .off |
||||
|
case .resetting: |
||||
|
state = .turningOff // 最接近的近似值 |
||||
|
case .unauthorized, .unsupported: |
||||
|
state = .off // 功能上等同于关闭 |
||||
|
default: |
||||
|
state = .unknown |
||||
|
} |
||||
|
|
||||
|
stateCallback?(.stateChanged, ["state": state.rawValue]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - CBCentralManagerDelegate |
||||
|
extension ClassicBluetoothHelper: CBCentralManagerDelegate { |
||||
|
func centralManagerDidUpdateState(_ central: CBCentralManager) { |
||||
|
let state: BluetoothState |
||||
|
|
||||
|
switch central.state { |
||||
|
case .poweredOn: |
||||
|
state = .on |
||||
|
case .poweredOff: |
||||
|
state = .off |
||||
|
case .resetting: |
||||
|
state = .turningOff |
||||
|
case .unauthorized, .unsupported: |
||||
|
state = .off |
||||
|
default: |
||||
|
state = .unknown |
||||
|
} |
||||
|
|
||||
|
stateCallback?(.stateChanged, ["state": state.rawValue]) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,4 @@ |
|||||
|
#import <Flutter/Flutter.h> |
||||
|
|
||||
|
@interface ClassicBluetoothPlugin : NSObject<FlutterPlugin> |
||||
|
@end |
||||
@ -0,0 +1,12 @@ |
|||||
|
#import "ClassicBluetoothPlugin.h" |
||||
|
#if __has_include(<classic_bluetooth/classic_bluetooth-Swift.h>) |
||||
|
#import <classic_bluetooth/classic_bluetooth-Swift.h> |
||||
|
#else |
||||
|
#import "classic_bluetooth-Swift.h" |
||||
|
#endif |
||||
|
|
||||
|
@implementation ClassicBluetoothPlugin |
||||
|
+ (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar>*)registrar { |
||||
|
[SwiftClassicBluetoothPlugin registerWithRegistrar:registrar]; |
||||
|
} |
||||
|
@end |
||||
@ -0,0 +1,108 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
import ExternalAccessory |
||||
|
|
||||
|
public class ClassicBluetoothPlugin: NSObject, FlutterPlugin { |
||||
|
private let TAG = "ClassicBluetoothPlugin" |
||||
|
private var methodChannel: FlutterMethodChannel? |
||||
|
private var eventChannel: FlutterEventChannel? |
||||
|
private var eventSink: FlutterEventSink? |
||||
|
private let bluetoothHelper = ClassicBluetoothHelper() |
||||
|
|
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
let channel = FlutterMethodChannel(name: "com.yunqiinnovation.classic_bluetooth/methods", binaryMessenger: registrar.messenger()) |
||||
|
let instance = ClassicBluetoothPlugin() |
||||
|
registrar.addMethodCallDelegate(instance, channel: channel) |
||||
|
|
||||
|
// 设置事件通道（用于蓝牙状态监控） |
||||
|
let eventChannel = FlutterEventChannel(name: "com.yunqiinnovation.classic_bluetooth/events", binaryMessenger: registrar.messenger()) |
||||
|
eventChannel.setStreamHandler(instance) |
||||
|
|
||||
|
instance.methodChannel = channel |
||||
|
instance.eventChannel = eventChannel |
||||
|
} |
||||
|
|
||||
|
public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) { |
||||
|
switch call.method { |
||||
|
case "getConnectedDevices": |
||||
|
getConnectedDevices(result: result) |
||||
|
case "isBluetoothEnabled": |
||||
|
result(bluetoothHelper.isBluetoothEnabled()) |
||||
|
default: |
||||
|
result(FlutterMethodNotImplemented) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func getConnectedDevices(result: @escaping FlutterResult) { |
||||
|
if !bluetoothHelper.isBluetoothEnabled() { |
||||
|
result([]) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
bluetoothHelper.getConnectedDevices { devices, error in |
||||
|
if let error = error { |
||||
|
result(FlutterError(code: "BLUETOOTH_ERROR", message: error.localizedDescription, details: nil)) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
result(devices) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 发送事件到Flutter |
||||
|
private func sendEvent(_ event: [String: Any]) { |
||||
|
if let eventSink = eventSink { |
||||
|
eventSink(event) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - FlutterStreamHandler |
||||
|
extension ClassicBluetoothPlugin: FlutterStreamHandler { |
||||
|
public func onListen(withArguments arguments: Any?, eventSink events: @escaping FlutterEventSink) -> FlutterError? { |
||||
|
self.eventSink = events |
||||
|
|
||||
|
// 开始监听蓝牙状态变化 |
||||
|
bluetoothHelper.startBluetoothStateMonitoring { [weak self] eventType, data in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
switch eventType { |
||||
|
case .stateChanged: |
||||
|
if let state = data["state"] as? String { |
||||
|
self.sendEvent([ |
||||
|
"type": "stateChanged", |
||||
|
"state": state, |
||||
|
"timestamp": Int(Date().timeIntervalSince1970 * 1000) |
||||
|
]) |
||||
|
} |
||||
|
case .deviceConnected: |
||||
|
if let device = data["device"] as? [String: String] { |
||||
|
self.sendEvent([ |
||||
|
"type": "deviceConnected", |
||||
|
"device": device, |
||||
|
"timestamp": Int(Date().timeIntervalSince1970 * 1000) |
||||
|
]) |
||||
|
} |
||||
|
case .deviceDisconnected: |
||||
|
if let device = data["device"] as? [String: String] { |
||||
|
self.sendEvent([ |
||||
|
"type": "deviceDisconnected", |
||||
|
"device": device, |
||||
|
"timestamp": Int(Date().timeIntervalSince1970 * 1000) |
||||
|
]) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 报告当前蓝牙状态 |
||||
|
bluetoothHelper.reportCurrentBluetoothState() |
||||
|
|
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
public func onCancel(withArguments arguments: Any?) -> FlutterError? { |
||||
|
eventSink = nil |
||||
|
bluetoothHelper.stopBluetoothStateMonitoring() |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,136 @@ |
|||||
|
import Foundation |
||||
|
import os.log |
||||
|
|
||||
|
/// iOS端的文件日志助手类，提供与Android端FileLogger类似的功能 |
||||
|
class FileLogger { |
||||
|
|
||||
|
private static let logQueue = DispatchQueue(label: "com.yunqiinnovation.classic_bluetooth.logger", qos: .background) |
||||
|
private static let dateFormatter: DateFormatter = { |
||||
|
let formatter = DateFormatter() |
||||
|
formatter.dateFormat = "yyyy-MM-dd HH:mm:ss.SSS" |
||||
|
return formatter |
||||
|
}() |
||||
|
|
||||
|
/// 日志类型 |
||||
|
enum LogLevel: String { |
||||
|
case debug = "DEBUG" |
||||
|
case info = "INFO" |
||||
|
case warning = "WARNING" |
||||
|
case error = "ERROR" |
||||
|
} |
||||
|
|
||||
|
/// 输出调试日志 |
||||
|
/// - Parameters: |
||||
|
/// - tag: 标签 |
||||
|
/// - message: 日志消息 |
||||
|
static func d(_ tag: String, _ message: String) { |
||||
|
log(level: .debug, tag: tag, message: message) |
||||
|
} |
||||
|
|
||||
|
/// 输出信息日志 |
||||
|
/// - Parameters: |
||||
|
/// - tag: 标签 |
||||
|
/// - message: 日志消息 |
||||
|
static func i(_ tag: String, _ message: String) { |
||||
|
log(level: .info, tag: tag, message: message) |
||||
|
} |
||||
|
|
||||
|
/// 输出警告日志 |
||||
|
/// - Parameters: |
||||
|
/// - tag: 标签 |
||||
|
/// - message: 日志消息 |
||||
|
static func w(_ tag: String, _ message: String) { |
||||
|
log(level: .warning, tag: tag, message: message) |
||||
|
} |
||||
|
|
||||
|
/// 输出错误日志 |
||||
|
/// - Parameters: |
||||
|
/// - tag: 标签 |
||||
|
/// - message: 日志消息 |
||||
|
/// - error: 可选的错误对象 |
||||
|
static func e(_ tag: String, _ message: String, _ error: Error? = nil) { |
||||
|
var fullMessage = message |
||||
|
if let error = error { |
||||
|
fullMessage += " - Error: \(error.localizedDescription)" |
||||
|
} |
||||
|
log(level: .error, tag: tag, message: fullMessage) |
||||
|
} |
||||
|
|
||||
|
/// 输出日志 |
||||
|
/// - Parameters: |
||||
|
/// - level: 日志级别 |
||||
|
/// - tag: 标签 |
||||
|
/// - message: 日志消息 |
||||
|
private static func log(level: LogLevel, tag: String, message: String) { |
||||
|
let timestamp = dateFormatter.string(from: Date()) |
||||
|
let logMessage = "[\(timestamp)] [\(level.rawValue)] [\(tag)]: \(message)" |
||||
|
|
||||
|
// 输出到控制台 |
||||
|
print(logMessage) |
||||
|
|
||||
|
// 异步写入日志文件 |
||||
|
logQueue.async { |
||||
|
writeToLogFile(logMessage) |
||||
|
} |
||||
|
|
||||
|
// 使用系统日志 |
||||
|
if #available(iOS 10.0, *) { |
||||
|
let osLog = OSLog(subsystem: "com.yunqiinnovation.classic_bluetooth", category: tag) |
||||
|
let osLogType: OSLogType |
||||
|
|
||||
|
switch level { |
||||
|
case .debug: |
||||
|
osLogType = .debug |
||||
|
case .info: |
||||
|
osLogType = .info |
||||
|
case .warning: |
||||
|
osLogType = .default |
||||
|
case .error: |
||||
|
osLogType = .error |
||||
|
} |
||||
|
|
||||
|
os_log("%{public}@", log: osLog, type: osLogType, message) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 将日志写入文件 |
||||
|
/// - Parameter message: 日志消息 |
||||
|
private static func writeToLogFile(_ message: String) { |
||||
|
guard let documentsDirectory = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first else { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
let logDirectoryURL = documentsDirectory.appendingPathComponent("Logs") |
||||
|
let dateString = DateFormatter().with { $0.dateFormat = "yyyy-MM-dd" }.string(from: Date()) |
||||
|
let logFileURL = logDirectoryURL.appendingPathComponent("bluetooth_\(dateString).log") |
||||
|
|
||||
|
do { |
||||
|
// 创建日志文件夹 |
||||
|
try FileManager.default.createDirectory(at: logDirectoryURL, withIntermediateDirectories: true) |
||||
|
|
||||
|
// 如果文件不存在，创建文件 |
||||
|
if !FileManager.default.fileExists(atPath: logFileURL.path) { |
||||
|
FileManager.default.createFile(atPath: logFileURL.path, contents: nil) |
||||
|
} |
||||
|
|
||||
|
// 追加日志内容 |
||||
|
if let fileHandle = FileHandle(forWritingAtPath: logFileURL.path) { |
||||
|
fileHandle.seekToEndOfFile() |
||||
|
if let data = "\(message)\n".data(using: .utf8) { |
||||
|
fileHandle.write(data) |
||||
|
} |
||||
|
fileHandle.closeFile() |
||||
|
} |
||||
|
} catch { |
||||
|
print("写入日志文件失败: \(error.localizedDescription)") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - DateFormatter 扩展 |
||||
|
extension DateFormatter { |
||||
|
func with(_ block: (DateFormatter) -> Void) -> DateFormatter { |
||||
|
block(self) |
||||
|
return self |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,11 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
|
||||
|
// 这个文件是为了满足Flutter插件的命名约定 |
||||
|
// 实际实现在ClassicBluetoothPlugin.swift中 |
||||
|
public class SwiftClassicBluetoothPlugin: NSObject, FlutterPlugin { |
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
// 调用真正的实现 |
||||
|
ClassicBluetoothPlugin.register(with: registrar) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,44 @@ |
|||||
|
# iOS蓝牙配置说明 |
||||
|
|
||||
|
## 所需权限 |
||||
|
|
||||
|
要在iOS中使用经典蓝牙功能，需要在应用的`Info.plist`中添加以下配置： |
||||
|
|
||||
|
### 1. 外部附件协议支持 |
||||
|
|
||||
|
在Info.plist中添加`UISupportedExternalAccessoryProtocols`数组，并添加你的设备支持的协议。如果不确定，可以先添加常见的协议： |
||||
|
|
||||
|
```xml |
||||
|
<key>UISupportedExternalAccessoryProtocols</key> |
||||
|
<array> |
||||
|
<string>com.example.protocol</string> |
||||
|
<!-- 添加其他你设备支持的协议 --> |
||||
|
</array> |
||||
|
``` |
||||
|
|
||||
|
### 2. 蓝牙使用说明 |
||||
|
|
||||
|
在iOS 13及以上版本中，需要添加蓝牙使用说明： |
||||
|
|
||||
|
```xml |
||||
|
<key>NSBluetoothAlwaysUsageDescription</key> |
||||
|
<string>我们需要使用蓝牙来连接您的音频设备</string> |
||||
|
``` |
||||
|
|
||||
|
对于iOS 13以下版本： |
||||
|
|
||||
|
```xml |
||||
|
<key>NSBluetoothPeripheralUsageDescription</key> |
||||
|
<string>我们需要使用蓝牙来连接您的音频设备</string> |
||||
|
``` |
||||
|
|
||||
|
## 常见问题 |
||||
|
|
||||
|
1. 如果看不到任何设备，请确保： |
||||
|
- 设备已配对并连接到iOS设备 |
||||
|
- 设备是MFi认证的（Made for iPhone/iPad），因为iOS对经典蓝牙有限制 |
||||
|
- 添加了正确的协议字符串 |
||||
|
|
||||
|
2. 蓝牙状态变化不及时： |
||||
|
- iOS系统对蓝牙状态变化的通知可能有延迟 |
||||
|
- 请确保用户已经授权蓝牙权限 |
||||
@ -0,0 +1,22 @@ |
|||||
|
# |
||||
|
# To learn more about a Podspec see http://guides.cocoapods.org/syntax/podspec.html. |
||||
|
# |
||||
|
Pod::Spec.new do |s| |
||||
|
s.name = 'classic_bluetooth' |
||||
|
s.version = '0.0.1' |
||||
|
s.summary = '经典蓝牙插件' |
||||
|
s.description = <<-DESC |
||||
|
一个用于管理经典蓝牙设备的Flutter插件。 |
||||
|
DESC |
||||
|
s.homepage = 'http://yunqiinnovation.com' |
||||
|
s.license = { :file => '../LICENSE' } |
||||
|
s.author = { 'YunQi Innovation' => 'support@yunqiinnovation.com' } |
||||
|
s.source = { :path => '.' } |
||||
|
s.source_files = 'Classes/**/*' |
||||
|
|
||||
|
s.ios.deployment_target = '11.0' |
||||
|
s.swift_version = '5.0' |
||||
|
|
||||
|
# 添加CoreBluetooth和ExternalAccessory框架依赖 |
||||
|
s.framework = 'CoreBluetooth', 'ExternalAccessory' |
||||
|
end |
||||
@ -0,0 +1,4 @@ |
|||||
|
#import <Flutter/Flutter.h> |
||||
|
|
||||
|
@interface DeepvoiceUtilsPlugin : NSObject<FlutterPlugin> |
||||
|
@end |
||||
@ -0,0 +1,12 @@ |
|||||
|
#import "DeepvoiceUtilsPlugin.h" |
||||
|
#if __has_include(<deepvoice_utils/deepvoice_utils-Swift.h>) |
||||
|
#import <deepvoice_utils/deepvoice_utils-Swift.h> |
||||
|
#else |
||||
|
#import "deepvoice_utils-Swift.h" |
||||
|
#endif |
||||
|
|
||||
|
@implementation DeepvoiceUtilsPlugin |
||||
|
+ (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar>*)registrar { |
||||
|
[SwiftDeepvoiceUtilsPlugin registerWithRegistrar:registrar]; |
||||
|
} |
||||
|
@end |
||||
@ -0,0 +1,30 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
|
||||
|
public class DeepvoiceUtilsPlugin: NSObject, FlutterPlugin { |
||||
|
private static let TAG = "DeepvoiceUtilsPlugin" |
||||
|
|
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
let channel = FlutterMethodChannel(name: "deepvoice_utils", binaryMessenger: registrar.messenger()) |
||||
|
let instance = DeepvoiceUtilsPlugin() |
||||
|
registrar.addMethodCallDelegate(instance, channel: channel) |
||||
|
|
||||
|
// 初始化文件日志 |
||||
|
FileLogger.initialize() |
||||
|
FileLogger.d(TAG, "DeepvoiceUtilsPlugin 已初始化") |
||||
|
} |
||||
|
|
||||
|
public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) { |
||||
|
switch call.method { |
||||
|
case "getPlatformVersion": |
||||
|
result("iOS \(UIDevice.current.systemVersion)") |
||||
|
default: |
||||
|
result(FlutterMethodNotImplemented) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public func detachFromEngine(for registrar: FlutterPluginRegistrar) { |
||||
|
// 关闭日志系统 |
||||
|
FileLogger.shutdown() |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,286 @@ |
|||||
|
import Foundation |
||||
|
import os.log |
||||
|
|
||||
|
/** |
||||
|
* 文件日志工具 |
||||
|
* 将日志同时输出到控制台和文件系统 |
||||
|
*/ |
||||
|
@objc public class FileLogger: NSObject { |
||||
|
|
||||
|
private static let TAG = "FileLogger" |
||||
|
private static var instance: FileLogger? |
||||
|
|
||||
|
private let logDir: URL |
||||
|
private var currentLogFile: URL? |
||||
|
private let logQueue = DispatchQueue(label: "com.yunqiinnovation.deepvoice_utils.logger", qos: .utility) |
||||
|
|
||||
|
private let dateFormatter: DateFormatter = { |
||||
|
let formatter = DateFormatter() |
||||
|
formatter.dateFormat = "yyyy-MM-dd" |
||||
|
return formatter |
||||
|
}() |
||||
|
|
||||
|
private let timeFormatter: DateFormatter = { |
||||
|
let formatter = DateFormatter() |
||||
|
formatter.dateFormat = "HH:mm:ss.SSS" |
||||
|
return formatter |
||||
|
}() |
||||
|
|
||||
|
/** |
||||
|
* 初始化日志系统 |
||||
|
*/ |
||||
|
private init(directory: URL) { |
||||
|
self.logDir = directory.appendingPathComponent("logs", isDirectory: true) |
||||
|
super.init() |
||||
|
|
||||
|
// 创建日志目录 |
||||
|
try? FileManager.default.createDirectory(at: logDir, withIntermediateDirectories: true) |
||||
|
|
||||
|
// 创建当天日志文件 |
||||
|
updateLogFile() |
||||
|
|
||||
|
// 记录初始化完成 |
||||
|
self.d(FileLogger.TAG, "文件日志系统已初始化") |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 更新日志文件为当天日期 |
||||
|
*/ |
||||
|
private func updateLogFile() { |
||||
|
let today = dateFormatter.string(from: Date()) |
||||
|
let logFile = logDir.appendingPathComponent("log_\(today).txt") |
||||
|
|
||||
|
if currentLogFile == nil || currentLogFile?.path != logFile.path { |
||||
|
currentLogFile = logFile |
||||
|
|
||||
|
// 如果是新文件，添加文件头 |
||||
|
if !FileManager.default.fileExists(atPath: logFile.path) { |
||||
|
let headerDate = DateFormatter() |
||||
|
headerDate.dateFormat = "yyyy-MM-dd HH:mm:ss" |
||||
|
let headerLine = "===== 日志开始: \(headerDate.string(from: Date())) =====\n" |
||||
|
|
||||
|
try? headerLine.write(to: logFile, atomically: true, encoding: .utf8) |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 将日志写入文件 |
||||
|
*/ |
||||
|
private func logToFile(level: String, tag: String, message: String) { |
||||
|
logQueue.async { [weak self] in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
do { |
||||
|
// 检查是否需要更新日志文件 |
||||
|
self.updateLogFile() |
||||
|
|
||||
|
guard let logFile = self.currentLogFile, |
||||
|
FileManager.default.fileExists(atPath: logFile.path) else { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 格式化日志内容 |
||||
|
let timestamp = self.timeFormatter.string(from: Date()) |
||||
|
let logLine = "\(timestamp) \(level)/\(tag): \(message)\n" |
||||
|
|
||||
|
// 获取文件句柄并追加内容 |
||||
|
if let fileHandle = try? FileHandle(forWritingTo: logFile) { |
||||
|
fileHandle.seekToEndOfFile() |
||||
|
if let data = logLine.data(using: .utf8) { |
||||
|
fileHandle.write(data) |
||||
|
} |
||||
|
fileHandle.closeFile() |
||||
|
} else { |
||||
|
// 如果无法获取文件句柄，尝试直接写入 |
||||
|
let existingContent = try? String(contentsOf: logFile, encoding: .utf8) |
||||
|
let newContent = (existingContent ?? "") + logLine |
||||
|
try newContent.write(to: logFile, atomically: true, encoding: .utf8) |
||||
|
} |
||||
|
} catch { |
||||
|
print("写入日志文件失败: \(error.localizedDescription)") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 调试级别日志 |
||||
|
*/ |
||||
|
private func d(_ tag: String, _ message: String) { |
||||
|
// 输出到控制台 |
||||
|
print("D/\(tag): \(message)") |
||||
|
|
||||
|
// 写入文件 |
||||
|
logToFile(level: "D", tag: tag, message: message) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 信息级别日志 |
||||
|
*/ |
||||
|
private func i(_ tag: String, _ message: String) { |
||||
|
// 输出到控制台 |
||||
|
print("I/\(tag): \(message)") |
||||
|
|
||||
|
// 写入文件 |
||||
|
logToFile(level: "I", tag: tag, message: message) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 警告级别日志 |
||||
|
*/ |
||||
|
private func w(_ tag: String, _ message: String) { |
||||
|
// 输出到控制台 |
||||
|
print("W/\(tag): \(message)") |
||||
|
|
||||
|
// 写入文件 |
||||
|
logToFile(level: "W", tag: tag, message: message) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 错误级别日志 |
||||
|
*/ |
||||
|
private func e(_ tag: String, _ message: String, _ error: Error? = nil) { |
||||
|
var fullMessage = message |
||||
|
|
||||
|
// 如果有错误，添加错误信息 |
||||
|
if let error = error { |
||||
|
fullMessage += "\n\(error.localizedDescription)" |
||||
|
} |
||||
|
|
||||
|
// 输出到控制台 |
||||
|
print("E/\(tag): \(fullMessage)") |
||||
|
|
||||
|
// 写入文件 |
||||
|
logToFile(level: "E", tag: tag, message: fullMessage) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 清理旧日志文件 |
||||
|
*/ |
||||
|
private func cleanOldLogs(daysToKeep: Int = 7) { |
||||
|
logQueue.async { |
||||
|
do { |
||||
|
let currentDate = Date() |
||||
|
let fileManager = FileManager.default |
||||
|
|
||||
|
let logFiles = try fileManager.contentsOfDirectory(at: self.logDir, includingPropertiesForKeys: [.creationDateKey], options: []) |
||||
|
|
||||
|
for fileURL in logFiles where fileURL.pathExtension == "txt" { |
||||
|
if let fileAttributes = try? fileManager.attributesOfItem(atPath: fileURL.path), |
||||
|
let creationDate = fileAttributes[.creationDate] as? Date { |
||||
|
|
||||
|
let age = Calendar.current.dateComponents([.day], from: creationDate, to: currentDate) |
||||
|
|
||||
|
if let days = age.day, days > daysToKeep { |
||||
|
try? fileManager.removeItem(at: fileURL) |
||||
|
print("D/\(FileLogger.TAG): 已删除旧日志文件: \(fileURL.lastPathComponent)") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} catch { |
||||
|
print("E/\(FileLogger.TAG): 清理旧日志失败: \(error.localizedDescription)") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MARK: - 公共静态方法 |
||||
|
|
||||
|
/** |
||||
|
* 初始化日志系统 |
||||
|
*/ |
||||
|
@objc public static func initialize() { |
||||
|
if instance == nil { |
||||
|
let documentsDirectory = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0] |
||||
|
instance = FileLogger(directory: documentsDirectory) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 初始化日志系统（使用自定义目录） |
||||
|
*/ |
||||
|
@objc public static func initialize(withDirectory directory: URL) { |
||||
|
if instance == nil { |
||||
|
instance = FileLogger(directory: directory) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 调试日志 |
||||
|
*/ |
||||
|
@objc public static func d(_ tag: String, _ message: String) { |
||||
|
// 如果实例不存在，先使用默认目录初始化 |
||||
|
if instance == nil { |
||||
|
initialize() |
||||
|
} |
||||
|
|
||||
|
instance?.d(tag, message) |
||||
|
|
||||
|
// 使用系统日志 |
||||
|
if #available(iOS 10.0, *) { |
||||
|
os_log("%{public}@", log: OSLog(subsystem: "com.yunqiinnovation.deepvoice_utils", category: tag), type: .debug, message) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 信息日志 |
||||
|
*/ |
||||
|
@objc public static func i(_ tag: String, _ message: String) { |
||||
|
if instance == nil { |
||||
|
initialize() |
||||
|
} |
||||
|
|
||||
|
instance?.i(tag, message) |
||||
|
|
||||
|
// 使用系统日志 |
||||
|
if #available(iOS 10.0, *) { |
||||
|
os_log("%{public}@", log: OSLog(subsystem: "com.yunqiinnovation.deepvoice_utils", category: tag), type: .info, message) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 警告日志 |
||||
|
*/ |
||||
|
@objc public static func w(_ tag: String, _ message: String) { |
||||
|
if instance == nil { |
||||
|
initialize() |
||||
|
} |
||||
|
|
||||
|
instance?.w(tag, message) |
||||
|
|
||||
|
// 使用系统日志 |
||||
|
if #available(iOS 10.0, *) { |
||||
|
os_log("%{public}@", log: OSLog(subsystem: "com.yunqiinnovation.deepvoice_utils", category: tag), type: .default, message) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 错误日志 |
||||
|
*/ |
||||
|
@objc public static func e(_ tag: String, _ message: String, _ error: Error? = nil) { |
||||
|
if instance == nil { |
||||
|
initialize() |
||||
|
} |
||||
|
|
||||
|
instance?.e(tag, message, error) |
||||
|
|
||||
|
// 使用系统日志 |
||||
|
if #available(iOS 10.0, *) { |
||||
|
let errorMessage = error != nil ? "\(message): \(error!.localizedDescription)" : message |
||||
|
os_log("%{public}@", log: OSLog(subsystem: "com.yunqiinnovation.deepvoice_utils", category: tag), type: .error, errorMessage) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 清理旧日志 |
||||
|
*/ |
||||
|
@objc public static func cleanOldLogs(daysToKeep: Int = 7) { |
||||
|
instance?.cleanOldLogs(daysToKeep: daysToKeep) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 关闭日志系统 |
||||
|
*/ |
||||
|
@objc public static func shutdown() { |
||||
|
instance = nil |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,11 @@ |
|||||
|
import Flutter |
||||
|
import UIKit |
||||
|
|
||||
|
// 这个类是为了满足Flutter插件的命名约定 |
||||
|
// 实际实现在DeepvoiceUtilsPlugin.swift中 |
||||
|
public class SwiftDeepvoiceUtilsPlugin: NSObject, FlutterPlugin { |
||||
|
public static func register(with registrar: FlutterPluginRegistrar) { |
||||
|
// 调用真正的实现 |
||||
|
DeepvoiceUtilsPlugin.register(with: registrar) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,54 @@ |
|||||
|
# DeepVoice Utils iOS 实现 |
||||
|
|
||||
|
## 简介 |
||||
|
|
||||
|
本模块是 `deepvoice_utils` 插件的 iOS 实现部分，主要提供以下功能： |
||||
|
|
||||
|
- 文件日志记录（FileLogger）：将日志记录到文件，方便调试和问题排查 |
||||
|
|
||||
|
## 使用说明 |
||||
|
|
||||
|
### 日志系统 |
||||
|
|
||||
|
该插件自动初始化日志系统，开发者无需手动调用初始化方法。日志文件默认保存在应用的 Documents 目录下的 logs 文件夹中，按日期分文件存储。 |
||||
|
|
||||
|
## 配置要求 |
||||
|
|
||||
|
- iOS 11.0 或更高版本 |
||||
|
- Swift 5.0 |
||||
|
- 确保应用拥有文件写入权限 |
||||
|
|
||||
|
## 开发说明 |
||||
|
|
||||
|
### 文件结构 |
||||
|
|
||||
|
- `DeepvoiceUtilsPlugin.swift`：插件主类，负责与 Flutter 交互 |
||||
|
- `FileLogger.swift`：日志系统实现，提供日志记录功能 |
||||
|
- `SwiftDeepvoiceUtilsPlugin.swift`：命名约定桥接类 |
||||
|
- `DeepvoiceUtilsPlugin.h/.m`：Objective-C 桥接文件 |
||||
|
|
||||
|
### 自定义日志目录 |
||||
|
|
||||
|
如需使用自定义日志目录，可以在应用启动时手动调用初始化方法： |
||||
|
|
||||
|
```swift |
||||
|
import deepvoice_utils |
||||
|
|
||||
|
// 在 AppDelegate 中 |
||||
|
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool { |
||||
|
// 自定义日志目录 |
||||
|
let customDirectory = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0].appendingPathComponent("custom_logs") |
||||
|
FileLogger.initialize(withDirectory: customDirectory) |
||||
|
|
||||
|
return true |
||||
|
} |
||||
|
``` |
||||
|
|
||||
|
### 清理旧日志 |
||||
|
|
||||
|
默认情况下，日志系统不会自动清理旧日志。如果需要清理，可以手动调用： |
||||
|
|
||||
|
```swift |
||||
|
// 清理7天前的日志 |
||||
|
FileLogger.cleanOldLogs(daysToKeep: 7) |
||||
|
``` |
||||
@ -0,0 +1,19 @@ |
|||||
|
# |
||||
|
# To learn more about a Podspec see http://guides.cocoapods.org/syntax/podspec.html. |
||||
|
# |
||||
|
Pod::Spec.new do |s| |
||||
|
s.name = 'deepvoice_utils' |
||||
|
s.version = '0.0.1' |
||||
|
s.summary = '深语工具库' |
||||
|
s.description = <<-DESC |
||||
|
提供文件日志记录等通用功能的Flutter插件。 |
||||
|
DESC |
||||
|
s.homepage = 'http://yunqiinnovation.com' |
||||
|
s.license = { :file => '../LICENSE' } |
||||
|
s.author = { 'YunQi Innovation' => 'support@yunqiinnovation.com' } |
||||
|
s.source = { :path => '.' } |
||||
|
s.source_files = 'Classes/**/*' |
||||
|
|
||||
|
s.ios.deployment_target = '11.0' |
||||
|
s.swift_version = '5.0' |
||||
|
end |
||||
@ -0,0 +1,249 @@ |
|||||
|
import Foundation |
||||
|
|
||||
|
/// 工具类型枚举 |
||||
|
enum ToolType { |
||||
|
case localFunction // 本地函数 |
||||
|
case mcpTool // MCP工具 |
||||
|
} |
||||
|
|
||||
|
/// 函数处理器协议 |
||||
|
protocol FunctionHandler { |
||||
|
/// 处理函数调用 |
||||
|
/// - Parameters: |
||||
|
/// - arguments: 函数参数，Dictionary格式 |
||||
|
/// - Returns: 函数执行结果，字符串 |
||||
|
func handle(arguments: [String: Any]) async throws -> String |
||||
|
} |
||||
|
|
||||
|
/// MCPClient - iOS版本，负责连接到Model Context Protocol服务器和管理本地函数 |
||||
|
public class MCPClient { |
||||
|
private let TAG = "MCPClient" |
||||
|
|
||||
|
// 本地函数Map，函数名 -> 处理器 |
||||
|
private var localFunctions = [String: FunctionHandler]() |
||||
|
|
||||
|
// 工具定义Map |
||||
|
private var toolDefinitions = [[String: Any]]() |
||||
|
|
||||
|
// 是否已连接 |
||||
|
private var isConnected = false |
||||
|
|
||||
|
// 系统功能处理器 |
||||
|
private var systemFunctionHandler: SystemFunctionHandler? |
||||
|
|
||||
|
// SSE会话 |
||||
|
private var sseSession: URLSession? |
||||
|
private var sseTask: URLSessionDataTask? |
||||
|
|
||||
|
/// 初始化MCPClient |
||||
|
/// - Parameter context: 可选的应用上下文 |
||||
|
init(_ context: Any? = nil) { |
||||
|
if let appContext = context { |
||||
|
systemFunctionHandler = SystemFunctionHandler(context: appContext) |
||||
|
|
||||
|
// 初始化语音功能 |
||||
|
initializeVoiceFunctions() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 初始化语音功能处理器 |
||||
|
private func initializeVoiceFunctions() { |
||||
|
do { |
||||
|
// 使用SystemFunctionHandler注册所有功能 |
||||
|
systemFunctionHandler?.registerAllFunctions(client: self) |
||||
|
|
||||
|
print("\(TAG) 语音功能初始化完成") |
||||
|
} catch { |
||||
|
print("\(TAG) 语音功能初始化失败: \(error.localizedDescription)") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 注册本地函数 |
||||
|
/// - Parameters: |
||||
|
/// - name: 函数名称 |
||||
|
/// - description: 函数描述 |
||||
|
/// - parameters: 函数参数定义 |
||||
|
/// - handler: 函数处理器 |
||||
|
/// - Returns: 是否注册成功 |
||||
|
func registerLocalFunction(name: String, description: String, parameters: [String: Any], handler: FunctionHandler) -> Bool { |
||||
|
do { |
||||
|
// 存储函数处理器 |
||||
|
localFunctions[name] = handler |
||||
|
|
||||
|
// 创建函数定义 |
||||
|
let functionDef: [String: Any] = [ |
||||
|
"name": name, |
||||
|
"description": description, |
||||
|
"parameters": parameters |
||||
|
] |
||||
|
|
||||
|
// 添加到工具定义列表 |
||||
|
let toolDef: [String: Any] = [ |
||||
|
"type": "function", |
||||
|
"function": functionDef |
||||
|
] |
||||
|
|
||||
|
// 更新或添加工具定义 |
||||
|
if let index = toolDefinitions.firstIndex(where: { |
||||
|
guard let function = $0["function"] as? [String: Any], |
||||
|
let functionName = function["name"] as? String else { |
||||
|
return false |
||||
|
} |
||||
|
return functionName == name |
||||
|
}) { |
||||
|
toolDefinitions[index] = toolDef |
||||
|
} else { |
||||
|
toolDefinitions.append(toolDef) |
||||
|
} |
||||
|
|
||||
|
print("\(TAG) 成功注册本地函数: \(name)") |
||||
|
return true |
||||
|
} catch { |
||||
|
print("\(TAG) 注册函数失败: \(error.localizedDescription)") |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 连接到MCP服务器(通过SSE) |
||||
|
/// - Parameter mcpServer: 服务器URL |
||||
|
/// - Returns: 是否连接成功 |
||||
|
func connectToSSE(_ mcpServer: String) -> Bool { |
||||
|
// 关闭现有连接 |
||||
|
closeConnection() |
||||
|
|
||||
|
// 如果URL为空，返回失败 |
||||
|
if mcpServer.isEmpty { |
||||
|
print("\(TAG) MCP服务器URL为空") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 解析服务器URL |
||||
|
guard let serverURL = URL(string: mcpServer) else { |
||||
|
print("\(TAG) 无效的MCP服务器URL: \(mcpServer)") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
print("\(TAG) 正在连接到MCP服务器: \(mcpServer)") |
||||
|
|
||||
|
// 创建会话配置 |
||||
|
let config = URLSessionConfiguration.default |
||||
|
config.timeoutIntervalForRequest = 30 |
||||
|
config.timeoutIntervalForResource = 60 |
||||
|
|
||||
|
// 创建会话 |
||||
|
let session = URLSession(configuration: config) |
||||
|
self.sseSession = session |
||||
|
|
||||
|
// 创建请求 |
||||
|
var request = URLRequest(url: serverURL) |
||||
|
request.httpMethod = "GET" |
||||
|
request.addValue("text/event-stream", forHTTPHeaderField: "Accept") |
||||
|
|
||||
|
// 创建并启动SSE任务 |
||||
|
let task = session.dataTask(with: request) { [weak self] data, response, error in |
||||
|
guard let self = self else { return } |
||||
|
|
||||
|
if let error = error { |
||||
|
print("\(self.TAG) MCP服务器连接失败: \(error.localizedDescription)") |
||||
|
self.isConnected = false |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
if let httpResponse = response as? HTTPURLResponse { |
||||
|
if httpResponse.statusCode == 200 { |
||||
|
print("\(self.TAG) 成功连接到MCP服务器") |
||||
|
self.isConnected = true |
||||
|
|
||||
|
// 处理服务器响应数据 |
||||
|
if let data = data, let responseString = String(data: data, encoding: .utf8) { |
||||
|
print("\(self.TAG) 收到初始响应: \(responseString)") |
||||
|
|
||||
|
// 在这里解析可用工具列表，暂时模拟 |
||||
|
// 实际情况下，需要实现SSE流解析 |
||||
|
} |
||||
|
} else { |
||||
|
print("\(self.TAG) MCP服务器连接失败，状态码: \(httpResponse.statusCode)") |
||||
|
self.isConnected = false |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
task.resume() |
||||
|
self.sseTask = task |
||||
|
|
||||
|
// 在iOS中，连接是异步的，此处返回true表示连接已开始 |
||||
|
// 实际连接状态通过isConnected属性检查 |
||||
|
return true |
||||
|
} |
||||
|
|
||||
|
/// 检查是否已连接 |
||||
|
/// - Returns: 是否已连接 |
||||
|
func isConnected() -> Bool { |
||||
|
return isConnected |
||||
|
} |
||||
|
|
||||
|
/// 关闭连接 |
||||
|
func closeConnection() { |
||||
|
sseTask?.cancel() |
||||
|
sseTask = nil |
||||
|
isConnected = false |
||||
|
} |
||||
|
|
||||
|
/// 获取工具定义列表，用于OpenAI请求 |
||||
|
/// - Returns: 工具定义列表 |
||||
|
func getToolMaps() -> [[String: Any]] { |
||||
|
return toolDefinitions |
||||
|
} |
||||
|
|
||||
|
/// 解析JSON参数 |
||||
|
/// - Parameter argumentsJson: JSON字符串 |
||||
|
/// - Returns: 解析后的参数字典 |
||||
|
func parseJsonArguments(_ argumentsJson: String) -> [String: Any] { |
||||
|
guard !argumentsJson.isEmpty, |
||||
|
let data = argumentsJson.data(using: .utf8), |
||||
|
let parsedArgs = try? JSONSerialization.jsonObject(with: data) as? [String: Any] else { |
||||
|
return [:] |
||||
|
} |
||||
|
return parsedArgs |
||||
|
} |
||||
|
|
||||
|
/// 调用工具 |
||||
|
/// - Parameters: |
||||
|
/// - name: 工具名称 |
||||
|
/// - arguments: 工具参数 |
||||
|
/// - Returns: 工具调用结果 |
||||
|
func callTool(name: String, arguments: [String: Any]) async -> [String: Any] { |
||||
|
do { |
||||
|
// 检查是否为本地函数 |
||||
|
if let handler = localFunctions[name] { |
||||
|
let result = try await handler.handle(arguments: arguments) |
||||
|
return [ |
||||
|
"name": name, |
||||
|
"context": result |
||||
|
] |
||||
|
} |
||||
|
|
||||
|
// 如果不是本地函数，尝试作为MCP工具调用 |
||||
|
if isConnected { |
||||
|
// 这里应该实现真正的MCP工具调用 |
||||
|
// 暂时返回一个模拟结果 |
||||
|
return [ |
||||
|
"name": name, |
||||
|
"context": "模拟MCP工具调用结果: \(name)(\(arguments))" |
||||
|
] |
||||
|
} |
||||
|
|
||||
|
return ["context": "无法处理工具调用：找不到工具 \(name) 或MCP未连接"] |
||||
|
} catch { |
||||
|
print("\(TAG) 调用工具失败: \(error.localizedDescription)") |
||||
|
return ["context": "工具调用失败: \(error.localizedDescription)"] |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 关闭MCP客户端 |
||||
|
func close() { |
||||
|
closeConnection() |
||||
|
localFunctions.removeAll() |
||||
|
toolDefinitions.removeAll() |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,514 @@ |
|||||
|
import Foundation |
||||
|
import UIKit |
||||
|
import CoreLocation |
||||
|
import Contacts |
||||
|
import EventKit |
||||
|
|
||||
|
/// 系统功能处理器 |
||||
|
class SystemFunctionHandler { |
||||
|
private let TAG = "SystemFunctionHandler" |
||||
|
private let context: Any |
||||
|
static let ACTION_EXIT_INTERACTION = "com.yunqiinnovation.deepsound.action.EXIT_INTERACTION" |
||||
|
|
||||
|
/// 初始化系统功能处理器 |
||||
|
/// - Parameter context: 应用上下文 |
||||
|
init(context: Any) { |
||||
|
self.context = context |
||||
|
} |
||||
|
|
||||
|
/// 注册所有系统功能 |
||||
|
/// - Parameter client: MCP客户端 |
||||
|
func registerAllFunctions(client: MCPClient) { |
||||
|
// 注册退出交互函数 |
||||
|
registerExitInteractionFunction(client) |
||||
|
|
||||
|
// 注册翻译模式函数 |
||||
|
registerTranslationModeFunction(client) |
||||
|
|
||||
|
// 注册发送短信函数 |
||||
|
registerSendTextMessageFunction(client) |
||||
|
|
||||
|
// 注册拨打电话函数 |
||||
|
registerMakePhoneCallFunction(client) |
||||
|
|
||||
|
// 注册打开应用函数 |
||||
|
registerOpenAppFunction(client) |
||||
|
|
||||
|
// 注册设置提醒函数 |
||||
|
registerSetReminderFunction(client) |
||||
|
|
||||
|
// 注册获取当前时间函数 |
||||
|
registerGetCurrentTimeFunction(client) |
||||
|
|
||||
|
// 注册获取当前位置函数 |
||||
|
registerGetCurrentLocationFunction(client) |
||||
|
|
||||
|
print("\(TAG) 所有系统功能注册完成") |
||||
|
} |
||||
|
|
||||
|
/// 注册退出交互函数 |
||||
|
private func registerExitInteractionFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [:], |
||||
|
"required": [] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = ExitInteractionHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "exit_interaction", |
||||
|
description: "结束当前交互", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册退出交互功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册翻译模式函数 |
||||
|
private func registerTranslationModeFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [:], |
||||
|
"required": [] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = TranslationModeHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "enter_translation_mode", |
||||
|
description: "用户请求进入实时翻译模式时，启动实时翻译功能", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册翻译模式功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册发送短信函数 |
||||
|
private func registerSendTextMessageFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [ |
||||
|
"contact": [ |
||||
|
"type": "string", |
||||
|
"description": "联系人姓名或电话号码" |
||||
|
], |
||||
|
"message": [ |
||||
|
"type": "string", |
||||
|
"description": "短信内容" |
||||
|
] |
||||
|
], |
||||
|
"required": ["contact", "message"] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = SendTextMessageHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "send_text_message", |
||||
|
description: "发送短信", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册发送短信功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册拨打电话函数 |
||||
|
private func registerMakePhoneCallFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [ |
||||
|
"contact": [ |
||||
|
"type": "string", |
||||
|
"description": "联系人姓名或电话号码" |
||||
|
] |
||||
|
], |
||||
|
"required": ["contact"] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = MakePhoneCallHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "make_phone_call", |
||||
|
description: "拨打电话", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册拨打电话功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册打开应用函数 |
||||
|
private func registerOpenAppFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [ |
||||
|
"app_name": [ |
||||
|
"type": "string", |
||||
|
"description": "应用名称" |
||||
|
] |
||||
|
], |
||||
|
"required": ["app_name"] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = OpenAppHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "open_app", |
||||
|
description: "当用户请求打开特定应用时，启动相应应用", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册打开应用功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册设置提醒函数 |
||||
|
private func registerSetReminderFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [ |
||||
|
"title": [ |
||||
|
"type": "string", |
||||
|
"description": "提醒标题" |
||||
|
], |
||||
|
"content": [ |
||||
|
"type": "string", |
||||
|
"description": "提醒内容" |
||||
|
], |
||||
|
"time": [ |
||||
|
"type": "string", |
||||
|
"description": "提醒时间，格式为'yyyy-MM-dd HH:mm'，如'2023-12-31 14:30'" |
||||
|
] |
||||
|
], |
||||
|
"required": ["title", "time"] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = SetReminderHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "set_reminder", |
||||
|
description: "设置提醒事项", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册设置提醒功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册获取当前时间函数 |
||||
|
private func registerGetCurrentTimeFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [ |
||||
|
"format": [ |
||||
|
"type": "string", |
||||
|
"description": "时间格式，可选，默认为标准格式" |
||||
|
] |
||||
|
], |
||||
|
"required": [] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = GetCurrentTimeHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "get_current_time", |
||||
|
description: "获取当前日期和时间", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册获取当前时间功能") |
||||
|
} |
||||
|
|
||||
|
/// 注册获取当前位置函数 |
||||
|
private func registerGetCurrentLocationFunction(_ client: MCPClient) { |
||||
|
// 定义函数参数 |
||||
|
let parameters: [String: Any] = [ |
||||
|
"type": "object", |
||||
|
"properties": [:], |
||||
|
"required": [] |
||||
|
] |
||||
|
|
||||
|
// 创建函数处理器 |
||||
|
let handler = GetCurrentLocationHandler() |
||||
|
|
||||
|
// 注册函数 |
||||
|
client.registerLocalFunction( |
||||
|
name: "get_current_location", |
||||
|
description: "获取当前地理位置", |
||||
|
parameters: parameters, |
||||
|
handler: handler |
||||
|
) |
||||
|
print("\(TAG) 已注册获取当前位置功能") |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 退出交互处理器 |
||||
|
class ExitInteractionHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
// 发送通知和广播 |
||||
|
NotificationCenter.default.post( |
||||
|
name: Notification.Name(SystemFunctionHandler.ACTION_EXIT_INTERACTION), |
||||
|
object: nil, |
||||
|
userInfo: ["timestamp": Date().timeIntervalSince1970] |
||||
|
) |
||||
|
|
||||
|
return "{\"result\": \"已结束当前交互\"}" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 翻译模式处理器 |
||||
|
class TranslationModeHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
// 发送通知 |
||||
|
NotificationCenter.default.post( |
||||
|
name: Notification.Name("com.yunqiinnovation.deepsound.ACTION_ENTER_TRANSLATION_MODE"), |
||||
|
object: nil, |
||||
|
userInfo: ["timestamp": Date().timeIntervalSince1970] |
||||
|
) |
||||
|
|
||||
|
return "{\"result\": \"已进入实时翻译模式\"}" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 发送短信处理器 |
||||
|
class SendTextMessageHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
guard let contact = arguments["contact"] as? String, |
||||
|
let message = arguments["message"] as? String else { |
||||
|
return "{\"result\": \"发送短信失败：缺少联系人或消息内容\"}" |
||||
|
} |
||||
|
|
||||
|
// iOS不允许应用直接发送短信，但可以打开短信应用 |
||||
|
if let url = URL(string: "sms:\(contact)&body=\(message.addingPercentEncoding(withAllowedCharacters: .urlQueryAllowed) ?? "")") { |
||||
|
DispatchQueue.main.async { |
||||
|
UIApplication.shared.open(url, options: [:], completionHandler: nil) |
||||
|
} |
||||
|
return "{\"result\": \"已打开短信应用，收件人：\(contact)\"}" |
||||
|
} |
||||
|
|
||||
|
return "{\"result\": \"发送短信失败：无法打开短信应用\"}" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 拨打电话处理器 |
||||
|
class MakePhoneCallHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
guard let contact = arguments["contact"] as? String else { |
||||
|
return "{\"result\": \"拨打电话失败：缺少联系人\"}" |
||||
|
} |
||||
|
|
||||
|
let phoneNumber = contact.replacingOccurrences(of: "[^+0-9]", with: "", options: .regularExpression) |
||||
|
|
||||
|
if let url = URL(string: "tel://\(phoneNumber)") { |
||||
|
DispatchQueue.main.async { |
||||
|
UIApplication.shared.open(url, options: [:], completionHandler: nil) |
||||
|
} |
||||
|
return "{\"result\": \"正在拨打\(contact)的电话\"}" |
||||
|
} |
||||
|
|
||||
|
return "{\"result\": \"拨打电话失败：无法拨打电话\"}" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 打开应用处理器 |
||||
|
class OpenAppHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
guard let appName = arguments["app_name"] as? String else { |
||||
|
return "{\"result\": \"打开应用失败：缺少应用名称\"}" |
||||
|
} |
||||
|
|
||||
|
// iOS没有直接打开其他应用的API，只能通过URL Scheme |
||||
|
// 这里只是模拟返回成功，实际上无法在iOS上实现这个功能 |
||||
|
return "{\"result\": \"打开应用功能在iOS上受限，无法直接打开\(appName)\"}" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 设置提醒处理器 |
||||
|
class SetReminderHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
guard let title = arguments["title"] as? String, |
||||
|
let timeStr = arguments["time"] as? String else { |
||||
|
return "{\"result\": \"设置提醒失败：缺少标题或时间\"}" |
||||
|
} |
||||
|
|
||||
|
let content = arguments["content"] as? String ?? "" |
||||
|
|
||||
|
// 解析时间 |
||||
|
let dateFormatter = DateFormatter() |
||||
|
dateFormatter.dateFormat = "yyyy-MM-dd HH:mm" |
||||
|
guard let date = dateFormatter.date(from: timeStr) else { |
||||
|
return "{\"result\": \"设置提醒失败：时间格式无效，请使用'yyyy-MM-dd HH:mm'格式\"}" |
||||
|
} |
||||
|
|
||||
|
// 使用EventKit添加提醒 |
||||
|
let eventStore = EKEventStore() |
||||
|
let event = EKEvent(eventStore: eventStore) |
||||
|
event.title = title |
||||
|
event.notes = content |
||||
|
event.startDate = date |
||||
|
event.endDate = date.addingTimeInterval(3600) // 1小时后 |
||||
|
event.calendar = eventStore.defaultCalendarForNewEvents |
||||
|
|
||||
|
do { |
||||
|
try eventStore.save(event, span: .thisEvent) |
||||
|
return "{\"result\": \"已创建提醒: \(title), 时间: \(timeStr)\"}" |
||||
|
} catch { |
||||
|
return "{\"result\": \"设置提醒失败：\(error.localizedDescription)\"}" |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 获取当前时间处理器 |
||||
|
class GetCurrentTimeHandler: FunctionHandler { |
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
let format = arguments["format"] as? String |
||||
|
|
||||
|
let calendar = Calendar.current |
||||
|
let date = Date() |
||||
|
|
||||
|
let dateFormatter = DateFormatter() |
||||
|
dateFormatter.locale = Locale(identifier: "zh_CN") |
||||
|
|
||||
|
let timeStr: String |
||||
|
switch format?.lowercased() { |
||||
|
case "short": |
||||
|
dateFormatter.dateFormat = "HH:mm" |
||||
|
timeStr = dateFormatter.string(from: date) |
||||
|
case "date": |
||||
|
dateFormatter.dateFormat = "yyyy年MM月dd日" |
||||
|
timeStr = dateFormatter.string(from: date) |
||||
|
case "time": |
||||
|
dateFormatter.dateFormat = "HH:mm:ss" |
||||
|
timeStr = dateFormatter.string(from: date) |
||||
|
case "full": |
||||
|
dateFormatter.dateFormat = "yyyy年MM月dd日 EEEE HH:mm:ss" |
||||
|
timeStr = dateFormatter.string(from: date) |
||||
|
case "iso": |
||||
|
dateFormatter.dateFormat = "yyyy-MM-dd'T'HH:mm:ss" |
||||
|
timeStr = dateFormatter.string(from: date) |
||||
|
default: |
||||
|
dateFormatter.dateFormat = "yyyy年MM月dd日 HH:mm" |
||||
|
timeStr = dateFormatter.string(from: date) |
||||
|
} |
||||
|
|
||||
|
// 获取星期几 |
||||
|
let weekDay: String |
||||
|
if format == "full" { |
||||
|
weekDay = "" // 已经包含在dateFormatter里 |
||||
|
} else { |
||||
|
let weekDayNum = calendar.component(.weekday, from: date) |
||||
|
switch weekDayNum { |
||||
|
case 1: weekDay = "星期日" |
||||
|
case 2: weekDay = "星期一" |
||||
|
case 3: weekDay = "星期二" |
||||
|
case 4: weekDay = "星期三" |
||||
|
case 5: weekDay = "星期四" |
||||
|
case 6: weekDay = "星期五" |
||||
|
case 7: weekDay = "星期六" |
||||
|
default: weekDay = "" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 格式化结果 |
||||
|
let resultTime: String |
||||
|
if format == "short" || format == "time" { |
||||
|
resultTime = timeStr |
||||
|
} else if format == "full" { |
||||
|
resultTime = timeStr // already contains weekday |
||||
|
} else { |
||||
|
resultTime = "\(timeStr) \(weekDay)" |
||||
|
} |
||||
|
|
||||
|
return "{\"result\": \"\(resultTime)\", \"time\": \"\(resultTime)\"}" |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/// 获取当前位置处理器 |
||||
|
class GetCurrentLocationHandler: FunctionHandler { |
||||
|
private let locationManager = CLLocationManager() |
||||
|
private var locationCompletionHandler: ((String, Bool) -> Void)? |
||||
|
|
||||
|
func handle(arguments: [String: Any]) async throws -> String { |
||||
|
return await withCheckedContinuation { continuation in |
||||
|
self.getLocation { message, success in |
||||
|
continuation.resume(returning: "{\"result\": \"\(message)\", \"success\": \(success)}") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private func getLocation(completion: @escaping (String, Bool) -> Void) { |
||||
|
locationCompletionHandler = completion |
||||
|
|
||||
|
// 检查权限 |
||||
|
let authStatus = CLLocationManager.authorizationStatus() |
||||
|
if authStatus == .denied || authStatus == .restricted { |
||||
|
completion("需要位置权限才能获取当前位置", false) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 检查位置服务是否开启 |
||||
|
if !CLLocationManager.locationServicesEnabled() { |
||||
|
completion("位置服务未启用，请打开GPS或网络定位", false) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 如果有最后的位置，使用它 |
||||
|
if let lastLocation = locationManager.location { |
||||
|
let message = "当前位置: 经度 \(lastLocation.coordinate.longitude), 纬度 \(lastLocation.coordinate.latitude)" |
||||
|
completion(message, true) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// 否则请求更新 |
||||
|
locationManager.delegate = self |
||||
|
locationManager.desiredAccuracy = kCLLocationAccuracyBest |
||||
|
locationManager.requestWhenInUseAuthorization() |
||||
|
locationManager.startUpdatingLocation() |
||||
|
|
||||
|
// 设置超时 |
||||
|
DispatchQueue.main.asyncAfter(deadline: .now() + 10) { [weak self] in |
||||
|
if let handler = self?.locationCompletionHandler { |
||||
|
self?.locationManager.stopUpdatingLocation() |
||||
|
handler("获取位置超时，请稍后重试", false) |
||||
|
self?.locationCompletionHandler = nil |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 扩展以接收位置更新 |
||||
|
extension GetCurrentLocationHandler: CLLocationManagerDelegate { |
||||
|
func locationManager(_ manager: CLLocationManager, didUpdateLocations locations: [CLLocation]) { |
||||
|
guard let location = locations.last, let handler = locationCompletionHandler else { return } |
||||
|
|
||||
|
manager.stopUpdatingLocation() |
||||
|
|
||||
|
let message = "当前位置: 经度 \(location.coordinate.longitude), 纬度 \(location.coordinate.latitude)" |
||||
|
handler(message, true) |
||||
|
locationCompletionHandler = nil |
||||
|
} |
||||
|
|
||||
|
func locationManager(_ manager: CLLocationManager, didFailWithError error: Error) { |
||||
|
manager.stopUpdatingLocation() |
||||
|
locationCompletionHandler?("获取位置失败: \(error.localizedDescription)", false) |
||||
|
locationCompletionHandler = nil |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue