prehistoric-systems/FMHY-SafeGuard
1
0
Fork 0
mirror of https://github.com/fmhy/FMHY-SafeGuard.git synced 2026-03-11 08:55:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
Kenneth Hendricks 2026-02-03 18:49:30 -05:00 committed by GitHub
parent 60b009e790
commit 69a22e092c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 43 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
src/js/background.js
View file

@ -60,6 +60,8 @@ let starredSites = [];
let unsafeReasons = {}; // Object to store reasons for unsafe sites
const approvedUrls = new Map(); // Map to store approved URLs per tab
const notesCache = new Map(); // Cache for fetched notes
let userTrustedDomains = new Set(); // User-defined trusted domains
let userUntrustedDomains = new Set(); // User-defined untrusted domains
// Base64 Starred Links (from rentry.co/FMHYB64) - stored encoded, decoded at runtime
const base64StarredLinksEncoded = [
@ -1232,6 +1234,15 @@ function getStatusFromLists(url) {
// Special handling for repository hosting sites
const urlObj = new URL(url);
const domain = urlObj.hostname;
const normalizedDomain = domain.replace(/^www\./, "").toLowerCase();
// Check user-defined domains first (highest priority)
if (userTrustedDomains.has(normalizedDomain)) {
return "safe";
}
if (userUntrustedDomains.has(normalizedDomain)) {
return "unsafe";
}
const isRepoSite = ["github.com", "gitlab.com", "sourceforge.net"].some(
(domain) =>
urlObj.hostname === domain || urlObj.hostname.endsWith("." + domain)
@ -1425,14 +1436,44 @@ async function initializeSettings() {
console.log("Settings initialized:", mergedSettings);
}
// Load user-defined trusted/untrusted domains from storage
async function loadUserDomains() {
try {
const { userTrustedDomains: trusted, userUntrustedDomains: untrusted } =
await browserAPI.storage.local.get(["userTrustedDomains", "userUntrustedDomains"]);
if (trusted && Array.isArray(trusted)) {
userTrustedDomains = new Set(trusted.map(d => d.toLowerCase().replace(/^www\./, "")));
console.log(`Loaded ${userTrustedDomains.size} user trusted domains`);
}
if (untrusted && Array.isArray(untrusted)) {
userUntrustedDomains = new Set(untrusted.map(d => d.toLowerCase().replace(/^www\./, "")));
console.log(`Loaded ${userUntrustedDomains.size} user untrusted domains`);
}
} catch (error) {
console.error("Error loading user domains:", error);
}
}
// Listen for storage changes to update user domains
browserAPI.storage.onChanged.addListener((changes, area) => {
if (area === "local") {
if (changes.userTrustedDomains || changes.userUntrustedDomains) {
loadUserDomains().then(() => {
console.log("User domains reloaded after settings change");
});
}
}
});
// Extension initialization
async function initializeExtension() {
console.log("Initializing extension...");
try {
await initializeSettings();
// Check if we need to update
await loadUserDomains();
if (await shouldUpdate()) {
await fetchFilterLists();
await fetchSafeSites();