html-to-article-json/lib/parse/embeds.js

Converting HTML to article-json

import setupText from './text';
import {parse} from 'embeds';

const transformers = {
  custom: ({height, width, secure, src}) => ({
    type: 'embed',
    embedType: 'custom',
    caption: [],
    attribution: [],
    width,
    height,
    secure,
    src
  }),
  vine: ({id, url}) => ({
    type: 'embed',
    embedType: 'vine',
    caption: [],
    attribution: [],
    id,
    url
  }),
  spotify: ({spotifyUri, width, height, url}) => ({
    type: 'embed',
    embedType: 'spotify',
    caption: [],
    attribution: [],
    spotifyUri,
    width,
    height,
    url
  }),
  facebook: ({embedAs, headline, date, text, url, user}) => ({
    type: 'embed',
    embedType: 'facebook',
    caption: [],
    attribution: [],
    headline: headline || '',
    url,
    date,
    user,
    text,
    embedAs
  }),
  instagram: ({date, id, text, url, user}) => ({
    type: 'embed',
    embedType: 'instagram',
    caption: [],
    attribution: [],
    date: date || '',
    user: user || '',
    id,
    text,
    url
  }),
  twitter: ({embedAs, url, date, user, id, text}) => ({
    type: 'embed',
    embedType: 'twitter',
    caption: [],
    attribution: [],
    embedAs,
    url,
    date,
    user,
    id,
    text
  }),
  youtube: ({youtubeId}) => ({
    type: 'embed',
    embedType: 'youtube',
    caption: [],
    attribution: [],
    youtubeId
  }),
  tumblr: ({url, did, text}) => ({
    type: 'embed',
    embedType: 'tumblr',
    caption: [],
    url,
    did,
    text
  }),
  tidal: ({dataId, dataType}) => ({
    type: 'embed',
    embedType: 'tidal',
    caption: [],
    dataId,
    dataType
  }),
  video: ({width, height, sources}) => ({
    type: 'embed',
    embedType: 'video',
    caption: [],
    attribution: [],
    width: width || '',
    height: height || '',
    sources
  }),
  image: ({width, height, src, alt}) => ({
    type: 'embed',
    embedType: 'image',
    caption: alt ? [{content: alt, type: 'text'}] : [],
    attribution: [],
    width: width || '',
    height: height || '',
    src
  })
};

export default opts => {
  const text = setupText(opts);
  const {customEmbedTypes = []} = opts;

  const getEmbed = (elms) => {
    for (let i = 0; i < elms.length; i++) {
      let elm = elms[i];

      if (elm && elm.tagName) {
        const obj = toEmbedObj(elm) || getEmbed(elm.childNodes);
        if (obj) {
          return obj;
        }
      }
    }
  };

  const parseCustomEmbed = elm => {
    for (let i = 0; i < customEmbedTypes.length; ++i) {
      const customEmbed = customEmbedTypes[i];
      const obj = customEmbed.parse(elm);
      if (obj) {
        return obj;
      }
    }
  };

  const parseBuiltinEmbed = elm => {
    const obj = parse(elm);
    return obj && transformers[obj.type] && transformers[obj.type](obj);
  };

  const toEmbedObj = elm => {
    return parseCustomEmbed(elm) || parseBuiltinEmbed(elm) || null;
  };

  const getCaption = (elms, result, opts) => {
    for (let i = 0; i < elms.length; i++) {
      let elm = elms[i];

      // ELEMENT_NODE
      if (elm.tagName && elm.tagName.toLowerCase() !== 'cite' && elm.childNodes.length) {
        getCaption(elm.childNodes, result, text(opts, elm));
      }

      // TEXT_NODE
      if (elm.nodeName === '#text' && elm.data.length > 0) {
        result.push(text(opts, elm));
      }
    }

    return result;
  };

  const getAttribution = (elms, result, opts) => {
    for (let i = 0; i < elms.length; i++) {
      let elm = elms[i];

      // ELEMENT_NODE
      if (elm.tagName && elm.childNodes.length) {
        getAttribution(elm.childNodes, result, text(opts, elm));
      }

      // TEXT_NODE
      if (elm.nodeName === '#text' && elm.data.length > 0) {
        result.push(text(opts, elm));
      }
    }

    return result;
  };

  const getFigureProps = ({attributes, elm}) => {
    const figureProps = {};
    if (!elm.attributes || !elm.attributes.length) {
      return figureProps;
    }

    attributes.forEach((name) => {
      const value = elm.getAttribute(name);
      if (value) {
        figureProps[name] = value;
      }
    });

    return figureProps;
  };

  return elm => {
    if (elm.tagName && elm.tagName.toLowerCase() === 'figure') {
      const obj = getEmbed(elm.childNodes);
      if (obj && opts.parseFigureProps) {
        obj.figureProps = getFigureProps({
          attributes: opts.parseFigureProps,
          elm
        });
      }
      if (obj) {
        const figcaptions = elm.getElementsByTagName('figcaption');
        obj.caption = getCaption(figcaptions, [], {});
        const cites = elm.getElementsByTagName('cite');
        obj.attribution = getAttribution(cites, [], {});
      }
      return obj;
    }

    return toEmbedObj(elm);
  };
};