3 Commits

6 changed files with 143 additions and 2 deletions

11
.dockerignore Normal file
View File

@@ -0,0 +1,11 @@
node_modules
client/node_modules
client/dist
dist
.git
.env
logs
generated
redis-data
test
docs

View File

@@ -35,6 +35,14 @@ EMAIL_SECURE=true
EMAIL_USER=
EMAIL_PASS=
EMAIL_FROM=
MAIL_PROVIDER=
NOVAMAILIO_MAIL_API_BASE=
NOVAMAILIO_MAIL_CREDENTIAL=
NOVAMAILIO_MAIL_FROM_NAME=
NOVAMAILIO_MAIL_FINGERPRINT=
NOVAMAILIO_MAIL_ORIGIN=
NOVAMAILIO_MAIL_REFERER=
NOVAMAILIO_MAIL_USER_AGENT=
# Zulip
ZULIP_CONFIG_MODE=dynamic

47
Dockerfile Normal file
View File

@@ -0,0 +1,47 @@
FROM node:22-bookworm-slim AS build
RUN npm install --global pnpm@9.15.4
WORKDIR /app
COPY package.json pnpm-lock.yaml pnpm-workspace.yaml ./
COPY client/package.json ./client/package.json
RUN pnpm install --frozen-lockfile
COPY nest-cli.json tsconfig.json tsconfig.build.json ./
COPY src ./src
RUN pnpm run build
FROM node:22-bookworm-slim AS runtime
ENV NODE_ENV=production
ENV SKIN_GENERATION_PYTHON=/opt/skin-generation-venv/bin/python
ENV PIP_INDEX_URL=https://mirrors.cloud.tencent.com/pypi/simple
RUN sed -i 's|deb.debian.org|mirrors.cloud.tencent.com|g' /etc/apt/sources.list.d/debian.sources \
&& apt-get update \
&& apt-get install --yes --no-install-recommends ca-certificates python3 python3-venv \
&& rm -rf /var/lib/apt/lists/*
RUN npm install --global pnpm@9.15.4
WORKDIR /app
COPY requirements-skin-generation.txt ./
RUN python3 -m venv /opt/skin-generation-venv \
&& /opt/skin-generation-venv/bin/pip install --no-cache-dir --requirement requirements-skin-generation.txt
COPY package.json pnpm-lock.yaml pnpm-workspace.yaml ./
COPY client/package.json ./client/package.json
RUN pnpm install --prod --frozen-lockfile && pnpm store prune
COPY --from=build /app/dist ./dist
COPY scripts ./scripts
RUN mkdir -p config generated/account-assets generated/skins logs redis-data \
&& chown -R node:node /app
USER node
EXPOSE 3000 3001
HEALTHCHECK --interval=15s --timeout=5s --start-period=30s --retries=4 \
CMD node -e "fetch('http://127.0.0.1:3000/health').then(r=>{if(!r.ok)process.exit(1)}).catch(()=>process.exit(1))"
CMD ["node", "dist/main.js"]

View File

@@ -0,0 +1,3 @@
numpy==2.0.2
openai==2.14.0
Pillow==11.3.0

View File

@@ -1,5 +1,6 @@
import { Controller, Get, Query } from '@nestjs/common';
import { Controller, Get, Header, Param, Query, Res } from '@nestjs/common';
import { ApiOperation, ApiQuery, ApiTags } from '@nestjs/swagger';
import { Response } from 'express';
import { RankingsService } from './rankings.service';
import { RankingCategoryId } from './rankings.types';
@@ -8,6 +9,17 @@ import { RankingCategoryId } from './rankings.types';
export class RankingsController {
constructor(private readonly rankingsService: RankingsService) {}
@Get('datawhale-honor/avatar/:memberId')
@Header('Cache-Control', 'public, max-age=86400, stale-while-revalidate=604800')
@ApiOperation({ summary: '代理 Datawhale 荣誉榜成员头像' })
async getDatawhaleMemberAvatar(
@Param('memberId') memberId: string,
@Res() res: Response,
): Promise<void> {
const avatar = await this.rankingsService.getMemberAvatar(memberId);
res.type(avatar.contentType).send(avatar.body);
}
@Get('datawhale-honor')
@ApiOperation({
summary: '获取Datawhale荣誉榜',

View File

@@ -1,4 +1,4 @@
import { BadGatewayException, Injectable, Logger, OnModuleInit } from '@nestjs/common';
import { BadGatewayException, Injectable, Logger, NotFoundException, OnModuleInit } from '@nestjs/common';
import { Cron } from '@nestjs/schedule';
import axios from 'axios';
import {
@@ -17,6 +17,12 @@ const DATAWHALE_WEEKLY_COMMITS_URL = 'https://mv.datawhale.cc/data/commits_weekl
const DATAWHALE_ASSET_BASE_URL = 'https://mv.datawhale.cc/';
const DEFAULT_CATEGORY: RankingCategoryId = 'weekly_commits';
const DEFAULT_LIMIT = 10;
const MAX_AVATAR_BYTES = 2 * 1024 * 1024;
export interface RankingAvatarPayload {
body: Buffer;
contentType: string;
}
const CATEGORIES: RankingCategory[] = [
{
@@ -115,6 +121,40 @@ export class RankingsService implements OnModuleInit {
return this.getCachedPayload(DEFAULT_CATEGORY, DEFAULT_LIMIT);
}
async getMemberAvatar(memberId: string): Promise<RankingAvatarPayload> {
const normalizedId = this.cleanString(memberId);
const member = this.members.find(item => this.cleanString(item.id) === normalizedId);
if (!member) {
throw new NotFoundException('榜单成员不存在');
}
const avatarUrl = this.sourceAvatarUrl(member);
if (!avatarUrl) {
throw new NotFoundException('榜单成员没有头像');
}
try {
const response = await axios.get<ArrayBuffer>(avatarUrl, {
responseType: 'arraybuffer',
timeout: 10000,
maxContentLength: MAX_AVATAR_BYTES,
maxBodyLength: MAX_AVATAR_BYTES,
});
const body = Buffer.from(response.data);
if (body.length === 0 || body.length > MAX_AVATAR_BYTES) {
throw new BadGatewayException('头像源返回的文件大小异常');
}
const contentType = this.detectImageContentType(body);
if (!contentType) {
throw new BadGatewayException('头像源返回了不支持的文件类型');
}
return { body, contentType };
} catch (error) {
if (error instanceof BadGatewayException) {
throw error;
}
throw new BadGatewayException(`头像源请求失败:${this.errorMessage(error)}`);
}
}
private getCachedPayload(
category: RankingCategoryId,
limit: number,
@@ -366,6 +406,13 @@ export class RankingsService implements OnModuleInit {
}
private avatarUrl(member: DatawhaleMemberRow): string {
const id = this.cleanString(member.id);
return id && this.sourceAvatarUrl(member)
? `/api/rankings/datawhale-honor/avatar/${encodeURIComponent(id)}`
: '';
}
private sourceAvatarUrl(member: DatawhaleMemberRow): string {
const avatar = this.cleanString(member.avatar);
if (!avatar) {
return '';
@@ -423,6 +470,19 @@ export class RankingsService implements OnModuleInit {
return Number.isFinite(parsed) ? parsed : 0;
}
private detectImageContentType(body: Buffer): string {
if (body.length >= 8 && body.subarray(0, 8).equals(Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]))) {
return 'image/png';
}
if (body.length >= 3 && body[0] === 0xff && body[1] === 0xd8 && body[2] === 0xff) {
return 'image/jpeg';
}
if (body.length >= 12 && body.toString('ascii', 0, 4) === 'RIFF' && body.toString('ascii', 8, 12) === 'WEBP') {
return 'image/webp';
}
return '';
}
private cleanString(value: unknown): string {
return String(value ?? '').trim();
}