Sam Rawlins
5cd4c0c99a
Change-Id: I2c69742379f7feeb2f37b257e428f058a672f950 Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/423322 Reviewed-by: Brian Wilkerson <brianwilkerson@google.com> Commit-Queue: Samuel Rawlins <srawlins@google.com>
106 lines
2.4 KiB
Dart
106 lines
2.4 KiB
Dart
// Copyright (c) 2022, the Dart project authors. Please see the AUTHORS file
|
|
// for details. All rights reserved. Use of this source code is governed by a
|
|
// BSD-style license that can be found in the LICENSE file.
|
|
|
|
/// A lightweight DOM model.
|
|
library;
|
|
|
|
import 'dart:convert';
|
|
|
|
const _htmlEscape = HtmlEscape(HtmlEscapeMode.element);
|
|
|
|
class Document extends Element {
|
|
static const Set<String> selfClosing = {'br', 'link', 'meta'};
|
|
|
|
Document() : super.tag('');
|
|
|
|
/// Return the full HTML text for the document.
|
|
String get outerHtml {
|
|
var buf = StringBuffer();
|
|
|
|
void emitNode(Node node) {
|
|
if (node is Element) {
|
|
buf.write('<${node.name}');
|
|
for (var attr in node.attributes.keys) {
|
|
buf.write(' $attr="${node.attributes[attr]}"');
|
|
}
|
|
if (node.nodes.length == 1 &&
|
|
node.nodes.first is Text &&
|
|
!(node.nodes.first as Text).text.contains('\n')) {
|
|
buf.write('>');
|
|
var text = node.nodes.first as Text;
|
|
buf.write(_htmlEscape.convert(text.text));
|
|
buf.write('</${node.name}>');
|
|
} else {
|
|
buf.write('>');
|
|
for (var child in node.nodes) {
|
|
emitNode(child);
|
|
}
|
|
if (!selfClosing.contains(node.name)) {
|
|
buf.write('</${node.name}>');
|
|
}
|
|
}
|
|
} else if (node is Text) {
|
|
buf.write(_htmlEscape.convert(node.text));
|
|
} else {
|
|
throw 'unknown node type: $node';
|
|
}
|
|
}
|
|
|
|
buf.write('<!DOCTYPE html>');
|
|
|
|
for (var child in nodes) {
|
|
emitNode(child);
|
|
}
|
|
|
|
buf.writeln();
|
|
|
|
return buf.toString();
|
|
}
|
|
}
|
|
|
|
class Element extends Node {
|
|
final String name;
|
|
|
|
Map<String, String> attributes = {};
|
|
|
|
Element.tag(this.name);
|
|
|
|
List<Element> get children => nodes.whereType<Element>().toList();
|
|
|
|
List<String> get classes {
|
|
var classes = attributes['class'];
|
|
if (classes != null) {
|
|
return classes.split(' ').toList();
|
|
} else {
|
|
return const [];
|
|
}
|
|
}
|
|
|
|
void append(Node child) {
|
|
child.parent = this;
|
|
nodes.add(child);
|
|
}
|
|
}
|
|
|
|
abstract class Node {
|
|
Element? parent;
|
|
|
|
final List<Node> nodes = [];
|
|
}
|
|
|
|
class Text extends Node {
|
|
static final RegExp tagRegex = RegExp(r'<[^>]+>');
|
|
|
|
final String text;
|
|
|
|
Text(this.text);
|
|
|
|
@override
|
|
List<Node> get nodes => const [];
|
|
|
|
String get textRemoveTags {
|
|
return text.replaceAll(tagRegex, '');
|
|
}
|
|
}
|