@pynickle/koishi-plugin-chat-summarizer 1.0.0

package/lib/index.d.ts

@@ -0,0 +1,6 @@
+import { Context } from 'koishi';
+import { Config } from './types';
+import { name, inject, ConfigSchema } from './config';
+export { name, inject };
+export { ConfigSchema as Config };
+export declare function apply(ctx: Context, config: Config): void;

package/lib/md-to-image.d.ts

@@ -0,0 +1,36 @@
+import { Context } from 'koishi';
+export declare class MarkdownToImageService {
+    private ctx;
+    private logger;
+    private isRendering;
+    private renderQueue;
+    constructor(ctx: Context);
+    /**
+     * 检查是否正在渲染，避免并发渲染
+     */
+    private waitForRenderSlot;
+    /**
+     * 释放渲染槽位
+     */
+    private releaseRenderSlot;
+    /**
+     * 生成字体 CSS - 使用 Google Fonts CDN
+     */
+    private generateFontCSS;
+    /**
+     * 将文本中的 emoji 转换为图片标签
+     */
+    private convertEmojiToImages;
+    /**
+     * 获取 emoji 的 Unicode 码点
+     */
+    private getEmojiCodePoint;
+    /**
+     * 将 markdown 内容转换为图片
+     */
+    convertToImage(markdownContent: string): Promise<Buffer>;
+    /**
+     * 简单的 markdown 到 HTML 转换
+     */
+    private markdownToHtml;
+}

package/lib/message-processor.d.ts

@@ -0,0 +1,68 @@
+import { Element } from 'koishi';
+export interface ProcessedMessage {
+    content: string;
+    messageType: 'text' | 'image' | 'mixed' | 'other';
+    imageUrls: string[];
+    fileUrls: Array<{
+        url: string;
+        fileName: string;
+    }>;
+    videoUrls: Array<{
+        url: string;
+        fileName: string;
+    }>;
+}
+export declare class MessageProcessor {
+    private includeImages;
+    constructor(includeImages?: boolean);
+    /**
+     * 处理消息元素数组
+     */
+    processElements(elements: Element[]): ProcessedMessage;
+    /**
+     * 检查消息是否包含图片
+     */
+    hasImages(elements: Element[]): boolean;
+    /**
+     * 检查消息是否包含文件
+     */
+    hasFiles(elements: Element[]): boolean;
+    /**
+     * 检查消息是否包含视频
+     */
+    hasVideos(elements: Element[]): boolean;
+    /**
+     * 提取所有图片 URL
+     */
+    extractImageUrls(elements: Element[]): string[];
+    /**
+     * 提取所有文件 URL
+     */
+    extractFileUrls(elements: Element[]): Array<{
+        url: string;
+        fileName: string;
+    }>;
+    /**
+     * 提取所有视频 URL
+     */
+    extractVideoUrls(elements: Element[]): Array<{
+        url: string;
+        fileName: string;
+    }>;
+    /**
+     * 获取纯文本内容（去除所有非文本元素）
+     */
+    getPlainText(elements: Element[]): string;
+    /**
+     * 检查消息是否只包含文本
+     */
+    isTextOnly(elements: Element[]): boolean;
+    /**
+     * 检查消息是否为空
+     */
+    isEmpty(elements: Element[]): boolean;
+    /**
+     * 解析 JSON 消息内容，特别处理 QQ 小程序分享卡片
+     */
+    private parseJsonMessage;
+}

package/lib/s3-uploader.d.ts

