koishi-plugin-nmc-radar 1.2.0 → 1.3.0

package/lib/index.d.ts

@@ -1,7 +1,14 @@
-import type { Context } from 'koishi';
-import { Schema } from 'koishi';
+import type { Awaitable, Context } from 'koishi';
+import { h, Schema } from 'koishi';
 export declare const name = "nmc-radar";
+export declare const inject: {
+    optional: string[];
+};
+type Resolver = (urls: string[], id: string) => Awaitable<h>;
+declare const resolvers: Record<string, Resolver>;
 export interface Config {
+    defaultResolver: keyof typeof resolvers;
 }
 export declare const Config: Schema<Config>;
-export declare function apply(ctx: Context): void;
+export declare function apply(ctx: Context, config: Config): void;
+export {};

package/lib/index.js

@@ -1,6 +1,8 @@
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
@@ -15,6 +17,14 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
@@ -22,9 +32,12 @@ var src_exports = {};
 __export(src_exports, {
   Config: () => Config,
   apply: () => apply,
+  inject: () => inject,
   name: () => name
 });
 module.exports = __toCommonJS(src_exports);
+var import_promises = require("node:fs/promises");
+var import_node_path = __toESM(require("node:path"));
 var import_jsdom = require("jsdom");
 var import_koishi = require("koishi");
 
@@ -308,21 +321,52 @@ var radars_default = {
 
 // src/index.ts
 var name = "nmc-radar";
-var Config = import_koishi.Schema.object({});
-function apply(ctx) {
-  ctx.command("radar <name:string>", "查看雷达图").alias("雷达").action(async ({ session }, name2) => {
+var inject = {
+  optional: ["ffmpeg"]
+};
+var resolvers = {
+  img: /* @__PURE__ */ __name((urls) => import_koishi.h.img(urls[0]), "img"),
+  imgs: /* @__PURE__ */ __name((urls) => (0, import_koishi.h)(import_koishi.h.Fragment, ...urls.map((url) => import_koishi.h.img(url))), "imgs"),
+  url: /* @__PURE__ */ __name((urls) => import_koishi.h.text(urls[0]), "url"),
+  urls: /* @__PURE__ */ __name((urls) => import_koishi.h.text(urls.join("\n")), "urls")
+};
+var Config = import_koishi.Schema.object({
+  defaultResolver: import_koishi.Schema.union(Object.keys(resolvers)).default("img").description("默认输出类型。")
+});
+function apply(ctx, config) {
+  const command = ctx.command("radar <name:string>", "查看雷达图", { checkUnknown: true }).alias("雷达").option("type", "--type <type:string> 输出类型", { type: Object.keys(resolvers) }).option("type", "--img 输出单张图片", { value: "img" }).option("type", "--imgs 输出多张图片", { value: "imgs" }).option("type", "--url 输出 URL", { value: "url" }).option("type", "--urls 输出 URL 列表", { value: "urls" }).action(async ({ session, options }, name2) => {
+    if (!(name2 in radars_default))
+      return void session?.send("雷达站不存在，可使用 radar.list 查看所有雷达站。");
     const url = radars_default[name2];
-    if (!url)
-      return void session?.send("该雷达图不存在");
-    const dom = new import_jsdom.JSDOM(await ctx.http.get(url));
-    const image = dom.window.document.querySelector("div[data-img]");
-    return (0, import_koishi.h)("img", { src: image?.dataset.img || "" });
+    const { window: { document } } = new import_jsdom.JSDOM(await ctx.http.get(url));
+    const nodes = document.querySelectorAll("div[data-img]");
+    const urls = Array.from(nodes).map((node) => node.dataset.img || "");
+    const resolver = resolvers[options?.type || config.defaultResolver];
+    return await resolver(urls, import_koishi.Random.id());
+  });
+  ctx.inject(["ffmpeg"], async (ctx2) => {
+    command.option("type", "--gif 输出 GIF 动画", { value: "gif" });
+    resolvers.gif = async (urls, id) => {
+      const baseDir = import_node_path.default.join(ctx2.baseDir, "temp", name, id);
+      await (0, import_promises.mkdir)(baseDir, { recursive: true });
+      await Promise.all(urls.map(async (url, i) => {
+        const response = await ctx2.http.get(url, { responseType: "stream" });
+        const filename = `${String(i + 1).padStart(3, "0")}.png`;
+        const filePath = import_node_path.default.join(baseDir, filename);
+        await (0, import_promises.writeFile)(filePath, response);
+      }));
+      const outputPath = import_node_path.default.join(baseDir, "..", `${id}.gif`);
+      await ctx2.ffmpeg.builder().input(import_node_path.default.join(baseDir, "%03d.png")).run("file", outputPath);
+      return import_koishi.h.img(`file:///${outputPath}`);
+    };
   });
+  command.subcommand(".list", "查看所有雷达站").action(() => Object.keys(radars_default).join(" "));
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Config,
   apply,
+  inject,
   name
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-nmc-radar",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "气象台雷达图插件",
   "typings": "lib/index.d.ts",
   "license": "MIT",
@@ -22,20 +22,16 @@
     "koishi": "4.18.11"
   },
   "dependencies": {
-    "@zorner/gifencoder": "^1.0.1",
-    "jsdom": "^29.1.1",
-    "probe-image-size": "^7.3.0"
+    "jsdom": "^29.1.1"
   },
   "devDependencies": {
-    "@koishijs/canvas": "^0.2.0",
-    "@types/gifencoder": "^2.0.3",
     "@types/jsdom": "^28.0.3",
-    "@types/probe-image-size": "^7"
+    "koishi-plugin-ffmpeg": "^1.1.0"
   },
   "koishi": {
     "service": {
       "optional": [
-        "canvas"
+        "ffmpeg"
       ]
     }
   }