mirror of
https://github.com/LLOneBot/LLOneBot.git
synced 2024-11-22 01:56:33 +00:00
refactor
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1,3 +1,4 @@
|
||||
node_modules/
|
||||
dist/
|
||||
.idea/
|
||||
.DS_Store
|
743
package-lock.json
generated
743
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
@@ -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",
|
||||
|
@@ -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];
|
||||
}
|
||||
}
|
||||
}
|
@@ -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
13
src/global.d.ts
vendored
@@ -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";
|
||||
|
||||
|
||||
|
@@ -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));
|
||||
|
166
src/main/main.ts
166
src/main/main.ts
@@ -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()
|
||||
}
|
||||
|
||||
|
||||
|
@@ -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,
|
||||
|
@@ -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)
|
||||
}
|
||||
}
|
||||
})
|
||||
|
@@ -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()
|
||||
})
|
||||
|
||||
}
|
||||
}
|
@@ -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;
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
|
@@ -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)
|
||||
}
|
||||
}
|
@@ -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
63
src/onebot11/utils.ts
Normal 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
|
||||
};
|
||||
}
|
@@ -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)
|
||||
|
436
src/renderer.ts
436
src/renderer.ts
@@ -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
|
||||
|
@@ -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)
|
@@ -50,7 +50,7 @@ module.exports = {
|
||||
]
|
||||
},
|
||||
optimization: {
|
||||
minimize: true,
|
||||
minimize: false,
|
||||
minimizer: [
|
||||
new TerserPlugin({
|
||||
extractComments: false,
|
||||
|
Reference in New Issue
Block a user