5chan/src/utils/getCommentMediaInfo.js

import extName from 'ext-name';
import { canEmbed } from '../components/Embed';

const getCommentMediaInfo = (comment) => {
  if (!comment?.thumbnailUrl && !comment?.link) {
    return;
  }

  if (comment?.link) {
    try {
      const url = new URL(comment.link);
      const host = url.hostname;
      let scrapedThumbnailUrl;

      if (['youtube.com', 'www.youtube.com', 'youtu.be'].includes(host)) {
        const videoId = host === 'youtu.be' ? url.pathname.slice(1) : url.searchParams.get('v');
        scrapedThumbnailUrl = `https://img.youtube.com/vi/${videoId}/sddefault.jpg`;

      } else if (host.includes('bitchute.com')) {
        const videoId = url.pathname.split('/')[2];
        scrapedThumbnailUrl = `https://static-3.bitchute.com/live/cover_images/F61vWF4shy8s/${videoId}_640x360.jpg`;

      } else if (host.includes('streamable.com')) {
        const videoId = url.pathname.split('/')[1];
        scrapedThumbnailUrl = `https://cdn-cf-east.streamable.com/image/${videoId}.jpg`;
      }

      if (canEmbed(url)) {
        return {
          url: comment.link,
          type: 'iframe',
          thumbnail: comment.thumbnailUrl || scrapedThumbnailUrl,
        };
      }

      const mime = extName(url.pathname.toLowerCase().replace('/', ''))[0]?.mime;

      if (mime?.startsWith('image')) {
        return {
          url: comment.link,
          type: 'image',
        };
      }

      if (mime?.startsWith('video')) {
        return {
          url: comment.link,
          type: 'video',
          thumbnail: comment.thumbnailUrl,
        };
      }

      if (mime?.startsWith('audio')) {
        return {
          url: comment.link,
          type: 'audio',
        };
      }
    } catch (error) {
      return;
    }
  }

  if (comment?.thumbnailUrl && comment?.thumbnailUrl !== comment?.link) {
    return {
      url: comment.link,
      type: 'webpage',
      thumbnail: comment.thumbnailUrl,
    };
  }

  if (comment?.link) {
    return {
      url: comment.link,
      type: 'webpage',
    };
  }
};

export default getCommentMediaInfo;