mioku-plugin-deerpipe 1.0.0

package/image.ts

@@ -0,0 +1,347 @@
+import * as fs from "fs/promises";
+import * as path from "path";
+import { fileURLToPath } from "url";
+import type { ScreenshotService } from "mioku";
+import { escapeHtml, getAvatarUrl, isNightMode } from "./utils";
+
+interface CalendarOptions {
+  year: number;
+  month: number;
+  todayDay: number;
+  records: Map<number, number>;
+  name: string;
+  userId: number;
+}
+
+interface RankRow {
+  rank: number;
+  name: string;
+  userId: number;
+  count: number;
+}
+
+interface DeerTheme {
+  pageBg: string;
+  textColor: string;
+  titleAccent: string;
+  countColor: string;
+  emptyColor: string;
+  dayHalo: string;
+  rankCountColor: string;
+  avatarBorder: string;
+}
+
+const DAY_THEME: DeerTheme = {
+  pageBg: "#ffffff",
+  textColor: "#000000",
+  titleAccent: "#d50000",
+  countColor: "#d50000",
+  emptyColor: "#888888",
+  dayHalo: "#ffffff",
+  rankCountColor: "#d50000",
+  avatarBorder: "#cccccc",
+};
+
+const NIGHT_THEME: DeerTheme = {
+  pageBg: "#1d2030",
+  textColor: "#f1ecdb",
+  titleAccent: "#ff6b6b",
+  countColor: "#ff8a65",
+  emptyColor: "#9a9a9a",
+  dayHalo: "#2a2f44",
+  rankCountColor: "#ff8a65",
+  avatarBorder: "#5a5f73",
+};
+
+function getTheme(): DeerTheme {
+  return isNightMode() ? NIGHT_THEME : DAY_THEME;
+}
+
+interface AssetUris {
+  defaultAvatar: string;
+  check: string;
+  deerpipe: string;
+}
+
+let assetsCache: AssetUris | null = null;
+
+async function loadAssets(): Promise<AssetUris> {
+  if (assetsCache) return assetsCache;
+  const here = path.dirname(fileURLToPath(import.meta.url));
+  const assetsDir = path.join(here, "assets");
+  const [avatar, check, deerpipe] = await Promise.all([
+    fs.readFile(path.join(assetsDir, "akkarin@80x80.png")),
+    fs.readFile(path.join(assetsDir, "check@96x100.png")),
+    fs.readFile(path.join(assetsDir, "deerpipe@100x82.png")),
+  ]);
+  assetsCache = {
+    defaultAvatar: `data:image/png;base64,${avatar.toString("base64")}`,
+    check: `data:image/png;base64,${check.toString("base64")}`,
+    deerpipe: `data:image/png;base64,${deerpipe.toString("base64")}`,
+  };
+  return assetsCache;
+}
+
+// Mon–Sun weeks, matching Python's calendar.monthcalendar(year, month)
+function buildMonthGrid(year: number, month: number): number[][] {
+  const firstDow = new Date(year, month - 1, 1).getDay(); // 0=Sun
+  const daysInMonth = new Date(year, month, 0).getDate();
+  // Convert to Monday-first index: Mon=0..Sun=6
+  const leadingEmpty = (firstDow + 6) % 7;
+  const weeks: number[][] = [];
+  let cur: number[] = new Array(leadingEmpty).fill(0);
+  for (let d = 1; d <= daysInMonth; d++) {
+    cur.push(d);
+    if (cur.length === 7) {
+      weeks.push(cur);
+      cur = [];
+    }
+  }
+  if (cur.length > 0) {
+    while (cur.length < 7) cur.push(0);
+    weeks.push(cur);
+  }
+  return weeks;
+}
+
+export async function generateCalendarImage(
+  screenshotService: ScreenshotService,
+  options: CalendarOptions,
+): Promise<string> {
+  const { year, month, records, name, userId } = options;
+  const assets = await loadAssets();
+  // 日历始终使用白天主题（按用户要求 🦌 / 🦌历 / 补🦌 不适配夜间模式）
+  const theme = DAY_THEME;
+  const weeks = buildMonthGrid(year, month);
+
+  const cellSize = 100;
+  const headerHeight = 100;
+  const imgW = 700;
+  const imgH = headerHeight + cellSize * weeks.length;
+
+  const cellsHtml = weeks
+    .map((week, weekIdx) =>
+      week
+        .map((day, dayIdx) => {
+          if (day === 0) return "";
+          const x = dayIdx * cellSize;
+          const y = headerHeight + weekIdx * cellSize;
+          const count = records.get(day) ?? 0;
+          const checked = count > 0;
+          const countText =
+            count > 1 ? (count > 999 ? "x999+" : `x${count}`) : "";
+          return `
+            <img class="stamp" src="${assets.deerpipe}" style="left:${x}px;top:${y}px" />
+            <div class="day-num" style="left:${x + 8}px;top:${y + 50}px">${day}</div>
+            ${
+              checked
+                ? `<img class="check" src="${assets.check}" style="left:${x}px;top:${y}px" />`
+                : ""
+            }
+            ${
+              countText
+                ? `<div class="count" style="left:${x + cellSize - 5}px;top:${y + cellSize - 25}px">${countText}</div>`
+                : ""
+            }
+          `;
+        })
+        .join(""),
+    )
+    .join("");
+
+  const html = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="utf-8" />
+    <style>
+      ${BASE_CSS}
+      .canvas {
+        position: relative;
+        width: ${imgW}px;
+        height: ${imgH}px;
+        background: ${theme.pageBg};
+        color: ${theme.textColor};
+        overflow: hidden;
+      }
+      .avatar {
+        position: absolute;
+        left: 10px;
+        top: 10px;
+        width: 80px;
+        height: 80px;
+        object-fit: cover;
+        border-radius: 50%;
+        border: 2px solid ${theme.avatarBorder};
+        box-sizing: border-box;
+      }
+      .title {
+        position: absolute;
+        left: 100px;
+        top: 10px;
+        font-size: 25px;
+        line-height: 30px;
+        color: ${theme.textColor};
+      }
+      .subtitle {
+        position: absolute;
+        left: 100px;
+        top: 45px;
+        font-size: 25px;
+        line-height: 30px;
+        color: ${theme.textColor};
+      }
+      .stamp { position: absolute; width: 100px; height: 82px; }
+      .check { position: absolute; width: 96px; height: 100px; }
+      .day-num {
+        position: absolute;
+        font-size: 36px;
+        line-height: 36px;
+        font-weight: 700;
+        color: ${theme.textColor};
+        text-shadow:
+          1px 0 0 ${theme.dayHalo},
+          -1px 0 0 ${theme.dayHalo},
+          0 1px 0 ${theme.dayHalo},
+          0 -1px 0 ${theme.dayHalo},
+          1px 1px 0 ${theme.dayHalo},
+          -1px -1px 0 ${theme.dayHalo},
+          1px -1px 0 ${theme.dayHalo},
+          -1px 1px 0 ${theme.dayHalo};
+      }
+      .count {
+        position: absolute;
+        transform: translateX(-100%);
+        font-size: 20px;
+        line-height: 20px;
+        color: ${theme.countColor};
+        font-weight: 700;
+        -webkit-text-stroke: 1px ${theme.countColor};
+      }
+    </style>
+  </head>
+  <body>
+    <div class="canvas">
+      <img class="avatar" src="${escapeHtml(getAvatarUrl(userId))}" onerror="this.onerror=null;this.src='${assets.defaultAvatar}'" />
+      <div class="title">${year}-${String(month).padStart(2, "0")} 🦌签到日历</div>
+      <div class="subtitle">@${escapeHtml(name)}</div>
+      ${cellsHtml}
+    </div>
+  </body></html>`;
+
+  return screenshotService.screenshot(html, {
+    width: imgW,
+    height: imgH,
+    fullPage: false,
+    type: "png",
+  });
+}
+
+export async function generateRankImage(
+  screenshotService: ScreenshotService,
+  rows: RankRow[],
+  _year: number,
+  _month: number,
+): Promise<string> {
+  const assets = await loadAssets();
+  const theme = getTheme();
+  const imgW = 400;
+  const headerHeight = 100;
+  const rowHeight = 100;
+  // Original: IMG_H = (len(rank) + 1) * 100 — empty rank still gets header + 1 row.
+  const visibleRows = rows.length || 1;
+  const imgH = headerHeight + rowHeight * visibleRows;
+
+  const rowsHtml = rows
+    .map((row, idx) => {
+      const y = headerHeight + idx * rowHeight;
+      return `
+        <img class="rank-avatar" src="${escapeHtml(getAvatarUrl(row.userId))}" onerror="this.onerror=null;this.src='${assets.defaultAvatar}'" style="top:${y + 10}px" />
+        <div class="rank-name" style="top:${y + 10}px">@${escapeHtml(row.name)}</div>
+        <div class="rank-count" style="top:${y + 50}px">x${row.count}</div>
+      `;
+    })
+    .join("");
+
+  const emptyHtml =
+    rows.length === 0
+      ? `<div class="rank-empty" style="top:${headerHeight + 30}px">本月还没人🦌过呢</div>`
+      : "";
+
+  const html = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="utf-8" />
+    <style>
+      ${BASE_CSS}
+      .canvas {
+        position: relative;
+        width: ${imgW}px;
+        height: ${imgH}px;
+        background: ${theme.pageBg};
+        color: ${theme.textColor};
+        overflow: hidden;
+      }
+      .rank-title {
+        position: absolute;
+        left: 0;
+        top: 25px;
+        width: 100%;
+        text-align: center;
+        font-size: 50px;
+        line-height: 50px;
+        color: ${theme.titleAccent};
+        font-weight: 700;
+        -webkit-text-stroke: 1px ${theme.titleAccent};
+      }
+      .rank-avatar {
+        position: absolute;
+        left: 10px;
+        width: 80px;
+        height: 80px;
+        object-fit: cover;
+        border-radius: 50%;
+        border: 2px solid ${theme.avatarBorder};
+        box-sizing: border-box;
+      }
+      .rank-name {
+        position: absolute;
+        left: 100px;
+        font-size: 25px;
+        line-height: 25px;
+        color: ${theme.textColor};
+      }
+      .rank-count {
+        position: absolute;
+        left: 100px;
+        font-size: 25px;
+        line-height: 25px;
+        color: ${theme.rankCountColor};
+      }
+      .rank-empty {
+        position: absolute;
+        left: 0;
+        width: 100%;
+        text-align: center;
+        font-size: 22px;
+        color: ${theme.emptyColor};
+      }
+    </style>
+  </head>
+  <body>
+    <div class="canvas">
+      <div class="rank-title">本月Top5🦌榜</div>
+      ${rowsHtml}
+      ${emptyHtml}
+    </div>
+  </body></html>`;
+
+  return screenshotService.screenshot(html, {
+    width: imgW,
+    height: imgH,
+    fullPage: false,
+    type: "png",
+  });
+}
+
+const BASE_CSS = `
+  * { box-sizing: border-box; margin: 0; padding: 0; }
+  body {
+    font-family: "MiSans", "PingFang SC", "Microsoft YaHei",
+      "Noto Sans CJK SC", "Hiragino Sans GB", "Apple Color Emoji",
+      "Noto Color Emoji", sans-serif;
+  }
+`;

