import { Template } from '@/lib/types/template';

export const VIDEO_EXTENSIONS = ['mp4', 'webm', 'ogg', 'mov', 'avi', 'mkv', 'flv'];
export const IMAGE_EXTENSIONS = ['jpg', 'jpeg', 'png', 'gif', 'webp', 'bmp', 'svg'];

/**
 * 验证媒体URL是否有效
 */
export const isValidMediaUrl = (url: string): boolean => {
  if (!url || url.trim() === '') return false;

  try {
    new URL(url);
    return true;
  } catch {
    // 如果不是有效的URL格式，检查是否是相对路径
    return url.startsWith('/') || url.startsWith('./') || url.startsWith('../');
  }
};

/**
 * 获取媒体类型：video 或 image
 */
export const getMediaType = (template: Template): 'video' | 'image' => {
  // 1. 优先检查 previewUrl 的文件扩展名
  if (template.previewUrl && isValidMediaUrl(template.previewUrl)) {
    const previewUrl = template.previewUrl.toLowerCase();

    // 检查是否为视频文件
    const videoMatch = new RegExp(`\\.(${VIDEO_EXTENSIONS.join('|')})$`).test(previewUrl);
    if (videoMatch) {
      return 'video';
    }

    // 检查是否为图片文件
    const imageMatch = new RegExp(`\\.(${IMAGE_EXTENSIONS.join('|')})$`).test(previewUrl);
    if (imageMatch) {
      return 'image';
    }
  }

  // 2. 检查 coverImageUrl 的文件扩展名
  if (template.coverImageUrl && isValidMediaUrl(template.coverImageUrl)) {
    const coverImageUrl = template.coverImageUrl.toLowerCase();
    const imageMatch = new RegExp(`\\.(${IMAGE_EXTENSIONS.join('|')})$`).test(coverImageUrl);
    if (imageMatch) {
      return 'image';
    }
  }

  // 3. 根据 previewUrl 是否存在来推断类型
  if (template.previewUrl && template.previewUrl.trim() !== '') {
    return 'video';
  }

  // 4. 如果有coverImageUrl，默认为图片
  if (template.coverImageUrl && template.coverImageUrl.trim() !== '') {
    return 'image';
  }

  // 5. 默认为图片
  return 'image';
};

/**
 * 判断模板是否为视频类型
 */
export const isVideoTemplate = (template: Template): boolean => {
  const mediaType = getMediaType(template);
  return mediaType === 'video';
};

/**
 * 判断模板是否为图片类型
 */
export const isImageTemplate = (template: Template): boolean => {
  const mediaType = getMediaType(template);
  return mediaType === 'image';
};

/**
 * 获取有效的媒体URL
 */
export const getEffectiveMediaUrl = (template: Template): string => {
  const mediaType = getMediaType(template);

  if (mediaType === 'video') {
    return template.previewUrl || '';
  } else {
    return template.coverImageUrl || '';
  }
};

/**
 * 检查媒体URL是否有效且可以加载
 */
export const canLoadMedia = (template: Template): boolean => {
  const mediaType = getMediaType(template);
  const url = getEffectiveMediaUrl(template);

  if (!url || !isValidMediaUrl(url)) {
    console.warn(`模板 "${template.title}" 的${mediaType}URL无效:`, url);
    return false;
  }

  return true;
};