[xml] Use io.Writer for xml.print(doc).

This commit is contained in:
Jeroen van Rijn
2021-12-05 02:52:22 +01:00
parent 9b2e67df67
commit 46a4927aca
3 changed files with 36 additions and 25 deletions
+28 -23
View File
@@ -10,64 +10,69 @@ package xml
List of contributors: List of contributors:
Jeroen van Rijn: Initial implementation. Jeroen van Rijn: Initial implementation.
*/ */
import "core:io"
import "core:fmt" import "core:fmt"
/* /*
Just for debug purposes. Just for debug purposes.
*/ */
print :: proc(doc: ^Document) { print :: proc(writer: io.Writer, doc: ^Document) -> (written: int, err: io.Error) {
assert(doc != nil) if doc == nil { return }
using fmt using fmt
println("[XML Prolog]")
written += wprintf(writer, "[XML Prolog]\n")
for attr in doc.prolog { for attr in doc.prolog {
printf("\t%v: %v\n", attr.key, attr.val) written += wprintf(writer, "\t%v: %v\n", attr.key, attr.val)
} }
printf("[Encoding] %v\n", doc.encoding) written += wprintf(writer, "[Encoding] %v\n", doc.encoding)
printf("[DOCTYPE] %v\n", doc.doctype.ident) written += wprintf(writer, "[DOCTYPE] %v\n", doc.doctype.ident)
if len(doc.doctype.rest) > 0 { if len(doc.doctype.rest) > 0 {
printf("\t%v\n", doc.doctype.rest) wprintf(writer, "\t%v\n", doc.doctype.rest)
} }
if doc.root != nil { if doc.root != nil {
println(" --- ") wprintln(writer, " --- ")
print_element(0, doc.root) print_element(writer, doc.root)
println(" --- ") wprintln(writer, " --- ")
} }
return written, .None
} }
print_element :: proc(indent: int, element: ^Element) { print_element :: proc(writer: io.Writer, element: ^Element, indent := 0) -> (written: int, err: io.Error) {
if element == nil { return } if element == nil { return }
using fmt using fmt
tab :: proc(indent: int) { tab :: proc(writer: io.Writer, indent: int) {
tabs := "\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t" tabs := "\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t"
i := max(0, min(indent, len(tabs))) i := max(0, min(indent, len(tabs)))
printf("%v", tabs[:i]) wprintf(writer, "%v", tabs[:i])
} }
tab(indent) tab(writer, indent)
if element.kind == .Element { if element.kind == .Element {
printf("<%v>\n", element.ident) wprintf(writer, "<%v>\n", element.ident)
if len(element.value) > 0 { if len(element.value) > 0 {
tab(indent + 1) tab(writer, indent + 1)
printf("[Value] %v\n", element.value) wprintf(writer, "[Value] %v\n", element.value)
} }
for attr in element.attribs { for attr in element.attribs {
tab(indent + 1) tab(writer, indent + 1)
printf("[Attr] %v: %v\n", attr.key, attr.val) wprintf(writer, "[Attr] %v: %v\n", attr.key, attr.val)
} }
for child in element.children { for child in element.children {
print_element(indent + 1, child) print_element(writer, child, indent + 1)
} }
} else if element.kind == .Comment { } else if element.kind == .Comment {
printf("[COMMENT] %v\n", element.value) wprintf(writer, "[COMMENT] %v\n", element.value)
} }
return written, .None
} }
+7 -1
View File
@@ -2,6 +2,7 @@ package xml_example
import "core:encoding/xml" import "core:encoding/xml"
import "core:mem" import "core:mem"
import "core:strings"
import "core:fmt" import "core:fmt"
Error_Handler :: proc(pos: xml.Pos, fmt: string, args: ..any) { Error_Handler :: proc(pos: xml.Pos, fmt: string, args: ..any) {
@@ -28,7 +29,12 @@ _main :: proc() {
doc, err := xml.parse(DOC, OPTIONS, FILENAME, Error_Handler) doc, err := xml.parse(DOC, OPTIONS, FILENAME, Error_Handler)
defer xml.destroy(doc) defer xml.destroy(doc)
xml.print(doc) buf: strings.Builder
defer strings.destroy_builder(&buf)
w := strings.to_writer(&buf)
xml.print(w, doc)
println(strings.to_string(buf))
if err != .None { if err != .None {
printf("Parse error: %v\n", err) printf("Parse error: %v\n", err)
+1 -1
View File
@@ -75,6 +75,7 @@ Option_Flag :: enum {
*/ */
Decode_SGML_Entities, Decode_SGML_Entities,
} }
Option_Flags :: bit_set[Option_Flag; u8]
Document :: struct { Document :: struct {
root: ^Element, root: ^Element,
@@ -122,7 +123,6 @@ Options :: struct {
flags: Option_Flags, flags: Option_Flags,
expected_doctype: string, expected_doctype: string,
} }
Option_Flags :: bit_set[Option_Flag]
Encoding :: enum { Encoding :: enum {
Unknown, Unknown,