cstyle/transformers/banda/main.go

diff --git a/cstyle/transformers/banda/main.go b/cstyle/transformers/banda/main.go
index ce55116..0d9fef8 100644
--- a/cstyle/transformers/banda/main.go
+++ b/cstyle/transformers/banda/main.go
@@ -10,3 +10,2 @@ func Init() cstyle.Transformer {
-		Selector: func(n *element.Node, c *cstyle.CSS) bool {
-			ps := c.PsuedoStyles[n.Properties.Id]
-			if ps["::before"] != nil || ps["::after"] != nil {
+		Selector: func(n *element.Node) bool {
+			if n.PseudoElements["::before"] != nil || n.PseudoElements["::after"] != nil {
@@ -19,2 +18 @@ func Init() cstyle.Transformer {
-			ps := c.PsuedoStyles[n.Properties.Id]
-			if ps["::before"] != nil {
+			if n.PseudoElements["::before"] != nil {
@@ -23 +21,2 @@ func Init() cstyle.Transformer {
-				before.Style("display", "inline")
+				before.CStyle, _ = c.GetStyles(&before)
+				before.CStyle["display"] = "inline"
@@ -25,2 +24,2 @@ func Init() cstyle.Transformer {
-				for k, v := range ps["::before"] {
-					before.Style(k, v)
+				for k, v := range n.PseudoElements["::before"] {
+					before.CStyle[k] = v
@@ -29 +28 @@ func Init() cstyle.Transformer {
-				before.InnerText = ps["::before"]["content"][1 : len(ps["::before"]["content"])-1]
+				before.InnerText = n.PseudoElements["::before"]["content"][1 : len(n.PseudoElements["::before"]["content"])-1]
@@ -38 +37 @@ func Init() cstyle.Transformer {
-			if ps["::after"] != nil {
+			if n.PseudoElements["::after"] != nil {
@@ -41 +40,2 @@ func Init() cstyle.Transformer {
-				after.Style("display", "inline")
+				after.CStyle, _ = c.GetStyles(&after)
+				after.CStyle["display"] = "inline"
@@ -43,2 +43,2 @@ func Init() cstyle.Transformer {
-				for k, v := range ps["::after"] {
-					after.Style(k, v)
+				for k, v := range n.PseudoElements["::after"] {
+					after.CStyle[k] = v
@@ -47 +47 @@ func Init() cstyle.Transformer {
-				after.InnerText = ps["::after"]["content"][1 : len(ps["::after"]["content"])-1]
+				after.InnerText = n.PseudoElements["::after"]["content"][1 : len(n.PseudoElements["::after"]["content"])-1]

package banda

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

func Init() cstyle.Transformer {
	return cstyle.Transformer{
		Selector: func(n *element.Node, c *cstyle.CSS) bool {
			ps := c.PsuedoStyles[n.Properties.Id]
			if ps["::before"] != nil || ps["::after"] != nil {
				return true
			} else {
				return false
			}
		},
		Handler: func(n *element.Node, c *cstyle.CSS) *element.Node {
			ps := c.PsuedoStyles[n.Properties.Id]
			if ps["::before"] != nil {
				before := n.CreateElement("before")
				before.Parent = n
				before.Style("display", "inline")

				for k, v := range ps["::before"] {
					before.Style(k, v)
				}

				before.InnerText = ps["::before"]["content"][1 : len(ps["::before"]["content"])-1]

				if len(n.Children) == 0 {
					n.AppendChild(&before)
				} else {
					n.InsertBefore(&before, n.Children[0])
				}
			}

			if ps["::after"] != nil {
				after := n.CreateElement("after")
				after.Parent = n
				after.Style("display", "inline")

				for k, v := range ps["::after"] {
					after.Style(k, v)
				}

				after.InnerText = ps["::after"]["content"][1 : len(ps["::after"]["content"])-1]

				n.AppendChild(&after)
			}

			return n
		},
	}
}