75 lines
1.5 KiB
JavaScript
75 lines
1.5 KiB
JavaScript
|
var sax = require("../lib/sax")
|
||
|
, printer = sax.createStream(false, {lowercasetags:true, trim:true})
|
||
|
, fs = require("fs")
|
||
|
|
||
|
function entity (str) {
|
||
|
return str.replace('"', '"')
|
||
|
}
|
||
|
|
||
|
printer.tabstop = 2
|
||
|
printer.level = 0
|
||
|
printer.indent = function () {
|
||
|
print("\n")
|
||
|
for (var i = this.level; i > 0; i --) {
|
||
|
for (var j = this.tabstop; j > 0; j --) {
|
||
|
print(" ")
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
printer.on("opentag", function (tag) {
|
||
|
this.indent()
|
||
|
this.level ++
|
||
|
print("<"+tag.name)
|
||
|
for (var i in tag.attributes) {
|
||
|
print(" "+i+"=\""+entity(tag.attributes[i])+"\"")
|
||
|
}
|
||
|
print(">")
|
||
|
})
|
||
|
|
||
|
printer.on("text", ontext)
|
||
|
printer.on("doctype", ontext)
|
||
|
function ontext (text) {
|
||
|
this.indent()
|
||
|
print(text)
|
||
|
}
|
||
|
|
||
|
printer.on("closetag", function (tag) {
|
||
|
this.level --
|
||
|
this.indent()
|
||
|
print("</"+tag+">")
|
||
|
})
|
||
|
|
||
|
printer.on("cdata", function (data) {
|
||
|
this.indent()
|
||
|
print("<![CDATA["+data+"]]>")
|
||
|
})
|
||
|
|
||
|
printer.on("comment", function (comment) {
|
||
|
this.indent()
|
||
|
print("<!--"+comment+"-->")
|
||
|
})
|
||
|
|
||
|
printer.on("error", function (error) {
|
||
|
console.error(error)
|
||
|
throw error
|
||
|
})
|
||
|
|
||
|
if (!process.argv[2]) {
|
||
|
throw new Error("Please provide an xml file to prettify\n"+
|
||
|
"TODO: read from stdin or take a file")
|
||
|
}
|
||
|
var xmlfile = require("path").join(process.cwd(), process.argv[2])
|
||
|
var fstr = fs.createReadStream(xmlfile, { encoding: "utf8" })
|
||
|
|
||
|
function print (c) {
|
||
|
if (!process.stdout.write(c)) {
|
||
|
fstr.pause()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
process.stdout.on("drain", function () {
|
||
|
fstr.resume()
|
||
|
})
|
||
|
|
||
|
fstr.pipe(printer)
|