104 lines
3.4 KiB
TypeScript
104 lines
3.4 KiB
TypeScript
import { tdFileId, tdFileId as td } from './types'
|
|
import { BinaryWriter } from '@mtcute/core/src/utils/binary/binary-writer'
|
|
import { encodeUrlSafeBase64, telegramRleEncode } from './utils'
|
|
import FileType = tdFileId.FileType
|
|
|
|
type InputUniqueLocation =
|
|
| Pick<td.RawWebRemoteFileLocation, '_' | 'url'>
|
|
| Pick<td.RawPhotoRemoteFileLocation, '_' | 'volumeId' | 'localId'>
|
|
| Pick<td.RawCommonRemoteFileLocation, '_' | 'id'>
|
|
|
|
/**
|
|
* Serialize an object with information about file
|
|
* to TDLib and Bot API compatible Unique File ID
|
|
*
|
|
* Unique File IDs can't be used to download or reuse files,
|
|
* but they are globally unique, meaning that the same file will
|
|
* have the same unique ID regardless of the user who created
|
|
* this ID (unlike normal File IDs, that also contain user-bound
|
|
* file access hash)
|
|
*
|
|
* @param location Information about file location
|
|
*/
|
|
export function toUniqueFileId(
|
|
location: Omit<td.RawFullRemoteFileLocation, '_'>
|
|
): string
|
|
export function toUniqueFileId(
|
|
type: FileType,
|
|
location: InputUniqueLocation
|
|
): string
|
|
export function toUniqueFileId(
|
|
first: FileType | Omit<td.RawFullRemoteFileLocation, '_'>,
|
|
second?: InputUniqueLocation
|
|
): string {
|
|
const inputType = typeof first === 'number' ? first : first.type
|
|
const inputLocation = typeof first === 'number' ? second! : first.location
|
|
|
|
let type
|
|
if (inputLocation._ === 'web') {
|
|
type = 0
|
|
} else {
|
|
switch (inputType) {
|
|
case td.FileType.Photo:
|
|
case td.FileType.ProfilePhoto:
|
|
case td.FileType.Thumbnail:
|
|
case td.FileType.EncryptedThumbnail:
|
|
case td.FileType.Wallpaper:
|
|
type = 1
|
|
break
|
|
case td.FileType.Video:
|
|
case td.FileType.VoiceNote:
|
|
case td.FileType.Document:
|
|
case td.FileType.Sticker:
|
|
case td.FileType.Audio:
|
|
case td.FileType.Animation:
|
|
case td.FileType.VideoNote:
|
|
case td.FileType.Background:
|
|
case td.FileType.DocumentAsFile:
|
|
type = 2
|
|
break
|
|
case td.FileType.SecureRaw:
|
|
case td.FileType.Secure:
|
|
type = 3
|
|
break
|
|
case td.FileType.Encrypted:
|
|
type = 4
|
|
break
|
|
case td.FileType.Temp:
|
|
type = 5
|
|
break
|
|
default:
|
|
throw new td.InvalidFileIdError(
|
|
`Invalid file type: ${inputType}`
|
|
)
|
|
}
|
|
}
|
|
|
|
let writer: BinaryWriter
|
|
switch (inputLocation._) {
|
|
case 'photo':
|
|
writer = BinaryWriter.alloc(16)
|
|
writer.int32(type)
|
|
writer.long(inputLocation.volumeId)
|
|
writer.int32(inputLocation.localId)
|
|
break
|
|
case 'web':
|
|
writer = BinaryWriter.alloc(
|
|
Buffer.byteLength(inputLocation.url, 'utf-8') + 8
|
|
)
|
|
writer.int32(type)
|
|
writer.string(inputLocation.url)
|
|
break
|
|
case 'common':
|
|
writer = BinaryWriter.alloc(12)
|
|
writer.int32(type)
|
|
writer.long(inputLocation.id)
|
|
break
|
|
default:
|
|
throw new td.UnsupportedError(
|
|
`Unique IDs are not supported for ${(inputLocation as any)._}`
|
|
)
|
|
}
|
|
|
|
return encodeUrlSafeBase64(telegramRleEncode(writer.result()))
|
|
}
|