1
0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-05 03:23:50 +08:00

feature word segmentation

tx分词API
This commit is contained in:
wdvxdr 2020-09-29 20:18:03 +08:00
parent 6d87095182
commit 3e033da6a6
5 changed files with 444 additions and 0 deletions

View File

@ -1240,3 +1240,21 @@ func (c *QQClient) buildAppInfoRequestPacket(id string) (uint16, []byte) {
packet := packets.BuildUniPacket(c.Uin, seq, "LightAppSvc.mini_app_info.GetAppInfoById", 1, c.OutGoingPacketSessionId, EmptyBytes, c.sigInfo.d2Key, payload) packet := packets.BuildUniPacket(c.Uin, seq, "LightAppSvc.mini_app_info.GetAppInfoById", 1, c.OutGoingPacketSessionId, EmptyBytes, c.sigInfo.d2Key, payload)
return seq, packet return seq, packet
} }
func (c *QQClient) buildWordSegmentationPacket(data []byte) (uint16, []byte) {
seq := c.nextSeq()
body := &oidb.D79ReqBody{
Uin: uint64(c.Uin),
Content: data,
Qua: []byte("and_537065262_8.4.5"),
}
b, _ := proto.Marshal(body)
req := &oidb.OIDBSSOPkg{
Command: 3449,
ServiceType: 1,
Bodybuffer: b,
}
payload, _ := proto.Marshal(req)
packet := packets.BuildUniPacket(c.Uin, seq, "OidbSvc.0xd79", 1, c.OutGoingPacketSessionId, EmptyBytes, c.sigInfo.d2Key, payload)
return seq, packet
}

View File

