koishi-plugin-luogu-saver-bot 0.1.0

package/lib/index.d.ts

@@ -0,0 +1,100 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "luogu-saver-bot";
+export interface Config {
+    endpoint?: string;
+    userAgent?: string;
+}
+export declare const Config: Schema<Config>;
+export type Article = {
+    id: string;
+    title: string;
+    content: string;
+    authorId: number;
+    category: number;
+    upvote?: number;
+    favorCount?: number;
+    solutionForPid?: string | null;
+    priority?: number;
+    deleted?: number;
+    tags: string[];
+    createdAt?: string;
+    updatedAt?: string;
+    deleteReason?: string;
+    contentHash?: string | null;
+    viewCount?: number;
+    renderedContent?: string | null;
+};
+export type StdResponse<A> = {
+    code: number;
+    message: string;
+    data: A;
+};
+export type ArticleHistory = {
+    id: number;
+    articleId: string;
+    version: number;
+    title: string;
+    content: string;
+    createdAt: string;
+};
+export type CountResponse = {
+    count: number;
+};
+export type TaskQuery = Record<string, any> | null;
+export type Task = {
+    id: string;
+    info?: string | null;
+    status: 0 | 1 | 2 | 3;
+    createdAt: string;
+    type: 'save' | 'ai_process';
+    target?: string | null;
+    payload: Record<string, any>;
+};
+export type TaskCreateBase = {
+    type: string;
+    payload: Record<string, any>;
+};
+export type TaskCreateSave = TaskCreateBase & {
+    type: 'save';
+    payload: {
+        target: string;
+        targetId: string;
+        metadata?: Record<string, any>;
+    };
+};
+export type TaskCreateAi = TaskCreateBase & {
+    type: 'ai_process';
+    payload: {
+        target: string;
+        metadata: Record<string, any>;
+    };
+};
+export type TaskCreateResponse = {
+    taskId: string;
+};
+declare class LuoguSaverClient {
+    private ctx;
+    endpoint: string;
+    userAgent: string;
+    constructor(ctx: Context, endpoint: string, userAgent: string);
+    private buildUrl;
+    private headers;
+    getArticle(id: string, extraHeaders?: Record<string, string>): Promise<Article>;
+    getRecent(opts?: {
+        count?: number;
+        updated_after?: string;
+        truncated_count?: number;
+    }, extraHeaders?: Record<string, string>): Promise<any>;
+    getCount(extraHeaders?: Record<string, string>): Promise<any>;
+    getRelevant(id: string, extraHeaders?: Record<string, string>): Promise<any>;
+    getHistory(id: string, extraHeaders?: Record<string, string>): Promise<any>;
+    createTask(body: TaskCreateSave | TaskCreateAi, extraHeaders?: Record<string, string>): Promise<string>;
+    getTask(id: string, extraHeaders?: Record<string, string>): Promise<Record<string, any>>;
+}
+declare module 'koishi' {
+    interface Context {
+        luogu_saver: LuoguSaverClient;
+    }
+}
+export declare function apply(ctx: Context, config?: Config): void;
+export {};

package/lib/index.js

