Merge branch 'task/#4' into task/#5

config.example.yml

@@ -23,6 +23,7 @@ templates:
 #      testResponse: 'testResponseCode'
 #      secretKey: 'groupSecretKey'
 #      apiKey: 'callbackApiKey'
+#      post_types: ['post','copy','reply','postpone','suggest']
 #      channels:
 #        - id: '@pogonia_test_chan'
 #          events:

src/config/types.ts

@@ -10,7 +10,7 @@ export interface Config extends Record<string, any> {
   http: HttpConfig;
   telegram: TelegramConfig;
   vk: VkConfig;
-  logger?: LoggerConfig;
-  templates?: TemplateConfig;
-  postgres?: PostgresConfig;
+  logger: LoggerConfig;
+  templates: TemplateConfig;
+  postgres: PostgresConfig;
 }

src/service/db/index.ts

@@ -9,13 +9,13 @@ export interface Storage {
     tgMessageId: number,
     groupId: number,
     channel: string
-  ): Promise<Event>;
+  ): Promise<Event | undefined>;
   getEventById(
     type: VkEvent,
     eventId: number,
     groupId: number,
     channel: string
-  ): Promise<Event>;
+  ): Promise<Event | undefined>;
   createEvent(
     type: VkEvent,
     eventId: number,
@@ -23,7 +23,7 @@ export interface Storage {
     channel: string,
     tgMessageId: number,
     text: Record<any, any>
-  ): Promise<Event>;
+  ): Promise<Event | undefined>;
   createOrUpdateLike(
     messageId: number,
     channel: string,
@@ -31,7 +31,11 @@ export interface Storage {
     text: string
   ): Promise<Like>;
   getLikesFor(channel: string, messageId: number): Promise<Like[]>;
-  getLikeBy(channel: string, messageId: number, author: number): Promise<Like>;
-  createPost(eventId: number, text: string): Promise<Post>;
-  findPostByEvent(eventId: number): Promise<Post>;
+  getLikeBy(
+    channel: string,
+    messageId: number,
+    author: number
+  ): Promise<Like | undefined>;
+  createPost(eventId: number, text: string): Promise<Post | undefined>;
+  findPostByEvent(eventId: number): Promise<Post | undefined>;
 }

src/service/db/postgres/entities/Event.ts

@@ -10,21 +10,21 @@ import { VkEvent } from "../../../vk/types";
 @Entity()
 export class Event {
   @PrimaryGeneratedColumn()
-  id: number;
+  id!: number;
   @Column()
-  type: VkEvent;
+  type!: VkEvent;
   @Column()
-  vkEventId: number;
+  vkEventId!: number;
   @Column()
-  vkGroupId: number;
+  vkGroupId!: number;
   @Column()
-  channel: string;
+  channel!: string;
   @Column()
-  tgMessageId: number;
+  tgMessageId!: number;
   @CreateDateColumn()
-  createdAt: Date;
+  createdAt!: Date;
   @UpdateDateColumn()
-  updatedAt: Date;
+  updatedAt!: Date;
   @Column("simple-json", { default: {}, nullable: false })
-  text: Record<any, any>;
+  text!: Record<any, any>;
 }

src/service/db/postgres/entities/Like.ts

