57 lines
1.6 KiB
TypeScript
57 lines
1.6 KiB
TypeScript
import yaml from 'yaml';
|
|
import { sortKeys } from './helpers';
|
|
import { YAMLMap, YAMLSeq, Pair, Node } from 'yaml/types';
|
|
|
|
function addComments(items: Array<Pair>, comments: Record<string, string>, prefix = '') {
|
|
items.forEach(item => {
|
|
if (item.key !== undefined) {
|
|
const itemKey = item.key.toString();
|
|
const key = prefix ? `${prefix}.${itemKey}` : itemKey;
|
|
if (comments[key]) {
|
|
const value = comments[key].split('\\n').join('\n ');
|
|
item.commentBefore = ` ${value}`;
|
|
}
|
|
if (Array.isArray(item.value?.items)) {
|
|
addComments(item.value.items, comments, key);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
const timestampTag = {
|
|
identify: (value: unknown) => value instanceof Date,
|
|
default: true,
|
|
tag: '!timestamp',
|
|
test: RegExp(
|
|
'^' +
|
|
'([0-9]{4})-([0-9]{2})-([0-9]{2})' + // YYYY-MM-DD
|
|
'T' + // T
|
|
'([0-9]{2}):([0-9]{2}):([0-9]{2}(\\.[0-9]+)?)' + // HH:MM:SS(.ss)?
|
|
'Z' + // Z
|
|
'$',
|
|
),
|
|
resolve: (str: string) => new Date(str),
|
|
stringify: (value: Node) => (value as Date).toISOString(),
|
|
} as const;
|
|
|
|
export default {
|
|
fromFile(content: string) {
|
|
if (content && content.trim().endsWith('---')) {
|
|
content = content.trim().slice(0, -3);
|
|
}
|
|
return yaml.parse(content, { customTags: [timestampTag] });
|
|
},
|
|
|
|
toFile(data: object, sortedKeys: string[] = [], comments: Record<string, string> = {}) {
|
|
const contents = yaml.createNode(data) as YAMLMap | YAMLSeq;
|
|
|
|
addComments(contents.items, comments);
|
|
|
|
contents.items.sort(sortKeys(sortedKeys, item => item.key?.toString()));
|
|
const doc = new yaml.Document();
|
|
doc.contents = contents;
|
|
|
|
return doc.toString();
|
|
},
|
|
};
|