mirror of
https://github.com/obsqrbtz/goose-highlighter.git
synced 2026-04-08 20:19:06 +03:00
599 lines
20 KiB
TypeScript
599 lines
20 KiB
TypeScript
import { HighlightList, HighlightWord, ExportData } from '../types.js';
|
|
import { StorageService } from '../services/StorageService.js';
|
|
import { MessageService } from '../services/MessageService.js';
|
|
import { DOMUtils } from '../utils/DOMUtils.js';
|
|
|
|
export class PopupController {
|
|
private lists: HighlightList[] = [];
|
|
private currentListIndex = 0;
|
|
private selectedCheckboxes = new Set<number>();
|
|
private globalHighlightEnabled = true;
|
|
private wordSearchQuery = '';
|
|
private matchCaseEnabled = false;
|
|
private matchWholeEnabled = false;
|
|
private exceptionsList: string[] = [];
|
|
private currentTabHost = '';
|
|
private activeTab = 'lists';
|
|
|
|
async initialize(): Promise<void> {
|
|
await this.loadData();
|
|
await this.getCurrentTab();
|
|
this.loadActiveTab();
|
|
this.translateTitles();
|
|
this.setupEventListeners();
|
|
this.render();
|
|
}
|
|
|
|
private async loadData(): Promise<void> {
|
|
const data = await StorageService.get();
|
|
this.lists = data.lists || [];
|
|
this.globalHighlightEnabled = data.globalHighlightEnabled ?? true;
|
|
this.matchCaseEnabled = data.matchCaseEnabled ?? false;
|
|
this.matchWholeEnabled = data.matchWholeEnabled ?? false;
|
|
this.exceptionsList = data.exceptionsList || [];
|
|
|
|
if (this.lists.length === 0) {
|
|
this.lists.push({
|
|
id: Date.now(),
|
|
name: chrome.i18n.getMessage('default_list_name') || 'Default List',
|
|
background: '#ffff00',
|
|
foreground: '#000000',
|
|
active: true,
|
|
words: []
|
|
});
|
|
}
|
|
}
|
|
|
|
private async getCurrentTab(): Promise<void> {
|
|
try {
|
|
const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
|
|
if (tab?.url) {
|
|
const url = new URL(tab.url);
|
|
this.currentTabHost = url.hostname;
|
|
}
|
|
} catch (e) {
|
|
console.warn('Could not get current tab:', e);
|
|
}
|
|
}
|
|
|
|
private loadActiveTab(): void {
|
|
const saved = localStorage.getItem('goose-highlighter-active-tab');
|
|
if (saved) {
|
|
this.activeTab = saved;
|
|
}
|
|
}
|
|
|
|
private translateTitles(): void {
|
|
document.querySelectorAll('[data-i18n-title]').forEach(element => {
|
|
const key = element.getAttribute('data-i18n-title');
|
|
if (key) {
|
|
const translation = chrome.i18n.getMessage(key);
|
|
if (translation) {
|
|
element.setAttribute('title', translation);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
private saveActiveTab(): void {
|
|
localStorage.setItem('goose-highlighter-active-tab', this.activeTab);
|
|
}
|
|
|
|
private switchTab(tabName: string): void {
|
|
this.activeTab = tabName;
|
|
this.saveActiveTab();
|
|
|
|
document.querySelectorAll('.tab-button').forEach(btn => {
|
|
btn.classList.toggle('active', btn.getAttribute('data-tab') === tabName);
|
|
});
|
|
|
|
document.querySelectorAll('.tab-content').forEach(content => {
|
|
content.classList.toggle('active', content.getAttribute('data-tab-content') === tabName);
|
|
});
|
|
}
|
|
|
|
private setupEventListeners(): void {
|
|
this.setupTabs();
|
|
this.setupListManagement();
|
|
this.setupWordManagement();
|
|
this.setupSettings();
|
|
this.setupExceptions();
|
|
this.setupImportExport();
|
|
this.setupTheme();
|
|
}
|
|
|
|
private setupTabs(): void {
|
|
document.querySelectorAll('.tab-button').forEach(button => {
|
|
button.addEventListener('click', () => {
|
|
const tabName = (button as HTMLElement).getAttribute('data-tab');
|
|
if (tabName) this.switchTab(tabName);
|
|
});
|
|
});
|
|
|
|
this.switchTab(this.activeTab);
|
|
}
|
|
|
|
private setupListManagement(): void {
|
|
const listSelect = document.getElementById('listSelect') as HTMLSelectElement;
|
|
|
|
listSelect.addEventListener('change', () => {
|
|
this.selectedCheckboxes.clear();
|
|
this.currentListIndex = +listSelect.value;
|
|
this.renderWords();
|
|
this.updateListForm();
|
|
});
|
|
|
|
// Apply button for list settings
|
|
document.getElementById('applyListSettingsBtn')?.addEventListener('click', () => {
|
|
this.applyListSettings();
|
|
});
|
|
|
|
document.getElementById('newListBtn')?.addEventListener('click', () => {
|
|
this.lists.push({
|
|
id: Date.now(),
|
|
name: chrome.i18n.getMessage('new_list_name') || 'New List',
|
|
background: '#ffff00',
|
|
foreground: '#000000',
|
|
active: true,
|
|
words: []
|
|
});
|
|
this.currentListIndex = this.lists.length - 1;
|
|
this.save();
|
|
});
|
|
|
|
document.getElementById('deleteListBtn')?.addEventListener('click', () => {
|
|
if (confirm(chrome.i18n.getMessage('confirm_delete_list') || 'Delete this list?')) {
|
|
this.lists.splice(this.currentListIndex, 1);
|
|
this.currentListIndex = Math.max(0, this.currentListIndex - 1);
|
|
this.save();
|
|
}
|
|
});
|
|
}
|
|
|
|
private setupWordManagement(): void {
|
|
const bulkPaste = document.getElementById('bulkPaste') as HTMLTextAreaElement;
|
|
const wordList = document.getElementById('wordList') as HTMLDivElement;
|
|
const wordSearch = document.getElementById('wordSearch') as HTMLInputElement;
|
|
|
|
document.getElementById('addWordsBtn')?.addEventListener('click', () => {
|
|
const words = bulkPaste.value.split(/\n+/).map(w => w.trim()).filter(Boolean);
|
|
const list = this.lists[this.currentListIndex];
|
|
for (const w of words) {
|
|
list.words.push({
|
|
wordStr: w,
|
|
background: '',
|
|
foreground: '',
|
|
active: true
|
|
});
|
|
}
|
|
bulkPaste.value = '';
|
|
this.save();
|
|
});
|
|
|
|
this.setupWordListEvents(wordList);
|
|
this.setupWordSelection();
|
|
|
|
wordSearch.addEventListener('input', (e) => {
|
|
this.wordSearchQuery = (e.target as HTMLInputElement).value;
|
|
this.renderWords();
|
|
});
|
|
}
|
|
|
|
private setupWordListEvents(wordList: HTMLDivElement): void {
|
|
wordList.addEventListener('change', (e) => {
|
|
const target = e.target as HTMLInputElement;
|
|
if (target.type === 'checkbox' && target.dataset.index != null) {
|
|
const index = +target.dataset.index;
|
|
if (target.checked) {
|
|
this.selectedCheckboxes.add(index);
|
|
} else {
|
|
this.selectedCheckboxes.delete(index);
|
|
}
|
|
this.renderWords();
|
|
}
|
|
});
|
|
|
|
wordList.addEventListener('input', (e) => {
|
|
const target = e.target as HTMLInputElement;
|
|
const index = +(target.dataset.bgEdit ?? target.dataset.fgEdit ?? -1);
|
|
if (index === -1) return;
|
|
|
|
const word = this.lists[this.currentListIndex].words[index];
|
|
if (target.dataset.bgEdit != null) word.background = target.value;
|
|
if (target.dataset.fgEdit != null) word.foreground = target.value;
|
|
|
|
this.save();
|
|
});
|
|
|
|
wordList.addEventListener('keydown', (e) => {
|
|
if (e.key === 'Enter') {
|
|
const target = e.target as HTMLInputElement;
|
|
const index = +(target.dataset.wordEdit ?? -1);
|
|
if (index === -1) return;
|
|
|
|
const word = this.lists[this.currentListIndex].words[index];
|
|
if (target.dataset.wordEdit != null) {
|
|
word.wordStr = target.value;
|
|
this.save();
|
|
}
|
|
}
|
|
});
|
|
|
|
let scrollTimeout: number;
|
|
wordList.addEventListener('scroll', () => {
|
|
if (scrollTimeout) return;
|
|
scrollTimeout = window.setTimeout(() => {
|
|
requestAnimationFrame(() => this.renderWords());
|
|
scrollTimeout = 0;
|
|
}, 16);
|
|
});
|
|
}
|
|
|
|
private setupWordSelection(): void {
|
|
document.getElementById('selectAllBtn')?.addEventListener('click', () => {
|
|
const list = this.lists[this.currentListIndex];
|
|
list.words.forEach((_, index) => {
|
|
this.selectedCheckboxes.add(index);
|
|
});
|
|
this.renderWords();
|
|
});
|
|
|
|
document.getElementById('deselectAllBtn')?.addEventListener('click', () => {
|
|
this.selectedCheckboxes.clear();
|
|
this.renderWords();
|
|
});
|
|
|
|
document.getElementById('deleteSelectedBtn')?.addEventListener('click', () => {
|
|
if (confirm(chrome.i18n.getMessage('confirm_delete_words') || 'Delete selected words?')) {
|
|
const list = this.lists[this.currentListIndex];
|
|
const toDelete = Array.from(this.selectedCheckboxes);
|
|
this.lists[this.currentListIndex].words = list.words.filter((_, i) => !toDelete.includes(i));
|
|
this.selectedCheckboxes.clear();
|
|
this.save();
|
|
this.renderWords();
|
|
}
|
|
});
|
|
|
|
document.getElementById('enableSelectedBtn')?.addEventListener('click', () => {
|
|
const list = this.lists[this.currentListIndex];
|
|
this.selectedCheckboxes.forEach(index => {
|
|
list.words[index].active = true;
|
|
});
|
|
this.save();
|
|
this.renderWords();
|
|
});
|
|
|
|
document.getElementById('disableSelectedBtn')?.addEventListener('click', () => {
|
|
const list = this.lists[this.currentListIndex];
|
|
this.selectedCheckboxes.forEach(index => {
|
|
list.words[index].active = false;
|
|
});
|
|
this.save();
|
|
this.renderWords();
|
|
});
|
|
}
|
|
|
|
private setupSettings(): void {
|
|
const globalToggle = document.getElementById('globalHighlightToggle') as HTMLInputElement;
|
|
const matchCase = document.getElementById('matchCase') as HTMLInputElement;
|
|
const matchWhole = document.getElementById('matchWhole') as HTMLInputElement;
|
|
|
|
globalToggle.addEventListener('change', () => {
|
|
this.globalHighlightEnabled = globalToggle.checked;
|
|
this.updateGlobalToggleState();
|
|
});
|
|
|
|
matchCase.addEventListener('change', () => {
|
|
this.matchCaseEnabled = matchCase.checked;
|
|
this.save();
|
|
});
|
|
|
|
matchWhole.addEventListener('change', () => {
|
|
this.matchWholeEnabled = matchWhole.checked;
|
|
this.save();
|
|
});
|
|
}
|
|
|
|
private setupExceptions(): void {
|
|
document.getElementById('toggleExceptionBtn')?.addEventListener('click', () => {
|
|
if (!this.currentTabHost) return;
|
|
|
|
const isException = this.exceptionsList.includes(this.currentTabHost);
|
|
|
|
if (isException) {
|
|
this.exceptionsList = this.exceptionsList.filter(domain => domain !== this.currentTabHost);
|
|
} else {
|
|
this.exceptionsList.push(this.currentTabHost);
|
|
}
|
|
|
|
this.updateExceptionButton();
|
|
this.renderExceptions();
|
|
this.save();
|
|
});
|
|
|
|
document.getElementById('clearExceptionsBtn')?.addEventListener('click', () => {
|
|
if (confirm(chrome.i18n.getMessage('confirm_clear_exceptions') || 'Clear all exceptions?')) {
|
|
this.exceptionsList = [];
|
|
this.updateExceptionButton();
|
|
this.renderExceptions();
|
|
this.save();
|
|
}
|
|
});
|
|
|
|
document.getElementById('exceptionsList')?.addEventListener('click', (e) => {
|
|
const target = e.target as HTMLElement;
|
|
if (target.classList.contains('exception-remove')) {
|
|
const domain = target.dataset.domain!;
|
|
this.exceptionsList = this.exceptionsList.filter(d => d !== domain);
|
|
this.updateExceptionButton();
|
|
this.renderExceptions();
|
|
this.save();
|
|
}
|
|
});
|
|
}
|
|
|
|
private setupImportExport(): void {
|
|
const importInput = document.getElementById('importInput') as HTMLInputElement;
|
|
|
|
document.getElementById('exportBtn')?.addEventListener('click', () => {
|
|
const exportData: ExportData = {
|
|
lists: this.lists,
|
|
exceptionsList: this.exceptionsList
|
|
};
|
|
const blob = new Blob([JSON.stringify(exportData, null, 2)], { type: 'application/json' });
|
|
const url = URL.createObjectURL(blob);
|
|
const a = document.createElement('a');
|
|
a.href = url;
|
|
a.download = 'highlight-lists.json';
|
|
a.click();
|
|
URL.revokeObjectURL(url);
|
|
});
|
|
|
|
document.getElementById('importBtn')?.addEventListener('click', () => {
|
|
importInput.click();
|
|
});
|
|
|
|
importInput.addEventListener('change', (e) => {
|
|
const file = (e.target as HTMLInputElement).files?.[0];
|
|
if (!file) return;
|
|
|
|
const reader = new FileReader();
|
|
reader.onload = (e) => {
|
|
try {
|
|
const data = JSON.parse(e.target?.result as string);
|
|
|
|
if (Array.isArray(data)) {
|
|
this.lists = data;
|
|
} else if (data && typeof data === 'object') {
|
|
if (Array.isArray(data.lists)) {
|
|
this.lists = data.lists;
|
|
}
|
|
if (Array.isArray(data.exceptionsList)) {
|
|
this.exceptionsList = data.exceptionsList;
|
|
}
|
|
}
|
|
|
|
this.currentListIndex = 0;
|
|
this.updateExceptionButton();
|
|
this.renderExceptions();
|
|
this.save();
|
|
} catch (err) {
|
|
alert(chrome.i18n.getMessage('invalid_json_error') + ': ' + (err as Error).message);
|
|
}
|
|
};
|
|
reader.readAsText(file);
|
|
});
|
|
}
|
|
|
|
private setupTheme(): void {
|
|
const toggle = document.getElementById('themeToggle') as HTMLInputElement;
|
|
const body = document.body;
|
|
|
|
const savedTheme = localStorage.getItem('theme');
|
|
if (savedTheme === 'light') {
|
|
body.classList.remove('dark');
|
|
body.classList.add('light');
|
|
toggle.checked = false;
|
|
} else {
|
|
body.classList.add('dark');
|
|
body.classList.remove('light');
|
|
toggle.checked = true;
|
|
}
|
|
|
|
toggle.addEventListener('change', () => {
|
|
if (toggle.checked) {
|
|
body.classList.add('dark');
|
|
body.classList.remove('light');
|
|
localStorage.setItem('theme', 'dark');
|
|
} else {
|
|
body.classList.remove('dark');
|
|
body.classList.add('light');
|
|
localStorage.setItem('theme', 'light');
|
|
}
|
|
});
|
|
}
|
|
|
|
private applyListSettings(): void {
|
|
const listName = document.getElementById('listName') as HTMLInputElement;
|
|
const listBg = document.getElementById('listBg') as HTMLInputElement;
|
|
const listFg = document.getElementById('listFg') as HTMLInputElement;
|
|
const listActive = document.getElementById('listActive') as HTMLInputElement;
|
|
|
|
this.lists[this.currentListIndex].name = listName.value;
|
|
this.lists[this.currentListIndex].background = listBg.value;
|
|
this.lists[this.currentListIndex].foreground = listFg.value;
|
|
this.lists[this.currentListIndex].active = listActive.checked;
|
|
|
|
this.save();
|
|
}
|
|
|
|
private async save(): Promise<void> {
|
|
await StorageService.set({
|
|
lists: this.lists,
|
|
globalHighlightEnabled: this.globalHighlightEnabled,
|
|
matchCaseEnabled: this.matchCaseEnabled,
|
|
matchWholeEnabled: this.matchWholeEnabled,
|
|
exceptionsList: this.exceptionsList
|
|
});
|
|
|
|
this.render();
|
|
MessageService.sendToAllTabs({ type: 'WORD_LIST_UPDATED' });
|
|
MessageService.sendToAllTabs({
|
|
type: 'GLOBAL_TOGGLE_UPDATED',
|
|
enabled: this.globalHighlightEnabled
|
|
});
|
|
MessageService.sendToAllTabs({
|
|
type: 'MATCH_OPTIONS_UPDATED',
|
|
matchCase: this.matchCaseEnabled,
|
|
matchWhole: this.matchWholeEnabled
|
|
});
|
|
MessageService.sendToAllTabs({ type: 'EXCEPTIONS_LIST_UPDATED' });
|
|
}
|
|
|
|
|
|
|
|
private async updateGlobalToggleState(): Promise<void> {
|
|
await StorageService.update('globalHighlightEnabled', this.globalHighlightEnabled);
|
|
MessageService.sendToAllTabs({
|
|
type: 'GLOBAL_TOGGLE_UPDATED',
|
|
enabled: this.globalHighlightEnabled
|
|
});
|
|
}
|
|
|
|
private render(): void {
|
|
this.renderLists();
|
|
this.renderWords();
|
|
this.renderExceptions();
|
|
this.updateExceptionButton();
|
|
this.updateFormValues();
|
|
}
|
|
|
|
private renderLists(): void {
|
|
const listSelect = document.getElementById('listSelect') as HTMLSelectElement;
|
|
listSelect.innerHTML = this.lists.map((list, index) =>
|
|
`<option value="${index}">${DOMUtils.escapeHtml(list.name)}</option>`
|
|
).join('');
|
|
listSelect.value = this.currentListIndex.toString();
|
|
this.updateListForm();
|
|
}
|
|
|
|
private updateListForm(): void {
|
|
const list = this.lists[this.currentListIndex];
|
|
(document.getElementById('listName') as HTMLInputElement).value = list.name;
|
|
(document.getElementById('listBg') as HTMLInputElement).value = list.background;
|
|
(document.getElementById('listFg') as HTMLInputElement).value = list.foreground;
|
|
(document.getElementById('listActive') as HTMLInputElement).checked = list.active;
|
|
}
|
|
|
|
private renderWords(): void {
|
|
const list = this.lists[this.currentListIndex];
|
|
const wordList = document.getElementById('wordList') as HTMLDivElement;
|
|
|
|
let filteredWords = list.words;
|
|
if (this.wordSearchQuery.trim()) {
|
|
const q = this.wordSearchQuery.trim().toLowerCase();
|
|
filteredWords = list.words.filter(w => w.wordStr.toLowerCase().includes(q));
|
|
}
|
|
|
|
const itemHeight = 32;
|
|
const itemSpacing = 2;
|
|
const totalItemHeight = itemHeight + itemSpacing;
|
|
const containerHeight = wordList.clientHeight || 250;
|
|
const scrollTop = wordList.scrollTop;
|
|
const startIndex = Math.floor(scrollTop / totalItemHeight);
|
|
const endIndex = Math.min(
|
|
startIndex + Math.ceil(containerHeight / totalItemHeight) + 2,
|
|
filteredWords.length
|
|
);
|
|
|
|
wordList.innerHTML = '';
|
|
|
|
const spacer = document.createElement('div');
|
|
spacer.style.position = 'relative';
|
|
spacer.style.height = `${filteredWords.length * totalItemHeight}px`;
|
|
spacer.style.width = '100%';
|
|
|
|
for (let i = startIndex; i < endIndex; i++) {
|
|
const w = filteredWords[i];
|
|
if (!w) continue;
|
|
|
|
const realIndex = list.words.indexOf(w);
|
|
const container = this.createWordItem(w, realIndex, i, itemHeight);
|
|
spacer.appendChild(container);
|
|
}
|
|
|
|
wordList.appendChild(spacer);
|
|
|
|
const wordCount = document.getElementById('wordCount');
|
|
if (wordCount) {
|
|
wordCount.textContent = filteredWords.length.toString();
|
|
}
|
|
}
|
|
|
|
private createWordItem(word: HighlightWord, realIndex: number, displayIndex: number, itemHeight: number): HTMLDivElement {
|
|
const container = document.createElement('div');
|
|
container.className = 'word-item';
|
|
if (word.active === false) {
|
|
container.classList.add('disabled');
|
|
}
|
|
container.style.cssText = `
|
|
position: absolute;
|
|
top: ${displayIndex * (itemHeight + 2)}px;
|
|
`;
|
|
|
|
const list = this.lists[this.currentListIndex];
|
|
|
|
container.innerHTML = `
|
|
<input type="checkbox" class="word-checkbox" data-index="${realIndex}" ${this.selectedCheckboxes.has(realIndex) ? 'checked' : ''} title="${chrome.i18n.getMessage('select_title') || 'Select'}">
|
|
<input type="text" value="${DOMUtils.escapeHtml(word.wordStr)}" data-word-edit="${realIndex}" placeholder="${chrome.i18n.getMessage('word_placeholder') || 'Word or phrase'}">
|
|
<input type="color" value="${word.background || list.background}" data-bg-edit="${realIndex}" title="${chrome.i18n.getMessage('background_color_title') || 'Background color'}">
|
|
<input type="color" value="${word.foreground || list.foreground}" data-fg-edit="${realIndex}" title="${chrome.i18n.getMessage('text_color_title') || 'Text color'}">
|
|
`;
|
|
|
|
return container;
|
|
}
|
|
|
|
private updateExceptionButton(): void {
|
|
const toggleBtn = document.getElementById('toggleExceptionBtn');
|
|
const btnText = document.getElementById('exceptionBtnText');
|
|
|
|
if (!toggleBtn || !btnText || !this.currentTabHost) return;
|
|
|
|
const isException = this.exceptionsList.includes(this.currentTabHost);
|
|
|
|
if (isException) {
|
|
btnText.textContent = chrome.i18n.getMessage('remove_exception') || 'Remove from Exceptions';
|
|
toggleBtn.className = 'danger';
|
|
const icon = toggleBtn.querySelector('i');
|
|
if (icon) icon.className = 'fa-solid fa-trash';
|
|
} else {
|
|
btnText.textContent = chrome.i18n.getMessage('add_exception') || 'Add to Exceptions';
|
|
toggleBtn.className = '';
|
|
const icon = toggleBtn.querySelector('i');
|
|
if (icon) icon.className = 'fa-solid fa-plus';
|
|
}
|
|
}
|
|
|
|
private renderExceptions(): void {
|
|
const container = document.getElementById('exceptionsList');
|
|
if (!container) return;
|
|
|
|
if (this.exceptionsList.length === 0) {
|
|
container.innerHTML = `<div class="exception-item">${chrome.i18n.getMessage('no_exceptions') || 'No exceptions'}</div>`;
|
|
return;
|
|
}
|
|
|
|
container.innerHTML = this.exceptionsList.map(domain =>
|
|
`<div class="exception-item">
|
|
<span class="exception-domain">${DOMUtils.escapeHtml(domain)}</span>
|
|
<button class="exception-remove" data-domain="${DOMUtils.escapeHtml(domain)}">${chrome.i18n.getMessage('remove') || 'Remove'}</button>
|
|
</div>`
|
|
).join('');
|
|
}
|
|
|
|
private updateFormValues(): void {
|
|
(document.getElementById('globalHighlightToggle') as HTMLInputElement).checked = this.globalHighlightEnabled;
|
|
(document.getElementById('matchCase') as HTMLInputElement).checked = this.matchCaseEnabled;
|
|
(document.getElementById('matchWhole') as HTMLInputElement).checked = this.matchWholeEnabled;
|
|
}
|
|
} |