mirror of
https://github.com/Instadapp/Swap-Aggregator-Subgraph.git
synced 2024-07-29 21:57:12 +00:00
98 lines
2.6 KiB
JavaScript
98 lines
2.6 KiB
JavaScript
"use strict";
|
|
|
|
exports.findPair = findPair;
|
|
exports.YAMLMap = void 0;
|
|
|
|
var _Collection = require("./Collection");
|
|
|
|
var _Pair = require("./Pair");
|
|
|
|
var _Scalar = require("./Scalar");
|
|
|
|
function findPair(items, key) {
|
|
const k = key instanceof _Scalar.Scalar ? key.value : key;
|
|
|
|
for (const it of items) {
|
|
if (it instanceof _Pair.Pair) {
|
|
if (it.key === key || it.key === k) return it;
|
|
if (it.key && it.key.value === k) return it;
|
|
}
|
|
}
|
|
|
|
return undefined;
|
|
}
|
|
|
|
class YAMLMap extends _Collection.Collection {
|
|
add(pair, overwrite) {
|
|
if (!pair) pair = new _Pair.Pair(pair);else if (!(pair instanceof _Pair.Pair)) pair = new _Pair.Pair(pair.key || pair, pair.value);
|
|
const prev = findPair(this.items, pair.key);
|
|
const sortEntries = this.schema && this.schema.sortMapEntries;
|
|
|
|
if (prev) {
|
|
if (overwrite) prev.value = pair.value;else throw new Error(`Key ${pair.key} already set`);
|
|
} else if (sortEntries) {
|
|
const i = this.items.findIndex(item => sortEntries(pair, item) < 0);
|
|
if (i === -1) this.items.push(pair);else this.items.splice(i, 0, pair);
|
|
} else {
|
|
this.items.push(pair);
|
|
}
|
|
}
|
|
|
|
delete(key) {
|
|
const it = findPair(this.items, key);
|
|
if (!it) return false;
|
|
const del = this.items.splice(this.items.indexOf(it), 1);
|
|
return del.length > 0;
|
|
}
|
|
|
|
get(key, keepScalar) {
|
|
const it = findPair(this.items, key);
|
|
const node = it && it.value;
|
|
return !keepScalar && node instanceof _Scalar.Scalar ? node.value : node;
|
|
}
|
|
|
|
has(key) {
|
|
return !!findPair(this.items, key);
|
|
}
|
|
|
|
set(key, value) {
|
|
this.add(new _Pair.Pair(key, value), true);
|
|
}
|
|
/**
|
|
* @param {*} arg ignored
|
|
* @param {*} ctx Conversion context, originally set in Document#toJSON()
|
|
* @param {Class} Type If set, forces the returned collection type
|
|
* @returns {*} Instance of Type, Map, or Object
|
|
*/
|
|
|
|
|
|
toJSON(_, ctx, Type) {
|
|
const map = Type ? new Type() : ctx && ctx.mapAsMap ? new Map() : {};
|
|
if (ctx && ctx.onCreate) ctx.onCreate(map);
|
|
|
|
for (const item of this.items) item.addToJSMap(ctx, map);
|
|
|
|
return map;
|
|
}
|
|
|
|
toString(ctx, onComment, onChompKeep) {
|
|
if (!ctx) return JSON.stringify(this);
|
|
|
|
for (const item of this.items) {
|
|
if (!(item instanceof _Pair.Pair)) throw new Error(`Map items must all be pairs; found ${JSON.stringify(item)} instead`);
|
|
}
|
|
|
|
return super.toString(ctx, {
|
|
blockItem: n => n.str,
|
|
flowChars: {
|
|
start: '{',
|
|
end: '}'
|
|
},
|
|
isMap: true,
|
|
itemIndent: ctx.indent || ''
|
|
}, onComment, onChompKeep);
|
|
}
|
|
|
|
}
|
|
|
|
exports.YAMLMap = YAMLMap; |