52 lines
1.7 KiB
TypeScript
52 lines
1.7 KiB
TypeScript
import { readFile, stat } from 'node:fs/promises'
|
|
import { basename } from 'node:path'
|
|
|
|
import { Adapters, BASE, create, FileDTO, Note, NoteMeta } from '@cryptgeon/shared'
|
|
import mime from 'mime'
|
|
import { AES, Hex, TypedArray } from 'occulto'
|
|
|
|
import { exit } from './utils.js'
|
|
|
|
type UploadOptions = Pick<Note, 'views' | 'expiration'> & { password?: string }
|
|
|
|
export async function upload(input: string | string[], options: UploadOptions) {
|
|
try {
|
|
const { password, ...noteOptions } = options
|
|
const derived = options.password ? await AES.derive(options.password) : undefined
|
|
const key = derived ? derived[0] : await AES.generateKey()
|
|
|
|
let contents: string
|
|
let type: NoteMeta['type']
|
|
if (typeof input === 'string') {
|
|
contents = await Adapters.Text.encrypt(input, key)
|
|
type = 'text'
|
|
} else {
|
|
const files: FileDTO[] = await Promise.all(
|
|
input.map(async (path) => {
|
|
const data = new Uint8Array(await readFile(path))
|
|
const stats = await stat(path)
|
|
const extension = path.substring(path.indexOf('.') + 1)
|
|
const type = mime.getType(extension) ?? 'application/octet-stream'
|
|
return {
|
|
name: basename(path),
|
|
size: stats.size,
|
|
contents: data,
|
|
type,
|
|
} satisfies FileDTO
|
|
})
|
|
)
|
|
contents = await Adapters.Files.encrypt(files, key)
|
|
type = 'file'
|
|
}
|
|
|
|
// Create the actual note and upload it.
|
|
const note: Note = { ...noteOptions, contents, meta: { type, derivation: derived?.[1] } }
|
|
const result = await create(note)
|
|
let url = `${BASE}/note/${result.id}`
|
|
if (!derived) url += `#${Hex.encode(key)}`
|
|
console.log(`Note created:\n\n${url}`)
|
|
} catch {
|
|
exit('Could not create note')
|
|
}
|
|
}
|