73 lines
1.5 KiB
V
73 lines
1.5 KiB
V
module html
|
|
|
|
import strings
|
|
|
|
enum CloseTagType {
|
|
in_name
|
|
new_tag
|
|
}
|
|
|
|
// Tag holds the information of an HTML tag.
|
|
[ref_only]
|
|
pub struct Tag {
|
|
pub mut:
|
|
name string
|
|
content string
|
|
children []&Tag
|
|
attributes map[string]string // attributes will be like map[name]value
|
|
last_attribute string
|
|
parent &Tag = 0
|
|
position_in_parent int
|
|
closed bool
|
|
close_type CloseTagType = .in_name
|
|
}
|
|
|
|
fn (mut tag Tag) add_parent(t &Tag, position int) {
|
|
tag.position_in_parent = position
|
|
tag.parent = t
|
|
}
|
|
|
|
fn (mut tag Tag) add_child(t &Tag) int {
|
|
tag.children << t
|
|
return tag.children.len
|
|
}
|
|
|
|
// text returns the text contents of the tag.
|
|
pub fn (tag Tag) text() string {
|
|
if tag.name.len >= 2 && tag.name[..2] == 'br' {
|
|
return '\n'
|
|
}
|
|
mut text_str := strings.new_builder(200)
|
|
text_str.write(tag.content.replace('\n', ''))
|
|
for child in tag.children {
|
|
text_str.write(child.text())
|
|
}
|
|
return text_str.str()
|
|
}
|
|
|
|
pub fn (tag &Tag) str() string {
|
|
mut html_str := strings.new_builder(200)
|
|
html_str.write('<$tag.name')
|
|
for key, value in tag.attributes {
|
|
html_str.write(' $key')
|
|
if value.len > 0 {
|
|
html_str.write('="$value"')
|
|
}
|
|
}
|
|
html_str.write(if tag.closed && tag.close_type == .in_name {
|
|
'/>'
|
|
} else {
|
|
'>'
|
|
})
|
|
html_str.write(tag.content)
|
|
if tag.children.len > 0 {
|
|
for child in tag.children {
|
|
html_str.write(child.str())
|
|
}
|
|
}
|
|
if !tag.closed || tag.close_type == .new_tag {
|
|
html_str.write('</$tag.name>')
|
|
}
|
|
return html_str.str()
|
|
}
|