This commit is contained in:
linyuchen
2024-02-11 02:58:34 +08:00
parent f9c376f6c5
commit d08601505b
19 changed files with 797 additions and 1615 deletions

1
.gitignore vendored
View File

@@ -1,3 +1,4 @@
node_modules/
dist/
.idea/
.DS_Store

743
package-lock.json generated

File diff suppressed because it is too large Load Diff

View File

@@ -5,12 +5,13 @@
"main": "dist/main.js",
"scripts": {
"test": "echo \"Error: no test specified\" && exit 1",
"postinstall": "ELECTRON_SKIP_BINARY_DOWNLOAD=1 npm install electron --no-save",
"build": "npm run build-main && npm run build-preload && npm run build-renderer",
"build-main": "webpack --config webpack.main.config.js",
"build-preload": "webpack --config webpack.preload.config.js",
"build-renderer": "webpack --config webpack.renderer.config.js",
"build-mac": "npm run build && cp manifest.json dist/ && npm run deploy-mac",
"deploy-mac": "cp dist/* ~/Library/Containers/com.tencent.qq/Data/Documents/LiteLoaderQQNT/plugins/LLOnebot/",
"deploy-mac": "cp dist/* ~/Library/Containers/com.tencent.qq/Data/LiteLoaderQQNT/plugins/LLOnebot/",
"build-win": "npm run build && cp manifest.json dist/ && npm run deploy-win",
"deploy-win": "cmd /c \"copy dist\\* %USERPROFILE%\\documents\\LiteLoaderQQNT\\plugins\\LLOnebot\\\""
},
@@ -26,7 +27,6 @@
"@types/express": "^4.17.20",
"@types/uuid": "^9.0.8",
"babel-loader": "^9.1.3",
"electron": "^27.0.2",
"ts-loader": "^9.5.0",
"typescript": "^5.2.2",
"webpack": "^5.89.0",

View File

