126 lines
3.1 KiB
JavaScript
126 lines
3.1 KiB
JavaScript
|
/**
|
|||
|
|
* Notion API 服务封装
|
|||
|
|
*
|
|||
|
|
* 所有 Notion 数据库读写操作走这里，带内存缓存。
|
|||
|
|
* 版权：国作登字-2026-A-00037559
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
'use strict';
|
|||
|
|
|
|||
|
|
var Client;
|
|||
|
|
try {
|
|||
|
|
Client = require('@notionhq/client').Client;
|
|||
|
|
} catch (_) {
|
|||
|
|
// 如果 @notionhq/client 未安装，提供降级方案
|
|||
|
|
Client = null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var notion = Client ? new Client({ auth: process.env.NOTION_TOKEN }) : null;
|
|||
|
|
|
|||
|
|
// ====== 内存缓存（5分钟有效）======
|
|||
|
|
var cache = new Map();
|
|||
|
|
var CACHE_TTL = 5 * 60 * 1000;
|
|||
|
|
|
|||
|
|
function getCached(key) {
|
|||
|
|
var item = cache.get(key);
|
|||
|
|
if (item && Date.now() - item.time < CACHE_TTL) return item.data;
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function setCache(key, data) {
|
|||
|
|
cache.set(key, { data: data, time: Date.now() });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function clearCache() {
|
|||
|
|
cache.clear();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ====== 查询数据库 ======
|
|||
|
|
async function queryDB(dbId, filter, sorts, pageSize) {
|
|||
|
|
if (!notion) {
|
|||
|
|
throw new Error('Notion client not initialized (NOTION_TOKEN missing or @notionhq/client not installed)');
|
|||
|
|
}
|
|||
|
|
if (!dbId) {
|
|||
|
|
throw new Error('Database ID not provided');
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
pageSize = pageSize || 50;
|
|||
|
|
var cacheKey = JSON.stringify({ dbId: dbId, filter: filter, sorts: sorts, pageSize: pageSize });
|
|||
|
|
var cached = getCached(cacheKey);
|
|||
|
|
if (cached) return cached;
|
|||
|
|
|
|||
|
|
var body = { database_id: dbId, page_size: pageSize };
|
|||
|
|
if (filter) body.filter = filter;
|
|||
|
|
if (sorts) body.sorts = sorts;
|
|||
|
|
|
|||
|
|
var result = await notion.databases.query(body);
|
|||
|
|
setCache(cacheKey, result);
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ====== 写入数据库 ======
|
|||
|
|
async function writeToDB(dbId, properties) {
|
|||
|
|
if (!notion) {
|
|||
|
|
throw new Error('Notion client not initialized');
|
|||
|
|
}
|
|||
|
|
return notion.pages.create({
|
|||
|
|
parent: { database_id: dbId },
|
|||
|
|
properties: properties
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ====== Notion 属性提取器 ======
|
|||
|
|
function extractRichText(prop) {
|
|||
|
|
if (!prop) return '';
|
|||
|
|
if (prop.type === 'rich_text' && prop.rich_text) {
|
|||
|
|
return prop.rich_text.map(function(t) { return t.plain_text; }).join('');
|
|||
|
|
}
|
|||
|
|
if (prop.type === 'title' && prop.title) {
|
|||
|
|
return prop.title.map(function(t) { return t.plain_text; }).join('');
|
|||
|
|
}
|
|||
|
|
return '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function extractTitle(props) {
|
|||
|
|
for (var key in props) {
|
|||
|
|
var val = props[key];
|
|||
|
|
if (val && val.type === 'title' && val.title) {
|
|||
|
|
return val.title.map(function(t) { return t.plain_text; }).join('');
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function extractNumber(prop) {
|
|||
|
|
if (!prop || prop.type !== 'number') return null;
|
|||
|
|
return prop.number;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function extractSelect(prop) {
|
|||
|
|
if (!prop || prop.type !== 'select') return '';
|
|||
|
|
return prop.select ? prop.select.name : '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function extractMultiSelect(prop) {
|
|||
|
|
if (!prop || prop.type !== 'multi_select') return [];
|
|||
|
|
return (prop.multi_select || []).map(function(s) { return s.name; });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function extractDate(prop) {
|
|||
|
|
if (!prop || prop.type !== 'date') return '';
|
|||
|
|
return prop.date ? prop.date.start : '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = {
|
|||
|
|
notion: notion,
|
|||
|
|
queryDB: queryDB,
|
|||
|
|
writeToDB: writeToDB,
|
|||
|
|
clearCache: clearCache,
|
|||
|
|
extractRichText: extractRichText,
|
|||
|
|
extractTitle: extractTitle,
|
|||
|
|
extractNumber: extractNumber,
|
|||
|
|
extractSelect: extractSelect,
|
|||
|
|
extractMultiSelect: extractMultiSelect,
|
|||
|
|
extractDate: extractDate
|
|||
|
|
};
|