Compare commits
3 Commits
fix/pr-1-p
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| 2a3125075f | |||
| f37136d8c3 | |||
| f98bf3c493 |
11
.dockerignore
Normal file
11
.dockerignore
Normal file
@@ -0,0 +1,11 @@
|
||||
node_modules
|
||||
client/node_modules
|
||||
client/dist
|
||||
dist
|
||||
.git
|
||||
.env
|
||||
logs
|
||||
generated
|
||||
redis-data
|
||||
test
|
||||
docs
|
||||
@@ -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
47
Dockerfile
Normal 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"]
|
||||
3
requirements-skin-generation.txt
Normal file
3
requirements-skin-generation.txt
Normal file
@@ -0,0 +1,3 @@
|
||||
numpy==2.0.2
|
||||
openai==2.14.0
|
||||
Pillow==11.3.0
|
||||
@@ -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荣誉榜',
|
||||
|
||||
@@ -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();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user