feat(shorturl).

Author: memset0

Diff for: package.json

@@ -100,6 +100,7 @@
     "pinia": "^2.0.22",
     "qface": "^1.2.0",
     "qrcode": "^1.5.1",
+    "randomstring": "^1.2.2",
     "rd": "^2.0.1",
     "rss-feed-emitter": "^3.2.3",
     "sharp": "^0.30.7",

Diff for: src/commands/shorturl.ts

@@ -0,0 +1,94 @@
+import { Context, Time, Session, segment } from 'koishi'
+import randomstring from 'randomstring'
+import Koa from 'koa'
+
+declare module 'koishi' {
+	interface Tables {
+		shorturl: Shorturl
+	}
+}
+
+export interface Shorturl {
+	id: number
+	key: string
+	url: string
+	count: number
+}
+
+export interface Config {
+	root: string
+	authority: number
+}
+
+export const KEY_LENGTH = 6
+export const PRE_LENGTH_CHECK = true
+
+
+export const name = 'shorturl'
+export const using = ['web', 'database'] as const
+
+export function shorturlMiddlewareFactory(ctx: Context) {
+	return async (net: Koa.Context, next: Koa.Next) => {
+		if (!net.url.startsWith('/s/')) {
+			return next()
+		}
+
+		const key = net.url.slice(3)
+		if (!PRE_LENGTH_CHECK || key.length === KEY_LENGTH) {
+			const data = (await ctx.database.get('shorturl', { key }))?.[0]
+			if (data) {
+				// await ctx.database.update('shorturl', {})
+				net.status = 302
+				net.redirect(data.url)
+			} else {
+				net.status = 404
+			}
+		} else {
+			net.status = 404
+		}
+
+		if (net.status === 404) {
+			net.body = '404 Not Found: The shorturl you requested is not found on this server.'
+		}
+	}
+}
+
+export default function (ctx: Context, config: Config) {
+	async function generateKey(): Promise<string> {
+		let key: string
+		do {
+			key = randomstring.generate({
+				length: KEY_LENGTH,
+				charset: 'alphanumeric'
+			})
+		} while ((await ctx.database.get('shorturl', { key })).length !== 0)
+		return key
+	}
+
+	function userShortcut(session: Session): string {
+		return `${session.username}(${session.platform}:${session.userId})`
+	}
+
+	const logger = ctx.logger('shorturl')
+
+	if (!config.root) { throw new Error('[shorturl] No root url configured.') }
+	config.authority = config.authority || 1
+
+	ctx.model.extend('shorturl', {
+		id: 'unsigned',
+		key: 'string',
+		url: 'string',
+		count: 'unsigned',
+	}, {
+		autoInc: true,
+	})
+
+	ctx.command('shorturl <url:string>', '短网址生成器', { authority: config.authority })
+		.action(async ({ session }, url: string) => {
+			if (!url || !url.length || url.length > 1000) { return session.execute('help shorturl') }
+			logger.info('shorturl', 'add', userShortcut(session), url)
+			const key = await generateKey()
+			await ctx.database.create('shorturl', { key, url, count: 0 })
+			return segment.quote(session.messageId) + config.root + key
+		})
+}

Diff for: src/services/web/app.ts

@@ -15,6 +15,8 @@ import bodyParser from 'koa-bodyparser'
 import { Config } from './index'
 import { UserMeta, ChannelMeta, getUser, getChannel } from '../../utils/usermeta'
 
+import { shorturlMiddlewareFactory } from '../../commands/shorturl'
+
 export { Context as RouteContext } from 'koa'
 
 const viteDist = path.join(__dirname, 'vite/dist')
@@ -89,6 +91,12 @@ export class WebService {
 		return target
 	}
 
+	getPluginMiddlewares():Array<Koa.Middleware> {
+		return [
+			shorturlMiddlewareFactory(this.ctx)
+		]
+	}
+
 	register(argv: PageArgument): string {
 		const { data } = argv
 
@@ -162,6 +170,10 @@ export class WebService {
 			gzip: true,
 		}))
 
+		for (const middleware of this.getPluginMiddlewares()) {
+			this.app.use(middleware)
+		}
+
 		this.app
 			.use(this.router.routes())
 			.use(this.router.allowedMethods())

Diff for: src/services/web/vite/components.d.ts

