cstyle/transformers/margin-block/main.go

diff --git a/cstyle/transformers/margin-block/main.go b/cstyle/transformers/margin-block/main.go
index 1ee65e2..45e2989 100644
--- a/cstyle/transformers/margin-block/main.go
+++ b/cstyle/transformers/margin-block/main.go
@@ -12 +12 @@ func Init() cstyle.Transformer {
-			return n.CStyle["margin-block"] != "" || n.CStyle["margin-block-start"] != "" || n.CStyle["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.CStyle["writing-mode"]
-			mb := parseMarginBlock(n.CStyle["margin-block"])
+			writingMode := n.Style["writing-mode"]
+			mb := parseMarginBlock(n.Style["margin-block"])
@@ -19,2 +19,2 @@ func Init() cstyle.Transformer {
-			if n.CStyle["margin-block-start"] != "" {
-				mb[0] = n.CStyle["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.CStyle["margin-block-end"] != "" {
-				mb[1] = n.CStyle["margin-block-end"]
+			if n.Style["margin-block-end"] != "" {
+				mb[1] = n.Style["margin-block-end"]
@@ -26,2 +25,0 @@ func Init() cstyle.Transformer {
-			// !ISSUE: Not working tline36 broke margin
-
@@ -29,2 +27,2 @@ func Init() cstyle.Transformer {
-				n.CStyle["margin-left"] = mb[0]
-				n.CStyle["margin-right"] = mb[1]
+				n.Style["margin-left"] = mb[0]
+				n.Style["margin-right"] = mb[1]
@@ -34,5 +32,5 @@ func Init() cstyle.Transformer {
-				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]
+				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 +38,2 @@ func Init() cstyle.Transformer {
-				if n.CStyle["margin-bottom"] == "" {
-					n.CStyle["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) bool {
			return n.CStyle["margin-block"] != "" || n.CStyle["margin-block-start"] != "" || n.CStyle["margin-block-end"] != ""
		},
		Handler: func(n *element.Node, c *cstyle.CSS) *element.Node {

			writingMode := n.CStyle["writing-mode"]
			mb := parseMarginBlock(n.CStyle["margin-block"])

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

			// !ISSUE: Not working tline36 broke margin

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