const { Plugin } = require('obsidian');
class JsonMdTableConverterPlugin extends Plugin {
async onload() {
// 注册指令
this.addCommand({
id: 'convert-json-md-table',
name: 'JSON/MD Table Converter',
callback: () => {
this.convertToJsonOrMd();
},
});
// 注册右键菜单
this.registerEvent(
this.app.workspace.on('editor-menu', (menu, editor, event) => {
menu.addItem((item) => {
item
.setTitle('JSON/MD Table Converter')
.setIcon('dice')
.onClick(() => {
this.convertToJsonOrMd();
});
});
})
);
}
convertToJsonOrMd() {
const activeLeaf = this.app.workspace.activeLeaf;
if (!activeLeaf) return;
const editor = activeLeaf.view.sourceMode.cmEditor;
const text = editor.getSelection() || editor.getValue();
let result = '';
try {
// 尝试将文本解析为JSON
const json = JSON.parse(text);
if (
Array.isArray(json) &&
json.length > 0 &&
typeof json[0] === 'object'
) {
// 转换为Markdown表格
result = this.jsonToMdTable(json);
} else {
// 如果不是数组或者数组元素不是对象，则提示用户
return new Notice('Invalid JSON: JSON must be an array of objects.');
}
} catch (e) {
const msg = this.isMarkdownTable(text);
if (msg !== 'success') {
return new Notice(msg);
}
// 解析失败，可能是Markdown表格
result = JSON.stringify(this.mdTableToJson(text), null, 2);
}
editor.replaceSelection(result);
}
jsonToMdTable(json) {
const fieldSet = new Set(); // 使用Set来存储不重复的字段
for (let i = 0; i < json.length; i++) {
const elementKeys = Object.keys(json[i]);
for (let j = 0; j < elementKeys.length; j++) {
fieldSet.add(elementKeys[j]); // 将字段添加到Set中
}
}
const headers = Array.from(fieldSet);
const headerRow = '| ' + headers.join(' | ') + ' |';
const separatorRow = '| ' + headers.map(() => ':--:').join(' | ') + ' |';
const bodyRows = json.map(
(obj) =>
'| ' + headers.map((header) => obj[header] || '').join(' | ') + ' |'
);
return [headerRow, separatorRow, ...bodyRows].join('\n');
}
mdTableToJson(mdTable) {
const rows = mdTable.split('\n').filter((row) => row.trim() !== '');
for (let i = 0; i < rows.length; i++) {
rows[i] = rows[i].replace(/^\|+|\|+$/g, '');
}
const headers = rows[0].split('|').map((header) => header.trim());
const jsonArray = rows.slice(2).map((row) => {
const values = row.split('|').map((value) => value.trim());
return headers.reduce((obj, header, index) => {
obj[header] = values[index];
return obj;
}, {});
});
return jsonArray;
}
isMarkdownTable(str) {
// 去除字符串首尾的空白字符，并按行分割
const lines = str.trim().split('\n');
const msg =
'Conversion failed: The selected text is neither valid JSON nor a Markdown table.';
// 检查至少有两行
if (lines.length < 2) {
return msg;
}
// 检查每行是否至少包含一个'|'
for (let line of lines) {
if (line.indexOf('|') === -1) {
return msg;
}
}
let separatorLine = lines[1].replaceAll(' ', '');
// 分隔符行应该由'-'或':'组成，并且由'|'分隔
if (separatorLine.startsWith('|')) { // 只去掉首尾的各一个'|'
separatorLine = separatorLine.slice(1);
}
if (separatorLine.endsWith('|')) {
separatorLine = separatorLine.slice(0, -1);
}
const separators = separatorLine.split('|');
for (let separator of separators) {
separator = separator.replaceAll(' ', '');
if (separator.startsWith(':')) {
separator = separator.slice(1);
}
if (separator.endsWith(':')) {
separator = separator.slice(0, -1);
}
if (!/^-+$/.test(separator)) {
return msg;
}
}
// 检查分隔符行与标题行和内容行的列数是否匹配
for (let line of lines) {
line = line.trim()
if (line.startsWith('|')) {
line = line.slice(1);
}
if (line.endsWith('|')) {
line = line.slice(0, -1);
}
const pipeCount = line.split('|').length;
if (pipeCount !== separators.length) {
return msg;
}
}
if (lines.length == 2) {
return 'Conversion failed: Markdown table should contain at least one row of data.';
}
return 'success';
}
}
module.exports = JsonMdTableConverterPlugin;