@@ -9,17 +9,17 @@ import {
 @Entity()
 export class Like {
   @PrimaryGeneratedColumn()
-  id: number;
+  id!: number;
   @Column()
-  messageId: number;
+  messageId!: number;
   @Column()
-  channel: string;
+  channel!: string;
   @Column({ type: "text" })
-  text: string;
+  text!: string;
   @Column()
-  author: number;
+  author!: number;
   @CreateDateColumn()
-  createdAt: Date;
+  createdAt!: Date;
   @UpdateDateColumn()
-  updatedAt: Date;
+  updatedAt!: Date;
 }

src/service/db/postgres/entities/Post.ts

@@ -9,13 +9,13 @@ import {
 @Entity()
 export class Post {
   @PrimaryGeneratedColumn()
-  id: number;
+  id!: number;
   @Column()
-  eventId: number;
+  eventId!: number;
   @Column({ type: "text" })
-  text: string;
+  text!: string;
   @CreateDateColumn()
-  createdAt: Date;
+  createdAt!: Date;
   @UpdateDateColumn()
-  updatedAt: Date;
+  updatedAt!: Date;
 }

src/service/db/postgres/index.ts

@@ -11,10 +11,10 @@ import { Post } from "./entities/Post";
 const entities = [path.join(__dirname, "./entities/*")];
 
 export class PostgresDB implements Storage {
-  private connection: Connection;
-  private events: Repository<Event>;
-  private likes: Repository<Like>;
-  private posts: Repository<Post>;
+  private connection!: Connection;
+  private events!: Repository<Event>;
+  private likes!: Repository<Like>;
+  private posts!: Repository<Post>;
 
   constructor(private config: PostgresConfig) {}
 

src/service/logger/index.ts

@@ -6,7 +6,7 @@ const config = prepareConfig();
 const logger = createLogger({
   transports: new transports.Console({
     format: format.simple(),
-    level: config.logger.level || "info",
+    level: config.logger?.level || "info",
   }),
 });
 

src/service/telegram/index.ts

@@ -40,7 +40,7 @@ export class TelegramService {
     if (isWebhookEnabled) {
       await this.bot.telegram
         .deleteWebhook()
-        .then(() => this.bot.telegram.setWebhook(this.webhook.url))
+        .then(() => this.bot.telegram.setWebhook(this.webhook.url!))
         .then(async () => {
           const info = await this.bot.telegram.getWebhookInfo();
           if (!info.url) {
@@ -71,7 +71,7 @@ export class TelegramService {
    * Checks webhook availability
    */
   private getWebhookAvailable = async (): Promise<boolean> => {
-    const isWebhookEnabled = this.webhook.enabled && this.webhook.url;
+    const isWebhookEnabled = !!this.webhook.enabled && !!this.webhook.url;
     // TODO: test this.webhook.url with axios instead of 'true'
     return isWebhookEnabled && true;
   };

src/service/template/index.ts

@@ -1,6 +1,6 @@
 import extract from "remark-extract-frontmatter";
 import frontmatter from "remark-frontmatter";
-import compiler from "retext-stringify";
+import stringify from "retext-stringify";
 import parser from "remark-parse";
 import unified from "unified";
 import { parse } from "yaml";
@@ -26,7 +26,7 @@ export class Template<
       }
 
       const processor = unified()
-        .use(compiler)
+        .use(stringify)
         .use(frontmatter)
         .use(extract, { yaml: parse })
         .use(removeFrontmatter)
@@ -43,7 +43,7 @@ export class Template<
   }
 
   /**
-   * Themes the tempalte with values
+   * Themes the template with values
    */
   public theme = (values: V) => {
     return hb.compile(this.template)(values);
@@ -54,6 +54,7 @@ export class Template<
    */
   public static registerHelpers() {
     hb.registerHelper("ifEq", function (arg1, arg2, options) {
+      // @ts-ignore
       return arg1 == arg2 ? options.fn(this) : options.inverse(this);
     });
   }

src/service/vk/handlers/MessageNewHandler.ts

@@ -37,7 +37,7 @@ export class MessageNewHandler extends VkEventHandler<Fields, Values> {
     const parsed = this.template.theme({
       user,
       group: this.group,
-      text: context.text,
+      text: context?.text || "",
     });
 
     const extras: ExtraReplyMessage = {
@@ -46,7 +46,7 @@ export class MessageNewHandler extends VkEventHandler<Fields, Values> {
 
     this.appendButtons(extras, user.id);
 
-    await this.telegram.sendMessageToChan(this.channel, parsed, extras);
+    await this.telegram.sendMessageToChan(this.channel.id, parsed, extras);
 
     await next();
   };

src/service/vk/handlers/PostNewHandler.ts

@@ -21,7 +21,7 @@ type UrlPrefix = string;
 type ExtraGenerator = (
   text: string,
   eventId?: number
-) => Promise<InlineKeyboardButton[]>;
+) => Promise<InlineKeyboardButton[] | undefined>;
 
 interface Fields {
   image?: boolean;
@@ -35,6 +35,7 @@ interface Values {
   user?: UsersUserFull;
   group: ConfigGroup;
   text: string;
+  type?: string;
 }
 
 type LikeCtx = Composer.Context<CallbackQueryUpdate> & { match: string[] };
@@ -52,12 +53,9 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
 
   public execute = async (context: WallPostContext, next: NextMiddleware) => {
     const id = context?.wall?.id;
+    const postType = context?.wall?.postType;
 
-    if (
-      context.isRepost ||
-      !PostNewHandler.isValidPostType(context?.wall?.postType) ||
-      !id
-    ) {
+    if (context.isRepost || !this.isValidPostType(postType) || !id) {
       await next();
       return;
     }
@@ -75,9 +73,9 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
       ? await this.getUserByID(String(context.wall.signerId))
       : undefined;
 
-    const text = context.wall.text.trim();
+    const text = context.wall?.text?.trim() || "";
 
-    const parsed = this.themeText(text, user);
+    const parsed = this.themeText(text, postType, user);
 
     const extras: ExtraReplyMessage = {
       disable_web_page_preview: true,
@@ -95,13 +93,17 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
     if (hasThumb) {
       const thumb = await images.find((img) => img.mediumSizeUrl);
       msg = await this.telegram.sendPhotoToChan(
-        this.channel,
-        this.trimTextForPhoto(text, user),
-        thumb.mediumSizeUrl,
+        this.channel.id,
+        this.trimTextForPhoto(text, postType, user),
+        thumb?.mediumSizeUrl!,
         extras
       );
     } else {
-      msg = await this.telegram.sendMessageToChan(this.channel, parsed, extras);
+      msg = await this.telegram.sendMessageToChan(
+        this.channel.id,
+        parsed,
+        extras
+      );
     }
 
     const event = await this.createEvent(
@@ -109,7 +111,8 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
       msg.message_id,
       context.wall.toJSON()
     );
-    await this.db.createPost(event.id, context.wall.text);
+
+    await this.db.createPost(event!.id, context?.wall?.text || "");
 
     await next();
   };
@@ -117,17 +120,25 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
   /**
    * Checks if event of type we can handle
    */
-  public static isValidPostType(type: string): boolean {
+  private isValidPostType(type?: string): boolean {
+    if (!type) {
+      return false;
+    }
+
+    if (!this.channel.post_types) {
       return type === "post";
     }
 
+    return this.channel.post_types.includes(type);
+  }
+
   /**
    * Creates extras
    */
   private createKeyboard = async (
     text: string,
     eventId?: number
-  ): Promise<InlineKeyboardMarkup> => {
+  ): Promise<InlineKeyboardMarkup | undefined> => {
     const { buttons } = this.template.fields;
 
     if (!buttons?.length) {
@@ -137,7 +148,10 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
     const rows = await Promise.all(
       buttons.map((button) => this.extrasGenerators[button](text, eventId))
     );
-    const inline_keyboard = rows.filter((el) => el && el.length);
+
+    const inline_keyboard = rows.filter(
+      (el) => el && el.length
+    ) as InlineKeyboardButton[][];
 
     if (!inline_keyboard.length) {
       return;
@@ -153,13 +167,13 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
     const links = this.template.fields.links;
 
     if (!links) {
-      return [];
+      return;
     }
 
     const urls = extractURLs(text);
 
     if (!urls) {
-      return [];
+      return;
     }
 
     return urls
@@ -170,7 +184,7 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
 
         return label ? { text: links[label], url: url.toString() } : undefined;
       })
-      .filter((el) => el);
+      .filter((el) => el) as InlineKeyboardButton[];
   };
 
   /**
@@ -179,7 +193,15 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
   private generateLikes: ExtraGenerator = async (text, eventId) => {
     if (eventId) {
       const event = await this.getEventById(eventId);
-      const likes = await this.db.getLikesFor(this.channel, event.tgMessageId);
+      if (!event) {
+        throw new Error(`Can't find event`);
+      }
+
+      const likes = await this.db.getLikesFor(
+        this.channel.id,
+        event.tgMessageId
+      );
+
       const withCount = likes.reduce(
         (acc, like) => ({
           ...acc,
@@ -209,7 +231,7 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
   };
 
   /**
-   * Adds needed listeners
+   * Adds needed listeners for telegram
    */
   protected onInit = () => {
     if (this.template.fields.likes) {
@@ -227,7 +249,7 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
    * Reacts to like button press
    */
   private onLikeAction = async (ctx: LikeCtx, next) => {
-    const id = ctx.update.callback_query.message.message_id;
+    const id = ctx.update.callback_query?.message?.message_id;
     const author = ctx.update.callback_query.from.id;
     const [, channel, emo] = ctx.match;
     const event = await this.getEventByTgMessageId(id);
@@ -237,7 +259,7 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
       !emo ||
       !id ||
       !event ||
-      channel != this.channel ||
+      channel != this.channel.id ||
       !this.likes.includes(emo)
     ) {
       await next();
@@ -261,7 +283,7 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
     const markup = await this.createKeyboard(post.text, event.id);
 
     await ctx.telegram.editMessageReplyMarkup(
-      ctx.chat.id,
+      ctx.chat?.id,
       id,
       ctx.inlineMessageId,
       markup
@@ -274,6 +296,9 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
     next();
   };
 
+  /**
+   * Creates or updates like for {author} on {messageId} with {emo}
+   */
   private createOrUpdateLike = async (
     author: number,
     messageId: number,
@@ -281,23 +306,31 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
   ) => {
     return await this.db.createOrUpdateLike(
       messageId,
-      this.channel,
+      this.channel.id,
       author,
       emo
     );
   };
 
+  /**
+   * Gets like by {author} on {messageId}
+   */
   private getLike = async (author: number, messageId: number) => {
-    return await this.db.getLikeBy(this.channel, messageId, author);
+    return await this.db.getLikeBy(this.channel.id, messageId, author);
   };
 
   /**
    * Applies template theming to photos
    */
-  private themeText = (text: string, user?: UsersUserFull): string => {
+  private themeText = (
+    text: string,
+    type?: string,
+    user?: UsersUserFull
+  ): string => {
     return this.template.theme({
       user,
       group: this.group,
+      type,
       text,
     });
   };
@@ -305,15 +338,24 @@ export class PostNewHandler extends VkEventHandler<Fields, Values> {
   /**
    * Calculates, how much should we cut off the text to match photo caption limitations
    */
-  private trimTextForPhoto = (text: string, user: UsersUserFull): string => {
-    // Full markup
-    const full = this.themeText(text, user);
-    // Rest info except text
-    const others = this.themeText("", user);
+  private trimTextForPhoto = (
+    text: string,
+    type?: string,
+    user?: UsersUserFull
+  ): string => {
+    const withText = this.themeText(text, type, user);
 
-    // How much rest markup takes
-    const diff = full.length - others.length;
+    if (withText.length < PHOTO_CAPTION_LIMIT) {
+      return withText;
+    }
 
-    return full.slice(0, PHOTO_CAPTION_LIMIT - diff);
+    const withoutText = this.themeText("", type, user);
+    const suffix = "...";
+    const trimmed = text.slice(
+      0,
+      PHOTO_CAPTION_LIMIT - withoutText.length - suffix.length
+    );
+
+    return this.themeText(`${trimmed}${suffix}`, type, user);
   };
 }

src/service/vk/handlers/VkEventHandler.ts

@@ -1,5 +1,5 @@
 import { NextMiddleware } from "middleware-io";
-import { ConfigGroup, GroupInstance, VkEvent } from "../types";
+import { ConfigGroup, GroupChannel, GroupInstance, VkEvent } from "../types";
 import { VkService } from "../index";
 import { TelegramService } from "../../telegram";
 import { Template } from "../../template";
@@ -13,7 +13,7 @@ export class VkEventHandler<
   public constructor(
     protected type: VkEvent,
     protected group: ConfigGroup,
-    protected channel: string,
+    protected channel: GroupChannel,
     protected instance: GroupInstance,
     protected vk: VkService,
     protected telegram: TelegramService,
@@ -60,7 +60,7 @@ export class VkEventHandler<
       this.type,
       id,
       this.group.id,
-      this.channel
+      this.channel.id
     );
   };
 
@@ -78,7 +78,7 @@ export class VkEventHandler<
       this.type,
       tgMessageId,
       this.group.id,
-      this.channel
+      this.channel.id
     );
   };
 
@@ -94,7 +94,7 @@ export class VkEventHandler<
       this.type,
       id,
       this.group.id,
-      this.channel,
+      this.channel.id,
       tgMessageId,
       text
     );

src/service/vk/handlers/index.ts

@@ -1,4 +1,4 @@
-import { ConfigGroup, GroupInstance, VkEvent } from "../types";
+import { ConfigGroup, GroupChannel, GroupInstance, VkEvent } from "../types";
 import { VkEventHandler } from "./VkEventHandler";
 import { MessageNewHandler } from "./MessageNewHandler";
 import { StubHandler } from "./StubHandler";
@@ -12,7 +12,7 @@ interface Handler {
   new (
     type: VkEvent,
     group: ConfigGroup,
-    channel: string,
+    channel: GroupChannel,
     instance: GroupInstance,
     vk: VkService,
     telegram: TelegramService,

src/service/vk/index.ts

@@ -29,7 +29,7 @@ export class VkService {
       throw new Error("No vk groups to handle. Specify them in config");
     }
 
-    this.endpoint = config.endpoint;
+    this.endpoint = config.endpoint || "/";
 
     this.groups = config.groups.reduce(
       (acc, group) => ({
@@ -121,7 +121,7 @@ export class VkService {
           const handler = new vkEventToHandler[event](
             event,
             group,
-            chan.id,
+            chan,
             instance,
             this,
             this.telegram,

src/service/vk/types.ts

@@ -1,4 +1,5 @@
 import { API, Upload, Updates } from "vk-io";
+import { WallPostType } from "vk-io/lib/api/schemas/objects";
 
 export interface VkConfig extends Record<string, any> {
   groups: ConfigGroup[];
@@ -17,6 +18,7 @@ export interface ConfigGroup {
 export interface GroupChannel {
   id: string;
   events: VkEvent[];
+  post_types: WallPostType;
 }
 
 export enum VkEvent {

src/utils/extract.ts

@@ -13,5 +13,5 @@ export const extractURLs = (text: string): URL[] => {
         return;
       }
     })
-    .filter((el) => el);
+    .filter((el) => el) as URL[];
 };

tsconfig.json

@@ -12,7 +12,8 @@
     "target": "es2017",
     "baseUrl": ".",
     "paths": {},
-    "lib": ["esnext"]
+    "lib": ["esnext"],
+    "strict": true
   },
   "include": [
     "./**/*",