Phase 7 Task 3: 数据安全与合规
- 创建 security_manager.py 安全模块
- SecurityManager: 安全管理主类
- 审计日志系统 - 记录所有数据操作
- 端到端加密 - AES-256-GCM 加密项目数据
- 数据脱敏 - 支持手机号、邮箱、身份证等敏感信息脱敏
- 数据访问策略 - 基于用户、角色、IP、时间的访问控制
- 访问审批流程 - 敏感数据访问需要审批
- 更新 schema.sql 添加安全相关数据库表
- audit_logs: 审计日志表
- encryption_configs: 加密配置表
- masking_rules: 脱敏规则表
- data_access_policies: 数据访问策略表
- access_requests: 访问请求表
- 更新 main.py 添加安全相关 API 端点
- GET /api/v1/audit-logs - 查询审计日志
- GET /api/v1/audit-logs/stats - 审计统计
- POST /api/v1/projects/{id}/encryption/enable - 启用加密
- POST /api/v1/projects/{id}/encryption/disable - 禁用加密
- POST /api/v1/projects/{id}/encryption/verify - 验证密码
- GET /api/v1/projects/{id}/encryption - 获取加密配置
- POST /api/v1/projects/{id}/masking-rules - 创建脱敏规则
- GET /api/v1/projects/{id}/masking-rules - 获取脱敏规则
- PUT /api/v1/masking-rules/{id} - 更新脱敏规则
- DELETE /api/v1/masking-rules/{id} - 删除脱敏规则
- POST /api/v1/projects/{id}/masking/apply - 应用脱敏
- POST /api/v1/projects/{id}/access-policies - 创建访问策略
- GET /api/v1/projects/{id}/access-policies - 获取访问策略
- POST /api/v1/access-policies/{id}/check - 检查访问权限
- POST /api/v1/access-requests - 创建访问请求
- POST /api/v1/access-requests/{id}/approve - 批准访问
- POST /api/v1/access-requests/{id}/reject - 拒绝访问
- 更新 requirements.txt 添加 cryptography 依赖
- 更新 STATUS.md 和 README.md 记录完成状态
This commit is contained in:
@@ -1,217 +1,198 @@
|
||||
// InsightFlow Chrome Extension - Background Script
|
||||
// 处理后台任务、右键菜单、消息传递
|
||||
// 处理扩展的后台逻辑
|
||||
|
||||
// 默认配置
|
||||
const DEFAULT_CONFIG = {
|
||||
serverUrl: 'http://122.51.127.111:18000',
|
||||
apiKey: '',
|
||||
defaultProjectId: ''
|
||||
};
|
||||
|
||||
// 初始化
|
||||
chrome.runtime.onInstalled.addListener(() => {
|
||||
// 创建右键菜单
|
||||
chrome.contextMenus.create({
|
||||
id: 'clipSelection',
|
||||
title: '保存到 InsightFlow',
|
||||
contexts: ['selection', 'page']
|
||||
});
|
||||
|
||||
// 初始化存储
|
||||
chrome.storage.sync.get(['insightflowConfig'], (result) => {
|
||||
if (!result.insightflowConfig) {
|
||||
chrome.storage.sync.set({ insightflowConfig: DEFAULT_CONFIG });
|
||||
}
|
||||
});
|
||||
console.log('[InsightFlow] Extension installed');
|
||||
|
||||
// 创建右键菜单
|
||||
chrome.contextMenus.create({
|
||||
id: 'insightflow-clip-selection',
|
||||
title: 'Clip selection to InsightFlow',
|
||||
contexts: ['selection']
|
||||
});
|
||||
|
||||
chrome.contextMenus.create({
|
||||
id: 'insightflow-clip-page',
|
||||
title: 'Clip page to InsightFlow',
|
||||
contexts: ['page']
|
||||
});
|
||||
|
||||
chrome.contextMenus.create({
|
||||
id: 'insightflow-clip-link',
|
||||
title: 'Clip link to InsightFlow',
|
||||
contexts: ['link']
|
||||
});
|
||||
});
|
||||
|
||||
// 处理右键菜单点击
|
||||
chrome.contextMenus.onClicked.addListener((info, tab) => {
|
||||
if (info.menuItemId === 'clipSelection') {
|
||||
clipPage(tab, info.selectionText);
|
||||
}
|
||||
if (info.menuItemId === 'insightflow-clip-selection') {
|
||||
clipSelection(tab);
|
||||
} else if (info.menuItemId === 'insightflow-clip-page') {
|
||||
clipPage(tab);
|
||||
} else if (info.menuItemId === 'insightflow-clip-link') {
|
||||
clipLink(tab, info.linkUrl);
|
||||
}
|
||||
});
|
||||
|
||||
// 处理扩展图标点击
|
||||
chrome.action.onClicked.addListener((tab) => {
|
||||
clipPage(tab);
|
||||
});
|
||||
|
||||
// 监听来自内容脚本的消息
|
||||
// 处理来自 popup 的消息
|
||||
chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
|
||||
if (request.action === 'clipPage') {
|
||||
clipPage(sender.tab, request.selectionText);
|
||||
sendResponse({ success: true });
|
||||
} else if (request.action === 'getConfig') {
|
||||
chrome.storage.sync.get(['insightflowConfig'], (result) => {
|
||||
sendResponse(result.insightflowConfig || DEFAULT_CONFIG);
|
||||
});
|
||||
return true; // 保持消息通道开放
|
||||
} else if (request.action === 'saveConfig') {
|
||||
chrome.storage.sync.set({ insightflowConfig: request.config }, () => {
|
||||
sendResponse({ success: true });
|
||||
});
|
||||
return true;
|
||||
} else if (request.action === 'fetchProjects') {
|
||||
fetchProjects().then(projects => {
|
||||
sendResponse({ success: true, projects });
|
||||
}).catch(error => {
|
||||
sendResponse({ success: false, error: error.message });
|
||||
});
|
||||
return true;
|
||||
}
|
||||
if (request.action === 'clipPage') {
|
||||
chrome.tabs.query({ active: true, currentWindow: true }, (tabs) => {
|
||||
if (tabs[0]) {
|
||||
clipPage(tabs[0]).then(sendResponse);
|
||||
}
|
||||
});
|
||||
return true;
|
||||
} else if (request.action === 'clipSelection') {
|
||||
chrome.tabs.query({ active: true, currentWindow: true }, (tabs) => {
|
||||
if (tabs[0]) {
|
||||
clipSelection(tabs[0]).then(sendResponse);
|
||||
}
|
||||
});
|
||||
return true;
|
||||
} else if (request.action === 'openClipper') {
|
||||
chrome.action.openPopup();
|
||||
}
|
||||
});
|
||||
|
||||
// 剪藏页面
|
||||
async function clipPage(tab, selectionText = null) {
|
||||
try {
|
||||
// 获取配置
|
||||
const config = await getConfig();
|
||||
|
||||
if (!config.apiKey) {
|
||||
showNotification('请先配置 API Key', '点击扩展图标打开设置');
|
||||
chrome.runtime.openOptionsPage();
|
||||
return;
|
||||
// 剪辑整个页面
|
||||
async function clipPage(tab) {
|
||||
try {
|
||||
// 向 content script 发送消息提取内容
|
||||
const response = await chrome.tabs.sendMessage(tab.id, { action: 'extractContent' });
|
||||
|
||||
if (response.success) {
|
||||
// 保存到本地存储
|
||||
await saveClip(response.data);
|
||||
return { success: true, message: 'Page clipped successfully' };
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('[InsightFlow] Failed to clip page:', error);
|
||||
return { success: false, error: error.message };
|
||||
}
|
||||
}
|
||||
|
||||
// 剪辑选中的内容
|
||||
async function clipSelection(tab) {
|
||||
try {
|
||||
const response = await chrome.tabs.sendMessage(tab.id, { action: 'getSelection' });
|
||||
|
||||
if (response.success && response.data) {
|
||||
const clipData = {
|
||||
url: tab.url,
|
||||
title: tab.title,
|
||||
content: response.data.text,
|
||||
context: response.data.context,
|
||||
contentType: 'selection',
|
||||
extractedAt: new Date().toISOString()
|
||||
};
|
||||
|
||||
await saveClip(clipData);
|
||||
return { success: true, message: 'Selection clipped successfully' };
|
||||
} else {
|
||||
return { success: false, error: 'No text selected' };
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('[InsightFlow] Failed to clip selection:', error);
|
||||
return { success: false, error: error.message };
|
||||
}
|
||||
}
|
||||
|
||||
// 剪辑链接
|
||||
async function clipLink(tab, linkUrl) {
|
||||
const clipData = {
|
||||
url: linkUrl,
|
||||
title: linkUrl,
|
||||
content: `Link: ${linkUrl}`,
|
||||
sourceUrl: tab.url,
|
||||
contentType: 'link',
|
||||
extractedAt: new Date().toISOString()
|
||||
};
|
||||
|
||||
// 获取页面内容
|
||||
const [{ result }] = await chrome.scripting.executeScript({
|
||||
target: { tabId: tab.id },
|
||||
func: extractPageContent,
|
||||
args: [selectionText]
|
||||
await saveClip(clipData);
|
||||
return { success: true, message: 'Link clipped successfully' };
|
||||
}
|
||||
|
||||
// 保存剪辑内容
|
||||
async function saveClip(data) {
|
||||
// 获取现有剪辑
|
||||
const result = await chrome.storage.local.get(['clips']);
|
||||
const clips = result.clips || [];
|
||||
|
||||
// 添加新剪辑
|
||||
clips.unshift({
|
||||
id: generateId(),
|
||||
...data,
|
||||
synced: false
|
||||
});
|
||||
|
||||
// 发送到 InsightFlow
|
||||
const response = await sendToInsightFlow(config, result);
|
||||
|
||||
if (response.success) {
|
||||
showNotification('保存成功', '内容已导入 InsightFlow');
|
||||
} else {
|
||||
showNotification('保存失败', response.error || '未知错误');
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Clip error:', error);
|
||||
showNotification('保存失败', error.message);
|
||||
}
|
||||
}
|
||||
|
||||
// 提取页面内容
|
||||
function extractPageContent(selectionText) {
|
||||
const data = {
|
||||
url: window.location.href,
|
||||
title: document.title,
|
||||
selection: selectionText,
|
||||
timestamp: new Date().toISOString()
|
||||
};
|
||||
|
||||
if (selectionText) {
|
||||
// 只保存选中的文本
|
||||
data.content = selectionText;
|
||||
data.contentType = 'selection';
|
||||
} else {
|
||||
// 保存整个页面
|
||||
// 获取主要内容
|
||||
const article = document.querySelector('article') ||
|
||||
document.querySelector('main') ||
|
||||
document.querySelector('.content') ||
|
||||
document.querySelector('#content');
|
||||
|
||||
if (article) {
|
||||
data.content = article.innerText;
|
||||
data.contentType = 'article';
|
||||
} else {
|
||||
// 获取 body 文本,但移除脚本和样式
|
||||
const bodyClone = document.body.cloneNode(true);
|
||||
const scripts = bodyClone.querySelectorAll('script, style, nav, header, footer, aside');
|
||||
scripts.forEach(el => el.remove());
|
||||
data.content = bodyClone.innerText;
|
||||
data.contentType = 'page';
|
||||
// 只保留最近 100 条
|
||||
if (clips.length > 100) {
|
||||
clips.pop();
|
||||
}
|
||||
|
||||
// 限制内容长度
|
||||
if (data.content.length > 50000) {
|
||||
data.content = data.content.substring(0, 50000) + '...';
|
||||
data.truncated = true;
|
||||
// 保存
|
||||
await chrome.storage.local.set({ clips });
|
||||
|
||||
// 尝试同步到服务器
|
||||
syncToServer();
|
||||
}
|
||||
|
||||
// 同步到服务器
|
||||
async function syncToServer() {
|
||||
const { serverUrl, apiKey } = await chrome.storage.sync.get(['serverUrl', 'apiKey']);
|
||||
|
||||
if (!serverUrl || !apiKey) {
|
||||
console.log('[InsightFlow] Server not configured, skipping sync');
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
// 获取元数据
|
||||
data.meta = {
|
||||
description: document.querySelector('meta[name="description"]')?.content || '',
|
||||
keywords: document.querySelector('meta[name="keywords"]')?.content || '',
|
||||
author: document.querySelector('meta[name="author"]')?.content || ''
|
||||
};
|
||||
|
||||
return data;
|
||||
}
|
||||
|
||||
// 发送到 InsightFlow
|
||||
async function sendToInsightFlow(config, data) {
|
||||
const url = `${config.serverUrl}/api/v1/plugins/chrome/clip`;
|
||||
|
||||
const payload = {
|
||||
url: data.url,
|
||||
title: data.title,
|
||||
content: data.content,
|
||||
content_type: data.contentType,
|
||||
meta: data.meta,
|
||||
project_id: config.defaultProjectId || null
|
||||
};
|
||||
|
||||
const response = await fetch(url, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'X-API-Key': config.apiKey
|
||||
},
|
||||
body: JSON.stringify(payload)
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
const error = await response.text();
|
||||
throw new Error(error);
|
||||
}
|
||||
|
||||
return await response.json();
|
||||
}
|
||||
|
||||
// 获取配置
|
||||
function getConfig() {
|
||||
return new Promise((resolve) => {
|
||||
chrome.storage.sync.get(['insightflowConfig'], (result) => {
|
||||
resolve(result.insightflowConfig || DEFAULT_CONFIG);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 获取项目列表
|
||||
async function fetchProjects() {
|
||||
const config = await getConfig();
|
||||
|
||||
if (!config.apiKey) {
|
||||
throw new Error('请先配置 API Key');
|
||||
}
|
||||
|
||||
const response = await fetch(`${config.serverUrl}/api/v1/projects`, {
|
||||
headers: {
|
||||
'X-API-Key': config.apiKey
|
||||
|
||||
const result = await chrome.storage.local.get(['clips']);
|
||||
const clips = result.clips || [];
|
||||
const unsyncedClips = clips.filter(c => !c.synced);
|
||||
|
||||
if (unsyncedClips.length === 0) return;
|
||||
|
||||
for (const clip of unsyncedClips) {
|
||||
try {
|
||||
const response = await fetch(`${serverUrl}/api/v1/plugins/chrome/import`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'X-API-Key': apiKey
|
||||
},
|
||||
body: JSON.stringify({
|
||||
token: apiKey,
|
||||
url: clip.url,
|
||||
title: clip.title,
|
||||
content: clip.content,
|
||||
html_content: clip.html || null
|
||||
})
|
||||
});
|
||||
|
||||
if (response.ok) {
|
||||
clip.synced = true;
|
||||
clip.syncedAt = new Date().toISOString();
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('[InsightFlow] Sync failed:', error);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error('获取项目列表失败');
|
||||
}
|
||||
|
||||
const data = await response.json();
|
||||
return data.projects || [];
|
||||
|
||||
// 更新存储
|
||||
await chrome.storage.local.set({ clips });
|
||||
}
|
||||
|
||||
// 显示通知
|
||||
function showNotification(title, message) {
|
||||
chrome.notifications.create({
|
||||
type: 'basic',
|
||||
iconUrl: 'icons/icon128.png',
|
||||
title,
|
||||
message
|
||||
});
|
||||
}
|
||||
// 生成唯一ID
|
||||
function generateId() {
|
||||
return Date.now().toString(36) + Math.random().toString(36).substr(2);
|
||||
}
|
||||
|
||||
// 定时同步(每5分钟)
|
||||
chrome.alarms.create('syncClips', { periodInMinutes: 5 });
|
||||
chrome.alarms.onAlarm.addListener((alarm) => {
|
||||
if (alarm.name === 'syncClips') {
|
||||
syncToServer();
|
||||
}
|
||||
});
|
||||
Reference in New Issue
Block a user