LLOneBot/src/ntqqapi/wrapper.ts

import {
  NodeIKernelBuddyService,
  NodeIKernelGroupService,
  NodeIKernelProfileService,
  NodeIKernelProfileLikeService,
  NodeIKernelMSFService,
  NodeIKernelMsgService,
  NodeIKernelUixConvertService,
  NodeIKernelRichMediaService,
  NodeIKernelTicketService,
  NodeIKernelTipOffService,
  NodeIKernelSearchService
} from './services'
import { constants } from 'node:os'
import { Dict } from 'cosmokit'
const Process = require('node:process')

export interface NodeIQQNTWrapperSession {
  getBuddyService(): NodeIKernelBuddyService
  getGroupService(): NodeIKernelGroupService
  getProfileService(): NodeIKernelProfileService
  getProfileLikeService(): NodeIKernelProfileLikeService
  getMsgService(): NodeIKernelMsgService
  getMSFService(): NodeIKernelMSFService
  getUixConvertService(): NodeIKernelUixConvertService
  getRichMediaService(): NodeIKernelRichMediaService
  getTicketService(): NodeIKernelTicketService
  getTipOffService(): NodeIKernelTipOffService
  getSearchService(): NodeIKernelSearchService
}

export interface WrapperApi {
  NodeIQQNTWrapperSession?: NodeIQQNTWrapperSession
}

export interface WrapperConstructor {
  [key: string]: unknown
}

const wrapperApi: WrapperApi = {}

export const wrapperConstructor: WrapperConstructor = {}

const constructor = [
  'NodeIKernelBuddyListener',
  'NodeIKernelGroupListener',
  'NodeQQNTWrapperUtil',
  'NodeIKernelMsgListener',
  'NodeIQQNTWrapperEngine',
  'NodeIGlobalAdapter',
  'NodeIDependsAdapter',
  'NodeIDispatcherAdapter',
  'NodeIKernelSessionListener',
  'NodeIKernelLoginService',
  'NodeIKernelLoginListener',
  'NodeIKernelProfileService',
  'NodeIKernelProfileListener',
]

Process.dlopenOrig = Process.dlopen

Process.dlopen = function (module: Dict, filename: string, flags = constants.dlopen.RTLD_LAZY) {
  const dlopenRet = this.dlopenOrig(module, filename, flags)
  for (const export_name in module.exports) {
    module.exports[export_name] = new Proxy(module.exports[export_name], {
      construct: (target, args) => {
        const ret = new target(...args)
        if (export_name === 'NodeIQQNTWrapperSession') wrapperApi.NodeIQQNTWrapperSession = ret
        return ret
      }
    })
    if (constructor.includes(export_name)) {
      wrapperConstructor[export_name] = module.exports[export_name]
    }
  }
  return dlopenRet
}

export function getSession() {
  return wrapperApi['NodeIQQNTWrapperSession']
}