@minato-bot/plugin-gugu 2.0.0

package/dist/index.mjs

@@ -0,0 +1,166 @@
+import { and, between, defineRelations, desc, eq } from "drizzle-orm";
+import { bigint, bigserial, pgTable, text, timestamp } from "drizzle-orm/pg-core";
+import path from "node:path";
+import process from "node:process";
+import { drizzle } from "drizzle-orm/bun-sql";
+import { migrate } from "drizzle-orm/bun-sql/migrator";
+import { randomInt } from "node:crypto";
+import { Plugin } from "@atri-bot/core";
+import dayjs from "dayjs";
+import { Structs } from "node-napcat-ts";
+import yargs from "yargs";
+//#region src/db.ts
+const timestamps = {
+	created_at: timestamp().notNull().defaultNow(),
+	updated_at: timestamp().notNull().defaultNow().$onUpdate(() => /* @__PURE__ */ new Date())
+};
+const Pigeons = pgTable("pigeons", {
+	user_id: bigserial("id", { mode: "number" }).primaryKey(),
+	pigeon_num: bigint("pigeon_num", { mode: "number" }).notNull(),
+	...timestamps
+});
+const PigeonHistories = pgTable("pigeon_histories", {
+	id: bigserial("id", { mode: "number" }).primaryKey(),
+	user_id: bigserial("user_id", { mode: "number" }).notNull(),
+	operation: bigint("operation", { mode: "number" }).notNull(),
+	prev_num: bigint("prev_num", { mode: "number" }).notNull(),
+	current_num: bigint("current_num", { mode: "number" }).notNull(),
+	reason: text(),
+	...timestamps
+});
+const Schema = {
+	Pigeons,
+	PigeonHistories
+};
+const Relations = defineRelations(Schema, (r) => ({ PigeonHistories: { Pigeon: r.one.Pigeons({
+	from: r.PigeonHistories.user_id,
+	to: r.Pigeons.user_id
+}) } }));
+//#endregion
+//#region src/dbUtils.ts
+if (!process.env.DATABASE_URL) throw new Error("DATABASE_URL 环境变量未设置, 请设置 DATABASE_URL 环境变量以连接数据库");
+const Drizzle = drizzle(process.env.DATABASE_URL, {
+	schema: Schema,
+	relations: Relations
+});
+async function migrateDrizzle() {
+	await migrate(Drizzle, {
+		migrationsFolder: path.join(import.meta.dir, "../drizzle"),
+		migrationsTable: "gugu_drizzle_migrations"
+	});
+}
+async function getUserPigeonInfo(user_id) {
+	const pigeonInfo = await Drizzle.query.Pigeons.findFirst({ where: { user_id } });
+	if (pigeonInfo) return pigeonInfo;
+	await Drizzle.insert(Schema.Pigeons).values({
+		user_id,
+		pigeon_num: 0
+	});
+	return await getUserPigeonInfo(user_id);
+}
+async function addUserPigeonNum(user_id, addNum, reason) {
+	const pigeonInfo = await getUserPigeonInfo(user_id);
+	if (addNum < 0) return false;
+	await Drizzle.update(Schema.Pigeons).set({ pigeon_num: pigeonInfo.pigeon_num + addNum }).where(eq(Schema.Pigeons.user_id, user_id));
+	await Drizzle.insert(Schema.PigeonHistories).values({
+		user_id,
+		operation: addNum,
+		prev_num: pigeonInfo.pigeon_num,
+		current_num: pigeonInfo.pigeon_num + addNum,
+		reason
+	});
+	return true;
+}
+async function reduceUserPigeonNum(user_id, reduceNum, reason) {
+	const pigeonInfo = await getUserPigeonInfo(user_id);
+	if (reduceNum <= 0 || pigeonInfo.pigeon_num - reduceNum < 0) return false;
+	await Drizzle.update(Schema.Pigeons).set({ pigeon_num: pigeonInfo.pigeon_num - reduceNum }).where(eq(Schema.Pigeons.user_id, user_id));
+	await Drizzle.insert(Schema.PigeonHistories).values({
+		user_id,
+		operation: reduceNum,
+		prev_num: pigeonInfo.pigeon_num + reduceNum,
+		current_num: pigeonInfo.pigeon_num,
+		reason
+	});
+	return true;
+}
+//#endregion
+//#region package.json
+var name = "@minato-bot/plugin-gugu";
+//#endregion
+//#region src/plugin.ts
+const guguPluginGuguRegexp = /咕咕/;
+async function handleGuguCommand(user_id, addRange) {
+	const today = dayjs();
+	if (await Drizzle.select().from(Schema.PigeonHistories).where(and(eq(Schema.PigeonHistories.user_id, user_id), eq(Schema.PigeonHistories.reason, "每日咕咕"), between(Schema.PigeonHistories.created_at, today.startOf("day").toDate(), today.endOf("day").toDate()))).then((res) => res.length > 0)) return [Structs.text(`今天已经咕咕过了! 明天再来吧!`)];
+	const addNum = randomInt(addRange[0], addRange[1]);
+	if (!await addUserPigeonNum(user_id, addNum, "每日咕咕")) return [Structs.text(`修改鸽子数失败!`)];
+	return [Structs.text(`咕咕成功! 获得 ${addNum} 只鸽子!`)];
+}
+const queryPigeonRegexp = /我的鸽子|查鸽子/;
+const queryPigeonCommander = yargs().option("user_id", {
+	type: "number",
+	description: "要查询的用户ID，默认为自己"
+});
+const pigeonRankRegexp = /鸽子排行/;
+const pigeonRankCommander = yargs().option("page", {
+	alias: "p",
+	type: "number",
+	description: "页码，默认为1",
+	default: 1
+}).option("size", {
+	alias: "s",
+	type: "number",
+	description: "每页数量，默认为10，最大20",
+	default: 10
+});
+const plugin = new Plugin(name).setDefaultConfig({
+	addRange: [1, 100],
+	firstAdd: 300
+}).onInstall(async ({ event, config, bot }) => {
+	await migrateDrizzle();
+	event.regCommandEvent({
+		trigger: guguPluginGuguRegexp,
+		callback: async ({ context }) => {
+			const user_id = context.user_id;
+			if (!await Drizzle.query.Pigeons.findFirst({ where: { user_id } })) {
+				await addUserPigeonNum(user_id, config.firstAdd, "初始赠送");
+				await bot.sendMsg(context, [Structs.text(`欢迎第一次咕咕! 作为初始奖励, 你获得了 ${config.firstAdd} 只鸽子!`)]);
+			}
+			const result = await handleGuguCommand(user_id, config.addRange);
+			await bot.sendMsg(context, result);
+		}
+	});
+	event.regCommandEvent({
+		trigger: queryPigeonRegexp,
+		commander: queryPigeonCommander,
+		callback: async ({ context, options }) => {
+			const result = await getUserPigeonInfo(options.user_id ?? context.user_id);
+			const username = await bot.getUsername({ user_id: result.user_id });
+			await bot.sendMsg(context, [Structs.text(`用户 ${username} 共有 ${result.pigeon_num} 只鸽子!`)]);
+		}
+	});
+	event.regCommandEvent({
+		trigger: pigeonRankRegexp,
+		commander: pigeonRankCommander,
+		callback: async ({ context, options }) => {
+			const page = Math.max(1, options.page);
+			const size = Math.max(1, Math.min(20, options.size));
+			const offset = (page - 1) * size;
+			const rows = await Drizzle.select().from(Schema.Pigeons).orderBy(desc(Schema.Pigeons.pigeon_num)).limit(size).offset(offset);
+			if (rows.length === 0) {
+				await bot.sendMsg(context, [Structs.text(`暂无鸽子数据！`)]);
+				return;
+			}
+			const total = await Drizzle.select().from(Schema.Pigeons);
+			const totalPages = Math.ceil(total.length / size);
+			const rankList = await Promise.all(rows.map(async (item, index) => {
+				const username = await bot.getUsername({ user_id: item.user_id });
+				return `${offset + index + 1}. 用户: ${username} 共有 ${item.pigeon_num} 只鸽子`;
+			}));
+			await bot.sendMsg(context, [Structs.text(`鸽子排行 (第 ${page} 页 / 共 ${totalPages} 页):\n${rankList.join("\n")}`)]);
+		}
+	});
+});
+//#endregion
+export { Drizzle, PigeonHistories, Pigeons, Relations, Schema, addUserPigeonNum, getUserPigeonInfo, guguPluginGuguRegexp, handleGuguCommand, migrateDrizzle, pigeonRankCommander, pigeonRankRegexp, plugin, queryPigeonCommander, queryPigeonRegexp, reduceUserPigeonNum, timestamps };

