hrms-api-salary/src/services/edm.ts

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<StorageFile, "title" | "description" | "author" | "keyword" | "category">
> & {
  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 };
}