import { AsyncLocalStorage } from "node:async_hooks";

export type QiaoqiaoToolContext = {
  channel: "qiaoqiao";
  accountId?: string;
  sessionKey?: string;
  senderOpenId?: string;
  chatId?: string;
};

const qiaoqiaoToolContext = new AsyncLocalStorage<QiaoqiaoToolContext>();

export function getCurrentQiaoqiaoToolContext(): QiaoqiaoToolContext | undefined {
  return qiaoqiaoToolContext.getStore();
}

export function runWithQiaoqiaoToolContext<T>(
  context: QiaoqiaoToolContext,
  fn: () => Promise<T>,
): Promise<T> {
  return qiaoqiaoToolContext.run(context, fn);
}