feat: SVG Badge API 구현

Jihyun3478 · Jihyun3478 · commit ab21edcddda5 · 2025-12-12T13:27:12.000+09:00
feat: SVG Badge API 구현

- Controller, Service, Repository 구현
diff --git a/src/controllers/svg.controller.ts b/src/controllers/svg.controller.ts
@@ -0,0 +1,33 @@
+import { NextFunction, Request, RequestHandler, Response } from "express";
+import logger from '@/configs/logger.config'
+import { GetSvgBadgeParams, GetSvgBadgeQuery } from "@/types";
+import { SvgService } from '@/services/svg.service';
+
+export class SvgController {
+    constructor(private svgService: SvgService) {}
+
+    getSvgBadge: RequestHandler<GetSvgBadgeParams, any, any, GetSvgBadgeQuery> = async (
+        req: Request<GetSvgBadgeParams, object, object, GetSvgBadgeQuery>,
+        res: Response,
+        next: NextFunction,
+    ) => {
+        try {
+            const { username } = req.params;
+            const { type = 'default', assets = 'views,likes,posts', withrank = 'false'} = req.query;
+
+            const svgString = await this.svgService.generateBadgeSvg(
+                username,
+                type,
+                assets,
+                withrank === 'true',
+            );
+
+            res.setHeader('Content-Type', 'image/svg+xml');
+            res.setHeader('Cache-Control', 'public, max-age=1800');
+            res.send(svgString);
+        } catch (error) {
+            logger.error('SVG Badge 생성 실패: ', error);
+            next(error);
+        }
+    }
+}
diff --git a/src/repositories/svg.repository.ts b/src/repositories/svg.repository.ts
@@ -0,0 +1,100 @@
+import { Pool } from 'pg';
+import logger from '@/configs/logger.config';
+import { DBError, NotFoundError } from '@/exception';
+import { getCurrentKSTDateString, getKSTDateStringWithOffset } from '@/utils/date.util';
+
+export class SvgRepository {
+    constructor(private pool: Pool) {}
+
+    async getUserBadgeData(username: string, withRank: boolean, dateRange: number = 30) {
+        try {
+            const pastDateKST = getKSTDateStringWithOffset(-dateRange * 24 * 60);
+            const cteQuery = this.buildBadgeCteQuery(dateRange, pastDateKST);
+            
+            const userStatsQuery = `
+                ${cteQuery}
+                SELECT
+                    u.username,
+                    COALESCE(SUM(ts.today_view), 0) AS total_views,
+                    COALESCE(SUM(ts.today_like), 0) AS total_likes,
+                    COUNT(DISTINCT CASE WHEN p.is_active = true THEN p.id END) AS total_posts,
+                    SUM(COALESCE(ts.today_view, 0) - COALESCE(ss.start_view, 0)) AS view_diff,
+                    SUM(COALESCE(ts.today_like, 0) - COALESCE(ss.start_like, 0)) AS like_diff,
+                    COUNT(DISTINCT CASE WHEN p.released_at >= '${pastDateKST}' AND p.is_active = true THEN p.id END) AS post_diff
+                FROM users_user u
+                LEFT JOIN posts_post p ON p.user_id = u.id
+                LEFT JOIN today_stats ts ON ts.post_id = p.id
+                LEFT JOIN start_stats ss ON ss.post_id = p.id
+                WHERE u.username = $1
+                GROUP BY u.username
+            `;
+            const userStatsResult = await this.pool.query(userStatsQuery, [username, dateRange]);
+
+            if (userStatsResult.rows.length === 0) {
+                throw new NotFoundError(`사용자를 찾을 수 없습니다: ${username}`);
+            }
+
+            const recentPostsQuery = `
+                ${cteQuery}
+                SELECT
+                    p.title,
+                    p.released_at,
+                    COALESCE(ts.today_view, 0) AS today_view,
+                    COALESCE(ts.today_like, 0) AS today_like,
+                    COALESCE(ts.today_view, 0) - COALESCE(ss.start_view, 0) AS view_diff
+                FROM posts_post p
+                JOIN users_user u ON u.id = p.user_id
+                LEFT JOIN today_stats ts ON ts.post_id = p.id
+                LEFT JOIN start_stats ss ON ss.post_id = p.id
+                WHERE u.username = $1
+                    AND p.is_active = true
+                ORDER BY p.released_at DESC
+                LIMIT 3
+            `;
+            const recentPostsResult = await this.pool.query(recentPostsQuery, [username]);
+
+            return {
+                ...userStatsResult.rows[0],
+                recent_posts: recentPostsResult.rows,
+            };
+        } catch (error) {
+            if (error instanceof NotFoundError) {
+                throw error;
+            }
+
+            logger.error('SvgRepository getUserBadgeData error: ', error);
+            throw new DBError('배지 데이터 조회 중 문제가 발생했습니다.');
+        }
+    }
+
+    private buildBadgeCteQuery(dateRange: number, pastDateKST?: string) {
+        const nowDateKST = 
+            new Date().getUTCHours() === 15
+            ? getKSTDateStringWithOffset(-24 * 60)
+            : getCurrentKSTDateString();
+
+        if (!pastDateKST) {
+            pastDateKST = getKSTDateStringWithOffset(-dateRange * 24 * 60);
+        }
+
+        return `
+            WITH
+            today_stats AS (
+                SELECT DISTINCT ON (post_id)
+                    post_id,
+                    daily_view_count AS today_view,
+                    daily_like_count AS today_like
+                FROM posts_postdailystatistics
+                WHERE date = '${nowDateKST}'
+            ),
+            start_stats AS (
+                SELECT DISTINCT ON (post_id)
+                    post_id,
+                    daily_view_count AS start_view,
+                    daily_like_count AS start_like
+                FROM posts_postdailystatistics
+                WHERE date = '${pastDateKST}'
+            )
+        `;
+    }
+}
diff --git a/src/routes/index.ts b/src/routes/index.ts
@@ -4,6 +4,7 @@ import PostRouter from './post.router';
 import NotiRouter from './noti.router';
 import LeaderboardRouter from './leaderboard.router';
 import TotalStatsRouter from './totalStats.router';