@@ -7,33 +7,20 @@ export {}
 
 declare module '@vue/runtime-core' {
   export interface GlobalComponents {
-    AAvatar: typeof import('@arco-design/web-vue')['Avatar']
-    AAvatarGroup: typeof import('@arco-design/web-vue')['AvatarGroup']
     ACard: typeof import('@arco-design/web-vue')['Card']
-    ACardMeta: typeof import('@arco-design/web-vue')['CardMeta']
-    ADescriptions: typeof import('@arco-design/web-vue')['Descriptions']
-    AImage: typeof import('@arco-design/web-vue')['Image']
-    AImagePreviewGroup: typeof import('@arco-design/web-vue')['ImagePreviewGroup']
     ALayout: typeof import('@arco-design/web-vue')['Layout']
     ALayoutContent: typeof import('@arco-design/web-vue')['LayoutContent']
     ALayoutFooter: typeof import('@arco-design/web-vue')['LayoutFooter']
     ALayoutHeader: typeof import('@arco-design/web-vue')['LayoutHeader']
     APageHeader: typeof import('@arco-design/web-vue')['PageHeader']
-    APopover: typeof import('@arco-design/web-vue')['Popover']
-    ASkeleton: typeof import('@arco-design/web-vue')['Skeleton']
-    ASkeletonShape: typeof import('@arco-design/web-vue')['SkeletonShape']
     ASpin: typeof import('@arco-design/web-vue')['Spin']
-    ATooltip: typeof import('@arco-design/web-vue')['Tooltip']
     Avatar: typeof import('./src/components/Avatar.vue')['default']
     ExternalLink: typeof import('./src/components/ExternalLink.vue')['default']
     Footer: typeof import('./src/layout/Footer.vue')['default']
     Header: typeof import('./src/layout/Header.vue')['default']
     HeaderAvatar: typeof import('./src/components/HeaderAvatar.vue')['default']
-    HeaderPopover: typeof import('./src/components/HeaderPopover.vue')['default']
-    HeaderPopup: typeof import('./src/components/HeaderPopup.vue')['default']
     IconFile: typeof import('@arco-design/web-vue/es/icon')['IconFile']
     IconGithub: typeof import('@arco-design/web-vue/es/icon')['IconGithub']
-    IconInfoCircle: typeof import('@arco-design/web-vue/es/icon')['IconInfoCircle']
     Markdown: typeof import('./src/components/Markdown.vue')['default']
     PageError: typeof import('./src/layout/PageError.vue')['default']
     PageHome: typeof import('./src/layout/PageHome.vue')['default']

Diff for: yarn.lock

@@ -1146,6 +1146,11 @@ array-union@^2.1.0:
   resolved "https://registry.npmmirror.com/array-union/-/array-union-2.1.0.tgz#b798420adbeb1de828d84acd8a2e23d3efe85e8d"
   integrity sha512-HGyxoOTYUyCM6stUe6EJgnd4EoewAI7zMdfqO+kGjnlZmBDz/cR5pf8r/cR4Wq60sL/p0IkcjUEEPwS3GFrIyw==
 
+[email protected]:
+  version "1.0.2"
+  resolved "https://registry.yarnpkg.com/array-uniq/-/array-uniq-1.0.2.tgz#5fcc373920775723cfd64d65c64bef53bf9eba6d"
+  integrity sha512-GVYjmpL05al4dNlKJm53mKE4w9OOLiuVHWorsIA3YVz+Hu0hcn6PtE3Ydl0EqU7v+7ABC4mjjWsnLUxbpno+CA==
+
 asap@^2.0.0:
   version "2.0.6"
   resolved "https://registry.yarnpkg.com/asap/-/asap-2.0.6.tgz#e50347611d7e690943208bbdafebcbc2fb866d46"
@@ -4295,6 +4300,19 @@ [email protected]:
     discontinuous-range "1.0.0"
     ret "~0.1.10"
 
+[email protected]:
+  version "2.0.3"
+  resolved "https://registry.yarnpkg.com/randombytes/-/randombytes-2.0.3.tgz#674c99760901c3c4112771a31e521dc349cc09ec"
+  integrity sha512-lDVjxQQFoCG1jcrP06LNo2lbWp4QTShEXnhActFBwYuHprllQV6VUpwreApsYqCgD+N1mHoqJ/BI/4eV4R2GYg==
+
+randomstring@^1.2.2:
+  version "1.2.2"
+  resolved "https://registry.yarnpkg.com/randomstring/-/randomstring-1.2.2.tgz#61b3d42624c7b1fcaaaba161c53452eb3d8fef1c"
+  integrity sha512-9FByiB8guWZLbE+akdQiWE3I1I6w7Vn5El4o4y7o5bWQ6DWPcEOp+aLG7Jezc8BVRKKpgJd2ppRX0jnKu1YCfg==
+  dependencies:
+    array-uniq "1.0.2"
+    randombytes "2.0.3"
+
 range-parser@~1.2.1:
   version "1.2.1"
   resolved "https://registry.npmmirror.com/range-parser/-/range-parser-1.2.1.tgz#3cf37023d199e1c24d1a55b84800c2f3e6468031"