twenty/packages/twenty-server/src/utils/image.ts

import { type AxiosInstance } from 'axios';
import { isNonEmptyString } from '@sniptt/guards';

const cropRegex = /([w|h])([0-9]+)/;

export type ShortCropSize = `${'w' | 'h'}${number}` | 'original';

export interface CropSize {
  type: 'width' | 'height';
  value: number;
}

export const getCropSize = (value: ShortCropSize): CropSize | null => {
  const match = value.match(cropRegex);

  if (value === 'original' || match === null) {
    return null;
  }

  return {
    type: match[1] === 'w' ? 'width' : 'height',
    value: +match[2],
  };
};

export const getImageBufferFromUrl = async (
  url: string,
  axiosInstance: AxiosInstance,
): Promise<Buffer> => {
  if (!isNonEmptyString(url) || url.trim().length === 0) {
    throw new Error('Invalid URL provided: URL must be a non-empty string');
  }

  try {
    const response = await axiosInstance.get(url, {
      responseType: 'arraybuffer',
      validateStatus: (status) => status >= 200 && status < 300,
      maxRedirects: 5,
      timeout: 10000,
    });

    if (!response.data) {
      throw new Error('Received empty response from image URL');
    }

    const bufferLength = Buffer.isBuffer(response.data)
      ? response.data.length
      : response.data.byteLength;

    if (bufferLength === 0) {
      throw new Error('Received empty response from image URL');
    }

    const contentType = response.headers['content-type'];

    if (isNonEmptyString(contentType) && !contentType.startsWith('image/')) {
      throw new Error(
        `Invalid content type: expected image/*, got ${contentType}`,
      );
    }

    return Buffer.from(response.data, 'binary');
  } catch (error) {
    const message = error instanceof Error ? error.message : 'Unknown error';

    throw new Error(`Failed to fetch image from ${url}: ${message}`);
  }
};