import { DecodedJwt, createDecoder } from "fast-jwt"; import HttpError from "../interfaces/http-error"; import HttpStatus from "../interfaces/http-status"; import { StorageFile, StorageFolder } from "../interfaces/storage-fs"; const jwtDecode = createDecoder({ complete: true }); if (!process.env.STORAGE_URL) { throw new Error("Requires STORAGE_URL env variable."); } if (!process.env.STORAGE_REALM_URL && !process.env.STORAGE_SECRET) { throw new Error("Requires STORAGE_REALM_URL and STORAGE_SECRET env variable."); } export type FileProps = Partial< Pick > & { metadata?: { [key: string]: unknown }; }; const STORAGE_URL = process.env.STORAGE_URL; const STORAGE_REALM_URL = process.env.STORAGE_REALM_URL!; const STORAGE_SECRET = process.env.STORAGE_SECRET!; let token: string | null = null; let decoded: DecodedJwt | null = null; /** * Check if token is expired or will expire in 30 seconds * @returns true if expire or can't get exp, false otherwise */ export function expireCheck(token: string, beforeExpire: number = 30) { decoded = jwtDecode(token); if (decoded && decoded.payload.exp) { return Date.now() / 1000 >= decoded.payload.exp - beforeExpire; } return true; } /** * Get token from id service if needed */ export async function getToken() { if (!token || expireCheck(token)) { const body = new URLSearchParams(); body.append("client_id", "ext-api"); body.append("client_secret", STORAGE_SECRET); body.append("grant_type", "client_credentials"); const res = await fetch(`${STORAGE_REALM_URL}/protocol/openid-connect/token`, { method: "POST", body: body, }).catch((e) => console.error(e)); if (!res) return; const data = await res.json(); if (data && data.access_token) { token = data.access_token; } } return token; } /** * @param path - Path that new folder will live * @param name - Name of the folder to create * @param recursive - Will create parent automatically */ export async function createFolder(path: string[], name: string, recursive: boolean = false) { if (recursive && path.length > 0) { await createFolder(path.slice(0, -1), path[path.length - 1], true); } const res = await fetch(`${STORAGE_URL}/storage/folder`, { method: "POST", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ path, name }), }).catch((e) => console.error(e)); if (!res || !res.ok) { return Boolean(console.error(res ? await res.json() : res)); } return true; } /** * @param path - Path that new file will live * @param file - Name of the file to create */ export async function createFile(path: string[], file: string, props?: FileProps) { const res = await fetch(`${STORAGE_URL}/storage/file`, { method: "POST", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ ...props, path, file, hidden: path.some((v) => v.startsWith(".")), }), }).catch((e) => console.error(e)); if (!res || !res.ok) { return Boolean(console.error(res ? await res.json() : res)); } return (await res.json()) as StorageFile & { uploadUrl: string }; } export async function list(operation: "file" | "folder", path: string[]) { const res = await fetch(`${STORAGE_URL}/storage/list`, { method: "POST", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ operation, path, hidden: true }), }).catch((e) => console.error(e)); if (!res || !res.ok) { if (res && res.status === HttpStatus.NOT_FOUND) { return []; } return Boolean(console.error(res ? await res.json() : res)); } return (await res.json()) as StorageFile & { uploadUrl: string }; } export async function listFolder(path: string[]) { return (await list("folder", path)) as StorageFolder[] | boolean; } export async function listFile(path: string[]) { return (await list("file", path)) as StorageFile[] | boolean; } export async function updateFile(path: string[], file: string, metadata: FileProps, destPath?: string[], destFile?: string) { const res = await fetch(`${STORAGE_URL}/storage/file`, { method: "PUT", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ from: { path, file }, to: destPath && destFile ? { path: destPath, file: destFile,} : undefined, ...metadata, upload: false, }), }).catch((e) => console.error(e)); if (!res || !res.ok) { if (res && res.status === HttpStatus.NOT_FOUND) { throw new HttpError(HttpStatus.NOT_FOUND, "ไม่พบไฟล์ในระบบ"); } return Boolean(console.error(res ? await res.json() : res)); } return Boolean(res); } export async function deleteFolder(path: string[], name: string) { const res = await fetch(`${STORAGE_URL}/storage/folder`, { method: "DELETE", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ path: [...path, name] }), }).catch((e) => console.error(e)); if (!res || !res.ok) { return Boolean(console.error(res ? await res.json() : res)); } return true; } export async function deleteFile(path: string[], file: string) { const res = await fetch(`${STORAGE_URL}/storage/file`, { method: "DELETE", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ path, file }), }).catch((e) => console.error(e)); if (!res || !res.ok) { return Boolean(console.error(res ? await res.json() : res)); } return true; } export async function downloadFile(path: string[], file: string) { const res = await fetch(`${STORAGE_URL}/storage/file/download`, { method: "POST", headers: { Authorization: `Bearer ${await getToken()}`, "Content-Type": "application/json", }, body: JSON.stringify({ path, file }), }).catch((e) => console.error(e)); if (!res || !res.ok) { if (res && res.status === HttpStatus.NOT_FOUND) { throw new HttpError(HttpStatus.NOT_FOUND, "ไม่พบไฟล์ในระบบ"); } return Boolean(console.error(res ? await res.json() : res)); } return (await res.json()) as StorageFile & { downloadUrl: string }; }