@@ -0,0 +1,145 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+
+// src/task.ts
+var TaskStatus = /* @__PURE__ */ ((TaskStatus3) => {
+  TaskStatus3[TaskStatus3["PENDING"] = 0] = "PENDING";
+  TaskStatus3[TaskStatus3["PROCESSING"] = 1] = "PROCESSING";
+  TaskStatus3[TaskStatus3["COMPLETED"] = 2] = "COMPLETED";
+  TaskStatus3[TaskStatus3["FAILED"] = 3] = "FAILED";
+  return TaskStatus3;
+})(TaskStatus || {});
+function statusToString(status) {
+  if (status == null) return "UNKNOWN";
+  return TaskStatus[status] ?? "UNKNOWN";
+}
+__name(statusToString, "statusToString");
+
+// src/index.ts
+var name = "luogu-saver-bot";
+var Config = import_koishi.Schema.object({
+  endpoint: import_koishi.Schema.string().description("自定义 API endpoint，结尾无需斜杠").role("input").default(""),
+  userAgent: import_koishi.Schema.string().description("自定义 User-Agent").role("input").default("Uptime-Kuma")
+});
+var LuoguSaverClient = class {
+  constructor(ctx, endpoint, userAgent) {
+    this.ctx = ctx;
+    this.endpoint = endpoint;
+    this.userAgent = userAgent;
+    if (!this.endpoint) this.endpoint = "";
+  }
+  static {
+    __name(this, "LuoguSaverClient");
+  }
+  buildUrl(path) {
+    const base = this.endpoint.replace(/\/$/, "");
+    if (!base) return path;
+    console.log(`${base}${path}`);
+    if (path.startsWith("/")) return `${base}${path}`;
+    return `${base}/${path}`;
+  }
+  headers(extra) {
+    return Object.assign({ "User-Agent": this.userAgent }, extra || {});
+  }
+  async getArticle(id, extraHeaders) {
+    const url = this.buildUrl(`/article/query/${encodeURIComponent(id)}`);
+    const res = await this.ctx.http.get(url, { headers: this.headers(extraHeaders) });
+    if (res.code !== 200) return null;
+    return res.data;
+  }
+  async getRecent(opts, extraHeaders) {
+    const params = new URLSearchParams();
+    if (opts?.count != null) params.set("count", String(opts.count));
+    if (opts?.updated_after) params.set("updated_after", opts.updated_after);
+    if (opts?.truncated_count != null) params.set("truncated_count", String(opts.truncated_count));
+    const path = `/article/recent${params.toString() ? `?${params.toString()}` : ""}`;
+    const url = this.buildUrl(path);
+    const res = await this.ctx.http.get(url, { headers: this.headers(extraHeaders) });
+    return res?.data?.data ?? null;
+  }
+  async getCount(extraHeaders) {
+    const url = this.buildUrl("/article/count");
+    const res = await this.ctx.http.get(url, { headers: this.headers(extraHeaders) });
+    return res?.data?.data ?? null;
+  }
+  async getRelevant(id, extraHeaders) {
+    const url = this.buildUrl(`/article/relevant/${encodeURIComponent(id)}`);
+    const res = await this.ctx.http.get(url, { headers: this.headers(extraHeaders) });
+    return res?.data?.data ?? null;
+  }
+  async getHistory(id, extraHeaders) {
+    const url = this.buildUrl(`/article/history/${encodeURIComponent(id)}`);
+    const res = await this.ctx.http.get(url, { headers: this.headers(extraHeaders) });
+    return res?.data?.data ?? null;
+  }
+  async createTask(body, extraHeaders) {
+    const url = this.buildUrl("/task/create");
+    const res = await this.ctx.http.post(url, body, { headers: this.headers(extraHeaders) });
+    if (res.code !== 200) return null;
+    return res.data.taskId;
+  }
+  async getTask(id, extraHeaders) {
+    const url = this.buildUrl(`/task/query/${encodeURIComponent(id)}`);
+    const res = await this.ctx.http.get(url, { headers: this.headers(extraHeaders) });
+    if (res.code !== 200) return null;
+    return res.data;
+  }
+};
+function apply(ctx, config = {}) {
+  const endpoint = config.endpoint || "";
+  const userAgent = config.userAgent || "Uptime-Kuma";
+  ctx.luogu_saver = new LuoguSaverClient(ctx, endpoint, userAgent);
+  ctx.command("获取文章信息 <id>", "获取文章信息").action(async ({ options }, id) => {
+    if (!id) return "请提供文章 ID";
+    const art = await ctx.luogu_saver.getArticle(id);
+    console.log(art);
+    if (!art) return "未找到文章";
+    return `${art.title} by ${art.authorId}`;
+  });
+  ctx.command("创建保存任务 <target> <targetId>", "创建类型为 save 的任务").action(async (_, target, targetId) => {
+    const body = { type: "save", payload: { target, targetId } };
+    const id = await ctx.luogu_saver.createTask(body);
+    if (!id) return "创建失败";
+    return `保存任务已创建，ID: ${id}`;
+  });
+  ctx.command("查询任务状态 <id>", "查询任务状态").action(async ({ options }, id) => {
+    if (!id) return "请提供任务 ID";
+    const task = await ctx.luogu_saver.getTask(id);
+    if (task == null) return "任务不存在或返回为空";
+    if (typeof task === "object" && "status" in task) return `任务 ${id} 状态: ${statusToString(task.status)}`;
+    return JSON.stringify(task);
+  });
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  name
+});

package/lib/task.d.ts

@@ -0,0 +1,7 @@
+export declare enum TaskStatus {
+    PENDING = 0,
+    PROCESSING = 1,
+    COMPLETED = 2,
+    FAILED = 3
+}
+export declare function statusToString(status: number | null | undefined): string;

package/package.json

@@ -0,0 +1,22 @@
+{
+  "name": "koishi-plugin-luogu-saver-bot",
+  "description": "洛谷保存站机器人",
+  "version": "0.1.0",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "scripts": {},
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "devDependencies": {},
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-luogu-saver-bot
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-luogu-saver-bot?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-luogu-saver-bot)
+
+洛谷保存站机器人