Kentico/gatsby-source-kontent

import { SourceNodesArgs } from 'gatsby';
import {
  CustomPluginOptions,
  KontentType,
  KontentTypeElementArrayItem,
  KontentTypeElementsObject,
} from './types';
import {
  getKontentTypeNodeStringForCodeName,
  getKontentTypeTypeName,
} from './naming';
import { loadAllKontentTypesCached } from './client';

const getKontentTypeArtifact = (
  api: SourceNodesArgs,
  kontentType: KontentType,
  includeRawContent: boolean,
): KontentType => {
  const nodeIdString = getKontentTypeNodeStringForCodeName(
    kontentType.system.codename,
  );
  const nodeData: KontentType = {
    ...kontentType,
    id: api.createNodeId(nodeIdString),
    children: [],
    internal: {
      type: getKontentTypeTypeName(),
      contentDigest: api.createContentDigest(kontentType),
    },
  };
  if (includeRawContent) {
    nodeData.internal.content = JSON.stringify(kontentType);
  }
  return nodeData;
};

const transformElementObjectToArray = (types: Array<KontentType>): void => {
  types.forEach(type => {
    (type.elements as KontentTypeElementArrayItem[]) = Object.keys(
      type.elements,
    ).map((key: string) => {
      (type.elements as KontentTypeElementsObject)[key].codename = key;
      return (type.elements as KontentTypeElementsObject)[key];
    });
  });
};

const sourceNodes = async (
  api: SourceNodesArgs,
  options: CustomPluginOptions,
): Promise<void> => {
  const kontentTypes = await loadAllKontentTypesCached(options, api.cache);
  transformElementObjectToArray(kontentTypes);
  for (const kontentType of kontentTypes) {
    const nodeData = getKontentTypeArtifact(
      api,
      kontentType,
      options.includeRawContent,
    );
    api.actions.createNode(nodeData);
  }
};

export { sourceNodes as kontentTypesSourceNodes };