fix: telegram plugin fixes

Author: lorenzocorallo

backend/package.json

@@ -1,6 +1,6 @@
 {
   "name": "backend",
-  "version": "0.5.1",
+  "version": "0.5.2",
   "description": "PoliNetwork backend server",
   "private": true,
   "keywords": [],

backend/src/auth/index.ts

@@ -4,11 +4,13 @@ import { db } from "@/db/db";
 import { SCHEMA } from "@/db";
 import { env } from "@/env";
 import { AUTH_PATH } from "@/constants";
+import { telegramPlugin } from "./plugins/telegram";
 
 export const auth = betterAuth({
   basePath: AUTH_PATH,
   baseURL: env.PUBLIC_URL,
   trustedOrigins: env.TRUSTED_ORIGINS,
+  plugins: [telegramPlugin()],
   database: drizzleAdapter(db, {
     provider: "pg",
     schema: {

backend/src/auth/plugins/telegram/index.ts

@@ -1,6 +1,7 @@
 import { DB, SCHEMA } from "@/db";
-import { z, type BetterAuthPlugin } from "better-auth";
-import { createAuthEndpoint } from "better-auth/api";
+import { type BetterAuthPlugin } from "better-auth";
+import { z } from "zod";
+import { createAuthEndpoint, sessionMiddleware } from "better-auth/api";
 import crypto from "crypto";
 import { eq } from "drizzle-orm";
 
@@ -36,6 +37,8 @@ export const telegramPlugin = () => {
         "/telegram/link/start",
         {
           method: "POST",
+          use: [sessionMiddleware],
+          body: linkBody
         },
         async (ctx) => {
           const userId = ctx.context.session?.user.id;
@@ -44,14 +47,7 @@ export const telegramPlugin = () => {
               message: "You must be authenticated",
             });
 
-          const { data, success } = linkBody.safeParse(ctx.body);
-          if (!success)
-            return ctx.error("BAD_REQUEST", {
-              message:
-                "'telegramUsername' must be provided and must be a string",
-            });
-
-          const { telegramUsername } = data;
+          const { telegramUsername } = ctx.body;
           const [{ code, ttl }] = await DB.insert(SCHEMA.TG.link)
             .values({
               code: crypto
@@ -73,6 +69,7 @@ export const telegramPlugin = () => {
         "/telegram/link/verify/:code",
         {
           method: "GET",
+          use: [sessionMiddleware]
         },
         async (ctx) => {
           const userId = ctx.context.session?.user.id;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polinetwork/backend",
-  "version": "0.5.1",
+  "version": "0.5.2",
   "description": "Utils to interact with the backend.",
   "repository": {
     "type": "git",