diff --git a/backend/src/app.module.ts b/backend/src/app.module.ts index 41608f7..a13eb9a 100644 --- a/backend/src/app.module.ts +++ b/backend/src/app.module.ts @@ -5,6 +5,7 @@ import { ScheduleModule } from "@nestjs/schedule"; import { ThrottlerModule } from "@nestjs/throttler"; import { redisStore } from "cache-manager-redis-yet"; import { ApiKeysModule } from "./api-keys/api-keys.module"; +import { AdminModule } from "./admin/admin.module"; import { AppController } from "./app.controller"; import { AppService } from "./app.service"; import { AuthModule } from "./auth/auth.module"; @@ -42,6 +43,7 @@ import { UsersModule } from "./users/users.module"; SessionsModule, ReportsModule, ApiKeysModule, + AdminModule, ScheduleModule.forRoot(), ThrottlerModule.forRootAsync({ imports: [ConfigModule], diff --git a/backend/src/auth/auth.module.ts b/backend/src/auth/auth.module.ts index 303727c..015222c 100644 --- a/backend/src/auth/auth.module.ts +++ b/backend/src/auth/auth.module.ts @@ -5,6 +5,9 @@ import { SessionsModule } from "../sessions/sessions.module"; import { UsersModule } from "../users/users.module"; import { AuthController } from "./auth.controller"; import { AuthService } from "./auth.service"; +import { AuthGuard } from "./guards/auth.guard"; +import { OptionalAuthGuard } from "./guards/optional-auth.guard"; +import { RolesGuard } from "./guards/roles.guard"; import { RbacService } from "./rbac.service"; import { RbacRepository } from "./repositories/rbac.repository"; @@ -16,7 +19,21 @@ import { RbacRepository } from "./repositories/rbac.repository"; DatabaseModule, ], controllers: [AuthController], - providers: [AuthService, RbacService, RbacRepository], - exports: [AuthService, RbacService, RbacRepository], + providers: [ + AuthService, + RbacService, + RbacRepository, + AuthGuard, + OptionalAuthGuard, + RolesGuard, + ], + exports: [ + AuthService, + RbacService, + RbacRepository, + AuthGuard, + OptionalAuthGuard, + RolesGuard, + ], }) export class AuthModule {} diff --git a/backend/src/auth/guards/optional-auth.guard.ts b/backend/src/auth/guards/optional-auth.guard.ts new file mode 100644 index 0000000..b517c23 --- /dev/null +++ b/backend/src/auth/guards/optional-auth.guard.ts @@ -0,0 +1,43 @@ +import { + CanActivate, + ExecutionContext, + Injectable, +} from "@nestjs/common"; +import { ConfigService } from "@nestjs/config"; +import { getIronSession } from "iron-session"; +import { JwtService } from "../../crypto/services/jwt.service"; +import { getSessionOptions, SessionData } from "../session.config"; + +@Injectable() +export class OptionalAuthGuard implements CanActivate { + constructor( + private readonly jwtService: JwtService, + private readonly configService: ConfigService, + ) {} + + async canActivate(context: ExecutionContext): Promise { + const request = context.switchToHttp().getRequest(); + const response = context.switchToHttp().getResponse(); + + const session = await getIronSession( + request, + response, + getSessionOptions(this.configService.get("SESSION_PASSWORD") as string), + ); + + const token = session.accessToken; + + if (!token) { + return true; + } + + try { + const payload = await this.jwtService.verifyJwt(token); + request.user = payload; + } catch { + // Ignore invalid tokens for optional auth + } + + return true; + } +} diff --git a/backend/src/categories/repositories/categories.repository.ts b/backend/src/categories/repositories/categories.repository.ts index a15b7c8..6545d64 100644 --- a/backend/src/categories/repositories/categories.repository.ts +++ b/backend/src/categories/repositories/categories.repository.ts @@ -1,5 +1,5 @@ import { Injectable } from "@nestjs/common"; -import { eq } from "drizzle-orm"; +import { eq, sql } from "drizzle-orm"; import { DatabaseService } from "../../database/database.service"; import { categories } from "../../database/schemas"; import type { CreateCategoryDto } from "../dto/create-category.dto"; @@ -16,6 +16,13 @@ export class CategoriesRepository { .orderBy(categories.name); } + async countAll() { + const result = await this.databaseService.db + .select({ count: sql`count(*)` }) + .from(categories); + return Number(result[0].count); + } + async findOne(id: string) { const result = await this.databaseService.db .select() diff --git a/backend/src/users/dto/update-user.dto.ts b/backend/src/users/dto/update-user.dto.ts index 7a4a0af..8b7eedf 100644 --- a/backend/src/users/dto/update-user.dto.ts +++ b/backend/src/users/dto/update-user.dto.ts @@ -5,4 +5,13 @@ export class UpdateUserDto { @IsString() @MaxLength(32) displayName?: string; + + @IsOptional() + @IsString() + @MaxLength(255) + bio?: string; + + @IsOptional() + @IsString() + avatarUrl?: string; }