@@ -0,0 +1,125 @@
+export interface S3Config {
+    region: string;
+    bucket: string;
+    accessKeyId: string;
+    secretAccessKey: string;
+    endpoint?: string;
+    pathPrefix: string;
+}
+export interface UploadResult {
+    success: boolean;
+    url?: string;
+    key?: string;
+    error?: string;
+    fileSize?: number;
+}
+export declare class S3Uploader {
+    private client;
+    private config;
+    constructor(config: S3Config);
+    /**
+     * 上传文件缓冲区到 S3
+     */
+    uploadBuffer(buffer: Buffer, key: string, contentType?: string): Promise<UploadResult>;
+    /**
+     * 上传本地文件到 S3
+     */
+    uploadFile(filePath: string, key: string, contentType?: string): Promise<UploadResult>;
+    /**
+     * 上传文本内容到 S3
+     */
+    uploadText(content: string, key: string, contentType?: string): Promise<UploadResult>;
+    /**
+     * 从 URL 下载图片并上传到 S3（使用 axios 确保兼容性）
+     */
+    uploadImageFromUrl(imageUrl: string, key: string, httpService?: any, maxSize?: number): Promise<UploadResult>;
+    /**
+     * 从 URL 下载文件并上传到 S3
+     */
+    uploadFileFromUrl(fileUrl: string, key: string, fileName?: string, maxSize?: number): Promise<UploadResult>;
+    /**
+     * 从 URL 下载视频并上传到 S3
+     */
+    uploadVideoFromUrl(videoUrl: string, key: string, fileName?: string, maxSize?: number): Promise<UploadResult>;
+    /**
+     * 批量上传聊天记录文件
+     */
+    uploadChatLogFiles(files: Array<{
+        filePath: string;
+        key: string;
+    }>): Promise<UploadResult[]>;
+    /**
+     * 生成公共 URL
+     */
+    private generatePublicUrl;
+    /**
+     * 根据文件扩展名获取内容类型
+     */
+    private getContentTypeFromExtension;
+    /**
+     * 获取图片内容类型
+     */
+    private getImageContentType;
+    /**
+     * 获取文件内容类型
+     */
+    private getFileContentType;
+    /**
+     * 获取视频内容类型
+     */
+    private getVideoContentType;
+    /**
+     * 生成用于存储的 S3 键名
+     */
+    static generateImageKey(messageId: string, originalUrl: string, guildId?: string, index?: number): string;
+    /**
+     * 生成用于文件存储的 S3 键名
+     */
+    static generateFileKey(messageId: string, originalUrl: string, fileName?: string, guildId?: string, index?: number): string;
+    /**
+     * 生成用于视频存储的 S3 键名
+     */
+    static generateVideoKey(messageId: string, originalUrl: string, fileName?: string, guildId?: string, index?: number): string;
+    /**
+     * 生成聊天记录文件的 S3 键名（JSON 格式）
+     */
+    static generateChatLogKey(date: Date, guildId?: string): string;
+    /**
+     * 提取图片文件扩展名
+     */
+    private static getImageExtension;
+    /**
+     * 提取文件扩展名
+     */
+    private static getFileExtension;
+    /**
+     * 提取视频文件扩展名
+     */
+    private static getVideoExtension;
+    /**
+     * 检查是否支持的图片格式
+     */
+    static isSupportedImageFormat(url: string, allowedTypes: string[]): boolean;
+    /**
+     * 测试 S3 连接
+     */
+    testConnection(): Promise<{
+        success: boolean;
+        error?: string;
+    }>;
+    /**
+     * 获取 S3 存储桶中的文件列表
+     */
+    listFiles(prefix?: string): Promise<{
+        success: boolean;
+        files?: string[];
+        error?: string;
+    }>;
+    /**
+     * 下载 S3 文件到本地
+     */
+    downloadFile(s3Key: string, localPath: string): Promise<{
+        success: boolean;
+        error?: string;
+    }>;
+}

package/lib/services.d.ts

@@ -0,0 +1,26 @@
+import { Context } from 'koishi';
+import { Config } from './types';
+import { S3Uploader } from './s3-uploader';
+export declare class LoggerService {
+    private config;
+    private logger;
+    constructor(ctx: Context, config: Config);
+    debug(message: string, data?: any): void;
+    info(message: string): void;
+    warn(message: string): void;
+    error(message: string, error?: any): void;
+}
+export declare class S3Service {
+    private config;
+    private logger;
+    private s3Uploader;
+    constructor(config: Config, logger: LoggerService);
+    init(): void;
+    getUploader(): S3Uploader | null;
+}
+export declare class MessageProcessorService {
+    private messageProcessor;
+    constructor(includeImages: boolean);
+    processElements(elements: any[]): any;
+    normalizeUserId(userId: string): string;
+}

