100 lines
2.5 KiB
TypeScript
100 lines
2.5 KiB
TypeScript
import { LarkBody, LarkCard } from "@egg/lark-msg-tool"
|
|
import loggerIns from "@egg/logger"
|
|
import { GitlabService, NetTool } from "@egg/net-tool"
|
|
import { PathCheckTool } from "@egg/path-tool"
|
|
import { v4 as uuid } from "uuid"
|
|
|
|
import cardMap from "../constant/card"
|
|
import { APP_CONFIG, APP_MAP } from "../constant/config"
|
|
import functionMap from "../constant/function"
|
|
import tempMap from "../constant/template"
|
|
import DB from "../db"
|
|
import { AttachService } from "../services"
|
|
import { Context } from "../types"
|
|
import genLarkService from "./genLarkService"
|
|
|
|
/**
|
|
* 获取之前的requestId。
|
|
*
|
|
* @param larkBody - 请求体。
|
|
* @returns 返回之前的requestId。
|
|
*/
|
|
const getPreRequestId = (larkBody: LarkBody) => {
|
|
// 在Action请求中会带上之前的requestId
|
|
const value = larkBody.actionValue
|
|
if (!value || !value.requestId) return ""
|
|
return value.requestId
|
|
}
|
|
|
|
/**
|
|
* 生成请求上下文。
|
|
*
|
|
* @param {Request} req - 请求对象。
|
|
* @returns {Promise<Context>} 返回包含请求上下文的对象。
|
|
*/
|
|
const genContext = async (req: Request, rId?: string) => {
|
|
let body: any = null
|
|
let text: string = ""
|
|
try {
|
|
text = await req.text()
|
|
body = JSON.parse(text)
|
|
} catch {
|
|
/* empty */
|
|
}
|
|
const larkBody = new LarkBody(body)
|
|
const searchParams = new URL(req.url).searchParams
|
|
const headers = new Headers(req.headers)
|
|
const app = searchParams.get("app") || "egg"
|
|
const appInfo = APP_MAP[app]
|
|
const requestId = rId || getPreRequestId(larkBody) || uuid()
|
|
const logger = loggerIns.child({ requestId })
|
|
const genResp = new NetTool({ requestId })
|
|
const larkService = genLarkService("egg", requestId)
|
|
// 设置回复消息功能
|
|
larkService.message.setReplyMessage(larkBody.messageId)
|
|
const gitlabService = new GitlabService({
|
|
baseUrl: APP_CONFIG.GITLAB_BASE_URL,
|
|
authKey: APP_CONFIG.GITLAB_AUTH_KEY,
|
|
requestId,
|
|
})
|
|
const attachService = new AttachService({ requestId })
|
|
const path = new PathCheckTool(req.url)
|
|
const db = new DB(logger)
|
|
const larkCard = new LarkCard(
|
|
"egg",
|
|
false,
|
|
requestId,
|
|
cardMap,
|
|
tempMap,
|
|
functionMap
|
|
)
|
|
|
|
return {
|
|
req,
|
|
path,
|
|
requestId,
|
|
logger,
|
|
genResp,
|
|
body,
|
|
text,
|
|
larkService,
|
|
larkBody,
|
|
larkCard,
|
|
attachService,
|
|
gitlabService,
|
|
searchParams,
|
|
headers,
|
|
app,
|
|
appInfo,
|
|
db,
|
|
} as Context
|
|
}
|
|
|
|
export const genContextManually = (rId?: string, app = "egg") =>
|
|
genContext(
|
|
new Request("https://lark-egg-preview.ai.xiaomi.com?app=" + app),
|
|
rId
|
|
)
|
|
|
|
export default genContext
|