Files
rgthree-rgthree-comfy/web/utils_dom.js
T
2023-12-01 23:45:54 -05:00

245 lines
8.0 KiB
JavaScript

const DIRECT_ATTRIBUTE_MAP = {
cellpadding: 'cellPadding',
cellspacing: 'cellSpacing',
colspan: 'colSpan',
frameborder: 'frameBorder',
height: 'height',
maxlength: 'maxLength',
nonce: 'nonce',
role: 'role',
rowspan: 'rowSpan',
type: 'type',
usemap: 'useMap',
valign: 'vAlign',
width: 'width',
};
const RGX_NUMERIC_STYLE_UNIT = 'px';
const RGX_NUMERIC_STYLE = /^((max|min)?(width|height)|margin|padding|(margin|padding)?(left|top|bottom|right)|fontsize|borderwidth)$/i;
const RGX_DEFAULT_VALUE_PROP = /input|textarea|select/i;
function localAssertNotFalsy(input, errorMsg = `Input is not of type.`) {
if (input == null) {
throw new Error(errorMsg);
}
return input;
}
const RGX_STRING_VALID = '[a-z0-9_-]';
const RGX_TAG = new RegExp(`^([a-z]${RGX_STRING_VALID}*)(\\.|\\[|\\#|$)`, 'i');
const RGX_ATTR_ID = new RegExp(`#(${RGX_STRING_VALID}+)`, 'gi');
const RGX_ATTR_CLASS = new RegExp(`(^|\\S)\\.([a-z0-9_\\-\\.]+)`, 'gi');
const RGX_STRING_CONTENT_TO_SQUARES = '(.*?)(\\[|\\])';
const RGX_ATTRS_MAYBE_OPEN = new RegExp(`\\[${RGX_STRING_CONTENT_TO_SQUARES}`, 'gi');
const RGX_ATTRS_FOLLOW_OPEN = new RegExp(`^${RGX_STRING_CONTENT_TO_SQUARES}`, 'gi');
export function createText(text) {
return document.createTextNode(text);
}
export function createElement(selectorOrText, attributes = null) {
let selector = selectorOrText.replace(/[\r\n]\s*/g, '');
let tag = getSelectorTag(selector);
if (!tag) {
tag = 'div';
}
const element = document.createElement(tag);
selector = selector.replace(RGX_TAG, '$2');
selector = selector.replace(RGX_ATTR_ID, '[id="$1"]');
selector = selector.replace(RGX_ATTR_CLASS, (match, p1, p2) => `${p1}[class="${p2.replace(/\./g, ' ')}"]`);
const attrs = getSelectorAttributes(selector);
if (attrs) {
attrs.forEach((attr) => {
let matches = attr.substring(1, attr.length - 1).split('=');
let key = localAssertNotFalsy(matches.shift());
let value = matches.join('=');
if (value === undefined) {
setAttribute(element, key, true);
}
else {
value = value.replace(/^['"](.*)['"]$/, '$1');
setAttribute(element, key, value);
}
});
}
if (attributes) {
setAttributes(element, attributes);
}
return element;
}
function getSelectorTag(str) {
return tryMatch(str, RGX_TAG);
}
function getSelectorAttributes(selector) {
RGX_ATTRS_MAYBE_OPEN.lastIndex = 0;
let attrs = [];
let result;
while (result = RGX_ATTRS_MAYBE_OPEN.exec(selector)) {
let attr = result[0];
if (attr.endsWith(']')) {
attrs.push(attr);
}
else {
attr = result[0]
+ getOpenAttributesRecursive(selector.substr(RGX_ATTRS_MAYBE_OPEN.lastIndex), 2);
RGX_ATTRS_MAYBE_OPEN.lastIndex += (attr.length - result[0].length);
attrs.push(attr);
}
}
return attrs;
}
function getOpenAttributesRecursive(selectorSubstring, openCount) {
let matches = selectorSubstring.match(RGX_ATTRS_FOLLOW_OPEN);
let result = '';
if (matches && matches.length) {
result = matches[0];
openCount += result.endsWith(']') ? -1 : 1;
if (openCount > 0) {
result += getOpenAttributesRecursive(selectorSubstring.substr(result.length), openCount);
}
}
return result;
}
function tryMatch(str, rgx, index = 1) {
var _a;
let found = '';
try {
found = ((_a = str.match(rgx)) === null || _a === void 0 ? void 0 : _a[index]) || '';
}
catch (e) {
found = '';
}
return found;
}
export function setAttributes(element, data) {
let attr;
for (attr in data) {
if (data.hasOwnProperty(attr)) {
setAttribute(element, attr, data[attr]);
}
}
}
function getChild(value) {
if (value instanceof Node) {
return value;
}
if (typeof value === 'string') {
if (value.match(/<.*?>.*?<\/[a-z0-9]+>/)) {
return document.createRange().createContextualFragment(value);
}
if (getSelectorTag(value)) {
return createElement(value);
}
return createText(value);
}
if (value && typeof value.toElement === 'function') {
return value.toElement();
}
return null;
}
export function setAttribute(element, attribute, value) {
const isRemoving = value == null;
if (attribute === 'default') {
attribute = RGX_DEFAULT_VALUE_PROP.test(element.nodeName) ? 'value' : 'text';
}
if (attribute === 'text') {
empty(element).appendChild(createText(value != null ? String(value) : ''));
}
else if (attribute === 'html') {
empty(element).innerHTML += value != null ? String(value) : '';
}
else if (attribute == 'style') {
element.style.cssText = isRemoving ? '' : (value != null ? String(value) : '');
}
else if (attribute == 'events') {
for (const [key, fn] of Object.entries(value)) {
addEvent(element, key, fn);
}
}
else if (attribute === 'parent') {
value.appendChild(element);
}
else if (attribute === 'child' || attribute === 'children') {
if (typeof value === 'string' && /^\[[^\[\]]+\]$/.test(value)) {
const parseable = value.replace(/^\[([^\[\]]+)\]$/, '["$1"]').replace(/,/g, '","');
try {
const parsed = JSON.parse(parseable);
value = parsed;
}
catch (e) {
console.error(e);
}
}
let children = value instanceof Array ? value : [value];
for (let child of children) {
child = getChild(child);
if (child instanceof Node) {
if (element instanceof HTMLTemplateElement) {
element.content.appendChild(child);
}
else {
element.appendChild(child);
}
}
}
}
else if (attribute == 'for') {
element.htmlFor = value != null ? String(value) : '';
if (isRemoving) {
element.removeAttribute('for');
}
}
else if (attribute === 'class' || attribute === 'className' || attribute === 'classes') {
element.className = isRemoving ? '' : Array.isArray(value) ? value.join(' ') : String(value);
}
else if (['checked', 'disabled', 'readonly', 'required', 'selected'].includes(attribute)) {
element[attribute] = !!value;
}
else if (DIRECT_ATTRIBUTE_MAP.hasOwnProperty(attribute)) {
if (isRemoving) {
element.removeAttribute(DIRECT_ATTRIBUTE_MAP[attribute]);
}
else {
element.setAttribute(DIRECT_ATTRIBUTE_MAP[attribute], String(value));
}
}
else if (isRemoving) {
element.removeAttribute(attribute);
}
else {
let oldVal = element.getAttribute(attribute);
if (oldVal !== value) {
element.setAttribute(attribute, String(value));
}
}
}
function addEvent(element, key, fn) {
element.addEventListener(key, fn);
}
function setStyles(element, styles = null) {
if (styles) {
for (let name in styles) {
setStyle(element, name, styles[name]);
}
}
return element;
}
function setStyle(element, name, value) {
name = (name.indexOf('float') > -1 ? 'cssFloat' : name);
if (name.indexOf('-') != -1) {
name = name.replace(/-\D/g, (match) => {
return match.charAt(1).toUpperCase();
});
}
if (value == String(Number(value)) && RGX_NUMERIC_STYLE.test(name)) {
value = value + RGX_NUMERIC_STYLE_UNIT;
}
if (name === 'display' && typeof value !== 'string') {
value = !!value ? null : 'none';
}
element.style[name] = value === null ? null : String(value);
return element;
}
;
function empty(element) {
while (element.firstChild) {
element.removeChild(element.firstChild);
}
return element;
}