cstyle/transformers/margin-block/main.go

diff --git a/cstyle/transformers/margin-block/main.go b/cstyle/transformers/margin-block/main.go
index e7bd169..096f286 100644
--- a/cstyle/transformers/margin-block/main.go
+++ b/cstyle/transformers/margin-block/main.go
@@ -11,2 +11,2 @@ func Init() cstyle.Transformer {
-		Selector: func(n *element.Node, c *cstyle.CSS) bool {
-			return n.Style("margin-block") != "" || n.Style("margin-block-start") != "" || n.Style("margin-block-end") != ""
+		Selector: func(n *element.Node) bool {
+			return n.CStyle["margin-block"] != "" || n.CStyle["margin-block-start"] != "" || n.CStyle["margin-block-end"] != ""
@@ -16,2 +16,2 @@ func Init() cstyle.Transformer {
-			writingMode := n.Style("writing-mode")
-			mb := parseMarginBlock(n.Style("margin-block"))
+			writingMode := n.CStyle["writing-mode"]
+			mb := parseMarginBlock(n.CStyle["margin-block"])
@@ -19,2 +19,2 @@ func Init() cstyle.Transformer {
-			if n.Style("margin-block-start") != "" {
-				mb[0] = n.Style("margin-block-start")
+			if n.CStyle["margin-block-start"] != "" {
+				mb[0] = n.CStyle["margin-block-start"]
@@ -22,2 +22,2 @@ func Init() cstyle.Transformer {
-			if n.Style("margin-block-end") != "" {
-				mb[1] = n.Style("margin-block-end")
+			if n.CStyle["margin-block-end"] != "" {
+				mb[1] = n.CStyle["margin-block-end"]
@@ -29,2 +29,2 @@ func Init() cstyle.Transformer {
-				n.Style("margin-left", mb[0])
-				n.Style("margin-right", mb[1])
+				n.CStyle["margin-left"] = mb[0]
+				n.CStyle["margin-right"] = mb[1]
@@ -34,5 +34,5 @@ func Init() cstyle.Transformer {
-				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])
+				n.CStyle["margin-left"] = mb[1]
+				n.CStyle["margin-right"] = mb[0]
+			} else if n.CStyle["margin"] == "" {
+				if n.CStyle["margin-top"] == "" {
+					n.CStyle["margin-top"] = mb[0]
@@ -40,2 +40,2 @@ func Init() cstyle.Transformer {
-				if n.Style("margin-bottom") == "" {
-					n.Style("margin-bottom", mb[0])
+				if n.CStyle["margin-bottom"] == "" {
+					n.CStyle["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.Style("margin-block") != "" || n.Style("margin-block-start") != "" || n.Style("margin-block-end") != ""
		},
		Handler: func(n *element.Node, c *cstyle.CSS) *element.Node {

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

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

			// !ISSUE: Not working line36 broke margin

			if writingMode == "vertical-lr" {
				n.Style("margin-left", mb[0])
				n.Style("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.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])
				}
				if n.Style("margin-bottom") == "" {
					n.Style("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{}
	}
}