GuanM/LLOneBot
1
0
Fork 0
mirror of https://github.com/LLOneBot/LLOneBot.git synced 2024-11-22 01:56:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: auto delete file when call get_file

Browse Source
This commit is contained in:
linyuchen 2024-03-01 22:00:38 +08:00
parent 57ef8ed3e4
commit 4aadd81e60
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/onebot11/action/GetFile.ts
View File

@ -19,6 +19,7 @@ export interface GetFileResponse{
export class GetFileBase extends BaseAction<GetFilePayload, GetFileResponse>{
protected async _handle(payload: GetFilePayload): Promise<GetFileResponse> {
const cache = fileCache.get(payload.file)
const {autoDeleteFile, enableLocalFile2Url, autoDeleteFileSecond} = getConfigUtil().getConfig()
if (!cache) {
throw new Error('file not found')
}
@ -31,11 +32,16 @@ export class GetFileBase extends BaseAction<GetFilePayload, GetFileResponse>{
file_size: cache.fileSize,
file_name: cache.fileName
}
if (getConfigUtil().getConfig().enableLocalFile2Url) {
if (enableLocalFile2Url) {
if (!cache.url) {
res.base64 = await fs.readFile(cache.filePath, 'base64')
}
}
if (autoDeleteFile) {
setTimeout(() => {
fs.unlink(cache.filePath)
}, autoDeleteFileSecond * 1000)
}
return res
}
}