1
0
Fork 0
mirror of https://github.com/muerwre/vk-tg-bot.git synced 2025-04-24 22:46:41 +07:00

added working message handler

This commit is contained in:
Fedor Katurov 2021-04-28 13:03:10 +07:00
parent 12c8e30a1e
commit 1754092f7c
6 changed files with 77 additions and 39 deletions

View file

@ -9,7 +9,7 @@ async function main() {
try { try {
const config = prepareConfig(); const config = prepareConfig();
const telegram = new TelegramService(config.telegram); const telegram = new TelegramService(config.telegram);
const vkService = new VkService(config.vk); const vkService = new VkService(config.vk, telegram);
const telegramApi = new TelegramApi(telegram).listen(); const telegramApi = new TelegramApi(telegram).listen();
await telegram.start(); await telegram.start();

View file

@ -1,12 +1,28 @@
import { VkEventHandler } from "./types"; import { VkEventHandler } from "./VkEventHandler";
import { ContextDefaultState, MessageContext } from "vk-io"; import { MessageContext } from "vk-io";
import { NextMiddleware } from "middleware-io"; import { NextMiddleware } from "middleware-io";
import logger from "../../logger";
import { ContextDefaultState } from "vk-io/lib/structures/contexts/context";
export class MessageNewHandler extends VkEventHandler {
public execute = async (
context: MessageContext<ContextDefaultState>,
next: NextMiddleware
) => {
if (context.isOutbox) {
await next();
return;
}
const users = await this.instance.api.users.get({
user_ids: [String(context.senderId)],
});
const from = users[0];
logger.debug(
`received message from ${from.first_name} ${from.last_name}: ${context.text}`
);
export class MessageNewHandler extends VkEventHandler<
MessageContext<ContextDefaultState>
> {
public execute = async (context, next: NextMiddleware) => {
console.log("received message!");
await next(); await next();
}; };
} }

View file

@ -0,0 +1,21 @@
import { NextMiddleware } from "middleware-io";
import { ConfigGroup, GroupInstance } from "../types";
import { VkService } from "../index";
import { TelegramService } from "../../telegram";
export abstract class VkEventHandler {
public constructor(
protected group: ConfigGroup,
protected instance: GroupInstance,
protected vk: VkService,
protected telegram: TelegramService
) {}
public execute: (
context: any,
next: NextMiddleware
) => Promise<void> = async (ctx, next) => {
console.log(`vk received unknown event`, ctx);
await next();
};
}

View file

@ -1,11 +1,14 @@
import { VkEvent } from "../types"; import { VkEvent } from "../types";
import { VkEventHandler } from "./types"; import { VkEventHandler } from "./VkEventHandler";
import { MessageNewHandler } from "./MessageNewHandler"; import { MessageNewHandler } from "./MessageNewHandler";
export const vkEventToHandler: Record<VkEvent, typeof VkEventHandler> = { type DerivedHandler = typeof VkEventHandler;
[VkEvent.GroupJoin]: VkEventHandler, interface Handler extends DerivedHandler {}
[VkEvent.GroupLeave]: VkEventHandler,
export const vkEventToHandler: Record<VkEvent, Handler> = {
[VkEvent.GroupJoin]: MessageNewHandler,
[VkEvent.GroupLeave]: MessageNewHandler,
[VkEvent.MessageNew]: MessageNewHandler, [VkEvent.MessageNew]: MessageNewHandler,
[VkEvent.PostSuggestion]: VkEventHandler, [VkEvent.PostSuggestion]: MessageNewHandler,
[VkEvent.WallPostNew]: VkEventHandler, [VkEvent.WallPostNew]: MessageNewHandler,
}; };

View file

@ -1,13 +0,0 @@
import { NextMiddleware } from "middleware-io";
import { ConfigGroup } from "../types";
export class VkEventHandler<T = any> {
public constructor(protected config: ConfigGroup) {}
public execute: (context: T, next: NextMiddleware) => Promise<void> = async (
ctx,
next
) => {
console.log(`vk received unknown event`, ctx);
await next();
};
}

View file

@ -4,15 +4,16 @@ import logger from "../logger";
import { Request, Response } from "express"; import { Request, Response } from "express";
import { flatten, has, keys } from "ramda"; import { flatten, has, keys } from "ramda";
import { NextFunction } from "connect"; import { NextFunction } from "connect";
import { VkEventHandler } from "./handlers/types"; import { VkEventHandler } from "./handlers/VkEventHandler";
import { vkEventToHandler } from "./handlers"; import { vkEventToHandler } from "./handlers";
import { TelegramService } from "../telegram";
export class VkService { export class VkService {
public endpoint: string = "/"; public endpoint: string = "/";
private readonly instances: Record<string, GroupInstance>; private readonly instances: Record<string, GroupInstance>;
private readonly groups: Record<number, ConfigGroup>; private readonly groups: Record<number, ConfigGroup>;
constructor(private config: VkConfig) { constructor(private config: VkConfig, private telegram: TelegramService) {
if (!config.groups.length) { if (!config.groups.length) {
throw new Error("No vk groups to handle. Specify them in config"); throw new Error("No vk groups to handle. Specify them in config");
} }
@ -63,7 +64,6 @@ export class VkService {
private createGroupInstance = (group: ConfigGroup): GroupInstance => { private createGroupInstance = (group: ConfigGroup): GroupInstance => {
const api = new API({ const api = new API({
token: group.apiKey, token: group.apiKey,
apiBaseUrl: this.config.endpoint,
}); });
const upload = new Upload({ api }); const upload = new Upload({ api });
const updates = new Updates({ const updates = new Updates({
@ -73,7 +73,14 @@ export class VkService {
webhookSecret: group.secretKey, webhookSecret: group.secretKey,
}); });
const handlers = this.setupHandlers(group); const instance = {
api,
upload,
updates,
};
const handlers = this.setupHandlers(group, instance);
handlers.forEach((channel) => { handlers.forEach((channel) => {
keys(channel).forEach((event) => { keys(channel).forEach((event) => {
console.log(`updates in ${String(event)}`); console.log(`updates in ${String(event)}`);
@ -81,22 +88,26 @@ export class VkService {
}); });
}); });
return { return instance;
api,
upload,
updates,
};
}; };
/** /**
* Setups handlers * Setups handlers
*/ */
private setupHandlers(group: ConfigGroup): Record<VkEvent, VkEventHandler>[] { private setupHandlers(
group: ConfigGroup,
instance: GroupInstance
): Record<VkEvent, VkEventHandler>[] {
return flatten( return flatten(
group.channels.map((chan) => group.channels.map((chan) =>
chan.events.reduce((acc, event) => { chan.events.reduce((acc, event) => {
const handler = vkEventToHandler[event]; const handler = new (vkEventToHandler as any)[event](
return { ...acc, [event]: new handler(group) }; group,
instance,
this,
this.telegram
);
return { ...acc, [event]: handler };
}, {} as Record<VkEvent, VkEventHandler>[]) }, {} as Record<VkEvent, VkEventHandler>[])
) )
); );