From 4c555457cf62cd75ea897483d2a0bd668ce22d8a Mon Sep 17 00:00:00 2001
From: Albert Toledo <toledoalbert@gmail.com>
Date: Mon, 23 Sep 2019 16:50:41 -0400
Subject: [PATCH] Adds ability to add custom classes when adding a new block
 via class property in data object

---
 src/util/blockStyleFn.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/util/blockStyleFn.js b/src/util/blockStyleFn.js
index 5db2999b..a7fd586f 100644
--- a/src/util/blockStyleFn.js
+++ b/src/util/blockStyleFn.js
@@ -4,9 +4,11 @@ import { Block } from './constants';
 Get custom classnames for each of the different block types supported.
 */
 
-const BASE_BLOCK_CLASS = 'md-block';
-
 export default (block) => {
+  const classFromData = block.getData().get('class');
+  const customClassToApply = classFromData ? `${classFromData} ` : '';
+  const BASE_BLOCK_CLASS = `${customClassToApply}md-block`;
+
   switch (block.getType()) {
     case Block.BLOCKQUOTE:
       return `${BASE_BLOCK_CLASS} ${BASE_BLOCK_CLASS}-quote md-RichEditor-blockquote`;