koishi-plugin-wordpress-notifier 1.0.0 → 1.1.0

package/lib/index.d.ts

@@ -1,5 +1,11 @@
 import { Context, Schema } from 'koishi';
+export declare const inject: string[];
 export declare const name = "wordpress-notifier";
+declare module 'koishi' {
+    interface Tables {
+        wordpress_posts: WordPressPostRecord;
+    }
+}
 export interface Config {
     wordpressUrl: string;
     interval: number;
@@ -22,5 +28,10 @@ export interface WordPressPost {
     categories: number[];
     tags: number[];
 }
+export interface WordPressPostRecord {
+    id: number;
+    postId: number;
+    pushedAt: Date;
+}
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Config = exports.name = void 0;
+exports.Config = exports.name = exports.inject = void 0;
 exports.apply = apply;
 const koishi_1 = require("koishi");
+exports.inject = ['database', 'http'];
 exports.name = 'wordpress-notifier';
 exports.Config = koishi_1.Schema.object({
     wordpressUrl: koishi_1.Schema.string().description('WordPress 网站地址（例如：https://example.com）'),
@@ -14,7 +15,11 @@ exports.Config = koishi_1.Schema.object({
 });
 function apply(ctx, config) {
     ctx.logger.info('WordPress 推送插件已加载');
-    const pushedPosts = new Set();
+    ctx.model.extend('wordpress_posts', {
+        id: 'unsigned',
+        postId: 'integer',
+        pushedAt: 'timestamp'
+    });
     async function fetchLatestPosts() {
         try {
             const url = `${config.wordpressUrl}/wp-json/wp/v2/posts?per_page=${config.maxArticles}&orderby=date&order=desc`;
@@ -28,11 +33,15 @@ function apply(ctx, config) {
             return [];
         }
     }
-    function isPostPushed(postId) {
-        return pushedPosts.has(postId);
+    async function isPostPushed(postId) {
+        const record = await ctx.database.get('wordpress_posts', { postId });
+        return record.length > 0;
     }
-    function markPostAsPushed(postId) {
-        pushedPosts.add(postId);
+    async function markPostAsPushed(postId) {
+        await ctx.database.create('wordpress_posts', {
+            postId,
+            pushedAt: new Date()
+        });
     }
     function formatPostMessage(post, mention = false) {
         const title = post.title.rendered.replace(/<[^>]*>/g, '');
@@ -55,18 +64,18 @@ function apply(ctx, config) {
         if (posts.length === 0)
             return;
         for (const post of posts) {
-            if (!isPostPushed(post.id)) {
+            if (!(await isPostPushed(post.id))) {
                 const message = formatPostMessage(post, true);
                 for (const target of config.targets) {
                     try {
-                        await ctx.broadcast([`onebot:${target}`], message);
+                        await ctx.broadcast([target], message);
                         ctx.logger.info(`已推送文章到 ${target}: ${post.title.rendered}`);
                     }
                     catch (error) {
                         ctx.logger.error(`推送文章到 ${target} 失败: ${error}`);
                     }
                 }
-                markPostAsPushed(post.id);
+                await markPostAsPushed(post.id);
             }
         }
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-wordpress-notifier",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "WordPress 文章自动推送到 QQ",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
@@ -34,7 +34,7 @@
   ],
   "repository": {
     "type": "git",
-    "url": "https://github.com/your-username/koishi-plugin-wordpress-notifier.git"
+    "url": "https://github.com/Lexo0522/koishi-plugin-wordpress-notifier.git"
   },
   "homepage": "https://github.com/your-username/koishi-plugin-wordpress-notifier#readme",
   "bugs": {