cstyle/transformers/margin-block/main.go

diff --git a/cstyle/transformers/margin-block/main.go b/cstyle/transformers/margin-block/main.go
index a82ddf4..e7bd169 100644
--- a/cstyle/transformers/margin-block/main.go
+++ b/cstyle/transformers/margin-block/main.go
@@ -12 +12 @@ func Init() cstyle.Transformer {
-			return n.GetStyle("margin-block") != "" || n.GetStyle("margin-block-start") != "" || n.GetStyle("margin-block-end") != ""
+			return n.Style("margin-block") != "" || n.Style("margin-block-start") != "" || n.Style("margin-block-end") != ""
@@ -16,2 +16,2 @@ func Init() cstyle.Transformer {
-			writingMode := n.GetStyle("writing-mode")
-			mb := parseMarginBlock(n.GetStyle("margin-block"))
+			writingMode := n.Style("writing-mode")
+			mb := parseMarginBlock(n.Style("margin-block"))
@@ -19,2 +19,2 @@ func Init() cstyle.Transformer {
-			if n.GetStyle("margin-block-start") != "" {
-				mb[0] = n.GetStyle("margin-block-start")
+			if n.Style("margin-block-start") != "" {
+				mb[0] = n.Style("margin-block-start")
@@ -22,2 +22,2 @@ func Init() cstyle.Transformer {
-			if n.GetStyle("margin-block-end") != "" {
-				mb[1] = n.GetStyle("margin-block-end")
+			if n.Style("margin-block-end") != "" {
+				mb[1] = n.Style("margin-block-end")
@@ -29,2 +29,2 @@ func Init() cstyle.Transformer {
-				n.SetStyle("margin-left", mb[0])
-				n.SetStyle("margin-right", mb[1])
+				n.Style("margin-left", mb[0])
+				n.Style("margin-right", mb[1])
@@ -34,5 +34,5 @@ func Init() cstyle.Transformer {
-				n.SetStyle("margin-left", mb[1])
-				n.SetStyle("margin-right", mb[0])
-			} else if n.GetStyle("margin") == "" {
-				if n.GetStyle("margin-top") == "" {
-					n.SetStyle("margin-top", mb[0])
+				n.Style("margin-left", mb[1])
+				n.Style("margin-right", mb[0])
+			} else if n.Style("margin") == "" {
+				if n.Style("margin-top") == "" {
+					n.Style("margin-top", mb[0])
@@ -40,2 +40,2 @@ func Init() cstyle.Transformer {
-				if n.GetStyle("margin-bottom") == "" {
-					n.SetStyle("margin-bottom", mb[0])
+				if n.Style("margin-bottom") == "" {
+					n.Style("margin-bottom", mb[0])

package marginblock

import (
	"grim/cstyle"
	"grim/element"
	"strings"
)

func Init() cstyle.Transformer {
	return cstyle.Transformer{
		Selector: func(n *element.Node, c *cstyle.CSS) bool {
			return n.GetStyle("margin-block") != "" || n.GetStyle("margin-block-start") != "" || n.GetStyle("margin-block-end") != ""
		},
		Handler: func(n *element.Node, c *cstyle.CSS) *element.Node {

			writingMode := n.GetStyle("writing-mode")
			mb := parseMarginBlock(n.GetStyle("margin-block"))

			if n.GetStyle("margin-block-start") != "" {
				mb[0] = n.GetStyle("margin-block-start")
			}
			if n.GetStyle("margin-block-end") != "" {
				mb[1] = n.GetStyle("margin-block-end")
			}

			// !ISSUE: Not working line36 broke margin

			if writingMode == "vertical-lr" {
				n.SetStyle("margin-left", mb[0])
				n.SetStyle("margin-right", mb[1])
			} else if writingMode == "vertical-rl" {
				// !ISSUE: This will not move everything over
				// + link: https://developer.mozilla.org/en-US/docs/Web/CSS/margin-block
				n.SetStyle("margin-left", mb[1])
				n.SetStyle("margin-right", mb[0])
			} else if n.GetStyle("margin") == "" {
				if n.GetStyle("margin-top") == "" {
					n.SetStyle("margin-top", mb[0])
				}
				if n.GetStyle("margin-bottom") == "" {
					n.SetStyle("margin-bottom", mb[0])
				}
			}

			return n
		},
	}
}

func parseMarginBlock(s string) []string {
	split := strings.Split(s, " ")

	switch len(split) {
	case 2:
		return split
	case 1:
		return []string{split[0], split[0]}
	default:
		return []string{}
	}
}