lxm
1 year ago
14 changed files with 462 additions and 140 deletions
@ -0,0 +1,223 @@ |
|||
import Foundation |
|||
import music_service |
|||
|
|||
class MusicSystemFunctionHandler { |
|||
|
|||
// 添加 registerAllFunctions 方法 |
|||
func registerAllFunctions(client: MCPClient) { |
|||
|
|||
|
|||
// 注册QQ音乐搜索和播放功能 |
|||
client.registerLocalFunction( |
|||
name: "music_searchandplay_songs", |
|||
description: "在音乐播放服务中搜索并播放歌曲，支持歌曲、专辑、歌手等搜索类型", |
|||
parameters: [ |
|||
"type": "object", |
|||
"properties": [ |
|||
"keyword": [ |
|||
"type": "string", |
|||
"description": "搜索关键词，歌曲名、专辑名或歌手名" |
|||
], |
|||
"searchType": [ |
|||
"type": "integer", |
|||
"description": "搜索类型：0-歌名搜索，1-歌手搜索，2-歌曲类型搜索", |
|||
"default": 0 |
|||
] |
|||
], |
|||
"required": ["keyword"] |
|||
], |
|||
handler: MusicSearchHandler() |
|||
) |
|||
|
|||
// 注册媒体控制函数 |
|||
registerMediaPlayFunction(client: client) |
|||
registerMediaPauseFunction(client: client) |
|||
registerMediaCloseFunction(client: client) |
|||
registerMediaPreviousFunction(client: client) |
|||
registerMediaNextFunction(client: client) |
|||
} |
|||
} |
|||
|
|||
// MARK: - Private Handler Classes |
|||
|
|||
private class MusicSearchHandler: FunctionHandler { |
|||
func handle(arguments: [String: Any]) async throws -> String { |
|||
guard let keyword = arguments["keyword"] as? String else { |
|||
let result: [String: Any] = ["success": false, "message": "缺少搜索关键词"] |
|||
let jsonData = try JSONSerialization.data(withJSONObject: result) |
|||
return String(data: jsonData, encoding: .utf8) ?? "{}" |
|||
} |
|||
let searchType = arguments["searchType"] as? Int ?? 0 |
|||
|
|||
return await withCheckedContinuation { continuation in |
|||
MusicService.shared.searchSongs(keyword: keyword, searchType: searchType) { searchResult in |
|||
switch searchResult { |
|||
case .success(let songs): |
|||
let songsJson = songs.map { song -> [String: Any] in |
|||
return [ |
|||
"id": song["id"] ?? 0, |
|||
"name": song["name"] ?? "unknown", |
|||
"singer": song["singer"] ?? "unknown", |
|||
"url": song["url"] ?? "", |
|||
"image": song["image"] ?? "", |
|||
"ts": song["ts"] ?? 0 |
|||
] |
|||
} |
|||
|
|||
let result: [String: Any] = [ |
|||
"success": true, |
|||
"broadcast": false, |
|||
"message": "搜索成功,正在播放", |
|||
"data": songsJson |
|||
] |
|||
|
|||
do { |
|||
let jsonData = try JSONSerialization.data(withJSONObject: result, options: []) |
|||
let jsonString = String(data: jsonData, encoding: .utf8) ?? "{}" |
|||
continuation.resume(returning: jsonString) |
|||
} catch { |
|||
continuation.resume(returning: "{\"success\": false, \"message\": \"JSON serialization failed\"}") |
|||
} |
|||
|
|||
case .failure(let error): |
|||
let result: [String: Any] = [ |
|||
"success": false, |
|||
"message": "搜索失败: \(error.localizedDescription)" |
|||
] |
|||
do { |
|||
let jsonData = try JSONSerialization.data(withJSONObject: result) |
|||
let jsonString = String(data: jsonData, encoding: .utf8) ?? "{}" |
|||
continuation.resume(returning: jsonString) |
|||
} catch { |
|||
continuation.resume(returning: "{\"success\": false, \"message\": \"Search failed and JSON serialization failed\"}") |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
/// 注册媒体播放函数 |
|||
private func registerMediaPlayFunction(client: MCPClient) { |
|||
let parameters: [String: Any] = [ |
|||
"type": "object", |
|||
"properties": [:], |
|||
"required": [] |
|||
] |
|||
|
|||
let handler = MediaPlayHandler() |
|||
_ = client.registerLocalFunction( |
|||
name: "music_resume", |
|||
description: "恢复音乐播放", |
|||
parameters: parameters, |
|||
handler: handler |
|||
) |
|||
} |
|||
|
|||
/// 注册媒体暂停函数 |
|||
private func registerMediaPauseFunction(client: MCPClient) { |
|||
let parameters: [String: Any] = [ |
|||
"type": "object", |
|||
"properties": [:], |
|||
"required": [] |
|||
] |
|||
|
|||
let handler = MediaPauseHandler() |
|||
_ = client.registerLocalFunction( |
|||
name: "music_pause", |
|||
description: "暂停音乐播放", |
|||
parameters: parameters, |
|||
handler: handler |
|||
) |
|||
} |
|||
|
|||
/// 注册媒体关闭函数 |
|||
private func registerMediaCloseFunction(client: MCPClient) { |
|||
let parameters: [String: Any] = [ |
|||
"type": "object", |
|||
"properties": [:], |
|||
"required": [] |
|||
] |
|||
|
|||
let handler = MediaCloseHandler() |
|||
_ = client.registerLocalFunction( |
|||
name: "music_close", |
|||
description: "关闭音乐播放", |
|||
parameters: parameters, |
|||
handler: handler |
|||
) |
|||
} |
|||
|
|||
|
|||
/// 注册媒体上一首函数 |
|||
private func registerMediaPreviousFunction(client: MCPClient) { |
|||
let parameters: [String: Any] = [ |
|||
"type": "object", |
|||
"properties": [:], |
|||
"required": [] |
|||
] |
|||
|
|||
let handler = MediaPreviousHandler() |
|||
_ = client.registerLocalFunction( |
|||
name: "music_previous", |
|||
description: "播放上一首", |
|||
parameters: parameters, |
|||
handler: handler |
|||
) |
|||
} |
|||
|
|||
/// 注册媒体下一首函数 |
|||
private func registerMediaNextFunction(client: MCPClient) { |
|||
let parameters: [String: Any] = [ |
|||
"type": "object", |
|||
"properties": [:], |
|||
"required": [] |
|||
] |
|||
|
|||
let handler = MediaNextHandler() |
|||
_ = client.registerLocalFunction( |
|||
name: "music_next", |
|||
description: "播放下一首", |
|||
parameters: parameters, |
|||
handler: handler |
|||
) |
|||
} |
|||
|
|||
/// 媒体播放处理器 |
|||
private class MediaPlayHandler: FunctionHandler { |
|||
func handle(arguments: [String: Any]) async throws -> String { |
|||
NotificationCenter.default.post(name: NSNotification.Name("MediaPlay"), object: nil) |
|||
return "{\"broadcast\": false,\"result\": \"已开始音乐播放\"}" |
|||
} |
|||
} |
|||
|
|||
/// 媒体暂停处理器 |
|||
private class MediaPauseHandler: FunctionHandler { |
|||
func handle(arguments: [String: Any]) async throws -> String { |
|||
NotificationCenter.default.post(name: NSNotification.Name("MediaStop"), object: nil) |
|||
return "{\"broadcast\": false,\"result\": \"已停止音乐播放\"}" |
|||
} |
|||
} |
|||
/// 媒体暂停处理器 |
|||
private class MediaCloseHandler: FunctionHandler { |
|||
func handle(arguments: [String: Any]) async throws -> String { |
|||
NotificationCenter.default.post(name: NSNotification.Name("MediaPause"), object: nil) |
|||
return "{\"broadcast\": false,\"result\": \"已暂停音乐播放\"}" |
|||
} |
|||
} |
|||
|
|||
/// 媒体上一首处理器 |
|||
private class MediaPreviousHandler: FunctionHandler { |
|||
func handle(arguments: [String: Any]) async throws -> String { |
|||
NotificationCenter.default.post(name: NSNotification.Name("MediaPrevious"), object: nil) |
|||
return "{\"broadcast\": false,\"result\": \"已切换到上一首\"}" |
|||
} |
|||
} |
|||
|
|||
/// 媒体下一首处理器 |
|||
private class MediaNextHandler: FunctionHandler { |
|||
func handle(arguments: [String: Any]) async throws -> String { |
|||
NotificationCenter.default.post(name: NSNotification.Name("MediaNext"), object: nil) |
|||
return "{\"broadcast\": false,\"result\": \"已切换到下一首\"}" |
|||
} |
|||
} |
|||
Loading…
Reference in new issue