Support for different types of atomic blocks

I have different types of atomic blocks but currently getData() returns nothing. I've managed to get things working by passing the data to getAtomicBlockChunk with the following changes;

https://github.com/jpuri/html-to-draftjs/blob/master/src/library/index.js (37)

if (customChunkGenerator) {
  const value = customChunkGenerator(nodeName, node);
  if (value) {
    const entityId = Entity.__create(
      value.type,
      value.mutability,
      value.data || {},
    );
    // return { chunk: getAtomicBlockChunk(entityId) };
    return { chunk: getAtomicBlockChunk(entityId, value.data) };
  }
}

https://github.com/jpuri/html-to-draftjs/blob/master/src/library/chunkBuilder.js (74)

// export const getAtomicBlockChunk = (entityId: number): Object => {
export const getAtomicBlockChunk = (entityId: number, data: any): Object => {
  return {
    text: '\r ',
    inlines: [new OrderedSet()],
    entities: [entityId],
    blocks: [{
      type: 'atomic',
      depth: 0,
      // data: new Map({})
      data: new Map(data)
    }],
  };
};

Would it be possible to add this by default please?

Thank you.

Start in src/library/index.js around the customChunkGenerator handling and src/library/chunkBuilder.js around getAtomicBlockChunk. Trace how custom atomic block data is created and represented, then verify that getData() preserves the supplied data by exercising the existing test setup. Done means different atomic block types retain their data without requiring callers to modify these files.