mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-16 14:31:16 +00:00
26 lines
705 B
TypeScript
26 lines
705 B
TypeScript
|
import type {Pair} from 'yaml';
|
||
|
|
import {parseDocument, visit} from 'yaml';
|
||
|
|
|
||
|
|
const fs = require('fs');
|
||
|
|
/**
|
||
|
|
* Modify yaml file https://eemeli.org/yaml/#modifying-nodes
|
||
|
|
* @param filePath
|
||
|
|
* @param keyObj
|
||
|
|
*/
|
||
|
|
export function modifyYaml(filePath: string, keyObj: {[k in string]: any}) {
|
||
|
|
try {
|
||
|
|
const doc = parseDocument(fs.readFileSync(filePath, 'utf8'));
|
||
|
|
const keysMap = new Map(Object.entries(keyObj));
|
||
|
|
visit(doc, {
|
||
|
|
Pair: (_node, pair: Pair<any, any>) => {
|
||
|
|
if (keysMap.has(pair?.key?.value)) {
|
||
|
|
pair.value.value = keysMap.get(pair.key.value);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
});
|
||
|
|
fs.writeFileSync(filePath, doc.toString(), 'utf8');
|
||
|
|
} catch (e) {
|
||
|
|
console.error(e);
|
||
|
|
}
|
||
|
|
}
|