@@ -1,21 +1,38 @@
import {Group, MessageElement, RawMessage, SelfInfo, User} from "./types";
import {SelfInfo} from "./types";
import { NTQQApi } from '../ntqqapi/ntcall';
import { Group, RawMessage, User } from "../ntqqapi/types";
export let groups: Group[] = []
export let friends: User[] = []
export let msgHistory: Record<string, RawMessage> = {} // msgId: RawMessage
export function getFriend(qq: string): User | undefined {
return friends.find(friend => friend.uin === qq)
export async function getFriend(qq: string): Promise<User | undefined> {
let friend = friends.find(friend => friend.uin === qq)
if (!friend){
friends = await NTQQApi.getFriends(true)
friend = friends.find(friend => friend.uin === qq)
}
return friend
}
export function getGroup(qq: string): Group | undefined {
return groups.find(group => group.uid === qq)
export async function getGroup(qq: string): Promise<Group | undefined> {
let group = groups.find(group => group.groupCode === qq)
if (!group){
groups = await NTQQApi.getGroups(true);
group = groups.find(group => group.groupCode === qq)
}
return group
}
export function getGroupMember(groupQQ: string, memberQQ: string) {
const group = getGroup(groupQQ)
export async function getGroupMember(groupQQ: string, memberQQ: string) {
const group = await getGroup(groupQQ)
if (group) {
return group.members?.find(member => member.uin === memberQQ)
let member = group.members?.find(member => member.uin === memberQQ)
if (!member){
group.members = await NTQQApi.getGroupMembers(groupQQ)
member = group.members?.find(member => member.uin === memberQQ)
}
return member
}
}
@@ -28,3 +45,14 @@ export let selfInfo: SelfInfo = {
export function getHistoryMsgBySeq(seq: string) {
return Object.values(msgHistory).find(msg => msg.msgSeq === seq)
}
export let uidMaps:Record<string, User> = {} // 一串加密的字符串(uid) -> qq号
export function getStrangerByUin(uin: string) {
for (const key in uidMaps) {
if (uidMaps[key].uin === uin) {
return uidMaps[key];
}
}
}

View File

@@ -1,143 +1,10 @@
import {OB11ApiName, OB11MessageData} from "../onebot11/types";
export enum AtType {
notAt = 0,
atAll = 1,
atUser = 2
}
export enum ChatType {
friend = 1,
group = 2,
temp = 100
}
export interface GroupMember {
avatarPath: string;
cardName: string;
cardType: number;
isDelete: boolean;
nick: string;
qid: string;
remark: string;
role: number; // 群主:4, 管理员:3群员:2
shutUpTime: number; // 禁言时间,单位是什么暂时不清楚
uid: string; // 加密的字符串
uin: string; // QQ号
}
export interface SelfInfo {
user_id: string;
nickname: string;
}
export interface User {
avatarUrl?: string;
bio?: string; // 签名
nickName: string;
uid?: string; // 加密的字符串
uin: string; // QQ号
}
export interface Group {
uid: string; // 群号
name: string;
members?: GroupMember[];
}
export interface Peer {
chatType: ChatType
name: string
uid: string // qq号
}
export interface PttElement {
canConvert2Text: boolean
duration: number // 秒数
fileBizId: null
fileId: number // 0
fileName: string // "e4d09c784d5a2abcb2f9980bdc7acfe6.amr"
filePath: string // "/Users//Library/Containers/com.tencent.qq/Data/Library/Application Support/QQ/nt_qq_a6b15c9820595d25a56c1633ce19ad40/nt_data/Ptt/2023-11/Ori/e4d09c784d5a2abcb2f9980bdc7acfe6.amr"
fileSize: string // "4261"
fileSubId: string // "0"
fileUuid: string // "90j3z7rmRphDPrdVgP9udFBaYar#oK0TWZIV"
formatType: string // 1
invalidState: number // 0
md5HexStr: string // "e4d09c784d5a2abcb2f9980bdc7acfe6"
playState: number // 0
progress: number // 0
text: string // ""
transferStatus: number // 0
translateStatus: number // 0
voiceChangeType: number // 0
voiceType: number // 0
waveAmplitudes: number[]
}
export interface ArkElement {
bytesData: string
}
export interface RawMessage {
msgId: string,
msgTime: string,
msgSeq: string,
senderUin: string; // 发送者QQ号
peerUid: string; // 群号 或者 QQ uid
peerUin: string; // 群号 或者 发送者QQ号
sendNickName: string;
sendMemberName?: string; // 发送者群名片
chatType: ChatType,
elements: {
replyElement: {
senderUid: string, // 原消息发送者QQ号
sourceMsgIsIncPic: boolean; // 原消息是否有图片
sourceMsgText: string;
replayMsgSeq: string; // 源消息的msgSeq可以通过这个找到源消息的msgId
},
textElement: {
atType: AtType
atUid: string,
content: string,
atNtUid: string
},
picElement: {
sourcePath: string // 图片本地路径
picWidth: number
picHeight: number
fileSize: number
fileName: string
fileUuid: string
},
pttElement: PttElement,
arkElement: ArkElement
}[]
}
export interface MessageElement {
raw: RawMessage
peer: Peer,
sender: {
uid: string // 一串加密的字符串
memberName: string
nickname: string
}
}
export interface PostDataSendMsg {
action: OB11ApiName
message_type?: "private" | "group"
params?: {
user_id: string,
group_id: string,
message: OB11MessageData[];
},
user_id: string,
group_id: string,
message?: OB11MessageData[];
}
export interface Config {
port: number
hosts: string[]

13
src/global.d.ts vendored
View File

@@ -1,13 +1,12 @@
import {
Config,
Group,
GroupMember,
MessageElement,
Peer,
PostDataSendMsg, PttElement, RawMessage,
SelfInfo,
User
} from "./common/types";
PostDataSendMsg,
SelfInfo} from "./common/types";
import { Group } from "./ntqqapi/types";
import { GroupMember } from "./ntqqapi/types";
import { MessageElement, PttElement, RawMessage } from "./ntqqapi/types";
import { User } from "./ntqqapi/types";
import { SendIPCMsgSession } from "./main/ipcsend";

View File

@@ -1,5 +1,5 @@
import {ipcMain, webContents} from 'electron';
import {PostDataSendMsg} from "../common/types";
import {OB11PostSendMsg} from "../onebot11/types"
import {CHANNEL_RECALL_MSG, CHANNEL_SEND_MSG,CHANNEL_SEND_BACK_MSG} from "../common/channels";
import {v4 as uuid4} from "uuid";
import {log} from "../common/utils";
@@ -23,7 +23,7 @@ export interface SendIPCMsgSession<T> {
data: T
}
export function sendIPCSendQQMsg(postData: PostDataSendMsg, handleSendResult: (data: OB11Return<any>) => void) {
export function sendIPCSendQQMsg(postData: OB11PostSendMsg, handleSendResult: (data: OB11Return<any>) => void) {
const onceSessionId = uuid4();
const handler = (event: any, session: SendIPCMsgSession<OB11Return<any>>) => {
// log("llonebot send msg ipcMain.once:" + JSON.stringify(sendResult));

View File

@@ -43,134 +43,14 @@ function onLoad() {
ipcMain.handle(CHANNEL_GET_CONFIG, (event: any, arg: any) => {
return getConfigUtil().getConfig()
})
ipcMain.handle(CHANNEL_DOWNLOAD_FILE, async (event: any, arg: { uri: string, fileName: string }): Promise<{
success: boolean,
errMsg: string,
path: string
}> => {
let filePath = path.join(CONFIG_DIR, arg.fileName)
let url = new URL(arg.uri);
if (url.protocol == "base64:") {
// base64转成文件
let base64Data = arg.uri.split("base64://")[1]
try {
const buffer = Buffer.from(base64Data, 'base64');
fs.writeFileSync(filePath, buffer);
} catch (e: any) {
return {
success: false,
errMsg: `base64文件下载失败,` + e.toString(),
path: ""
}
}
} else if (url.protocol == "http:" || url.protocol == "https:") {
// 下载文件
let res = await fetch(url)
if (!res.ok) {
return {
success: false,
errMsg: `${url}下载失败,` + res.statusText,
path: ""
}
}
let blob = await res.blob();
let buffer = await blob.arrayBuffer();
try {
fs.writeFileSync(filePath, Buffer.from(buffer));
} catch (e: any) {
return {
success: false,
errMsg: `${url}下载失败,` + e.toString(),
path: ""
}
}
}
else{
return {
success: false,
errMsg: `不支持的file协议,` + url.protocol,
path: ""
}
}
if (isGIF(filePath)) {
fs.renameSync(filePath, filePath + ".gif");
filePath += ".gif";
}
return {
success: true,
errMsg: "",
path: filePath
};
})
ipcMain.on(CHANNEL_SET_CONFIG, (event: any, arg: Config) => {
getConfigUtil().setConfig(arg)
})
ipcMain.on(CHANNEL_START_HTTP_SERVER, (event: any, arg: any) => {
startExpress(getConfigUtil().getConfig().port)
})
ipcMain.on(CHANNEL_UPDATE_GROUPS, (event: any, arg: Group[]) => {
for (const group of arg) {
let existGroup = groups.find(g => g.uid == group.uid)
if (existGroup) {
if (!existGroup.members) {
existGroup.members = []
}
existGroup.name = group.name
for (const member of group.members || []) {
let existMember = existGroup.members?.find(m => m.uin == member.uin)
if (existMember) {
existMember.nick = member.nick
existMember.cardName = member.cardName
} else {
existGroup.members?.push(member)
}
}
} else {
groups.push(group)
}
}
groups.length = 0
groups.push(...arg)
})
ipcMain.on(CHANNEL_UPDATE_FRIENDS, (event: any, arg: User[]) => {
friends.length = 0
friends.push(...arg)
})
ipcMain.on(CHANNEL_POST_ONEBOT_DATA, (event: any, arg: any) => {
postMsg(arg);
})
ipcMain.on(CHANNEL_LOG, (event: any, arg: any) => {
log(arg)
})
ipcMain.handle(CHANNEL_SET_SELF_INFO, (event: any, arg: SelfInfo) => {
selfInfo.user_id = arg.user_id;
selfInfo.nickname = arg.nickname;
running = true;
})
ipcMain.on(CHANNEL_DELETE_FILE, (event: any, arg: string[]) => {
for (const path of arg) {
fs.unlinkSync(path);
}
})
ipcMain.handle(CHANNEL_GET_RUNNING_STATUS, (event: any, arg: any) => {
return running;
})
ipcMain.handle(CHANNEL_FILE2BASE64, async (event: any, path: string): Promise<{err: string, data: string}> => {
return await file2base64(path);
})
ipcMain.handle(CHANNEL_GET_HISTORY_MSG, (event: any, arg: string): RawMessage | undefined => {
return msgHistory[arg] || null;
})
function postRawMsg(msgList:RawMessage[]) {
const {debug, reportSelfMessage} = getConfigUtil().getConfig();
@@ -208,11 +88,47 @@ function onLoad() {
}
})
setTimeout(()=>{
NTQQApi.getSelfInfo().then(r=>{
log(r);
})
}, 10000)
async function getSelfInfo(){
if (!selfInfo.user_id){
setTimeout(()=>{
getSelfInfo().then()
})
}
const _ = await NTQQApi.getSelfInfo()
if (_.uin){
log("get self info success", _)
selfInfo.user_id = _.uin
let nickName = _.uin
try{
const userInfo = (await NTQQApi.getUserInfo(_.uid))
if (userInfo){
nickName = userInfo.nickName
}
}
catch(e){
log("get self nickname failed", e.toString())
}
selfInfo.nickname = nickName
try{
// let _friends = await NTQQApi.getFriends(true)
// log("friends api:", _friends)
// for (let f of _friends){
// friends.push(f)
// }
let _groups = await NTQQApi.getGroups(true)
log("groups api:", _groups)
for (let g of _groups){
g.members = (await NTQQApi.getGroupMembers(g.uid))
groups.push(g)
}
}catch(e){
log("!!!初始化失败", e.stack.toString())
}
startExpress(getConfigUtil().getConfig().port)
}
}
getSelfInfo().then()
}

View File

@@ -32,7 +32,7 @@ export class SendMsgElementConstructor {
};
}
reply(msgSeq: string, msgId: string, senderUin: string, senderUinStr: string): SendReplyElement {
static reply(msgSeq: string, msgId: string, senderUin: string, senderUinStr: string): SendReplyElement {
return {
elementType: ElementType.REPLY,
elementId: "",
@@ -45,7 +45,7 @@ export class SendMsgElementConstructor {
}
}
async pic(picPath: string): Promise<SendPicElement>{
static async pic(picPath: string): Promise<SendPicElement>{
const {md5, fileName, path, fileSize} = await NTQQApi.uploadFile(picPath);
const imageSize = await NTQQApi.getImageSize(picPath);
const picElement = {
@@ -71,7 +71,7 @@ export class SendMsgElementConstructor {
};
}
async ptt(pttPath: string):Promise<SendPttElement> {
static async ptt(pttPath: string):Promise<SendPttElement> {
const {md5, fileName, path, fileSize} = await NTQQApi.uploadFile(pttPath);
return {
elementType: ElementType.PTT,

View File

@@ -1,15 +1,19 @@
import {BrowserWindow} from 'electron';
import {getConfigUtil, log} from "../common/utils";
import {NTQQApiClass} from "./ntcall";
import {RawMessage} from "../common/types";
import {msgHistory} from "../common/data";
import {NTQQApiClass, sendMessagePool} from "./ntcall";
import { Group } from "./types";
import { RawMessage } from "./types";
import {groups, msgHistory} from "../common/data";
export let hookApiCallbacks: Record<string, (apiReturn: any)=>void>={}
export enum ReceiveCmd {
UPDATE_MSG = "nodeIKernelMsgListener/onMsgInfoListUpdate",
NEW_MSG = "nodeIKernelMsgListener/onRecvMsg",
SELF_SEND_MSG = "nodeIKernelMsgListener/onAddSendMsg"
SELF_SEND_MSG = "nodeIKernelMsgListener/onAddSendMsg",
USER_INFO = "nodeIKernelProfileListener/onProfileDetailInfoChanged",
GROUPS = "nodeIKernelGroupListener/onGroupListUpdate",
GROUPS_UNIX = "onGroupListUpdate"
}
interface NTQQApiReturnData<PayloadType=unknown> extends Array<any> {
@@ -28,13 +32,14 @@ interface NTQQApiReturnData<PayloadType=unknown> extends Array<any> {
let receiveHooks: Array<{
method: ReceiveCmd,
hookFunc: (payload: unknown) => void
hookFunc: (payload: any) => void
}> = []
export function hookNTQQApiReceive(window: BrowserWindow) {
const originalSend = window.webContents.send;
const patchSend = (channel: string, ...args: NTQQApiReturnData) => {
// 判断是否是列表
log(`received ntqq api message: ${channel}`, JSON.stringify(args))
if (args?.[1] instanceof Array) {
for (let receiveData of args?.[1]) {
const ntQQApiMethodName = receiveData.cmdName;
@@ -75,6 +80,25 @@ export function registerReceiveHook<PayloadType>(method: ReceiveCmd, hookFunc: (
})
}
function updateGroups(_groups: Group[]){
for(let group of _groups){
let existGroup = groups.find(g=>g.groupCode == group.groupCode)
if (!existGroup){
groups.push(group)
}
else{
Object.assign(existGroup, group);
}
}
}
registerReceiveHook<{groupList: Group[]}>(ReceiveCmd.GROUPS, (payload)=>updateGroups(payload.groupList))
registerReceiveHook<{groupList: Group[]}>(ReceiveCmd.GROUPS_UNIX, (payload)=>updateGroups(payload.groupList))
registerReceiveHook<any>(ReceiveCmd.USER_INFO, (payload)=>{
log("user info", payload);
})
registerReceiveHook<{ msgList: Array<RawMessage> }>(ReceiveCmd.UPDATE_MSG, (payload) => {
for (const message of payload.msgList) {
msgHistory[message.msgId] = message;
@@ -84,7 +108,25 @@ registerReceiveHook<{ msgList: Array<RawMessage> }>(ReceiveCmd.UPDATE_MSG, (payl
registerReceiveHook<{ msgList: Array<RawMessage> }>(ReceiveCmd.NEW_MSG, (payload) => {
for (const message of payload.msgList) {
log("收到新消息push到历史记录", message)
msgHistory[message.msgId] = message;
if (!msgHistory[message.msgId]){
msgHistory[message.msgId] = message
}
else{
Object.assign(msgHistory[message.msgId], message)
}
}
})
registerReceiveHook<{msgRecord: RawMessage}>(ReceiveCmd.SELF_SEND_MSG, ({msgRecord})=>{
const message = msgRecord;
const peerUid = message.peerUid;
// log("收到自己发送成功的消息", Object.keys(sendMessagePool), message);
const sendCallback = sendMessagePool[peerUid];
if (sendCallback){
try{
sendCallback(message);
}catch(e){
log("receive self msg error", e.stack)
}
}
})

View File

@@ -2,7 +2,11 @@ import {ipcMain} from "electron";
import {v4 as uuidv4} from "uuid";
import {hookApiCallbacks} from "./hook";
import {log} from "../common/utils";
import {ChatType, Group, GroupMember, User} from "../common/types";
import { ChatType } from "./types";
import { Group } from "./types";
import { GroupMember } from "./types";
import { RawMessage } from "./types";
import { User } from "./types";
import {SendMessageElement} from "./types";
interface IPCReceiveEvent {
@@ -48,9 +52,9 @@ enum NTQQApiChannel {
IPC_UP_1 = "IPC_UP_1",
}
interface Peer {
export interface Peer {
chatType: ChatType
peerUid: string // 是uid还是QQ号
peerUid: string // 如果是群聊uid为群号私聊uid就是加密的字符串
guildId?: ""
}
@@ -70,6 +74,13 @@ function callNTQQApi<ReturnType>(channel: NTQQApiChannel, className: NTQQApiClas
})
}
export let sendMessagePool: Record<string, ((sendSuccessMsg: RawMessage)=>void) | null> = {}// peerUid: callbackFunnc
interface GeneralCallResult{
result:0,
errMsg: string
}
export class NTQQApi {
// static likeFriend = defineNTQQApi<void>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.LIKE_FRIEND)
@@ -91,14 +102,14 @@ export class NTQQApi {
}
static getFriends(forced = false) {
return callNTQQApi(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.FRIENDS, [{force_update: forced}, undefined])
return callNTQQApi<GeneralCallResult>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.FRIENDS, [{force_update: forced}, undefined])
}
static getGroups(forced = false) {
return callNTQQApi<Group[]>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.GROUPS, [{force_update: forced}, undefined])
return callNTQQApi<GeneralCallResult>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.GROUPS, [{force_update: forced}, undefined])
}
static async getGroupMembers(groupQQ: string, num = 3000) {
static async getGroupMembers(groupQQ: string, num = 5000) {
const sceneId = callNTQQApi(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.GROUP_MEMBER_SCENE, [{
groupCode: groupQQ,
scene: "groupMemberList_MainWindow"
@@ -114,13 +125,11 @@ export class NTQQApi {
}
static async getUserInfo(uid: string) {
const result = await callNTQQApi<[{
payload: { profiles: Map<string, User> }
}]>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.USER_INFO,
const result = await callNTQQApi<GeneralCallResult>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.USER_INFO,
[{force: true, uids: [uid]}, undefined])
return new Promise<User>(resolve => {
resolve(result[0].payload.profiles.get(uid))
})
log("get user info result", result);
return result[0].payload.profiles.get(uid);
}
static getFileType(filePath: string) {
@@ -134,7 +143,7 @@ export class NTQQApi {
}
static copyFile(filePath: string, destPath: string) {
return callNTQQApi<string>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.FS_API, NTQQApiMethod.FILE_COPY, [filePath, destPath])
return callNTQQApi<string>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.FS_API, NTQQApiMethod.FILE_COPY, [{ fromPath: filePath, toPath: destPath }])
}
static getImageSize(filePath: string) {
@@ -152,7 +161,7 @@ export class NTQQApi {
static async uploadFile(filePath: string) {
const md5 = await NTQQApi.getFileMd5(filePath);
const fileName = `${md5}.${(await NTQQApi.getFileType(filePath)).ext}`;
const mediaPath = await callNTQQApi<string>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.FS_API, NTQQApiMethod.MEDIA_FILE_PATH, [{
const mediaPath = await callNTQQApi<string>(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.MEDIA_FILE_PATH, [{
path_info: {
md5HexStr: md5,
fileName: fileName,
@@ -164,6 +173,7 @@ export class NTQQApi {
file_uuid: ""
}
}])
log("media path", mediaPath)
await NTQQApi.copyFile(filePath, mediaPath);
const fileSize = await NTQQApi.getFileSize(filePath);
return {
@@ -178,11 +188,49 @@ export class NTQQApi {
return callNTQQApi(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.RECALL_MSG, [{peer, msgIds}, null])
}
static sendMsg(peer: Peer, msgElements: SendMessageElement){
return callNTQQApi(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.SEND_MSG, [{
msgId: "0",
peer, msgElements,
msgAttributeInfos: new Map(),
}, null])
static sendMsg(peer: Peer, msgElements: SendMessageElement[]){
const sendTimeout = 10 * 1000
return new Promise<RawMessage>((resolve, reject)=>{
const peerUid = peer.peerUid;
let usingTime = 0;
let success = false;
const checkSuccess = ()=>{
if (!success){
sendMessagePool[peerUid] = null;
reject("发送超时")
}
}
setTimeout(checkSuccess, sendTimeout);
const checkLastSend = ()=>{
let lastSending = sendMessagePool[peerUid]
if (sendTimeout < usingTime){
sendMessagePool[peerUid] = null;
reject("发送超时")
}
if (!!lastSending){
// log("有正在发送的消息,等待中...")
usingTime += 100;
setTimeout(checkLastSend, 100);
}
else{
log("可以进行发送消息,设置发送成功回调", sendMessagePool)
sendMessagePool[peerUid] = (rawMessage: RawMessage)=>{
success = true;
sendMessagePool[peerUid] = null;
resolve(rawMessage);
}
}
}
checkLastSend()
callNTQQApi(NTQQApiChannel.IPC_UP_2, NTQQApiClass.NT_API, NTQQApiMethod.SEND_MSG, [{
msgId: "0",
peer, msgElements,
msgAttributeInfos: new Map(),
}, null]).then()
})
}
}

View File

@@ -1,3 +1,60 @@
export interface User {
uid: string; // 加密的字符串
uin: string; // QQ号
nick: string;
avatarUrl?: string;
longNick: string; // 签名
raw: {
remark: string;
};
}
export interface Group{
groupCode: string,
maxMember: number,
memberCount: number,
groupName: string,
groupStatus: 0,
memberRole: 2,
isTop: boolean,
toppedTimestamp: "0",
privilegeFlag: number, //65760
isConf: boolean,
hasModifyConfGroupFace: boolean,
hasModifyConfGroupName: boolean,
remarkName: string,
hasMemo: boolean,
groupShutupExpireTime: string, //"0",
personShutupExpireTime: string, //"0",
discussToGroupUin: string, //"0",
discussToGroupMaxMsgSeq: number,
discussToGroupTime: number,
groupFlagExt: number, //1073938496,
authGroupType: number, //0,
groupCreditLevel: number, //0,
groupFlagExt3: number, //0,
groupOwnerId: {
"memberUin": string, //"0",
"memberUid": string, //"u_fbf8N7aeuZEnUiJAbQ9R8Q"
},
members: GroupMember[] // 原始数据是没有这个的,为了方便自己加了这个字段
}
export interface GroupMember {
avatarPath: string;
cardName: string;
cardType: number;
isDelete: boolean;
nick: string;
qid: string;
remark: string;
role: number; // 群主:4, 管理员:3群员:2
shutUpTime: number; // 禁言时间,单位是什么暂时不清楚
uid: string; // 加密的字符串
uin: string; // QQ号
}
export enum ElementType {
TEXT = 1,
PIC = 2,
@@ -68,3 +125,89 @@ export interface SendReplyElement {
}
export type SendMessageElement = SendTextElement | SendPttElement | SendPicElement | SendReplyElement
export enum AtType {
notAt = 0,
atAll = 1,
atUser = 2
}
export enum ChatType {
friend = 1,
group = 2,
temp = 100
}
export interface PttElement {
canConvert2Text: boolean;
duration: number; // 秒数
fileBizId: null;
fileId: number; // 0
fileName: string; // "e4d09c784d5a2abcb2f9980bdc7acfe6.amr"
filePath: string; // "/Users//Library/Containers/com.tencent.qq/Data/Library/Application Support/QQ/nt_qq_a6b15c9820595d25a56c1633ce19ad40/nt_data/Ptt/2023-11/Ori/e4d09c784d5a2abcb2f9980bdc7acfe6.amr"
fileSize: string; // "4261"
fileSubId: string; // "0"
fileUuid: string; // "90j3z7rmRphDPrdVgP9udFBaYar#oK0TWZIV"
formatType: string; // 1
invalidState: number; // 0
md5HexStr: string; // "e4d09c784d5a2abcb2f9980bdc7acfe6"
playState: number; // 0
progress: number; // 0
text: string; // ""
transferStatus: number; // 0
translateStatus: number; // 0
voiceChangeType: number; // 0
voiceType: number; // 0
waveAmplitudes: number[];
}
export interface ArkElement {
bytesData: string;
}
export interface RawMessage {
msgId: string;
msgTime: string;
msgSeq: string;
senderUin: string; // 发送者QQ号
peerUid: string; // 群号 或者 QQ uid
peerUin: string; // 群号 或者 发送者QQ号
sendNickName: string;
sendMemberName?: string; // 发送者群名片
chatType: ChatType;
elements: {
replyElement: {
senderUid: string; // 原消息发送者QQ号
sourceMsgIsIncPic: boolean; // 原消息是否有图片
sourceMsgText: string;
replayMsgSeq: string; // 源消息的msgSeq可以通过这个找到源消息的msgId
};
textElement: {
atType: AtType;
atUid: string;
content: string;
atNtUid: string;
};
picElement: {
sourcePath: string; // 图片本地路径
picWidth: number;
picHeight: number;
fileSize: number;
fileName: string;
fileUuid: string;
};
pttElement: PttElement;
arkElement: ArkElement;
}[];
}
export interface MessageElement {
raw: RawMessage;
peer: any;
sender: {
uid: string; // 一串加密的字符串
memberName: string;
nickname: string;
};
}

View File

@@ -1,5 +1,5 @@
import {OB11MessageDataType, OB11GroupMemberRole, OB11Message, OB11MessageData} from "./types";
import {AtType, ChatType, RawMessage} from "../common/types";
import {OB11MessageDataType, OB11GroupMemberRole, OB11Message, OB11MessageData, OB11Group, OB11GroupMember, Friend} from "./types";
import { AtType, ChatType, Group, GroupMember, RawMessage, User } from '../ntqqapi/types';
import {getFriend, getGroupMember, getHistoryMsgBySeq, msgHistory, selfInfo} from "../common/data";
import {file2base64, getConfigUtil} from "../common/utils";
@@ -28,13 +28,13 @@ export class OB11Constructor {
}
if (msg.chatType == ChatType.group) {
resMsg.group_id = msg.peerUin
const member = getGroupMember(msg.peerUin, msg.senderUin);
const member = await getGroupMember(msg.peerUin, msg.senderUin);
if (member) {
resMsg.sender.role = OB11Constructor.groupMemberRole(member.role);
}
} else if (msg.chatType == ChatType.friend) {
resMsg.sub_type = "friend"
const friend = getFriend(msg.senderUin);
const friend = await getFriend(msg.senderUin);
if (friend) {
resMsg.sender.nickname = friend.nickName;
}
@@ -54,7 +54,7 @@ export class OB11Constructor {
message_data["data"]["qq"] = "all"
} else {
let uid = element.textElement.atUid
let atMember = getGroupMember(msg.peerUin, uid)
let atMember = await getGroupMember(msg.peerUin, uid)
message_data["data"]["mention"] = atMember?.uin
message_data["data"]["qq"] = atMember?.uin
}
@@ -119,11 +119,48 @@ export class OB11Constructor {
return resMsg;
}
static groupMemberRole(role: number): OB11GroupMemberRole {
static friend(friend: User): Friend{
return {
user_id: friend.uin,
nickname: friend.nickName,
remark: friend.raw.remark
}
}
static friends(friends: User[]): Friend[]{
return friends.map(OB11Constructor.friend)
}
static groupMemberRole(role: number): OB11GroupMemberRole | undefined {
return {
4: OB11GroupMemberRole.owner,
3: OB11GroupMemberRole.admin,
2: OB11GroupMemberRole.member
}[role]
}
static groupMember(group_id: string, member: GroupMember): OB11GroupMember{
return {
group_id,
user_id: member.uin,
nickname: member.nick,
card: member.cardName
}
}
static groupMembers(group: Group): OB11GroupMember[]{
return group.members.map(m=>OB11Constructor.groupMember(group.groupCode, m))
}
static group(group: Group): OB11Group{
return {
group_id: group.groupCode,
group_name: group.groupName
}
}
static groups(groups: Group[]): OB11Group[]{
return groups.map(OB11Constructor.group)
}
}

View File

@@ -1,4 +1,11 @@
import {AtType, RawMessage} from "../common/types";
import {SelfInfo} from "../common/types";
import { AtType } from "../ntqqapi/types";
import { RawMessage } from "../ntqqapi/types";
import { User } from "../ntqqapi/types";
export interface Friend extends SelfInfo{
remark?: string
}
export enum OB11UserSex{
male = "male",
@@ -12,6 +19,27 @@ export enum OB11GroupMemberRole{
member = "member",
}
export interface OB11GroupMember {
group_id: string
user_id: string
nickname: string
card?: string
sex?: OB11UserSex
age?: number
join_time?: number
last_sent_time?: number
level?: number
role?: OB11GroupMemberRole
title?: string
}
export interface OB11Group{
group_id: string
group_name: string
member_count?: number
max_member_count?: number
}
interface OB11Sender {
user_id: string,
nickname: string,
@@ -45,9 +73,11 @@ export interface OB11Message {
}
export type OB11ApiName =
"send_private_msg"
"send_msg"
| "send_private_msg"
| "send_group_msg"
| "get_group_list"
| "get_group_info"
| "get_friend_list"
| "delete_msg"
| "get_login_info"
@@ -103,3 +133,10 @@ export type OB11MessageData = {
id: string,
}
}
export interface OB11PostSendMsg {
message_type?: "private" | "group"
user_id: string,
group_id?: string,
message: OB11MessageData[] | string | OB11MessageData;
}

63
src/onebot11/utils.ts Normal file
View File

@@ -0,0 +1,63 @@
import { CONFIG_DIR, isGIF } from "../common/utils";
import * as path from 'path';
import { NTQQApi } from '../ntqqapi/ntcall';
const fs = require("fs").promises;
export async function uri2local(fileName: string, uri: string){
let filePath = path.join(CONFIG_DIR, fileName)
let url = new URL(uri);
if (url.protocol == "base64:") {
// base64转成文件
let base64Data = uri.split("base64://")[1]
try {
const buffer = Buffer.from(base64Data, 'base64');
await fs.writeFile(filePath, buffer);
} catch (e: any) {
return {
success: false,
errMsg: `base64文件下载失败,` + e.toString(),
path: ""
}
}
} else if (url.protocol == "http:" || url.protocol == "https:") {
// 下载文件
let res = await fetch(url)
if (!res.ok) {
return {
success: false,
errMsg: `${url}下载失败,` + res.statusText,
path: ""
}
}
let blob = await res.blob();
let buffer = await blob.arrayBuffer();
try {
await fs.writeFile(filePath, Buffer.from(buffer));
} catch (e: any) {
return {
success: false,
errMsg: `${url}下载失败,` + e.toString(),
path: ""
}
}
} else if (url.protocol === "file:"){
await fs.copyFile(url.pathname, filePath);
// filePath = (await NTQQApi.uploadFile(url.pathname)).path;
}
else{
return {
success: false,
errMsg: `不支持的file协议,` + url.protocol,
path: ""
}
}
if (isGIF(filePath)) {
await fs.rename(filePath, filePath + ".gif");
filePath += ".gif";
}
return {
success: true,
errMsg: "",
path: filePath
};
}

View File

@@ -1,6 +1,9 @@
// Electron 主进程 与 渲染进程 交互的桥梁
import {Config, Group, PostDataSendMsg, RawMessage, SelfInfo, User} from "./common/types";
import {Config, SelfInfo} from "./common/types";
import { Group } from "./ntqqapi/types";
import { RawMessage } from "./ntqqapi/types";
import { User } from "./ntqqapi/types";
import {
CHANNEL_DOWNLOAD_FILE,
CHANNEL_GET_CONFIG,
@@ -46,12 +49,7 @@ contextBridge.exposeInMainWorld("llonebot", {
data: msgResult,
});
},
listenSendMessage: (handle: (jsonData: SendIPCMsgSession<PostDataSendMsg>) => void) => {
ipcRenderer.send(CHANNEL_LOG, "发送消息API已注册");
ipcRenderer.on(CHANNEL_SEND_MSG, (event: any, args: SendIPCMsgSession<PostDataSendMsg>) => {
handle(args)
})
},
listenRecallMessage: (handle: (jsonData: {message_id: string}) => void) => {
ipcRenderer.on(CHANNEL_RECALL_MSG, (event: any, args: {message_id: string}) => {
handle(args)

View File

@@ -1,440 +1,5 @@
/// <reference path="./global.d.ts" />
import {AtType, ChatType, Group, MessageElement, Peer, PostDataSendMsg, RawMessage, User} from "./common/types";
import {OB11SendMsgReturn} from "./onebot11/types";
import {ipcRenderer} from "electron";
import {CHANNEL_GET_HISTORY_MSG} from "./common/channels";
import { SendIPCMsgSession } from "./main/ipcsend";
let groups: Group[] = []
let friends: User[] = []
let uid_maps: Record<string, User> = {} // 一串加密的字符串 -> qq号
function getStrangerByUin(uin: string) {
for (const key in uid_maps) {
if (uid_maps[key].uin === uin) {
return uid_maps[key];
}
}
}
async function getHistoryMsg(msgId: string) {
return await window.llonebot.getHistoryMsg(msgId);
}
async function getUserInfo(uid: string): Promise<User> {
let user = uid_maps[uid]
if (!user) {
// 从服务器获取用户信息
user = await window.LLAPI.getUserInfo(uid)
uid_maps[uid] = user
}
return user
}
async function getFriends() {
let _friends = await window.LLAPI.getFriendsList(false)
for (let friend of _friends) {
let existFriend = friends.find(f => f.uin == friend.uin)
if (!existFriend) {
friends.push(friend)
}
}
window.llonebot.updateFriends(friends)
return friends
}
async function getFriend(qq: string) {
let friend = friends.find(friend => friend.uin == qq)
if (!friend) {
await getFriends();
friend = friends.find(friend => friend.uin == qq);
}
return friend;
}
async function getGroup(qq: string) {
let group = groups.find(group => group.uid == qq)
if (!group) {
await getGroups();
group = groups.find(group => group.uid == qq)
}
return group
}
async function getGroups() {
let __groups = await window.LLAPI.getGroupsList(false)
for (let group of __groups) {
group.members = [];
let existGroup = groups.find(g => g.uid == group.uid)
if (!existGroup) {
// console.log("更新群列表", groups)
groups.push(group)
}
}
window.llonebot.updateGroups(groups)
return groups
}
async function getGroupMembers(group_qq: string, forced: boolean = false) {
let group = await getGroup(group_qq)
if (!group?.members || group!.members!.length == 0 || forced) {
let res = (await window.LLAPI.getGroupMemberList(group_qq, 5000))
// console.log(`更新群${group}成员列表 await`, _res)
// window.LLAPI.getGroupMemberList(group_qq + "_groupMemberList_MainWindow", 5000).then(res =>{
let members = res.result.infos.values();
console.log("getGroupMemberList api response", res)
if (members && forced) {
group.members = []
}
for (const member of members) {
if (!group!.members!.find(m => m.uid == member.uid)) {
group!.members!.push(member)
}
uid_maps[member.uid] = {
uin: member.uin,
uid: member.uid,
nickName: member.nick
};
}
window.llonebot.updateGroups(groups)
console.log(`更新群${group.name}成员列表`, group)
// })
}
return group?.members
}
async function getGroupMember(group_qq: string, member_uid: string) {
let members = await getGroupMembers(group_qq)
if (members) {
let member = members.find(member => member.uid == member_uid)
if (!member) {
members = await getGroupMembers(group_qq, true)
member = members?.find(member => member.uid == member_uid)
}
return member
}
}
async function listenSendMessage(session: SendIPCMsgSession<PostDataSendMsg>) {
const postData = session.data
console.log("收到发送消息请求", postData);
let sendMsgResult: OB11SendMsgReturn = {
retcode: 0,
status: 0,
data: {
message_id: ""
},
message: "发送成功"
}
if (postData.action == "send_private_msg" || postData.action == "send_group_msg") {
let peer: Peer | null = null;
if (!postData.params) {
postData.params = {
message: postData.message,
user_id: postData.user_id,
group_id: postData.group_id
}
}
if (postData.action == "send_private_msg") {
let friend = await getFriend(postData.params.user_id)
if (friend) {
console.log("好友消息", postData)
peer = {
chatType: ChatType.friend,
name: friend.nickName,
uid: friend.uid
}
} else {
// 临时消息
console.log("发送临时消息", postData)
let receiver = getStrangerByUin(postData.params.user_id);
if (receiver) {
peer = {
chatType: ChatType.temp,
name: receiver.nickName,
uid: receiver.uid
}
} else {
sendMsgResult.status = -1;
sendMsgResult.retcode = -1;
sendMsgResult.message = `发送失败,未找到对象${postData.params.user_id},检查他是否为好友或是群友`;
}
}
} else if (postData.action == "send_group_msg") {
let group = await getGroup(postData.params.group_id)
if (group) {
peer = {
chatType: ChatType.group,
name: group.name,
uid: group.uid
}
} else {
sendMsgResult.status = -1;
sendMsgResult.retcode = -1;
sendMsgResult.message = `发送失败,未找到群${postData.params.group_id}`;
console.log("未找到群, 发送群消息失败", postData)
}
}
if (peer) {
let sendFiles: string[] = [];
for (let message of postData.params.message) {
if (message.type == "at") {
// @ts-ignore
message.type = "text"
message.atType = AtType.atUser
let atUid = message.data?.qq || message.atUid
if (atUid == "all") {
message.atType = AtType.atAll
atUid = "0";
message.content = `@全体成员`
} else {
let group = await getGroup(postData.params.group_id)
let atMember = group.members.find(member => member.uin == atUid)
message.atNtUid = atMember.uid
message.atUid = atUid
message.content = `@${atMember.cardName || atMember.nick}`
}
} else if (message.type == "text") {
message.content = message.data?.text || message.content
} else if (message.type == "image" || message.type == "voice" || message.type == "record") {
// 收到的是uri格式的需要转成本地的, uri格式有三种http, file, base64
let url = message.data?.file || message.file
let uri = new URL(url);
let ext: string;
if (message.type == "image") {
ext = ".png"
}
if (message.type == "voice" || message.type == "record") {
message.type = "voice"
ext = ".amr"
}
let localFilePath = `${Date.now()}${ext}`
if (uri.protocol == "file:") {
localFilePath = url.split("file://")[1]
} else {
const {errMsg, path} = await window.llonebot.downloadFile({
uri: url,
fileName: `${Date.now()}${ext}`
})
console.log("下载文件结果", errMsg, path)
if (errMsg) {
console.log("下载文件失败", errMsg);
sendMsgResult.status = -1;
sendMsgResult.retcode = -1;
sendMsgResult.message = `发送失败,下载文件失败,${errMsg}`;
break;
} else {
localFilePath = path;
}
sendFiles.push(localFilePath);
}
message.file = localFilePath
} else if (message.type == "reply") {
let msgId = message.data?.id || message.msgId
const rawMsg: RawMessage = await getHistoryMsg(msgId)
if (rawMsg){
message.msgId = msgId
message.msgSeq = rawMsg.msgSeq
}
}
}
console.log("发送消息", postData)
if (sendMsgResult.status !== 0) {
window.llonebot.sendSendMsgResult(session.id, sendMsgResult)
return;
}
window.LLAPI.sendMessage(peer, postData.params.message).then(
(res: MessageElement) => {
console.log("消息发送成功:", res, peer, postData.params.message)
if (sendFiles.length) {
window.llonebot.deleteFile(sendFiles);
}
sendMsgResult.data.message_id = res.raw.msgId;
window.llonebot.sendSendMsgResult(session.id, sendMsgResult)
},
err => {
sendMsgResult.status = -1;
sendMsgResult.retcode = -1;
sendMsgResult.message = `发送失败,${err}`;
window.llonebot.sendSendMsgResult(session.id, sendMsgResult)
console.log("消息发送失败", postData, err)
})
} else {
console.log(sendMsgResult, postData);
window.llonebot.sendSendMsgResult(session.id, sendMsgResult)
}
}
}
function recallMessage(msgId: string) {
getHistoryMsg(msgId).then((msg: RawMessage)=>{
const peer: Peer ={
chatType: msg.chatType,
name: "",
uid: msg.peerUin
}
window.LLAPI.recallMessage(peer, [msgId]).then()
})
}
let chatListEle: HTMLCollectionOf<Element>
async function getGroupsMembers(groupsArg: Group[]) {
// 批量获取群成员列表
let failedGroups: Group[] = []
for (const group of groupsArg) {
let handledGroup = await getGroupMembers(group.uid, true)
if (handledGroup.length == 0) {
failedGroups.push(group)
}
}
if (failedGroups.length > 0) {
console.log("获取群成员列表失败,重试", failedGroups.map(group => group.name))
setTimeout(() => {
getGroupsMembers(failedGroups).then()
}, 1000)
} else {
window.llonebot.log("全部群成员获取完毕")
}
}
async function initAccountInfo() {
let accountInfo = await window.LLAPI.getAccountInfo();
window.llonebot.log("getAccountInfo " + JSON.stringify(accountInfo));
if (!accountInfo.uid) {
return false;
}
let selfInfo = await window.LLAPI.getUserInfo(accountInfo.uid);
window.llonebot.setSelfInfo({
user_id: accountInfo.uin,
nickname: selfInfo.nickName
});
window.llonebot.log("selfInfo " + JSON.stringify(selfInfo));
return true;
}
function onLoad() {
window.llonebot.log("llonebot render onLoad");
window.llonebot.getRunningStatus().then(running => {
if (running) {
return;
}
initAccountInfo().then(
(initSuccess) => {
if (!initSuccess) {
return;
}
if (friends.length == 0) {
getFriends().then(() => {
});
}
if (groups.length == 0) {
getGroups().then(() => {
getGroupsMembers(groups).then(() => {
});
});
}
window.llonebot.log("llonebot render start");
window.llonebot.startExpress();
window.llonebot.listenSendMessage((session: SendIPCMsgSession<PostDataSendMsg>) => {
listenSendMessage(session).then().catch(err => console.log("listenSendMessage err", err))
})
window.llonebot.listenRecallMessage((arg: { message_id: string }) => {
// console.log("listenRecallMessage", arg)
recallMessage(arg.message_id)
})
window.llonebot.log("llonebot loaded");
// window.LLAPI.add_qmenu((qContextMenu: Node) => {
// let btn = document.createElement("a")
// btn.className = "q-context-menu-item q-context-menu-item--normal vue-component"
// btn.setAttribute("aria-disabled", "false")
// btn.setAttribute("role", "menuitem")
// btn.setAttribute("tabindex", "-1")
// btn.onclick = () => {
// // window.LLAPI.getPeer().then(peer => {
// // // console.log("current peer", peer)
// // if (peer && peer.chatType == "group") {
// // getGroupMembers(peer.uid, true).then(()=> {
// // console.log("获取群成员列表成功", groups);
// // alert("获取群成员列表成功")
// // })
// // }
// // })
// async function func() {
// for (const group of groups) {
// await getGroupMembers(group.uid, true)
// }
// }
//
// func().then(() => {
// console.log("获取群成员列表结果", groups);
// // 找到members数量为空的群
// groups.map(group => {
// if (group.members.length == 0) {
// console.log(`${group.name}群成员为空`)
// }
// })
// window.llonebot.updateGroups(groups)
// })
// }
// btn.innerText = "获取群成员列表"
// console.log(qContextMenu)
// // qContextMenu.appendChild(btn)
// })
//
// window.LLAPI.on("context-msg-menu", (event, target, msgIds) => {
// console.log("msg menu", event, target, msgIds);
// })
//
// // console.log("getAccountInfo", LLAPI.getAccountInfo());
// function getChatListEle() {
// chatListEle = document.getElementsByClassName("viewport-list__inner")
// console.log("chatListEle", chatListEle)
// if (chatListEle.length == 0) {
// setTimeout(getChatListEle, 500)
// } else {
// try {
// // 选择要观察的目标节点
// const targetNode = chatListEle[0];
//
// // 创建一个观察器实例并传入回调函数
// const observer = new MutationObserver(function (mutations) {
// mutations.forEach(function (mutation) {
// // console.log("chat list changed", mutation.type); // 输出 mutation 的类型
// // 获得当前聊天窗口
// window.LLAPI.getPeer().then(peer => {
// // console.log("current peer", peer)
// if (peer && peer.chatType == "group") {
// getGroupMembers(peer.uid, false).then()
// }
// })
// });
// });
//
// // 配置观察选项
// const config = {attributes: true, childList: true, subtree: true};
//
// // 传入目标节点和观察选项
// observer.observe(targetNode, config);
//
// } catch (e) {
// window.llonebot.log(e)
// }
// }
// }
//
// // getChatListEle();
}
);
});
}
// 打开设置界面时触发
async function onSettingWindowCreated(view: Element) {
@@ -584,7 +149,6 @@ async function onSettingWindowCreated(view: Element) {
}
setTimeout(onLoad, 5000)
export {
onSettingWindowCreated

View File

@@ -1,16 +1,21 @@
import {getConfigUtil, log} from "../common/utils";
// const express = require("express");
import express = require("express");
import {Request} from 'express';
const express = require("express");
import {Request, text} from 'express';
import {Response} from 'express';
const JSONbig = require('json-bigint');
const JSONbig = require('json-bigint')({storeAsString: true});
import {sendIPCRecallQQMsg, sendIPCSendQQMsg} from "../main/ipcsend";
import {PostDataSendMsg} from "../common/types";
import {friends, groups, msgHistory, selfInfo} from "../common/data";
import {OB11ApiName, OB11Message, OB11Return, OB11MessageData} from "../onebot11/types";
import {AtType, ChatType, Group, SelfInfo} from "../common/types";
import {friends, getGroup, getGroupMember, getStrangerByUin, groups, msgHistory, selfInfo} from "../common/data";
import { OB11ApiName, OB11Message, OB11Return, OB11MessageData, OB11Group, OB11GroupMember, OB11PostSendMsg, OB11MessageDataType, Friend } from '../onebot11/types';
import {OB11Constructor} from "../onebot11/constructor";
import { NTQQApi } from "../ntqqapi/ntcall";
import { Peer } from "../ntqqapi/ntcall";
import { ElementType, SendMessageElement } from "../ntqqapi/types";
import { SendMsgElementConstructor } from "../ntqqapi/constructor";
import { uri2local } from "../onebot11/utils";
import { v4 as uuid4 } from 'uuid';
// @SiberianHusky 2021-08-15
@@ -51,16 +56,20 @@ function checkSendMessage(sendMsgList: OB11MessageData[]) {
// ==end==
function constructReturnData(status: number, data: any = {}, message: string = "") {
function constructReturnData<T>(data: T, status: number=0, message: string = ""): OB11Return<T> {
return {
status: status,
retcode: status,
data: data,
message: message
}
}
function constructErrorReturn(err: string){
return constructReturnData(null, -1, err);
}
function handlePost(jsonData: any, handleSendResult: (data: OB11Return<any>) => void) {
log("API receive post:" + JSON.stringify(jsonData))
if (!jsonData.params) {
@@ -74,9 +83,7 @@ function handlePost(jsonData: any, handleSendResult: (data: OB11Return<any>) =>
message: ''
}
if (jsonData.action == "get_login_info") {
resData["data"] = selfInfo
} else if (jsonData.action == "send_private_msg" || jsonData.action == "send_group_msg") {
if (jsonData.action == "send_private_msg" || jsonData.action == "send_group_msg") {
if (jsonData.action == "send_private_msg") {
jsonData.message_type = "private"
} else {
@@ -99,8 +106,8 @@ function handlePost(jsonData: any, handleSendResult: (data: OB11Return<any>) =>
return {
group_id: group.uid,
group_name: group.name,
member_count: group.members.length,
group_members: group.members.map(member => {
member_count: group.members?.length,
group_members: group.members?.map(member => {
return {
user_id: member.uin,
user_name: member.cardName || member.nick,
@@ -115,18 +122,18 @@ function handlePost(jsonData: any, handleSendResult: (data: OB11Return<any>) =>
resData["data"] = {
group_id: group.uid,
group_name: group.name,
member_count: group.members.length,
member_count: group.members?.length,
}
}
} else if (jsonData.action == "get_group_member_info") {
let member = groups.find(group => group.uid == jsonData.params.group_id)?.members?.find(member => member.uin == jsonData.params.user_id)
resData["data"] = {
user_id: member.uin,
user_name: member.nick,
user_display_name: member.cardName || member.nick,
nickname: member.nick,
card: member.cardName,
role: OB11Constructor.groupMemberRole(member.role),
user_id: member?.uin,
user_name: member?.nick,
user_display_name: member?.cardName || member?.nick,
nickname: member?.nick,
card: member?.cardName,
role: member && OB11Constructor.groupMemberRole(member.role),
}
} else if (jsonData.action == "get_group_member_list") {
let group = groups.find(group => group.uid == jsonData.params.group_id)
@@ -159,112 +166,93 @@ function handlePost(jsonData: any, handleSendResult: (data: OB11Return<any>) =>
}
export function startExpress(port: number) {
const app = express();
// 中间件用于解析POST请求的请求体
app.use(express.urlencoded({extended: true, limit: "500mb"}));
app.use(express.json({
limit: '500mb',
verify: (req: any, res: any, buf: any, encoding: any) => {
req.rawBody = buf;
}
}));
app.use((req: any, res: any, next: any) => {
try {
req.body = JSONbig.parse(req.rawBody.toString());
next();
} catch (error) {
// next(error);
next();
}
const expressAPP = express();
expressAPP.use(express.urlencoded({extended: true, limit: "500mb"}));
expressAPP.use((req, res, next) => {
let data = '';
req.on('data', chunk => {
data += chunk.toString();
});
async function registerRouter<PayloadType, ReturnDataType>(action: OB11ApiName, handle: (payload: PayloadType) => Promise<OB11Return<ReturnDataType>>) {
let url = action.toString()
if (!action.startsWith("/")){
url = "/" + action
}
async function _handle(res: Response, payload: PayloadType) {
res.send(await handle(payload))
req.on('end', () => {
if (data) {
try {
// log("receive raw", data)
req.body = JSONbig.parse(data);
} catch (e) {
return next(e);
}
}
next();
});
});
// expressAPP.use(express.json({
// limit: '500mb',
// verify: (req: any, res: any, buf: any, encoding: any) => {
// req.rawBody = buf;
// }
// }));
app.post(url, (req: Request, res: Response) => {
_handle(res, req.body).then()
});
app.get(url, (req: Request, res: Response) => {
_handle(res, req.query as any).then()
});
}
export function startExpress(port: number) {
function parseToOnebot12(action: OB11ApiName) {
app.post('/' + action, (req: Request, res: Response) => {
let jsonData: PostDataSendMsg = req.body;
jsonData.action = action
let resData = handlePost(jsonData, (data: OB11Return<any>) => {
res.send(data)
})
if (resData) {
res.send(resData)
}
});
}
// function parseToOnebot12(action: OB11ApiName) {
// expressAPP.post('/' + action, (req: Request, res: Response) => {
// let jsonData: PostDataSendMsg = req.body;
// jsonData.action = action
// let resData = handlePost(jsonData, (data: OB11Return<any>) => {
// res.send(data)
// })
// if (resData) {
// res.send(resData)
// }
// });
// }
const actionList: OB11ApiName[] = ["get_login_info", "send_private_msg", "send_group_msg",
"get_group_list", "get_friend_list", "delete_msg", "get_group_member_list", "get_group_member_info"]
for (const action of actionList) {
parseToOnebot12(action as OB11ApiName)
}
// for (const action of actionList) {
// parseToOnebot12(action as OB11ApiName)
// }
app.get('/', (req: Request, res: Response) => {
expressAPP.get('/', (req: Request, res: Response) => {
res.send('llonebot已启动');
})
// 处理POST请求的路由
app.post('/', (req: Request, res: Response) => {
let jsonData: PostDataSendMsg = req.body;
let resData = handlePost(jsonData, (data: OB11Return<any>) => {
res.send(data)
})
if (resData) {
res.send(resData)
}
});
app.post('/send_msg', (req: Request, res: Response) => {
let jsonData: PostDataSendMsg = req.body;
if (jsonData.message_type == "private") {
jsonData.action = "send_private_msg"
} else if (jsonData.message_type == "group") {
jsonData.action = "send_group_msg"
} else {
if (jsonData.params.group_id) {
jsonData.action = "send_group_msg"
} else {
jsonData.action = "send_private_msg"
}
}
let resData = handlePost(jsonData, (data: OB11Return<any>) => {
res.send(data)
})
if (resData) {
res.send(resData)
}
})
// expressAPP.post('/', (req: Request, res: Response) => {
// let jsonData: PostDataSendMsg = req.body;
// let resData = handlePost(jsonData, (data: OB11Return<any>) => {
// res.send(data)
// })
// if (resData) {
// res.send(resData)
// }
// });
// expressAPP.post('/send_msg', (req: Request, res: Response) => {
// let jsonData: PostDataSendMsg = req.body;
// if (jsonData.message_type == "private") {
// jsonData.action = "send_private_msg"
// } else if (jsonData.message_type == "group") {
// jsonData.action = "send_group_msg"
// } else {
// if (jsonData.params?.group_id) {
// jsonData.action = "send_group_msg"
// } else {
// jsonData.action = "send_private_msg"
// }
// }
// let resData = handlePost(jsonData, (data: OB11Return<any>) => {
// res.send(data)
// })
// if (resData) {
// res.send(resData)
// }
// })
registerRouter<{ message_id: string }, OB11Message>("get_msg", async (payload) => {
const msg = msgHistory[payload.message_id.toString()]
if (msg) {
const msgData = await OB11Constructor.message(msg);
return constructReturnData(0, msgData)
} else {
return constructReturnData(1, {}, "消息不存在")
}
}).then(()=>{
})
app.listen(port, "0.0.0.0", () => {
expressAPP.listen(port, "0.0.0.0", () => {
console.log(`llonebot started 0.0.0.0:${port}`);
});
}
@@ -292,3 +280,195 @@ export function postMsg(msg: OB11Message) {
});
}
}
let routers: Record<string, (payload: any)=>Promise<OB11Return<any>>> = {};
function registerRouter<PayloadType, ReturnDataType>(action: OB11ApiName, handle: (payload: PayloadType) => Promise<OB11Return<ReturnDataType | null>>) {
let url = action.toString()
if (!action.startsWith("/")){
url = "/" + action
}
async function _handle(res: Response, payload: PayloadType) {
log("receive post data", url, payload)
try{
const result = await handle(payload)
res.send(result)
}
catch(e){
log(e.stack);
res.send(constructErrorReturn(e.stack.toString()))
}
}
expressAPP.post(url, (req: Request, res: Response) => {
_handle(res, req.body).then()
});
expressAPP.get(url, (req: Request, res: Response) => {
_handle(res, req.query as any).then()
});
routers[url] = handle
}
registerRouter<{ message_id: string }, OB11Message>("get_msg", async (payload) => {
log("history msg ids", Object.keys(msgHistory));
const msg = msgHistory[payload.message_id.toString()]
if (msg) {
const msgData = await OB11Constructor.message(msg);
return constructReturnData(msgData)
} else {
return constructErrorReturn("消息不存在")
}
})
registerRouter<{}, SelfInfo>("get_login_info", async (payload)=>{
return constructReturnData(selfInfo);
})
registerRouter<{}, Friend[]>("get_friend_list", async (payload)=>{
return constructReturnData(OB11Constructor.friends(friends));
})
registerRouter<{}, OB11Group[]>("get_group_list", async (payload)=>{
return constructReturnData(OB11Constructor.groups(groups));
})
registerRouter<{group_id: number}, OB11Group[]>("get_group_info", async (payload)=>{
const group = await getGroup(payload.group_id.toString())
if (group){
return constructReturnData(OB11Constructor.groups(groups));
}
else{
return constructErrorReturn(`${payload.group_id}不存在`)
}
})
registerRouter<{group_id: number}, OB11GroupMember[]>("get_group_member_list", async (payload)=>{
const group = await getGroup(payload.group_id.toString());
if (group){
return constructReturnData(OB11Constructor.groupMembers(group));
}
else{
return constructErrorReturn(`${payload.group_id}不存在`)
}
})
registerRouter<{group_id: number, user_id: number}, OB11GroupMember>("get_group_member_info", async (payload)=>{
const member = await getGroupMember(payload.group_id.toString(), payload.user_id.toString())
if (member){
return constructReturnData(OB11Constructor.groupMember(payload.group_id.toString(), member))
}
else{
return constructErrorReturn(`群成员${payload.user_id}不存在`)
}
})
const handleSendMsg = async (payload)=>{
const peer: Peer = {
chatType: ChatType.friend,
peerUid: ""
}
let group: Group | undefined = undefined;
if(payload?.group_id){
group = groups.find(g=>g.uid == payload.group_id?.toString())
if (!group){
return constructErrorReturn(`${payload.group_id}不存在`)
}
peer.chatType = ChatType.group
// peer.name = group.name
peer.peerUid = group.uid
}
else if (payload?.user_id){
const friend = friends.find(f=>f.uin == payload.user_id.toString())
if (friend){
// peer.name = friend.nickName
peer.peerUid = friend.uid
}
else{
peer.chatType = ChatType.temp
const tempUser = getStrangerByUin(payload.user_id.toString())
if (!tempUser){
return constructErrorReturn(`找不到私聊对象${payload.user_id}`)
}
// peer.name = tempUser.nickName
peer.peerUid = tempUser.uid
}
}
if (typeof payload.message === "string"){
payload.message = [{
type: OB11MessageDataType.text,
data: {
text: payload.message
}
}] as OB11MessageData[]
}
else if (!Array.isArray(payload.message)){
payload.message = [payload.message]
}
const sendElements: SendMessageElement[] = []
for (let sendMsg of payload.message){
switch(sendMsg.type){
case OB11MessageDataType.text: {
const text = sendMsg.data?.text;
if (text){
sendElements.push(SendMsgElementConstructor.text(sendMsg.data!.text))
}
}break;
case OB11MessageDataType.at: {
let atQQ = sendMsg.data?.qq;
if (atQQ){
atQQ = atQQ.toString()
if (atQQ === "all"){
sendElements.push(SendMsgElementConstructor.at(atQQ, atQQ, AtType.atAll, "全体成员"))
}
else{
const atMember = group?.members.find(m=>m.uin == atQQ)
if (atMember){
sendElements.push(SendMsgElementConstructor.at(atQQ, atMember.uid, AtType.atUser, atMember.cardName || atMember.nick))
}
}
}
}break;
case OB11MessageDataType.reply: {
let replyMsgId = sendMsg.data.id;
if (replyMsgId){
replyMsgId = replyMsgId.toString()
const replyMsg = msgHistory[replyMsgId]
if (replyMsg){
sendElements.push(SendMsgElementConstructor.reply(replyMsg.msgSeq, replyMsgId, replyMsg.senderUin, replyMsg.senderUin))
}
}
}break;
case OB11MessageDataType.image: {
const file = sendMsg.data?.file
if (file){
const picPath = await (await uri2local(uuid4(), file)).path
if (picPath){
sendElements.push(await SendMsgElementConstructor.pic(picPath))
}
}
}break;
case OB11MessageDataType.voice: {
const file = sendMsg.data?.file
if (file){
const voicePath = await (await uri2local(uuid4(), file)).path
if (voicePath){
sendElements.push(await SendMsgElementConstructor.ptt(voicePath))
}
}
}
}
}
log("send msg:", peer, sendElements)
try{
const returnMsg = await NTQQApi.sendMsg(peer, sendElements)
return constructReturnData({message_id: returnMsg.msgId})
}catch(e){
return constructErrorReturn(e.toString())
}
}
registerRouter<OB11PostSendMsg, {message_id: string}>("send_msg", handleSendMsg)
registerRouter<OB11PostSendMsg, {message_id: string}>("send_private_msg", handleSendMsg)
registerRouter<OB11PostSendMsg, {message_id: string}>("send_group_msg", handleSendMsg)

View File

@@ -50,7 +50,7 @@ module.exports = {
]
},
optimization: {
minimize: true,
minimize: false,
minimizer: [
new TerserPlugin({
extractComments: false,