[js] 给导入的日记补上属性

/**
 * @description 使用 fetch 直接请求 API，查找内容为 YYYY-MM-DD 格式且类型为 'd' 的块，并为其添加 custom-YYYYMMDD 属性。消息显示块内容。
 */
async function addDateAttributesToDailyNoteBlocksDirect() {
    // --- 配置 ---
    const SIYUAN_API_URL = `${window.location.protocol}//${window.location.hostname}:${window.location.port}`;
    const API_TOKEN = typeof window.siyuan?.config?.apiToken !== 'undefined' ? window.siyuan.config.apiToken : "YOUR_API_TOKEN";


    const headers = {
        'Authorization': `Token ${API_TOKEN}`,
        'Content-Type': 'application/json',
    };

    // --- 封装的 API 请求函数 ---
    async function fetchAPI(endpoint, data) {
        try {
            const response = await fetch(`${SIYUAN_API_URL}${endpoint}`, {
                method: 'POST',
                headers: headers,
                body: JSON.stringify(data),
            });
            const result = await response.json();
            if (!response.ok) {
                throw new Error(result.msg || `HTTP 错误! 状态码: ${response.status}`);
            }
            if (result.code !== 0) {
                 throw new Error(result.msg || `API 错误码: ${result.code}`);
            }
            return result.data;
        } catch (error) {
            console.error(`调用 API ${endpoint} 失败:`, error);
            throw error;
        }
    }

    // --- 封装的通知函数 ---
    async function pushNotification(message, isError = false, timeout = 5000) {
        const endpoint = isError ? '/api/notification/pushErrMsg' : '/api/notification/pushMsg';
        try {
            await fetchAPI(endpoint, { msg: message, timeout: timeout });
        } catch (notificationError) {
            console.warn(`发送通知失败: ${notificationError.message}`);
            if (isError) {
                alert(`错误: ${message}`);
            } else {
                alert(message);
            }
        }
    }

    // --- 主要逻辑开始 ---
    console.log("开始查找并处理日记块属性...");
    await pushNotification("开始查找并处理日记块属性...", false, 3000);

    // 1. 查找块
    const query = "SELECT id, content FROM blocks WHERE type = 'd' AND content LIKE '____-__-__'";
    let blocks = [];
    try {
        blocks = await fetchAPI('/api/query/sql', { stmt: query });
    } catch (error) {
        pushNotification(`查找日期块时出错: ${error.message}`, true);
        return;
    }

    if (!blocks || blocks.length === 0) {
        console.log("未找到符合条件的日期块。");
        pushNotification("未找到符合条件的日期块。", false);
        return;
    }

    console.log(`找到 ${blocks.length} 个候选块，开始精确匹配和处理...`);

    // 2. 准备统计和正则
    let updatedCount = 0;
    const errors = [];
    const dateRegex = /^\d{4}-\d{2}-\d{2}$/;

    // 3. 遍历处理
    for (const block of blocks) {
        const { id, content } = block; // 同时获取 id 和 content

        // 精确匹配格式
        if (!dateRegex.test(content)) {
            console.log(`块内容 "${content}" (ID: ${id}) 非标准 YYYY-MM-DD，跳过。`); // 保留ID供调试
            continue;
        }

        try {
            const dateValue = content.replace(/-/g, '');
            const attrName = `custom-dailynote-${dateValue}`;
            const attrs = { [attrName]: dateValue };

            await fetchAPI('/api/attr/setBlockAttrs', { id: id, attrs: attrs }); // 设置属性仍需 ID
            updatedCount++;
            // 修改日志：显示 content
            console.log(`成功为块 "${content}" 添加属性 ${attrName}=${dateValue}`);
            // 可选延时
            // await new Promise(resolve => setTimeout(resolve, 30));
        } catch (error) {
            // 修改日志和错误记录：显示 content
            console.error(`为块 "${content}" 添加属性时出错:`, error);
            errors.push({ id: id, content: content, error: error.message || error }); // 错误记录中保留 id 和 content
            // 暂时不在这里为每个错误发送通知，汇总后发送
        }
    }

    // 4. 最终报告
    const successMessage = `处理完成。共检查 ${blocks.length} 个候选块，成功更新 ${updatedCount} 个块。`;
    console.log(successMessage);
    pushNotification(successMessage, false);

    if (errors.length > 0) {
        // 修改错误报告：指出是哪些日期的块出错了
        const errorContents = errors.map(e => e.content).join(', ');
        const errorMessage = `有 ${errors.length} 个块在更新属性时出错 (例如: ${errorContents.substring(0, 50)}${errorContents.length > 50 ? '...' : ''})。详情请见控制台。`;
        console.error("出错的块详情:", errors);
        pushNotification(errorMessage, true, 8000);
    }
}

// --- 创建界面按钮 --- (这部分代码不变)
function createProcessButton() {
    const buttonId = 'add-daily-note-date-attrs-button';
    if (document.getElementById(buttonId)) {
        console.log("处理按钮已存在。");
        return;
    }

    const button = document.createElement('button');
    button.id = buttonId;
    button.textContent = '为日记块加日期属性';
    button.style.position = 'fixed';
    button.style.top = '70px';
    button.style.right = '20px';
    button.style.zIndex = '1001';
    button.style.padding = '8px 12px';
    button.style.border = 'none';
    button.style.backgroundColor = '#2d7ff9';
    button.style.color = 'white';
    button.style.borderRadius = '5px';
    button.style.cursor = 'pointer';
    button.style.boxShadow = '0 2px 5px rgba(0,0,0,0.2)';
    button.title = '点击处理所有内容为 YYYY-MM-DD 的文档块';

    button.onclick = async () => {
        if (!confirm("确定要为所有符合条件的日记块添加日期属性吗？这个操作会修改块数据。")) {
            return;
        }
        button.disabled = true;
        button.textContent = '处理中...';
        button.style.backgroundColor = '#aaa';

        try {
            await addDateAttributesToDailyNoteBlocksDirect();
        } catch (e) {
            console.error("执行过程中发生意外错误:", e);
            alert(`发生意外错误: ${e.message}`);
             try {
                await fetch(`${window.location.protocol}//${window.location.hostname}:${window.location.port}/api/notification/pushErrMsg`, {
                     method: 'POST',
                     headers: {'Content-Type': 'application/json', 'Authorization': `Token ${window.siyuan?.config?.apiToken || 'YOUR_API_TOKEN'}`},
                     body: JSON.stringify({ msg: `执行意外出错: ${e.message}`, timeout: 7000 }),
                });
             } catch(notifyErr) {}
        } finally {
            button.disabled = false;
            button.textContent = '为日记块加日期属性';
            button.style.backgroundColor = '#2d7ff9';
        }
    };

    document.body.appendChild(button);
    console.log("添加日期属性按钮已创建并添加到页面右上角。");
}

// --- 初始化 --- (这部分代码不变)
setTimeout(() => {
    if (typeof window.siyuan?.config !== 'undefined') {
        createProcessButton();
    } else {
        console.warn("似乎不在思源环境中，按钮未创建。");
    }
}, 1500);

​