package/index.ts

@@ -0,0 +1,86 @@
+import type { ScreenshotService } from "mioku";
+import { definePlugin, type MiokiContext } from "mioki";
+import { initDeerDatabase, type DeerDatabase } from "./db";
+import { handleDeerCommand, parseDeerCommand } from "./commands";
+
+const deerpipePlugin = definePlugin({
+  name: "deerpipe",
+  version: "1.0.0",
+  description: "🦌管签到插件，支持自🦌、帮🦌、补🦌、🦌历、🦌榜",
+
+  async setup(ctx: MiokiContext) {
+    ctx.logger.info("deerpipe 插件正在初始化...");
+
+    const screenshotService = ctx.services?.screenshot as
+      | ScreenshotService
+      | undefined;
+
+    if (!screenshotService) {
+      ctx.logger.warn("screenshot 服务未加载，deerpipe 插件无法生成图片");
+      return () => {
+        ctx.logger.info("deerpipe 插件已卸载");
+      };
+    }
+
+    let db: DeerDatabase;
+    try {
+      db = await initDeerDatabase();
+    } catch (error) {
+      ctx.logger.error(`deerpipe 数据库初始化失败: ${error}`);
+      return () => {
+        ctx.logger.info("deerpipe 插件已卸载");
+      };
+    }
+
+    // 启动时清掉非本月的旧数据，与原版 cleanup 行为一致
+    try {
+      const now = new Date();
+      await db.cleanupOtherMonths(now.getFullYear(), now.getMonth() + 1);
+    } catch (error) {
+      ctx.logger.warn(`deerpipe 启动期数据清理失败: ${error}`);
+    }
+
+    // 每周一 4:00 清理跨月数据
+    ctx.cron("0 4 * * 1", async () => {
+      try {
+        const now = new Date();
+        await db.cleanupOtherMonths(now.getFullYear(), now.getMonth() + 1);
+        ctx.logger.info("deerpipe 跨月数据已清理");
+      } catch (error) {
+        ctx.logger.error(`deerpipe 定时清理失败: ${error}`);
+      }
+    });
+
+    ctx.handle("message", async (event: any) => {
+      const text = ctx.text(event);
+      if (!text) return;
+
+      const cmd = parseDeerCommand(text, event.message ?? []);
+      if (cmd.type === "none") return;
+
+      try {
+        await handleDeerCommand(cmd, {
+          ctx,
+          db,
+          screenshot: screenshotService,
+          event,
+        });
+      } catch (error) {
+        ctx.logger.error(`deerpipe 命令执行失败: ${error}`);
+        try {
+          await event.reply(`🦌管插件出错了: ${error}`, true);
+        } catch {
+          // 忽略二次失败
+        }
+      }
+    });
+
+    ctx.logger.info("deerpipe 插件初始化完成");
+
+    return () => {
+      ctx.logger.info("deerpipe 插件已卸载");
+    };
+  },
+});
+
+export default deerpipePlugin;

