code-center/src/routes/n5-link/short-link.ts

import { desc, eq, count, or, like, and } from 'drizzle-orm';
import { app, db, schema } from '@/app.ts';
import { nanoid } from 'nanoid';
import z from 'zod';

// 列表
app.route({
  path: 'n5-short-link',
  key: 'list',
  middleware: ['auth'],
  description: '获取短链列表, 参数: page, pageSize, search, sort',
}).define(async (ctx) => {
  const tokenUser = ctx.state.tokenUser;
  const uid = tokenUser.id;
  const { page = 1, pageSize = 20, search, sort = 'DESC' } = ctx.query || {};

  const offset = (page - 1) * pageSize;
  const orderByField = sort === 'ASC' ? schema.shortLink.updatedAt : desc(schema.shortLink.updatedAt);

  let whereCondition: any = eq(schema.shortLink.userId, uid);
  if (search) {
    whereCondition = and(
      eq(schema.shortLink.userId, uid),
      or(
        like(schema.shortLink.title, `%${search}%`),
        like(schema.shortLink.slug, `%${search}%`),
      )
    );
  }

  const [list, totalCount] = await Promise.all([
    db.select().from(schema.shortLink).where(whereCondition).limit(pageSize).offset(offset).orderBy(orderByField),
    db.select({ count: count() }).from(schema.shortLink).where(whereCondition),
  ]);

  ctx.body = {
    list,
    pagination: { page, current: page, pageSize, total: totalCount[0]?.count || 0 },
  };
  return ctx;
}).addTo(app);

// 获取单个
app.route({
  path: 'n5-short-link',
  key: 'get',
  description: '获取单个短链, 参数: id 或 slug',
  middleware: ['auth'],
}).define(async (ctx) => {
  const { id, slug } = ctx.query || {};
  if (!id && !slug) {
    ctx.throw(400, 'id 或 slug 参数缺失');
  }
  const condition = id ? eq(schema.shortLink.id, id) : eq(schema.shortLink.slug, slug);
  const existing = await db.select().from(schema.shortLink).where(condition).limit(1);
  if (existing.length === 0) {
    ctx.throw(404, '短链不存在');
  }
  ctx.body = existing[0];
  return ctx;
}).addTo(app);

// 创建
app.route({
  path: 'n5-short-link',
  key: 'create',
  middleware: ['auth'],
  description: `创建短链`,
  metadata: {
    args: {
      data: z.object({
        slug: z.string().optional().describe('对外唯一业务ID'),
        title: z.string().optional().describe('标题'),
        description: z.string().optional().describe('描述'),
        tags: z.array(z.string()).optional().describe('标签数组'),
        data: z.record(z.string(), z.any()).optional().describe('附加数据对象'),
        type: z.enum(['link', 'agent']).optional().describe('类型'),
        version: z.string().optional().describe('版本号'),
      }).describe('创建短链参数对象'),
    }
  }
}).define(async (ctx) => {
  const tokenUser = ctx.state.tokenUser;
  const { userId, createdAt, updatedAt, id: _id, ...rest } = ctx.query.data || {};
  if (!rest.slug) {
    ctx.throw(400, 'slug 参数缺失');
  }
  const code = rest.code || nanoid(8);
  const result = await db.insert(schema.shortLink).values({
    ...rest,
    code,
    userId: tokenUser.id,
  }).returning();
  ctx.body = result[0];
  return ctx;
}).addTo(app);

// 更新
app.route({
  path: 'n5-short-link',
  key: 'update',
  middleware: ['auth'],
  description: `更新短链`,
  metadata: {
    args: {
      data: z.object({
        id: z.string().optional().describe('短链ID'),
        title: z.string().optional().describe('标题'),
        description: z.string().optional().describe('描述'),
        tags: z.array(z.string()).optional().describe('标签数组'),
        data: z.record(z.string(), z.any()).optional().describe('附加数据对象'),
        type: z.enum(['link', 'agent']).optional().describe('类型'),
        version: z.string().optional().describe('版本号'),
      }).describe('更新短链参数对象'),
    }
  }
}).define(async (ctx) => {
  const tokenUser = ctx.state.tokenUser;
  const { id, userId, createdAt, updatedAt, ...rest } = ctx.query.data || {};
  if (!id) {
    ctx.throw(400, 'id 参数缺失');
  }
  const existing = await db.select().from(schema.shortLink).where(eq(schema.shortLink.id, id)).limit(1);
  if (existing.length === 0) {
    ctx.throw(404, '短链不存在');
  }
  if (existing[0].userId !== tokenUser.id) {
    ctx.throw(403, '没有权限更新该短链');
  }
  const result = await db.update(schema.shortLink).set({ ...rest }).where(eq(schema.shortLink.id, id)).returning();
  ctx.body = result[0];
  return ctx;
}).addTo(app);

// 删除
app.route({
  path: 'n5-short-link',
  key: 'delete',
  middleware: ['auth'],
  description: '删除短链, 参数: id 短链ID',
}).define(async (ctx) => {
  const tokenUser = ctx.state.tokenUser;
  const { id } = ctx.query.data || {};
  if (!id) {
    ctx.throw(400, 'id 参数缺失');
  }
  const existing = await db.select().from(schema.shortLink).where(eq(schema.shortLink.id, id)).limit(1);
  if (existing.length === 0) {
    ctx.throw(404, '短链不存在');
  }
  if (existing[0].userId !== tokenUser.id) {
    ctx.throw(403, '没有权限删除该短链');
  }
  await db.delete(schema.shortLink).where(eq(schema.shortLink.id, id));
  ctx.body = { success: true };
  return ctx;
}).addTo(app);