package/lib/statistics.d.ts

@@ -0,0 +1,74 @@
+import { Logger } from 'koishi';
+import { InteractionStatistics, ParsedMessage } from './types';
+export declare class StatisticsService {
+    private logger;
+    constructor(logger: Logger);
+    /**
+     * 解析 JSONL 内容为消息数组
+     */
+    parseMessages(jsonlContent: string): ParsedMessage[];
+    /**
+     * 生成完整的统计数据
+     */
+    generateStatistics(messages: ParsedMessage[], topN?: number): InteractionStatistics;
+    /**
+     * 统计每个用户发言数量，排序取 TOP N
+     */
+    calculateActivityRanking(messages: ParsedMessage[], topN?: number): Array<{
+        username: string;
+        messageCount: number;
+        rank: number;
+    }>;
+    /**
+     * 按小时统计消息数量分布
+     */
+    calculateHourlyDistribution(messages: ParsedMessage[]): Array<{
+        hour: number;
+        count: number;
+        percentage: number;
+    }>;
+    /**
+     * 解析 @/回复 关系，统计互动频次
+     */
+    calculateInteractions(messages: ParsedMessage[]): {
+        mentions: Array<{
+            from: string;
+            to: string;
+            count: number;
+        }>;
+        replies: Array<{
+            from: string;
+            to: string;
+            count: number;
+        }>;
+    };
+    /**
+     * 计算基础统计指标
+     */
+    calculateBasicStats(messages: ParsedMessage[], hourlyDistribution: Array<{
+        hour: number;
+        count: number;
+        percentage: number;
+    }>): {
+        totalMessages: number;
+        uniqueUsers: number;
+        avgMessagesPerUser: number;
+        peakHour: number;
+    };
+    /**
+     * 格式化小时为显示字符串
+     */
+    formatHour(hour: number): string;
+    /**
+     * 获取时段描述
+     */
+    getPeriodDescription(hour: number): string;
+    /**
+     * 生成活跃时段摘要
+     */
+    generatePeakHourSummary(hourlyDistribution: Array<{
+        hour: number;
+        count: number;
+        percentage: number;
+    }>): string;
+}

package/lib/types.d.ts

