2017-01-14 09:48:52 -05:00

53 lines
1.1 KiB
JavaScript

// This file contains methods responsible for dealing with comments.
/**
* Share comments amongst siblings.
*/
export function shareCommentsWithSiblings() {
// NOTE: this assumes numbered keys
if (typeof this.key === "string") return;
const node = this.node;
if (!node) return;
const trailing = node.trailingComments;
const leading = node.leadingComments;
if (!trailing && !leading) return;
let prev = this.getSibling(this.key - 1);
let next = this.getSibling(this.key + 1);
if (!prev.node) prev = next;
if (!next.node) next = prev;
prev.addComments("trailing", leading);
next.addComments("leading", trailing);
}
export function addComment(type, content, line?) {
this.addComments(type, [{
type: line ? "CommentLine" : "CommentBlock",
value: content
}]);
}
/**
* Give node `comments` of the specified `type`.
*/
export function addComments(type: string, comments: Array) {
if (!comments) return;
const node = this.node;
if (!node) return;
const key = `${type}Comments`;
if (node[key]) {
node[key] = node[key].concat(comments);
} else {
node[key] = comments;
}
}