Add upload attachment ability into sdk

This commit is contained in:
2025-01-10 21:40:59 +08:00
parent d8f51e305b
commit b5765877af
8 changed files with 416 additions and 3 deletions

View File

@ -7,7 +7,7 @@
"name": "LittleSheep",
"email": "littlesheep.code@hotmail.com"
},
"version": "0.0.1",
"version": "0.0.2",
"tsup": {
"entry": [
"src/index.ts"

View File

@ -46,3 +46,141 @@ export async function listAttachment(id: string[]): Promise<SnAttachment[]> {
})
return resp.data.data
}
type MultipartProgress = {
value: number | null
current: number
total: number
}
type MultipartInfo = {
rid: string
fileChunks: Record<string, number>
isUploaded: boolean
}
class UploadAttachmentTask {
private content: File
private pool: string
private multipartSize: number = 0
private multipartInfo: MultipartInfo | null = null
private multipartProgress: MultipartProgress = { value: null, current: 0, total: 0 }
loading: boolean = false
success: boolean = false
error: string | null = null
constructor(content: File, pool: string) {
if (!content || !pool) {
throw new Error('Content and pool are required.')
}
this.content = content
this.pool = pool
}
public async submit(): Promise<void> {
this.loading = true
const limit = 3
try {
await this.createMultipartPlaceholder()
console.log(`[Paperclip] Multipart placeholder has been created with rid ${this.multipartInfo?.rid}`)
this.multipartProgress.value = 0
this.multipartProgress.current = 0
const chunks = Object.keys(this.multipartInfo?.fileChunks || {})
this.multipartProgress.total = chunks.length
const uploadChunks = async (chunk: string): Promise<void> => {
try {
await this.uploadSingleMultipart(chunk)
this.multipartProgress.current++
console.log(
`[Paperclip] Uploaded multipart ${this.multipartProgress.current}/${this.multipartProgress.total}`,
)
this.multipartProgress.value = this.multipartProgress.current / this.multipartProgress.total
} catch (err) {
console.log(`[Paperclip] Upload multipart ${chunk} failed, retrying in 3 seconds...`)
await this.delay(3000)
await uploadChunks(chunk)
}
}
for (let i = 0; i < chunks.length; i += limit) {
const chunkSlice = chunks.slice(i, i + limit)
await Promise.all(chunkSlice.map(uploadChunks))
}
if (this.multipartInfo?.isUploaded) {
console.log(`[Paperclip] Entire file has been uploaded in ${this.multipartProgress.total} chunk(s)`)
this.success = true
}
} catch (e) {
console.error(e)
this.error = e instanceof Error ? e.message : String(e)
} finally {
this.loading = false
}
}
private async createMultipartPlaceholder(): Promise<void> {
const mimetypeMap: Record<string, string> = {
mp4: 'video/mp4',
mov: 'video/quicktime',
mp3: 'audio/mp3',
wav: 'audio/wav',
m4a: 'audio/m4a',
}
const fileExtension = this.content.name.split('.').pop() || ''
const mimetype = mimetypeMap[fileExtension]
const nameArray = this.content.name.split('.')
nameArray.pop()
const resp = await sni.post('/cgi/uc/attachments/multipart', {
pool: this.pool,
size: this.content.size,
name: this.content.name,
alt: nameArray.join('.'),
mimetype,
})
const data = await resp.data
this.multipartSize = data.chunkSize
this.multipartInfo = data.meta
}
private async uploadSingleMultipart(chunkId: string): Promise<void> {
if (!this.multipartInfo) return
const chunkIdx = this.multipartInfo.fileChunks[chunkId]
const chunk = this.content.slice(chunkIdx * this.multipartSize, (chunkIdx + 1) * this.multipartSize)
const data = new FormData()
data.set('file', chunk)
const resp = await sni.post(`/cgi/uc/attachments/multipart/${this.multipartInfo.rid}/${chunkId}`, data, {
timeout: 3 * 60 * 1000,
})
this.multipartInfo = resp.data
}
private delay(ms: number): Promise<void> {
return new Promise((resolve) => setTimeout(resolve, ms))
}
public static formatBytes(bytes: number, decimals = 2): string {
if (!+bytes) return '0 Bytes'
const k = 1024
const dm = decimals < 0 ? 0 : decimals
const sizes = ['Bytes', 'KiB', 'MiB', 'GiB', 'TiB', 'PiB', 'EiB', 'ZiB', 'YiB']
const i = Math.floor(Math.log(bytes) / Math.log(k))
return `${parseFloat((bytes / Math.pow(k, i)).toFixed(dm))} ${sizes[i]}`
}
}

View File

@ -0,0 +1,24 @@
export interface MaRelease {
id: number
created_at: Date
updated_at: Date
deleted_at?: Date
version: string
type: number
channel: string
assets: Record<string, any>
product_id: number
meta: MaReleaseMeta
}
export interface MaReleaseMeta {
id: number
created_at: Date
updated_at: Date
deleted_at?: Date
title: string
description: string
content: string
attachments: string[]
release_id: number
}