fbredius/storybook

import React from 'react';
import { PartialStoryFn } from '@storybook/csf';

export function prepareForInline(storyFn: PartialStoryFn<any>) {
  const html = storyFn();
  if (typeof html === 'string') {
    // eslint-disable-next-line react/no-danger
    return <div dangerouslySetInnerHTML={{ __html: html }} />;
  }
  return (
    <div ref={(node?: HTMLDivElement): never | null => (node ? node.appendChild(html) : null)} />
  );
}