@@ -0,0 +1,216 @@
+export interface ForwardTarget {
+    groupId: string;
+    name?: string;
+}
+export interface GroupConfig {
+    groupId: string;
+    name?: string;
+    monitorEnabled?: boolean;
+    summaryEnabled?: boolean;
+    summaryTime?: string;
+    pushEnabled?: boolean;
+    pushTime?: string;
+    pushToSelf?: boolean;
+    forwardGroups?: ForwardTarget[];
+    systemPrompt?: string;
+    userPromptTemplate?: string;
+}
+export interface PushGroupConfig {
+    groupId: string;
+    channelId?: string;
+    platform?: string;
+}
+export interface Config {
+    s3: {
+        enabled: boolean;
+        bucket: string;
+        accessKeyId: string;
+        secretAccessKey: string;
+        endpoint?: string;
+        pathPrefix: string;
+    };
+    chatLog: {
+        enabled: boolean;
+        includeImages: boolean;
+        maxFileSize: number;
+        autoUploadTime: string;
+        retentionDays: number;
+        dbRetentionHours: number;
+    };
+    monitor: {
+        groups: GroupConfig[];
+        excludedUsers: string[];
+        excludeBots: boolean;
+    };
+    admin: {
+        adminIds: string[];
+    };
+    ai: {
+        enabled: boolean;
+        apiUrl: string;
+        apiKey: string;
+        apiMode?: 'chat.completions' | 'responses';
+        model?: string;
+        maxTokens?: number;
+        timeout?: number;
+        systemPrompt?: string;
+        userPromptTemplate?: string;
+        useFileMode?: boolean;
+        fileName?: string;
+        defaultSummaryTime?: string;
+        defaultPushTime?: string;
+    };
+    debug: boolean;
+}
+export interface ChatRecord {
+    id?: number;
+    messageId: string;
+    guildId?: string;
+    channelId: string;
+    userId: string;
+    username: string;
+    content: string;
+    originalElements: string;
+    timestamp: number;
+    messageType: 'text' | 'image' | 'mixed' | 'other';
+    imageUrls?: string;
+    fileUrls?: string;
+    videoUrls?: string;
+    uploadedAt?: number;
+    isUploaded: boolean;
+}
+export interface ImageRecord {
+    id?: number;
+    originalUrl: string;
+    s3Url: string;
+    s3Key: string;
+    fileSize: number;
+    uploadedAt: number;
+    messageId: string;
+}
+export interface FileRecord {
+    id?: number;
+    originalUrl: string;
+    s3Url: string;
+    s3Key: string;
+    fileName: string;
+    fileSize: number;
+    uploadedAt: number;
+    messageId: string;
+}
+export interface VideoRecord {
+    id?: number;
+    originalUrl: string;
+    s3Url: string;
+    s3Key: string;
+    fileName: string;
+    fileSize: number;
+    uploadedAt: number;
+    messageId: string;
+}
+export interface ChatLogFileRecord {
+    id?: number;
+    guildId?: string;
+    date: string;
+    filePath: string;
+    s3Key: string;
+    s3Url?: string;
+    fileSize: number;
+    recordCount: number;
+    uploadedAt: number;
+    status: 'pending' | 'uploading' | 'uploaded' | 'failed';
+    error?: string;
+    summaryImageUrl?: string;
+    summaryGeneratedAt?: number;
+}
+export interface PluginStats {
+    totalMessages: number;
+    todayMessages: number;
+    imageRecords: number;
+    uploadedMessages: number;
+}
+export interface AISummaryOutput {
+    summary: {
+        overview: string;
+        highlights: string[];
+        atmosphere: string;
+    };
+    hotTopics: Array<{
+        topic: string;
+        description: string;
+        participants: string[];
+        heatLevel: 'high' | 'medium' | 'low';
+    }>;
+    importantInfo: Array<{
+        type: 'announcement' | 'link' | 'resource' | 'decision' | 'other';
+        content: string;
+        source?: string;
+    }>;
+    quotes: Array<{
+        content: string;
+        author: string;
+    }>;
+}
+export interface InteractionStatistics {
+    activityRanking: Array<{
+        username: string;
+        messageCount: number;
+        rank: number;
+    }>;
+    hourlyDistribution: Array<{
+        hour: number;
+        count: number;
+        percentage: number;
+    }>;
+    interactions: {
+        mentions: Array<{
+            from: string;
+            to: string;
+            count: number;
+        }>;
+        replies: Array<{
+            from: string;
+            to: string;
+            count: number;
+        }>;
+    };
+    basicStats: {
+        totalMessages: number;
+        uniqueUsers: number;
+        avgMessagesPerUser: number;
+        peakHour: number;
+    };
+}
+export interface DailyReport {
+    date: string;
+    guildId: string;
+    aiContent: AISummaryOutput;
+    statistics: InteractionStatistics;
+    metadata: {
+        generatedAt: number;
+        aiModel: string;
+    };
+}
+export interface ParsedMessage {
+    timestamp: number;
+    time: string;
+    messageId: string;
+    guildId?: string;
+    channelId: string;
+    userId: string;
+    username: string;
+    content: string;
+    messageType: string;
+    imageUrls: string[];
+    fileUrls: string[];
+    videoUrls: string[];
+}
+declare module 'koishi' {
+    interface Tables {
+        chat_records: ChatRecord;
+        image_records: ImageRecord;
+        file_records: FileRecord;
+        video_records: VideoRecord;
+        chat_log_files: ChatLogFileRecord;
+    }
+}

