1 files changed, 60 insertions, 0 deletions
diff --git a/sd-plugin/src/main/java/org/intellij/sdk/language/SdBlock.java b/sd-plugin/src/main/java/org/intellij/sdk/language/SdBlock.java
new file mode 100644
index 00000000000..d32baf86d7a
--- /dev/null
+++ b/sd-plugin/src/main/java/org/intellij/sdk/language/SdBlock.java
@@ -0,0 +1,60 @@
+package org.intellij.sdk.language;
+
+import com.intellij.formatting.Alignment;
+import com.intellij.formatting.Block;
+import com.intellij.formatting.Indent;
+import com.intellij.formatting.Spacing;
+import com.intellij.formatting.SpacingBuilder;
+import com.intellij.formatting.Wrap;
+import com.intellij.formatting.WrapType;
+import com.intellij.lang.ASTNode;
+import com.intellij.psi.TokenType;
+import com.intellij.psi.formatter.common.AbstractBlock;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class SdBlock extends AbstractBlock {
+    
+    private final SpacingBuilder spacingBuilder;
+    
+    protected SdBlock(@NotNull ASTNode node, @Nullable Wrap wrap, @Nullable Alignment alignment,
+                          SpacingBuilder spacingBuilder) {
+        super(node, wrap, alignment);
+        this.spacingBuilder = spacingBuilder;
+    }
+    
+    @Override
+    protected List<Block> buildChildren() {
+        List<Block> blocks = new ArrayList<>();
+        ASTNode child = myNode.getFirstChildNode();
+        while (child != null) {
+            if (child.getElementType() != TokenType.WHITE_SPACE) {
+                Block block = new SdBlock(child, Wrap.createWrap(WrapType.NONE, false), Alignment.createAlignment(),
+                                              spacingBuilder);
+                blocks.add(block);
+            }
+            child = child.getTreeNext();
+        }
+        return blocks;
+    }
+    
+    @Override
+    public Indent getIndent() {
+        return Indent.getNoneIndent();
+    }
+    
+    @Nullable
+    @Override
+    public Spacing getSpacing(@Nullable Block child1, @NotNull Block child2) {
+        return spacingBuilder.getSpacing(this, child1, child2);
+    }
+    
+    @Override
+    public boolean isLeaf() {
+        return myNode.getFirstChildNode() == null;
+    }
+    
+}