api: send_music

ProtobufBot · Aug 12, 2022 · 7a08c12 · 7a08c12
1 parent 0d32be0
commit 7a08c12
Show file tree

Hide file tree

Showing 5 changed files with 1,258 additions and 347 deletions.
diff --git a/onebot_idl b/onebot_idl
diff --git a/pkg/bot/api_handler.go b/pkg/bot/api_handler.go
@@ -591,6 +591,36 @@ func HandleSetGroupSignIn(cli *client.QQClient, req *onebot.SetGroupSignInReq) *
 	return nil
 }
 
+func HandleSendMusic(cli *client.QQClient, req *onebot.SendMusicReq) *onebot.SendMusicResp {
+	musicType := map[string]int{
+		"qq":    message.QQMusic,
+		"163":   message.CloudMusic,
+		"migu":  message.MiguMusic,
+		"kugou": message.KugouMusic,
+		"kuwo":  message.KuwoMusic,
+	}
+	music := &message.MusicShareElement{
+		MusicType:  musicType[req.Type],
+		Title:      req.Title,
+		Brief:      req.Brief,
+		Summary:    req.Summary,
+		Url:        req.Url,
+		PictureUrl: req.PictureUrl,
+		MusicUrl:   req.MusicUrl,
+	}
+	if req.GroupId != 0 {
+		if _, err := cli.SendGroupMusicShare(req.GroupId, music); err != nil {
+			return nil
+		}
+		return &onebot.SendMusicResp{}
+	}
+	if req.UserId != 0 {
+		cli.SendFriendMusicShare(req.UserId, music)
+		return &onebot.SendMusicResp{}
+	}
+	return nil
+}
+
 func HandleSendGroupPoke(cli *client.QQClient, req *onebot.SendGroupPokeReq) *onebot.SendGroupPokeResp {
 	group := cli.FindGroup(req.GroupId)
 	if group == nil {

diff --git a/pkg/bot/remote.go b/pkg/bot/remote.go
@@ -380,6 +380,14 @@ func handleApiFrame(cli *client.QQClient, req *onebot.Frame, isApiAllow func(one
 		resp.Data = &onebot.Frame_SetGroupSignInResp{
 			SetGroupSignInResp: HandleSetGroupSignIn(cli, data.SetGroupSignInReq),
 		}
+	case *onebot.Frame_SendMusicReq:
+		resp.FrameType = onebot.Frame_TSendMusicResp
+		if resp.Ok = isApiAllow(onebot.Frame_TSendMusicReq); !resp.Ok {
+			return
+		}
+		resp.Data = &onebot.Frame_SendMusicResp{
+			SendMusicResp: HandleSendMusic(cli, data.SendMusicReq),
+		}
 	case *onebot.Frame_SendGroupPokeReq:
 		resp.FrameType = onebot.Frame_TSendGroupPokeResp
 		if resp.Ok = isApiAllow(onebot.Frame_TSendGroupPokeReq); !resp.Ok {