@ -146,6 +146,7 @@ func NewClientMd5(uin int64, passwordMd5 [16]byte) *QQClient {
"OidbSvc.0x6d6_2": decodeOIDB6d6Response, "OidbSvc.0x6d6_2": decodeOIDB6d6Response,
"OidbSvc.0x88d_0": decodeGroupInfoResponse, "OidbSvc.0x88d_0": decodeGroupInfoResponse,
"OidbSvc.0xe07_0": decodeImageOcrResponse, "OidbSvc.0xe07_0": decodeImageOcrResponse,
"OidbSvc.0xd79": decodeWordSegmentation,
"SummaryCard.ReqSummaryCard": decodeSummaryCardResponse, "SummaryCard.ReqSummaryCard": decodeSummaryCardResponse,
"PttCenterSvr.ShortVideoDownReq": decodePttShortVideoDownResponse, "PttCenterSvr.ShortVideoDownReq": decodePttShortVideoDownResponse,
"LightAppSvc.mini_app_info.GetAppInfoById": decodeAppInfoResponse, "LightAppSvc.mini_app_info.GetAppInfoById": decodeAppInfoResponse,
@ -256,6 +257,21 @@ func (c *QQClient) GetGroupHonorInfo(groupCode int64, honorType HonorType) (*Gro
return &ret, nil return &ret, nil
} }
func (c *QQClient) GetWordSegmentation(text string) ([]string, error) {
rsp, err := c.sendAndWait(c.buildWordSegmentationPacket([]byte(text)))
if err != nil {
return nil, err
}
if data, ok := rsp.([][]byte); ok {
var ret []string
for _, val := range data {
ret = append(ret, string(val))
}
return ret, nil
}
return nil, errors.New("decode error")
}
func (c *QQClient) GetSummaryInfo(target int64) (*SummaryCardInfo, error) { func (c *QQClient) GetSummaryInfo(target int64) (*SummaryCardInfo, error) {
rsp, err := c.sendAndWait(c.buildSummaryCardRequestPacket(target)) rsp, err := c.sendAndWait(c.buildSummaryCardRequestPacket(target))
if err != nil { if err != nil {

View File

@ -1037,6 +1037,22 @@ func decodeMultiApplyDownResponse(c *QQClient, _ uint16, payload []byte) (interf
return &mt, nil return &mt, nil
} }
// OidbSvc.0xd79
func decodeWordSegmentation(_ *QQClient, _ uint16, payload []byte) (interface{}, error) {
pkg := oidb.OIDBSSOPkg{}
rsp := &oidb.D79RspBody{}
if err := proto.Unmarshal(payload, &pkg); err != nil {
return nil, err
}
if err := proto.Unmarshal(pkg.Bodybuffer, rsp); err != nil {
return nil, err
}
if rsp.Content != nil {
return rsp.Content.SliceContent, nil
}
return nil, errors.New("no word receive")
}
// OidbSvc.0x6d6_2 // OidbSvc.0x6d6_2
func decodeOIDB6d6Response(c *QQClient, _ uint16, payload []byte) (interface{}, error) { func decodeOIDB6d6Response(c *QQClient, _ uint16, payload []byte) (interface{}, error) {
pkg := oidb.OIDBSSOPkg{} pkg := oidb.OIDBSSOPkg{}

View File

@ -0,0 +1,25 @@
syntax = "proto3";
option go_package = ".;oidb";
message D79ReqBody {
uint64 seq = 1;
uint64 uin = 2;
uint32 compress_flag = 3;
bytes content = 4;
uint64 sender_uin = 5;
bytes qua = 6;
bytes word_ext = 7;
}
message D79RspBody {
uint32 ret = 1;
uint64 seq = 2;
uint64 uin = 3;
uint32 compress_flag = 4;
D79Content content = 5;
}
message D79Content {
repeated bytes slice_content = 1;
}

View File

@ -0,0 +1,369 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// protoc-gen-go v1.25.0
// protoc v3.13.0
// source: oidb0xD79.proto
package oidb
import (
proto "github.com/golang/protobuf/proto"
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
// This is a compile-time assertion that a sufficiently up-to-date version
// of the legacy proto package is being used.
const _ = proto.ProtoPackageIsVersion4
type D79ReqBody struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Seq uint64 `protobuf:"varint,1,opt,name=seq,proto3" json:"seq,omitempty"`
Uin uint64 `protobuf:"varint,2,opt,name=uin,proto3" json:"uin,omitempty"`
CompressFlag uint32 `protobuf:"varint,3,opt,name=compress_flag,json=compressFlag,proto3" json:"compress_flag,omitempty"`
Content []byte `protobuf:"bytes,4,opt,name=content,proto3" json:"content,omitempty"`
SenderUin uint64 `protobuf:"varint,5,opt,name=sender_uin,json=senderUin,proto3" json:"sender_uin,omitempty"`
Qua []byte `protobuf:"bytes,6,opt,name=qua,proto3" json:"qua,omitempty"`
WordExt []byte `protobuf:"bytes,7,opt,name=word_ext,json=wordExt,proto3" json:"word_ext,omitempty"`
}
func (x *D79ReqBody) Reset() {
*x = D79ReqBody{}
if protoimpl.UnsafeEnabled {
mi := &file_oidb0xD79_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *D79ReqBody) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*D79ReqBody) ProtoMessage() {}
func (x *D79ReqBody) ProtoReflect() protoreflect.Message {
mi := &file_oidb0xD79_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use D79ReqBody.ProtoReflect.Descriptor instead.
func (*D79ReqBody) Descriptor() ([]byte, []int) {
return file_oidb0xD79_proto_rawDescGZIP(), []int{0}
}
func (x *D79ReqBody) GetSeq() uint64 {
if x != nil {
return x.Seq
}
return 0
}
func (x *D79ReqBody) GetUin() uint64 {
if x != nil {
return x.Uin
}
return 0
}
func (x *D79ReqBody) GetCompressFlag() uint32 {
if x != nil {
return x.CompressFlag
}
return 0
}
func (x *D79ReqBody) GetContent() []byte {
if x != nil {
return x.Content
}
return nil
}
func (x *D79ReqBody) GetSenderUin() uint64 {
if x != nil {
return x.SenderUin
}
return 0
}
func (x *D79ReqBody) GetQua() []byte {
if x != nil {
return x.Qua
}
return nil
}
func (x *D79ReqBody) GetWordExt() []byte {
if x != nil {
return x.WordExt
}
return nil
}
type D79RspBody struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Ret uint32 `protobuf:"varint,1,opt,name=ret,proto3" json:"ret,omitempty"`
Seq uint64 `protobuf:"varint,2,opt,name=seq,proto3" json:"seq,omitempty"`
Uin uint64 `protobuf:"varint,3,opt,name=uin,proto3" json:"uin,omitempty"`
CompressFlag uint32 `protobuf:"varint,4,opt,name=compress_flag,json=compressFlag,proto3" json:"compress_flag,omitempty"`
Content *D79Content `protobuf:"bytes,5,opt,name=content,proto3" json:"content,omitempty"`
}
func (x *D79RspBody) Reset() {
*x = D79RspBody{}
if protoimpl.UnsafeEnabled {
mi := &file_oidb0xD79_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *D79RspBody) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*D79RspBody) ProtoMessage() {}
func (x *D79RspBody) ProtoReflect() protoreflect.Message {
mi := &file_oidb0xD79_proto_msgTypes[1]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use D79RspBody.ProtoReflect.Descriptor instead.
func (*D79RspBody) Descriptor() ([]byte, []int) {
return file_oidb0xD79_proto_rawDescGZIP(), []int{1}
}
func (x *D79RspBody) GetRet() uint32 {
if x != nil {
return x.Ret
}
return 0
}
func (x *D79RspBody) GetSeq() uint64 {
if x != nil {
return x.Seq
}
return 0
}
func (x *D79RspBody) GetUin() uint64 {
if x != nil {
return x.Uin
}
return 0
}
func (x *D79RspBody) GetCompressFlag() uint32 {
if x != nil {
return x.CompressFlag
}
return 0
}
func (x *D79RspBody) GetContent() *D79Content {
if x != nil {
return x.Content
}
return nil
}
type D79Content struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
SliceContent [][]byte `protobuf:"bytes,1,rep,name=slice_content,json=sliceContent,proto3" json:"slice_content,omitempty"`
}
func (x *D79Content) Reset() {
*x = D79Content{}
if protoimpl.UnsafeEnabled {
mi := &file_oidb0xD79_proto_msgTypes[2]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *D79Content) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*D79Content) ProtoMessage() {}
func (x *D79Content) ProtoReflect() protoreflect.Message {
mi := &file_oidb0xD79_proto_msgTypes[2]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use D79Content.ProtoReflect.Descriptor instead.
func (*D79Content) Descriptor() ([]byte, []int) {
return file_oidb0xD79_proto_rawDescGZIP(), []int{2}
}
func (x *D79Content) GetSliceContent() [][]byte {
if x != nil {
return x.SliceContent
}
return nil
}
var File_oidb0xD79_proto protoreflect.FileDescriptor
var file_oidb0xD79_proto_rawDesc = []byte{
0x0a, 0x0f, 0x6f, 0x69, 0x64, 0x62, 0x30, 0x78, 0x44, 0x37, 0x39, 0x2e, 0x70, 0x72, 0x6f, 0x74,
0x6f, 0x22, 0xbb, 0x01, 0x0a, 0x0a, 0x44, 0x37, 0x39, 0x52, 0x65, 0x71, 0x42, 0x6f, 0x64, 0x79,
0x12, 0x10, 0x0a, 0x03, 0x73, 0x65, 0x71, 0x18, 0x01, 0x20, 0x01, 0x28, 0x04, 0x52, 0x03, 0x73,
0x65, 0x71, 0x12, 0x10, 0x0a, 0x03, 0x75, 0x69, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, 0x04, 0x52,
0x03, 0x75, 0x69, 0x6e, 0x12, 0x23, 0x0a, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x72, 0x65, 0x73, 0x73,
0x5f, 0x66, 0x6c, 0x61, 0x67, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0d, 0x52, 0x0c, 0x63, 0x6f, 0x6d,
0x70, 0x72, 0x65, 0x73, 0x73, 0x46, 0x6c, 0x61, 0x67, 0x12, 0x18, 0x0a, 0x07, 0x63, 0x6f, 0x6e,
0x74, 0x65, 0x6e, 0x74, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x63, 0x6f, 0x6e, 0x74,
0x65, 0x6e, 0x74, 0x12, 0x1d, 0x0a, 0x0a, 0x73, 0x65, 0x6e, 0x64, 0x65, 0x72, 0x5f, 0x75, 0x69,
0x6e, 0x18, 0x05, 0x20, 0x01, 0x28, 0x04, 0x52, 0x09, 0x73, 0x65, 0x6e, 0x64, 0x65, 0x72, 0x55,
0x69, 0x6e, 0x12, 0x10, 0x0a, 0x03, 0x71, 0x75, 0x61, 0x18, 0x06, 0x20, 0x01, 0x28, 0x0c, 0x52,
0x03, 0x71, 0x75, 0x61, 0x12, 0x19, 0x0a, 0x08, 0x77, 0x6f, 0x72, 0x64, 0x5f, 0x65, 0x78, 0x74,
0x18, 0x07, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x77, 0x6f, 0x72, 0x64, 0x45, 0x78, 0x74, 0x22,
0x8e, 0x01, 0x0a, 0x0a, 0x44, 0x37, 0x39, 0x52, 0x73, 0x70, 0x42, 0x6f, 0x64, 0x79, 0x12, 0x10,
0x0a, 0x03, 0x72, 0x65, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0d, 0x52, 0x03, 0x72, 0x65, 0x74,
0x12, 0x10, 0x0a, 0x03, 0x73, 0x65, 0x71, 0x18, 0x02, 0x20, 0x01, 0x28, 0x04, 0x52, 0x03, 0x73,
0x65, 0x71, 0x12, 0x10, 0x0a, 0x03, 0x75, 0x69, 0x6e, 0x18, 0x03, 0x20, 0x01, 0x28, 0x04, 0x52,
0x03, 0x75, 0x69, 0x6e, 0x12, 0x23, 0x0a, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x72, 0x65, 0x73, 0x73,
0x5f, 0x66, 0x6c, 0x61, 0x67, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0d, 0x52, 0x0c, 0x63, 0x6f, 0x6d,
0x70, 0x72, 0x65, 0x73, 0x73, 0x46, 0x6c, 0x61, 0x67, 0x12, 0x25, 0x0a, 0x07, 0x63, 0x6f, 0x6e,
0x74, 0x65, 0x6e, 0x74, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0b, 0x2e, 0x44, 0x37, 0x39,
0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x52, 0x07, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74,
0x22, 0x31, 0x0a, 0x0a, 0x44, 0x37, 0x39, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x12, 0x23,
0x0a, 0x0d, 0x73, 0x6c, 0x69, 0x63, 0x65, 0x5f, 0x63, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x18,
0x01, 0x20, 0x03, 0x28, 0x0c, 0x52, 0x0c, 0x73, 0x6c, 0x69, 0x63, 0x65, 0x43, 0x6f, 0x6e, 0x74,
0x65, 0x6e, 0x74, 0x42, 0x08, 0x5a, 0x06, 0x2e, 0x3b, 0x6f, 0x69, 0x64, 0x62, 0x62, 0x06, 0x70,
0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_oidb0xD79_proto_rawDescOnce sync.Once
file_oidb0xD79_proto_rawDescData = file_oidb0xD79_proto_rawDesc
)
func file_oidb0xD79_proto_rawDescGZIP() []byte {
file_oidb0xD79_proto_rawDescOnce.Do(func() {
file_oidb0xD79_proto_rawDescData = protoimpl.X.CompressGZIP(file_oidb0xD79_proto_rawDescData)
})
return file_oidb0xD79_proto_rawDescData
}
var file_oidb0xD79_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
var file_oidb0xD79_proto_goTypes = []interface{}{
(*D79ReqBody)(nil), // 0: D79ReqBody
(*D79RspBody)(nil), // 1: D79RspBody
(*D79Content)(nil), // 2: D79Content
}
var file_oidb0xD79_proto_depIdxs = []int32{
2, // 0: D79RspBody.content:type_name -> D79Content
1, // [1:1] is the sub-list for method output_type
1, // [1:1] is the sub-list for method input_type
1, // [1:1] is the sub-list for extension type_name
1, // [1:1] is the sub-list for extension extendee
0, // [0:1] is the sub-list for field type_name
}
func init() { file_oidb0xD79_proto_init() }
func file_oidb0xD79_proto_init() {
if File_oidb0xD79_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_oidb0xD79_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*D79ReqBody); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_oidb0xD79_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*D79RspBody); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_oidb0xD79_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*D79Content); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_oidb0xD79_proto_rawDesc,
NumEnums: 0,
NumMessages: 3,
NumExtensions: 0,
NumServices: 0,
},
GoTypes: file_oidb0xD79_proto_goTypes,
DependencyIndexes: file_oidb0xD79_proto_depIdxs,
MessageInfos: file_oidb0xD79_proto_msgTypes,
}.Build()
File_oidb0xD79_proto = out.File
file_oidb0xD79_proto_rawDesc = nil
file_oidb0xD79_proto_goTypes = nil
file_oidb0xD79_proto_depIdxs = nil
}