balebaazoo 1.2.3 → 1.4.0

package/dist/index.js

@@ -414,7 +414,13 @@ var Api = class {
     return this.call("answerPreCheckoutQuery", asParams(params));
   }
   inquireTransaction(params) {
-    return this.call("inquireTransaction", asParams(params));
+    const transactionId = params.transaction_id || params.provider_payment_charge_id;
+    if (!transactionId) {
+      throw new Error("transaction_id is required for inquireTransaction");
+    }
+    return this.call("inquireTransaction", {
+      transaction_id: transactionId
+    });
   }
 };
 function asParams(params) {
@@ -471,6 +477,13 @@ function matchUpdate(update) {
   const matches = [];
   if (update.message) {
     matches.push("message");
+    if (update.message.animation) matches.push("message:animation");
+    if (update.message.new_chat_members?.length) {
+      matches.push("message:new_chat_members");
+    }
+    if (update.message.left_chat_member) {
+      matches.push("message:left_chat_member");
+    }
     if (update.message.text !== void 0) matches.push("message:text");
     if (update.message.photo) matches.push("message:photo");
     if (update.message.document) matches.push("message:document");
@@ -530,20 +543,32 @@ function matchesChatType(ctx, chatType) {
 
 // src/composer.ts
 var Composer = class _Composer {
-  handler;
   registered = [];
+  dirty = true;
+  snapshot = [];
+  cachedHandler;
   constructor(...middleware) {
     this.registered.push(...middleware.map(normalizeMiddleware));
-    this.handler = async (ctx, next) => {
-      await runMiddleware(this.registered, ctx);
-      await next();
-    };
+    this.markDirty();
+  }
+  markDirty() {
+    this.dirty = true;
+    this.cachedHandler = void 0;
   }
   middleware() {
-    return this.handler;
+    if (!this.cachedHandler || this.dirty) {
+      this.snapshot = [...this.registered];
+      this.cachedHandler = async (ctx, next) => {
+        await runMiddleware(this.snapshot, ctx);
+        await next();
+      };
+      this.dirty = false;
+    }
+    return this.cachedHandler;
   }
   use(...middleware) {
     this.registered.push(...middleware.map(normalizeMiddleware));
+    this.markDirty();
     return this;
   }
   on(filter, ...middleware) {
@@ -567,6 +592,7 @@ var Composer = class _Composer {
       );
     });
     composer.registered.push(...middleware);
+    this.markDirty();
     return this;
   }
   command(command, ...middleware) {
@@ -621,6 +647,17 @@ var Composer = class _Composer {
       ...middleware
     );
   }
+  static compose(...composers) {
+    const result = new _Composer();
+    for (const item of composers) {
+      if (item instanceof _Composer) {
+        result.use(item);
+      } else {
+        result.use(item);
+      }
+    }
+    return result;
+  }
 };
 function escapeRegex(value) {
   return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
@@ -633,6 +670,8 @@ var Context = class {
   botInfo;
   callbackQueryAnswered = false;
   _updateTypes;
+  session = {};
+  state = {};
   constructor(options) {
     this.api = options.api;
     this.update = options.update;
@@ -731,6 +770,54 @@ var Context = class {
     const chatId = this.requireChatId();
     return this.api.sendInvoice({ chat_id: chatId, ...params });
   }
+  targetMessage() {
+    return this.callbackQuery?.message ?? this.message;
+  }
+  async editMessageText(text, extra) {
+    const message = this.targetMessage();
+    if (!message) {
+      throw new Error("No message to edit in context");
+    }
+    return this.api.editMessageText({
+      chat_id: message.chat.id,
+      message_id: message.message_id,
+      text,
+      ...extra
+    });
+  }
+  async deleteMessage(params) {
+    const message = this.targetMessage();
+    const chatId = params?.chat_id ?? message?.chat.id ?? this.chatId;
+    const messageId = params?.message_id ?? message?.message_id;
+    if (chatId === void 0 || messageId === void 0) {
+      throw new Error("No message to delete in context");
+    }
+    return this.api.deleteMessage({ chat_id: chatId, message_id: messageId });
+  }
+  async forwardMessage(toChatId, params) {
+    const message = this.targetMessage();
+    if (!message && !params?.from_chat_id) {
+      throw new Error("No source message in context");
+    }
+    return this.api.forwardMessage({
+      chat_id: toChatId,
+      from_chat_id: params?.from_chat_id ?? message.chat.id,
+      message_id: params?.message_id ?? message.message_id,
+      ...params
+    });
+  }
+  async copyMessage(toChatId, params) {
+    const message = this.targetMessage();
+    if (!message && !params?.from_chat_id) {
+      throw new Error("No source message in context");
+    }
+    return this.api.copyMessage({
+      chat_id: toChatId,
+      from_chat_id: params?.from_chat_id ?? message.chat.id,
+      message_id: params?.message_id ?? message.message_id,
+      ...params
+    });
+  }
   async sendChatAction(action) {
     const chatId = this.requireChatId();
     return this.api.sendChatAction({ chat_id: chatId, action });
@@ -840,14 +927,7 @@ var PollingRunner = class {
           { signal }
         );
         this.backoffAttempt = 0;
-        for (const update of updates) {
-          this.offset = update.update_id + 1;
-          try {
-            await bot.handleUpdate(update);
-          } catch (error) {
-            onError(error);
-          }
-        }
+        await this.processUpdates(bot, updates, options, onError);
         if (updates.length === 0 && signal.aborted) {
           break;
         }
@@ -866,6 +946,34 @@ var PollingRunner = class {
       }
     }
   }
+  async processUpdates(bot, updates, options, onError) {
+    const concurrency = Math.max(1, options.concurrency ?? 1);
+    if (concurrency === 1) {
+      for (const update of updates) {
+        this.offset = update.update_id + 1;
+        try {
+          await bot.handleUpdate(update);
+        } catch (error) {
+          onError(error);
+        }
+      }
+      return;
+    }
+    let index = 0;
+    const workers = Array.from({ length: Math.min(concurrency, updates.length) }, async () => {
+      while (index < updates.length) {
+        const current = index++;
+        const update = updates[current];
+        this.offset = update.update_id + 1;
+        try {
+          await bot.handleUpdate(update);
+        } catch (error) {
+          onError(error);
+        }
+      }
+    });
+    await Promise.all(workers);
+  }
   setOffset(offset) {
     this.offset = offset;
   }
@@ -933,7 +1041,7 @@ async function webhookFromJson(bot) {
 // src/bot.ts
 var Bot = class extends Composer {
   api;
-  botInfo;
+  _botInfo;
   polling = new PollingRunner();
   autoAnswer;
   catchHandler;
@@ -941,17 +1049,23 @@ var Bot = class extends Composer {
     const api = new Api({ token, ...options });
     super();
     this.api = api;
-    this.botInfo = options.botInfo;
+    this._botInfo = options.botInfo;
     this.autoAnswer = options.autoAnswerCallback ?? true;
     if (this.autoAnswer) {
       this.use(autoAnswerCallback());
     }
   }
+  get telegram() {
+    return this.api;
+  }
+  get botInfo() {
+    return this._botInfo;
+  }
   async init() {
-    if (!this.botInfo) {
-      this.botInfo = await this.api.getMe();
+    if (!this._botInfo) {
+      this._botInfo = await this.api.getMe();
     }
-    return this.botInfo;
+    return this._botInfo;
   }
   catch(handler) {
     this.catchHandler = handler;
@@ -974,7 +1088,7 @@ var Bot = class extends Composer {
     return new Context({
       api: this.api,
       update,
-      botInfo: this.botInfo
+      botInfo: this._botInfo
     });
   }
   async start(options = {}) {
@@ -1005,7 +1119,7 @@ var Bot = class extends Composer {
     }
     if (options.onStart) {
       try {
-        await options.onStart(this.botInfo);
+        await options.onStart(this._botInfo);
       } catch (error) {
         onError(error);
         return;
@@ -1016,6 +1130,13 @@ var Bot = class extends Composer {
   launch(options) {
     return this.start(options);
   }
+  async switchToPolling(options = {}) {
+    await this.api.deleteWebhook();
+    await this.start({
+      ...options,
+      dropPendingUpdates: options.dropPendingUpdates ?? true
+    });
+  }
   async stop() {
     await this.polling.stop();
   }
@@ -1136,6 +1257,132 @@ function setupGracefulShutdown(bot, options = {}) {
   }
 }
 
-export { Api, BaleAPIError, BaleError, BaleNetworkError, Bot, Composer, Context, DEFAULT_API_BASE, InlineKeyboard, InputFile, PollingRunner, ReplyKeyboard, autoAnswerCallback, bold, createWebhookHandler, errorHandler, extractCommand, isFilePath, isMiddlewareObject, italic, link, matchUpdate, matchesAnyFilter, matchesChatType, matchesFilter, md, normalizeMiddleware, removeKeyboard, runMiddleware, setupGracefulShutdown, spoiler, webhookFromJson };
+// src/scenes/index.ts
+var Scene = class extends Composer {
+  id;
+  enterMiddleware = [];
+  leaveMiddleware = [];
+  constructor(id, ...middleware) {
+    super(...middleware);
+    this.id = id;
+  }
+  enter(...middleware) {
+    this.enterMiddleware.push(...middleware);
+    return this;
+  }
+  leave(...middleware) {
+    this.leaveMiddleware.push(...middleware);
+    return this;
+  }
+  async runEnter(ctx) {
+    await runMiddleware(this.enterMiddleware, ctx);
+  }
+  async runLeave(ctx) {
+    await runMiddleware(this.leaveMiddleware, ctx);
+  }
+};
+var Stage = class extends Composer {
+  scenes = /* @__PURE__ */ new Map();
+  register(...scenes) {
+    for (const scene of scenes) {
+      this.scenes.set(scene.id, scene);
+    }
+    return this;
+  }
+  getScene(id) {
+    return this.scenes.get(id);
+  }
+  async enter(ctx, sceneId, state = {}) {
+    const scene = this.scenes.get(sceneId);
+    if (!scene) {
+      throw new Error(`Scene "${sceneId}" is not registered`);
+    }
+    const session2 = ctx.session;
+    const currentId = session2.__scene?.id;
+    if (currentId && currentId !== sceneId) {
+      const current = this.scenes.get(currentId);
+      if (current) {
+        await leaveScene(ctx, current);
+      }
+    }
+    await enterScene(ctx, sceneId, scene, state);
+  }
+  async leave(ctx) {
+    const session2 = ctx.session;
+    const sceneId = session2.__scene?.id;
+    if (!sceneId) return;
+    const scene = this.scenes.get(sceneId);
+    if (scene) {
+      await leaveScene(ctx, scene);
+    }
+  }
+  middleware() {
+    const sceneMiddleware = super.middleware();
+    return async (ctx, next) => {
+      const session2 = ctx.session;
+      const sceneId = session2.__scene?.id;
+      if (!sceneId) {
+        await sceneMiddleware(ctx, next);
+        return;
+      }
+      const scene = this.scenes.get(sceneId);
+      if (!scene) {
+        delete session2.__scene;
+        await sceneMiddleware(ctx, next);
+        return;
+      }
+      await scene.middleware()(ctx, next);
+    };
+  }
+};
+async function enterScene(ctx, sceneId, scene, initialState = {}) {
+  const session2 = ctx.session;
+  session2.__scene = { ...initialState, id: sceneId };
+  await scene.runEnter(ctx);
+}
+async function leaveScene(ctx, scene) {
+  const session2 = ctx.session;
+  await scene.runLeave(ctx);
+  delete session2.__scene;
+}
+
+// src/session/memory.ts
+function memorySessionStore() {
+  const data = /* @__PURE__ */ new Map();
+  return {
+    get: (key) => data.get(key),
+    set: (key, value) => {
+      data.set(key, value);
+    },
+    delete: (key) => {
+      data.delete(key);
+    }
+  };
+}
+
+// src/session/index.ts
+function session(options) {
+  const store = options.store ?? memorySessionStore();
+  return async (ctx, next) => {
+    const key = options.getSessionKey(ctx);
+    if (!key) {
+      await next();
+      return;
+    }
+    let current = store.get(key) ?? options.defaultSession(ctx);
+    Object.defineProperty(ctx, "session", {
+      configurable: true,
+      enumerable: true,
+      get: () => current,
+      set: (value) => {
+        current = value;
+      }
+    });
+    await next();
+    store.set(key, current);
+  };
+}
+
+export { Api, BaleAPIError, BaleError, BaleNetworkError, Bot, Composer, Context, DEFAULT_API_BASE, InlineKeyboard, InputFile, PollingRunner, ReplyKeyboard, Scene, Stage, autoAnswerCallback, bold, createWebhookHandler, enterScene, errorHandler, extractCommand, isFilePath, isMiddlewareObject, italic, leaveScene, link, matchUpdate, matchesAnyFilter, matchesChatType, matchesFilter, md, memorySessionStore, normalizeMiddleware, removeKeyboard, runMiddleware, session, setupGracefulShutdown, spoiler, webhookFromJson };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map