dan/node-html-parser
1
0
Fork 0
mirror of https://github.com/danbulant/node-html-parser synced 2026-05-19 04:18:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #14 from Sharcoux/master

Browse source 
Consider attributes modification in toString
This commit is contained in:
taoqf 2020-01-30 09:49:57 +08:00 committed by GitHub
commit 1ca18d01b2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 79 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
src/index.ts
View file

@ -1,4 +1,4 @@
import { decode } from 'he';
import { decode, encode } from 'he';
export enum NodeType {
ELEMENT_NODE = 1,
@ -504,6 +504,44 @@ export class HTMLElement extends Node {
this._rawAttrs = attrs;
return attrs;
}
/**
* Set an attribute value to the HTMLElement
* @param {string} key The attribute name
* @param {string} value The value to set, or null / undefined to remove an attribute
*/
setAttribute(key: string, value: string) {
//Update the attributes map
const attrs = this.attributes;
if(value===undefined || value===null) delete attrs[key];
else attrs[key] = value+'';
//Update the raw attributes
if(this._rawAttrs) {
if(value===undefined || value===null) delete this._rawAttrs[key];
else this._rawAttrs[key] = encode(value+'');
}
//Update rawString
this.rawAttrs = Object.keys(attrs).map(attr => attr+'='+encode(attrs[attr])).join(' ');
}
/**
* Replace all the attributes of the HTMLElement by the provided attributes
* @param {Attributes} attributes the new attribute set
*/
setAttributes(attributes: Attributes) {
//Update the attributes map
if(this.attributes) {
Object.keys(this.attributes).forEach(key => delete this.attributes[key]);
Object.keys(attributes).forEach(key => this.attributes[key] = attributes[key]+'');
}
//Update the raw attributes map
if(this.rawAttributes) {
Object.keys(this.rawAttributes).forEach(key => delete this.rawAttributes[key]);
Object.keys(attributes).forEach(key => this.rawAttributes[key] = encode(attributes[key]+''));
}
//Update rawString
this.rawAttrs = Object.keys(attributes).map(attr => attr+'='+encode(attributes[attr]+'')).join(' ');
}
}
interface MatherFunction { func: any; tagName: string; classes: string | string[]; attr_key: any; value: any; }

40
test/html.js
View file

@ -321,6 +321,46 @@ describe('HTML Parser', function () {
});
});
describe('#setAttribute', function () {
it('should edit the attributes of the element', function () {
var root = parseHTML('<p a=12></p>');
root.firstChild.setAttribute('a', 13);
root.firstChild.attributes.should.eql({
'a': '13',
});
root.firstChild.toString().should.eql('<p a=13></p>');
});
it('should add an attribute to the element', function () {
var root = parseHTML('<p a=12></p>');
root.firstChild.setAttribute('b', 13);
root.firstChild.attributes.should.eql({
'a': '12',
'b': '13',
});
root.firstChild.toString().should.eql('<p a=12 b=13></p>');
});
it('should remove an attribute from the element', function () {
var root = parseHTML('<p a=12 b=13 c=14></p>');
root.firstChild.setAttribute('b', null);
root.firstChild.setAttribute('c');
root.firstChild.attributes.should.eql({
'a': '12',
});
root.firstChild.toString().should.eql('<p a=12></p>');
});
});
describe('#setAttributes', function () {
it('should return attributes of the element', function () {
var root = parseHTML('<p a=12 data-id="!$$&amp;" yAz=\'1\' class="" disabled></p>');
root.firstChild.setAttributes({c: 12});
root.firstChild.attributes.should.eql({
'c': '12',
});
root.firstChild.toString().should.eql('<p c=12></p>');
});
});
describe('#querySelector()', function () {
it('should return correct elements in DOM tree', function () {
var root = parseHTML('<a id="id" data-id="myid"><div><span class="a b"></span><span></span><span></span></div></a>');