131 lines
4.1 KiB
TypeScript
131 lines
4.1 KiB
TypeScript
import './style.css';
|
|
import type { PinData } from '../../components/pin-collector';
|
|
import { startBatchDownload } from '../background';
|
|
|
|
document.querySelector<HTMLDivElement>('#app')!.innerHTML = `
|
|
<div class="popup">
|
|
<h2>🎨 画板收集器</h2>
|
|
<div class="actions">
|
|
<button id="collectBtn">📥 收集</button>
|
|
<button id="clearBtn">🗑 清空</button>
|
|
<button id="start-btn">▶ 开始加载</button>
|
|
<button id="stop-btn">⏹ 停止</button>
|
|
</div>
|
|
<div id="status"></div>
|
|
<div id="pinList"></div>
|
|
</div>
|
|
`;
|
|
|
|
// 按钮事件
|
|
document.getElementById('collectBtn')!.addEventListener('click', async () => {
|
|
showStatus('收集中...', 'info');
|
|
try {
|
|
await loadAndRender();
|
|
|
|
// 下载
|
|
const result = (await browser.storage.local.get('collectedPins')) as {
|
|
collectedPins?: PinData[];
|
|
};
|
|
const pins: PinData[] = result.collectedPins || [];
|
|
if (pins.length > 0) {
|
|
showStatus(`开始下载 ${pins.length} 张...`, 'info');
|
|
browser.runtime.sendMessage({ type: 'DOWNLOAD_PINS', pins, concurrency: 3 });
|
|
} else {
|
|
showStatus('没有可下载的图片', 'error');
|
|
}
|
|
} catch (e) {
|
|
const errMsg = e instanceof Error ? e.message : String(e);
|
|
showStatus(`错误: ${errMsg}`, 'error');
|
|
}
|
|
});
|
|
|
|
document.getElementById('clearBtn')!.addEventListener('click', async () => {
|
|
await browser.storage.local.remove('collectedPins');
|
|
renderPins([]);
|
|
showStatus('已清空', 'info');
|
|
});
|
|
|
|
// 首次打开 popup 时加载并渲染
|
|
loadAndRender();
|
|
|
|
// ---- 辅助函数 ----
|
|
|
|
async function loadAndRender(): Promise<void> {
|
|
const result = (await browser.storage.local.get('collectedPins')) as {
|
|
collectedPins?: PinData[];
|
|
};
|
|
const pins: PinData[] = result.collectedPins || [];
|
|
renderPins(pins);
|
|
}
|
|
|
|
|
|
function renderPins(pins: PinData[]): void {
|
|
const listEl = document.getElementById('pinList')!;
|
|
if (pins.length === 0) {
|
|
listEl.innerHTML = '<p class="empty">暂无收集的图片</p>';
|
|
return;
|
|
}
|
|
listEl.innerHTML = pins
|
|
.map(
|
|
(pin) => `
|
|
<div class="pin-item">
|
|
<img src="${pin.imgSmallSrc}" loading="lazy" alt="${escapeHtml(pin.alt)}" />
|
|
<div class="pin-info">
|
|
<div class="pin-author">${escapeHtml(pin.author || '未知作者')} · ${escapeHtml(pin.time || '')}</div>
|
|
<div class="pin-tags">${pin.tags.map(t => `#${escapeHtml(t)}`).join(' ')}</div>
|
|
<a class="pin-link" href="https://huaban.com${pin.url}" target="_blank">查看详情</a>
|
|
<a class="pin-link" href="${escapeHtml(pin.imgSrc)}" target="_blank">查看原图</a>
|
|
</div>
|
|
</div>
|
|
`
|
|
)
|
|
.join('');
|
|
}
|
|
|
|
function showStatus(msg: string, type: 'info' | 'success' | 'error'): void {
|
|
const statusEl = document.getElementById('status')!;
|
|
statusEl.textContent = msg;
|
|
statusEl.className = `status ${type}`;
|
|
setTimeout(() => (statusEl.textContent = ''), 8000);
|
|
}
|
|
|
|
function escapeHtml(str: string): string {
|
|
const div = document.createElement('div');
|
|
div.textContent = str;
|
|
return div.innerHTML;
|
|
}
|
|
|
|
let activeTabId: number | null = null;
|
|
// 获取当前标签页 ID
|
|
browser.tabs.query({ active: true, currentWindow: true }, ([tab]) => {
|
|
activeTabId = tab.id!;
|
|
document.getElementById('start-btn')?.addEventListener('click', () => {
|
|
browser.tabs.sendMessage(activeTabId!, { action: 'start' });
|
|
});
|
|
document.getElementById('stop-btn')?.addEventListener('click', () => {
|
|
browser.tabs.sendMessage(activeTabId!, { action: 'stop' });
|
|
});
|
|
});
|
|
// 接收进度回调
|
|
browser.runtime.onMessage.addListener(async (msg) => {
|
|
if (msg.type === 'progress') {
|
|
if (msg.error) {
|
|
document.getElementById('status')!.textContent = `❌ 错误: ${msg.error}`;
|
|
} else {
|
|
document.getElementById('status')!.textContent =
|
|
`第 ${msg.requestCount} 次 · 共 ${msg.totalItems} 条`;
|
|
}
|
|
if (msg.done) {
|
|
await loadAndRender();
|
|
document.getElementById('status')!.textContent =
|
|
`✅ 加载完成 · 共 ${msg.totalItems} 条`;
|
|
}
|
|
}
|
|
if (msg.type === 'DOWNLOAD_DONE') {
|
|
showStatus(
|
|
`✅ 下载完成: ${msg.success}/${msg.total}` +
|
|
(msg.failed > 0 ? ` (${msg.failed} 失败)` : ''),
|
|
'success'
|
|
);
|
|
}
|
|
}); |