package/package.json

@@ -0,0 +1,70 @@
+{
+  "name": "mioku-plugin-deerpipe",
+  "version": "1.0.0",
+  "description": "🦌管签到插件，支持自🦌、帮🦌、补🦌、🦌历、🦌榜",
+  "main": "index.ts",
+  "type": "module",
+  "keywords": [
+    "mioku"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/mioku-lab/mioku.git"
+  },
+  "mioku": {
+    "services": [
+      "screenshot",
+      "help"
+    ],
+    "help": {
+      "title": "🦌管签到",
+      "description": "每月🦌管签到，含日历与排行榜",
+      "commands": [
+        {
+          "cmd": "🦌",
+          "desc": "签到一次（可加 @某人 帮其签到）",
+          "usage": "🦌 / 🦌 @张三",
+          "role": "member"
+        },
+        {
+          "cmd": "补🦌 <日>",
+          "desc": "补签本月之前未签到的某天",
+          "usage": "补🦌 5",
+          "role": "member"
+        },
+        {
+          "cmd": "🦌历",
+          "desc": "查看本月🦌签到日历（可加 @某人）",
+          "usage": "🦌历 / 🦌历 @张三",
+          "role": "member"
+        },
+        {
+          "cmd": "🦌榜",
+          "desc": "查看本月本群🦌签到排行榜（仅群组）",
+          "usage": "🦌榜",
+          "role": "member"
+        },
+        {
+          "cmd": "帮🦌 <on|off> [@某人]",
+          "desc": "允许/禁止被别人帮🦌；@某人需管理员",
+          "usage": "帮🦌 off",
+          "role": "member"
+        },
+        {
+          "cmd": "禁🦌 @某人 [时长]",
+          "desc": "禁止某人在一段时间内🦌，省略时长视为解禁（仅管理员）",
+          "usage": "禁🦌 @张三 1d",
+          "role": "admin"
+        }
+      ]
+    }
+  },
+  "peerDependencies": {
+    "mioku": "^0.8.0",
+    "mioki": "^0.16.0"
+  },
+  "devDependencies": {
+    "mioku": "workspace:*",
+    "mioki": "^0.16.0"
+  }
+}