+import SvgRouter from './svg.router';
 import WebhookRouter from './webhook.router';
 
 const router: Router = express.Router();
@@ -17,6 +18,7 @@ router.use('/', PostRouter);
 router.use('/', NotiRouter);
 router.use('/', LeaderboardRouter);
 router.use('/', TotalStatsRouter);
+router.use('/', SvgRouter);
 router.use('/', WebhookRouter);
 
 export default router;
diff --git a/src/routes/svg.router.ts b/src/routes/svg.router.ts
@@ -0,0 +1,68 @@
+import pool from '@/configs/db.config';
+import express, { Router } from 'express';
+import { SvgRepository } from '@/repositories/svg.repository';
+import { SvgService } from '@/services/svg.service';
+import { SvgController } from '@/controllers/svg.controller';
+import { validateRequestDto } from '@/middlewares/validation.middleware';
+import { GetSvgBadgeQueryDto } from '@/types';
+
+const router: Router = express.Router();
+
+const svgRepository = new SvgRepository(pool);
+const svgService = new SvgService(svgRepository);
+const svgController = new SvgController(svgService);
+
+/**
+ * @swagger
+ * /api/{username}/badge:
+ *   get:
+ *     summary: 사용자 배지 SVG 조회
+ *     tags:
+ *       - SVG
+ *     security: []
+ *     parameters:
+ *       - in: path
+ *         name: username
+ *         required: true
+ *         schema:
+ *           type: string
+ *         description: 조회할 사용자명
+ *         example: six-standard
+ *       - in: query
+ *         name: type
+ *         schema:
+ *           type: string
+ *           enum: [default, simple]
+ *           default: default
+ *       - in: query
+ *         name: assets
+ *         schema:
+ *           type: string
+ *           default: views,likes,posts
+ *           example: views,likes,posts
+ *       - in: query
+ *         name: withrank
+ *         schema:
+ *           type: string
+ *           enum: [true, false]
+ *           default: false
+ *     responses:
+ *       '200':
+ *         description: SVG 배지 생성 성공
+ *         content:
+ *           image/svg+xml:
+ *             schema:
+ *               type: string
+ *               example: <svg>...</svg>
+ *       '404':
+ *         description: 사용자를 찾을 수 없음
+ *       '500':
+ *         description: 서버 오류 / 데이터 베이스 조회 오류
+ */
+router.get(
+  '/:username/badge',
+  validateRequestDto(GetSvgBadgeQueryDto, 'query'),
+  svgController.getSvgBadge as any,
+);
+
+export default router;
diff --git a/src/services/svg.service.ts b/src/services/svg.service.ts
@@ -0,0 +1,58 @@
+import logger from '@/configs/logger.config';
+import { SvgBadgeType } from '@/types';
+import { SvgRepository } from '@/repositories/svg.repository';
+
+export class SvgService {
+    constructor(private svgRepo: SvgRepository) {}
+
+    async generateBadgeSvg(
+        username: string,
+        type: SvgBadgeType,
+        assets: string,
+        withRank: boolean,
+    ): Promise<string> {
+        try {
+            const data = await this.svgRepo.getUserBadgeData(username, withRank);
+
+            if (type === 'simple') {
+                return this.generateSimpleSvg(data, assets);
+            } else {
+                return this.generateDefaultSvg(data, assets, withRank);
+            }
+        } catch (error) {
+            logger.error('SvgService generateBadgeSvg error: ', error);
+            throw error;
+        }
+    }
+
+    private generateSimpleSvg(data: any, assets: string): string {
+        return `<svg width="400" height="120" xmlns="http://www.w3.org/2000/svg">
+            <rect width="400" height="120" fill="#1E1E1E"/>
+            <text x="20" y="40" fill="white" font-size="20">${data.username}</text>
+            <text x="20" y="70" fill="white">Views: ${data.total_views}</text>
+            <text x="20" y="100" fill="white">Likes: ${data.total_likes}</text>
+        </svg>`;
+    }
+
+    private generateDefaultSvg(data: any, assets: string, withRank: boolean): string {
+        return `<svg width="600" height="300" xmlns="http://www.w3.org/2000/svg">
+            <rect width="600" height="300" fill="#1E1E1E"/>
+            <text x="20" y="30" fill="white" font-size="18">${data.username}</text>
+            <text x="20" y="60" fill="white">Views: ${data.total_views}</text>
+            <text x="20" y="90" fill="white">Recent Posts: ${data.recent_posts?.length || 0}</text>
+            ${withRank ? `<text x="20" y="120" fill="white">Rank: #${data.view_rank || 'N/A'}</text>` : ''}
+        </svg>`;
+    }
+
+    private formatNumber(num: number): string {
+        if (num >= 1000000) {
+            return (num / 1000000).toFixed(1) + 'm';
+        }
+
+        if (num >= 1000) {
+            return (num / 1000).toFixed(1) + 'k';
+        }
+
+        return num.toString();
+    }
+}
diff --git a/src/types/dto/requests/svgRequest.type.ts b/src/types/dto/requests/svgRequest.type.ts
@@ -0,0 +1,33 @@
+import { IsEnum, IsOptional, IsString } from 'class-validator';
+
+export type SvgBadgeType = 'default' | 'simple';
+
+export interface GetSvgBadgeParams {
+    username: string;
+}
+
+export interface GetSvgBadgeQuery {
+    type?: SvgBadgeType;
+    assets?: string;
+    withrank?: string;
+}
+
+export class GetSvgBadgeQueryDto {
+    @IsOptional()
+    @IsEnum(['default', 'simple'])
+    type?: SvgBadgeType;
+
+    @IsOptional()
+    @IsString()
+    assets?: string;
+
+    @IsOptional()
+    @IsEnum(['true', 'false'])
+    withrank?: string;
+
+    constructor(type?: SvgBadgeType, assets?: string, withrank?: string) {
+        this.type = type || 'default';
+        this.assets = assets || 'views,likes,posts'
+        this.withrank = withrank || 'false';
+    }
+}
diff --git a/src/types/index.ts b/src/types/index.ts
@@ -38,6 +38,10 @@ export type { TotalStatsItem } from '@/types/dto/responses/totalStatsResponse.ty
 export { GetTotalStatsQueryDto } from '@/types/dto/requests/getTotalStatsQuery.type';
 export { TotalStatsResponseDto } from '@/types/dto/responses/totalStatsResponse.type';
 
+// SVG 관련
+export type { GetSvgBadgeParams, GetSvgBadgeQuery, SvgBadgeType } from '@/types/dto/requests/svgRequest.type';
+export { GetSvgBadgeQueryDto } from '@/types/dto/requests/svgRequest.type';
+
 // Sentry 관련
 export type { SentryIssueStatus } from '@/types/models/Sentry.type';
 export type { SentryProject, SentryIssue, SentryWebhookData } from '@/types/models/Sentry.type';