package/drizzle/20260326111244_lumpy_gravity/migration.sql

@@ -0,0 +1,17 @@
+CREATE TABLE "pigeon_histories" (
+	"id" bigserial PRIMARY KEY,
+	"user_id" bigserial,
+	"operation" bigint NOT NULL,
+	"prev_num" bigint NOT NULL,
+	"current_num" bigint NOT NULL,
+	"reason" text,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "pigeons" (
+	"id" bigserial PRIMARY KEY,
+	"pigeon_num" bigint NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL
+);

package/drizzle/20260326111244_lumpy_gravity/snapshot.json

@@ -0,0 +1,199 @@
+{
+  "version": "8",
+  "dialect": "postgres",
+  "id": "ec10cb7e-96b0-4cf3-8896-f89c6c9525b8",
+  "prevIds": [
+    "00000000-0000-0000-0000-000000000000"
+  ],
+  "ddl": [
+    {
+      "isRlsEnabled": false,
+      "name": "pigeon_histories",
+      "entityType": "tables",
+      "schema": "public"
+    },
+    {
+      "isRlsEnabled": false,
+      "name": "pigeons",
+      "entityType": "tables",
+      "schema": "public"
+    },
+    {
+      "type": "bigserial",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "id",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "bigserial",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "user_id",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "bigint",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "operation",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "bigint",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "prev_num",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "bigint",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "current_num",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "text",
+      "typeSchema": null,
+      "notNull": false,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "reason",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "timestamp",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": "now()",
+      "generated": null,
+      "identity": null,
+      "name": "created_at",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "timestamp",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": "now()",
+      "generated": null,
+      "identity": null,
+      "name": "updated_at",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeon_histories"
+    },
+    {
+      "type": "bigserial",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "id",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeons"
+    },
+    {
+      "type": "bigint",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": null,
+      "generated": null,
+      "identity": null,
+      "name": "pigeon_num",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeons"
+    },
+    {
+      "type": "timestamp",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": "now()",
+      "generated": null,
+      "identity": null,
+      "name": "created_at",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeons"
+    },
+    {
+      "type": "timestamp",
+      "typeSchema": null,
+      "notNull": true,
+      "dimensions": 0,
+      "default": "now()",
+      "generated": null,
+      "identity": null,
+      "name": "updated_at",
+      "entityType": "columns",
+      "schema": "public",
+      "table": "pigeons"
+    },
+    {
+      "columns": [
+        "id"
+      ],
+      "nameExplicit": false,
+      "name": "pigeon_histories_pkey",
+      "schema": "public",
+      "table": "pigeon_histories",
+      "entityType": "pks"
+    },
+    {
+      "columns": [
+        "id"
+      ],
+      "nameExplicit": false,
+      "name": "pigeons_pkey",
+      "schema": "public",
+      "table": "pigeons",
+      "entityType": "pks"
+    }
+  ],
+  "renames": []
+}

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "@minato-bot/plugin-gugu",
+  "type": "module",
+  "version": "2.0.0",
+  "description": "gugu for atri framework",
+  "author": "huan_kong",
+  "license": "MIT",
+  "main": "./dist/index.mjs",
+  "types": "./dist/index.d.mts",
+  "files": [
+    "./dist/**/*",
+    "./drizzle/**/*"
+  ],
+  "scripts": {
+    "build": "tsdown",
+    "lint": "eslint",
+    "typecheck": "tsc --noEmit -p tsconfig.json --composite false",
+    "drizzle": "drizzle-kit"
+  },
+  "dependencies": {
+    "@atri-bot/core": "^2.1.3",
+    "dayjs": "^1.11.20",
+    "drizzle-orm": "^1.0.0-beta.19",
+    "node-napcat-ts": "^0.4.23",
+    "yargs": "^18.0.0"
+  },
+  "devDependencies": {
+    "@types/yargs": "^17.0.35",
+    "drizzle-kit": "^1.0.0-beta.19"
+  }
+}