package/types.ts

@@ -0,0 +1,23 @@
+export interface DeerUser {
+  scene: string;
+  userId: number;
+  canBeHelped: boolean;
+  noDeerUntil: number | null;
+}
+
+export interface DeerCheckInResult {
+  ok: boolean;
+  records: Map<number, number>;
+}
+
+export interface DeerRankEntry {
+  userId: number;
+  count: number;
+}
+
+export interface DeerScene {
+  key: string;
+  isGroup: boolean;
+  groupId?: number;
+  privateUserId?: number;
+}

package/utils.ts

@@ -0,0 +1,144 @@
+import type { MiokiContext } from "mioki";
+import * as fs from "fs/promises";
+import type { DeerScene } from "./types";
+
+export function getAvatarUrl(userId: number): string {
+  return `https://q1.qlogo.cn/g?b=qq&nk=${userId}&s=640`;
+}
+
+export function getAtUserId(message: any[]): number | undefined {
+  if (!Array.isArray(message)) return undefined;
+  for (const seg of message) {
+    if (seg?.type !== "at") continue;
+    const raw = seg?.qq ?? seg?.data?.qq;
+    if (raw === "all" || raw == null) continue;
+    const qq = Number(raw);
+    if (Number.isFinite(qq)) return qq;
+  }
+  return undefined;
+}
+
+export function resolveScene(event: any): DeerScene {
+  if (event?.message_type === "group" && event?.group_id != null) {
+    return {
+      key: `g:${event.group_id}`,
+      isGroup: true,
+      groupId: Number(event.group_id),
+    };
+  }
+  return {
+    key: `p:${event.user_id}`,
+    isGroup: false,
+    privateUserId: Number(event.user_id),
+  };
+}
+
+export async function resolveUserName(
+  ctx: MiokiContext,
+  event: any,
+  userId: number,
+): Promise<string> {
+  const selfId = Number(event?.self_id);
+  if (event?.message_type === "group" && event?.group_id != null) {
+    try {
+      const member = await ctx
+        .pickBot(selfId)
+        .getGroupMemberInfo(Number(event.group_id), userId);
+      const name =
+        String(member?.card || "").trim() ||
+        String(member?.nickname || "").trim();
+      if (name) return name;
+    } catch {
+      // fall through
+    }
+  }
+  if (Number(event?.user_id) === userId && event?.sender) {
+    const name =
+      String(event.sender?.card || "").trim() ||
+      String(event.sender?.nickname || "").trim();
+    if (name) return name;
+  }
+  try {
+    const stranger = (await ctx
+      .pickBot(selfId)
+      .api("get_stranger_info", { user_id: userId })) as
+      | { nickname?: string }
+      | undefined;
+    const name = String(stranger?.nickname || "").trim();
+    if (name) return name;
+  } catch {
+    // ignore
+  }
+  return String(userId);
+}
+
+const DURATION_RE = /(\d+)\s*([smhdwy秒分时天周年]|min|hour|day|week|year)/gi;
+
+export function parseDuration(text: string): number | null {
+  const t = String(text || "").trim();
+  if (!t) return null;
+  let total = 0;
+  let matched = false;
+  for (const m of t.matchAll(DURATION_RE)) {
+    matched = true;
+    const value = parseInt(m[1], 10);
+    const unit = m[2].toLowerCase();
+    if (unit === "s" || unit === "秒") total += value;
+    else if (unit === "m" || unit === "min" || unit === "分") total += value * 60;
+    else if (unit === "h" || unit === "hour" || unit === "时") total += value * 3600;
+    else if (unit === "d" || unit === "day" || unit === "天") total += value * 86400;
+    else if (unit === "w" || unit === "week" || unit === "周") total += value * 604800;
+    else if (unit === "y" || unit === "year" || unit === "年") total += value * 31536000;
+  }
+  if (!matched) {
+    if (/^\d+$/.test(t)) return parseInt(t, 10);
+    return null;
+  }
+  return total;
+}
+
+export function formatDateTime(timestamp: number): string {
+  const d = new Date(timestamp);
+  const pad = (n: number) => String(n).padStart(2, "0");
+  return `${d.getFullYear()}-${pad(d.getMonth() + 1)}-${pad(d.getDate())} ${pad(
+    d.getHours(),
+  )}:${pad(d.getMinutes())}`;
+}
+
+export function escapeHtml(value: string): string {
+  return String(value)
+    .replace(/&/g, "&amp;")
+    .replace(/</g, "&lt;")
+    .replace(/>/g, "&gt;")
+    .replace(/"/g, "&quot;")
+    .replace(/'/g, "&#39;");
+}
+
+// 19:00–07:00 视为夜间模式，与 help 插件保持一致
+export function isNightMode(): boolean {
+  const hour = new Date().getHours();
+  return hour >= 19 || hour < 7;
+}
+
+export async function replyImage(
+  event: any,
+  segment: { image: (file: string) => any } | undefined,
+  imagePath: string,
+  text?: any[],
+): Promise<void> {
+  const imageSeg = segment?.image
+    ? segment.image(imagePath)
+    : { type: "image", file: imagePath };
+  const payload = text ? [...text, imageSeg] : [imageSeg];
+  try {
+    await event.reply(payload, true);
+  } catch {
+    const buf = await fs.readFile(imagePath);
+    const base64 = `base64://${buf.toString("base64")}`;
+    const fallbackSeg = segment?.image
+      ? segment.image(base64)
+      : { type: "image", file: base64 };
+    const fallback = text ? [...text, fallbackSeg] : [fallbackSeg];
+    await event.reply(fallback, true);
+  }
+}