render.go

package main

import (
	"strings"
)

// 渲染block type的json结构
func RenderBlock(n *NodeTree, child string, versionName string, pageUID string) string {
	pageInfo := []AttrStringStruct{{
		Key: "version",
		Value: versionName,
	}}

	pageInfo = append(pageInfo, AttrStringStruct{
		Key: "pageUID",
		Value: pageUID,
	})

	attrWithType := []AttrStringStruct{{
		Key:   "type",
		Value: n.Type,
	}}

	attrWithData := []AttrInterfaceStruct{{
		Key:   "data",
		Value: n.Data,
	}}

	attrWithKey := []AttrInterfaceStruct{{
		Key:   "key",
		Value: n.Key,
	}}

	return H("Block", child, AttrString(pageInfo)+AttrString(attrWithType)+AttrInterface(attrWithData)+AttrInterface(attrWithKey))
}

// 渲染document type的json结构
func RenderDocument(n *NodeTree, child string) string {
	attr := []AttrStringStruct{{
		Key:   "type",
		Value: n.Type,
	}}
	attrWithKey := []AttrInterfaceStruct{{
		Key:   "key",
		Value: n.Key,
	}}
	return H("Document", child, AttrString(attr)+AttrInterface(attrWithKey))
}

// 渲染inline type的json结构
func RenderInline(n *NodeTree, child string) string {
	typeAttr := []AttrStringStruct{{
		Key:   "type",
		Value: n.Type,
	}}

	attrWithKey := []AttrInterfaceStruct{{
		Key:   "key",
		Value: n.Key,
	}}

	text := ""

	if (n.Data != Data{}) {
		attr := []AttrInterfaceStruct{{
			Key:   "data",
			Value: n.Data,
		}}
		text += H("Inline", child, AttrString(typeAttr)+AttrInterface(attr)+AttrInterface(attrWithKey))
	} else {
		text += H("Inline", child, AttrString(typeAttr)+AttrInterface(attrWithKey))
	}
	return text
}

// 渲染mark type的json结构
func RenderMark(n *NodeTree, child string) string {
	attrWithKey := []AttrInterfaceStruct{{
		Key:   "key",
		Value: n.Key,
	}}
	return H("Mark", child, ""+AttrInterface(attrWithKey))
}

// 渲染text type的json结构
func RenderText(n *NodeTree, child string) string {
	text := ""
	if len(n.Ranges) > 0 {
		for i := range n.Ranges {
			attrWithKey := []AttrInterfaceStruct{{
				Key:   "key",
				Value: n.Key + string(i),
			}}
			content := n.Ranges[i].Text

			if strings.ContainsAny(content, "{&}&<&>&`&\n") {
				content = transfer(content, '{', '}', '<', '>', '`')
			}

			if len(n.Ranges[i].Marks) > 0 {
				attr := []AttrInterfaceStruct{{
					Key:   "marks",
					Value: n.Ranges[i].Marks,
				}}
				text += H("Text", content, AttrInterface(attr)+AttrInterface(attrWithKey))
			} else {
				text += H("Text", content, ""+AttrInterface(attrWithKey))
			}
		}
	}

	return text
}

// 解析element tag的props部分，
func transfer(s string, char ...byte) string {
	for i := 0; i < len(s); i++ {
		for _, c := range char {
			if s[i] == c {
				s = s[:i] + `{"` + string(c) + `"}` + s[i+1:]
				i += 4
				break
			}
			if s[i] == '\n' {
				if i == len(s)-1 && s != string('\n') {
					s = s[:i] + `<span style={{display: "block"}}><br/></span>` + s[i+1:]
					return s
				} else {
					s = s[:i] + "<br/>" + s[i+1:]
					i += 4
				}
				break
			}
		}
	}
	return s
}