package/lib/utils.d.ts

@@ -0,0 +1,47 @@
+/**
+ * 公共工具函数模块
+ */
+/**
+ * 将时间戳格式化为 UTC+8 可读时间
+ */
+export declare const formatDateInUTC8: (timestamp: number) => string;
+/**
+ * 将时间戳格式化为简化格式（不包含毫秒）
+ */
+export declare const formatDateSimple: (timestamp: number) => string;
+/**
+ * 获取 UTC+8 时区的日期字符串 (YYYY-MM-DD)
+ */
+export declare const getDateStringInUTC8: (timestamp: number) => string;
+/**
+ * 获取当前 UTC+8 时间的 Date 对象
+ */
+export declare const getCurrentTimeInUTC8: () => Date;
+/**
+ * 统一的错误处理器
+ */
+export declare const handleError: (error: any, defaultMessage?: string) => string;
+/**
+ * 安全的 JSON 解析
+ */
+export declare const safeJsonParse: <T>(jsonString: string | null | undefined, defaultValue: T) => T;
+/**
+ * 安全的 JSON 序列化
+ */
+export declare const safeJsonStringify: (obj: any) => string;
+/**
+ * 延迟执行函数
+ */
+export declare const delay: (ms: number) => Promise<void>;
+/**
+ * 文件大小格式化
+ */
+export declare const formatFileSize: (bytes: number) => string;
+/**
+ * 批量处理函数，支持并发限制
+ */
+export declare const processBatch: <T, R>(items: T[], processor: (item: T) => Promise<R>, concurrency?: number) => Promise<R[]>;
+/**
+ * 替换URL中的域名
+ */
+export declare const replaceImageUrl: (originalUrl: string) => string;

package/package.json

@@ -0,0 +1,60 @@
+{
+  "name": "@pynickle/koishi-plugin-chat-summarizer",
+  "description": "Koishi 聊天记录收集和上传插件",
+  "version": "1.0.0",
+  "main": "lib/index.cjs",
+  "typings": "lib/index.d.ts",
+  "type": "module",
+  "files": [
+    "lib"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "koishi",
+    "plugin",
+    "chat"
+  ],
+  "scripts": {
+    "lint": "npx oxlint ./src",
+    "lint:fix": "npx oxlint ./src --fix",
+    "fmt": "npx oxfmt ./src",
+    "build": "node esbuild.config.js && tsc --emitDeclarationOnly",
+    "semantic-release": "semantic-release"
+  },
+  "dependencies": {
+    "@ai-sdk/openai": "^3.0.41",
+    "@aws-sdk/client-s3": "^3.1011.0",
+    "@aws-sdk/lib-storage": "^3.1011.0",
+    "ai": "^6.0.116",
+    "axios": "^1.13.6",
+    "github-markdown-css": "^5.9.0"
+  },
+  "devDependencies": {
+    "@semantic-release/changelog": "^6.0.3",
+    "@semantic-release/git": "^10.0.1",
+    "@semantic-release/npm": "^13.1.5",
+    "@types/node": "^24.12.0",
+    "conventional-changelog-conventionalcommits": "^9.3.0",
+    "esbuild": "^0.27.4",
+    "oxfmt": "^0.41.0",
+    "oxlint": "^1.56.0",
+    "rimraf": "^6.1.3",
+    "semantic-release": "^25.0.3",
+    "typescript": "^5.9.3"
+  },
+  "peerDependencies": {
+    "koishi": "^4.18.0"
+  },
+  "koishi": {
+    "description": {
+      "zh": "Koishi 聊天记录收集和上传插件"
+    },
+    "service": {
+      "required": [
+        "database",
+        "http",
+        "puppeteer"
+      ]
+    }
+  }
+}