julian/chrom-tool
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

refactor backend

Browse Source
This commit is contained in:
Julian Freeman
2026-04-16 13:35:37 -04:00
parent f7915e2890
commit 42c3d488d2
8 changed files with 628 additions and 608 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
public/tauri.svg
View File

@@ -1,6 +0,0 @@
<svg width="206" height="231" viewBox="0 0 206 231" fill="none" xmlns="http://www.w3.org/2000/svg">
<path d="M143.143 84C143.143 96.1503 133.293 106 121.143 106C108.992 106 99.1426 96.1503 99.1426 84C99.1426 71.8497 108.992 62 121.143 62C133.293 62 143.143 71.8497 143.143 84Z" fill="#FFC131"/>
<ellipse cx="84.1426" cy="147" rx="22" ry="22" transform="rotate(180 84.1426 147)" fill="#24C8DB"/>
<path fill-rule="evenodd" clip-rule="evenodd" d="M166.738 154.548C157.86 160.286 148.023 164.269 137.757 166.341C139.858 160.282 141 153.774 141 147C141 144.543 140.85 142.121 140.558 139.743C144.975 138.204 149.215 136.139 153.183 133.575C162.73 127.404 170.292 118.608 174.961 108.244C179.63 97.8797 181.207 86.3876 179.502 75.1487C177.798 63.9098 172.884 53.4021 165.352 44.8883C157.82 36.3744 147.99 30.2165 137.042 27.1546C126.095 24.0926 114.496 24.2568 103.64 27.6274C92.7839 30.998 83.1319 37.4317 75.8437 46.1553C74.9102 47.2727 74.0206 48.4216 73.176 49.5993C61.9292 50.8488 51.0363 54.0318 40.9629 58.9556C44.2417 48.4586 49.5653 38.6591 56.679 30.1442C67.0505 17.7298 80.7861 8.57426 96.2354 3.77762C111.685 -1.01901 128.19 -1.25267 143.769 3.10474C159.348 7.46215 173.337 16.2252 184.056 28.3411C194.775 40.457 201.767 55.4101 204.193 71.404C206.619 87.3978 204.374 103.752 197.73 118.501C191.086 133.25 180.324 145.767 166.738 154.548ZM41.9631 74.275L62.5557 76.8042C63.0459 72.813 63.9401 68.9018 65.2138 65.1274C57.0465 67.0016 49.2088 70.087 41.9631 74.275Z" fill="#FFC131"/>
<path fill-rule="evenodd" clip-rule="evenodd" d="M38.4045 76.4519C47.3493 70.6709 57.2677 66.6712 67.6171 64.6132C65.2774 70.9669 64 77.8343 64 85.0001C64 87.1434 64.1143 89.26 64.3371 91.3442C60.0093 92.8732 55.8533 94.9092 51.9599 97.4256C42.4128 103.596 34.8505 112.392 30.1816 122.756C25.5126 133.12 23.9357 144.612 25.6403 155.851C27.3449 167.09 32.2584 177.598 39.7906 186.112C47.3227 194.626 57.153 200.784 68.1003 203.846C79.0476 206.907 90.6462 206.743 101.502 203.373C112.359 200.002 122.011 193.568 129.299 184.845C130.237 183.722 131.131 182.567 131.979 181.383C143.235 180.114 154.132 176.91 164.205 171.962C160.929 182.49 155.596 192.319 148.464 200.856C138.092 213.27 124.357 222.426 108.907 227.222C93.458 232.019 76.9524 232.253 61.3736 227.895C45.7948 223.538 31.8055 214.775 21.0867 202.659C10.3679 190.543 3.37557 175.59 0.949823 159.596C-1.47592 143.602 0.768139 127.248 7.41237 112.499C14.0566 97.7497 24.8183 85.2327 38.4045 76.4519ZM163.062 156.711L163.062 156.711C162.954 156.773 162.846 156.835 162.738 156.897C162.846 156.835 162.954 156.773 163.062 156.711Z" fill="#24C8DB"/>
</svg>
Side by Side

Before

Width:  |  Height:  |  Size: 2.5 KiB

1
public/vite.svg
View File

@@ -1 +0,0 @@
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="31.88" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 257"><defs><linearGradient id="IconifyId1813088fe1fbc01fb466" x1="-.828%" x2="57.636%" y1="7.652%" y2="78.411%"><stop offset="0%" stop-color="#41D1FF"></stop><stop offset="100%" stop-color="#BD34FE"></stop></linearGradient><linearGradient id="IconifyId1813088fe1fbc01fb467" x1="43.376%" x2="50.316%" y1="2.242%" y2="89.03%"><stop offset="0%" stop-color="#FFEA83"></stop><stop offset="8.333%" stop-color="#FFDD35"></stop><stop offset="100%" stop-color="#FFA800"></stop></linearGradient></defs><path fill="url(#IconifyId1813088fe1fbc01fb466)" d="M255.153 37.938L134.897 252.976c-2.483 4.44-8.862 4.466-11.382.048L.875 37.958c-2.746-4.814 1.371-10.646 6.827-9.67l120.385 21.517a6.537 6.537 0 0 0 2.322-.004l117.867-21.483c5.438-.991 9.574 4.796 6.877 9.62Z"></path><path fill="url(#IconifyId1813088fe1fbc01fb467)" d="M185.432.063L96.44 17.501a3.268 3.268 0 0 0-2.634 3.014l-5.474 92.456a3.268 3.268 0 0 0 3.997 3.378l24.777-5.718c2.318-.535 4.413 1.507 3.936 3.838l-7.361 36.047c-.495 2.426 1.782 4.5 4.151 3.78l15.304-4.649c2.372-.72 4.652 1.36 4.15 3.788l-11.698 56.621c-.732 3.542 3.979 5.473 5.943 2.437l1.313-2.028l72.516-144.72c1.215-2.423-.88-5.186-3.54-4.672l-25.505 4.922c-2.396.462-4.435-1.77-3.759-4.114l16.646-57.705c.677-2.35-1.37-4.583-3.769-4.113Z"></path></svg>
Side by Side

Before

Width:  |  Height:  |  Size: 1.5 KiB

6
src-tauri/src/commands.rs Normal file
View File

@@ -0,0 +1,6 @@
use crate::{models::ScanResponse, scanner};
#[tauri::command]
pub fn scan_browsers() -> Result<ScanResponse, String> {
scanner::scan_browsers()
}

605
src-tauri/src/lib.rs
View File

@@ -1,608 +1,13 @@
use std::{
collections::{BTreeMap, BTreeSet},
env, fs,
path::{Path, PathBuf},
};
use base64::{engine::general_purpose::STANDARD, Engine as _};
use serde::Serialize;
use serde_json::Value;
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct ScanResponse {
browsers: Vec<BrowserView>,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct BrowserView {
browser_id: String,
browser_name: String,
data_root: String,
profiles: Vec<ProfileSummary>,
extensions: Vec<ExtensionSummary>,
bookmarks: Vec<BookmarkSummary>,
stats: BrowserStats,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct BrowserStats {
profile_count: usize,
extension_count: usize,
bookmark_count: usize,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct ProfileSummary {
id: String,
name: String,
email: Option<String>,
avatar_data_url: Option<String>,
avatar_label: String,
path: String,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct ExtensionSummary {
id: String,
name: String,
version: Option<String>,
icon_data_url: Option<String>,
profile_ids: Vec<String>,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct BookmarkSummary {
url: String,
title: String,
profile_ids: Vec<String>,
}
struct BrowserDefinition {
id: &'static str,
name: &'static str,
local_app_data_segments: &'static [&'static str],
}
struct TempExtension {
id: String,
name: String,
version: Option<String>,
icon_data_url: Option<String>,
profile_ids: BTreeSet<String>,
}
struct TempBookmark {
url: String,
title: String,
profile_ids: BTreeSet<String>,
}
#[tauri::command]
fn scan_browsers() -> Result<ScanResponse, String> {
let local_app_data = local_app_data_dir().ok_or_else(|| {
"Unable to resolve the LOCALAPPDATA directory for the current user.".to_string()
})?;
let browsers = browser_definitions()
.into_iter()
.filter_map(|definition| scan_browser(&local_app_data, definition))
.collect();
Ok(ScanResponse { browsers })
}
mod commands;
mod models;
mod scanner;
mod utils;
#[cfg_attr(mobile, tauri::mobile_entry_point)]
pub fn run() {
tauri::Builder::default()
.plugin(tauri_plugin_opener::init())
.invoke_handler(tauri::generate_handler![scan_browsers])
.invoke_handler(tauri::generate_handler![commands::scan_browsers])
.run(tauri::generate_context!())
.expect("error while running tauri application");
}
fn browser_definitions() -> Vec<BrowserDefinition> {
vec![
BrowserDefinition {
id: "chrome",
name: "Google Chrome",
local_app_data_segments: &["Google", "Chrome", "User Data"],
},
BrowserDefinition {
id: "edge",
name: "Microsoft Edge",
local_app_data_segments: &["Microsoft", "Edge", "User Data"],
},
BrowserDefinition {
id: "brave",
name: "Brave",
local_app_data_segments: &["BraveSoftware", "Brave-Browser", "User Data"],
},
]
}
fn local_app_data_dir() -> Option<PathBuf> {
env::var_os("LOCALAPPDATA").map(PathBuf::from).or_else(|| {
env::var_os("USERPROFILE")
.map(PathBuf::from)
.map(|path| path.join("AppData").join("Local"))
})
}
fn scan_browser(local_app_data: &Path, definition: BrowserDefinition) -> Option<BrowserView> {
let root = definition
.local_app_data_segments
.iter()
.fold(local_app_data.to_path_buf(), |path, segment| {
path.join(segment)
});
if !root.is_dir() {
return None;
}
let local_state = read_json_file(&root.join("Local State")).unwrap_or(Value::Null);
let profile_cache = local_state
.get("profile")
.and_then(|value| value.get("info_cache"))
.and_then(Value::as_object);
let mut profile_ids = BTreeSet::new();
collect_profile_ids_from_fs(&root, &mut profile_ids);
if let Some(cache) = profile_cache {
for profile_id in cache.keys() {
profile_ids.insert(profile_id.to_string());
}
}
let mut profiles = Vec::new();
let mut extensions = BTreeMap::<String, TempExtension>::new();
let mut bookmarks = BTreeMap::<String, TempBookmark>::new();
for profile_id in profile_ids {
let profile_path = root.join(&profile_id);
if !profile_path.is_dir() {
continue;
}
let profile_info = profile_cache.and_then(|cache| cache.get(&profile_id));
profiles.push(build_profile_summary(
&root,
&profile_path,
&profile_id,
profile_info,
));
scan_extensions_for_profile(&profile_path, &profile_id, &mut extensions);
scan_bookmarks_for_profile(&profile_path, &profile_id, &mut bookmarks);
}
let profiles = sort_profiles(profiles);
let extensions = extensions
.into_values()
.map(|entry| ExtensionSummary {
id: entry.id,
name: entry.name,
version: entry.version,
icon_data_url: entry.icon_data_url,
profile_ids: entry.profile_ids.into_iter().collect(),
})
.collect::<Vec<_>>();
let bookmarks = bookmarks
.into_values()
.map(|entry| BookmarkSummary {
url: entry.url,
title: entry.title,
profile_ids: entry.profile_ids.into_iter().collect(),
})
.collect::<Vec<_>>();
Some(BrowserView {
browser_id: definition.id.to_string(),
browser_name: definition.name.to_string(),
data_root: root.display().to_string(),
stats: BrowserStats {
profile_count: profiles.len(),
extension_count: extensions.len(),
bookmark_count: bookmarks.len(),
},
profiles,
extensions: sort_extensions(extensions),
bookmarks: sort_bookmarks(bookmarks),
})
}
fn collect_profile_ids_from_fs(root: &Path, profile_ids: &mut BTreeSet<String>) {
let Ok(entries) = fs::read_dir(root) else {
return;
};
for entry in entries.flatten() {
let Ok(file_type) = entry.file_type() else {
continue;
};
if !file_type.is_dir() {
continue;
}
let name = entry.file_name().to_string_lossy().to_string();
if name == "Default" || name.starts_with("Profile ") {
profile_ids.insert(name);
}
}
}
fn build_profile_summary(
root: &Path,
profile_path: &Path,
profile_id: &str,
profile_info: Option<&Value>,
) -> ProfileSummary {
let name = first_non_empty([
profile_info
.and_then(|value| value.get("name"))
.and_then(Value::as_str),
profile_info
.and_then(|value| value.get("gaia_name"))
.and_then(Value::as_str),
Some(profile_id),
])
.unwrap_or(profile_id)
.to_string();
let email = first_non_empty([
profile_info
.and_then(|value| value.get("user_name"))
.and_then(Value::as_str),
None,
])
.map(str::to_string);
let avatar_data_url = resolve_profile_avatar(root, profile_path, profile_info);
let avatar_label = name
.chars()
.find(|character| !character.is_whitespace())
.map(|character| character.to_uppercase().collect::<String>())
.unwrap_or_else(|| "?".to_string());
ProfileSummary {
id: profile_id.to_string(),
name,
email,
avatar_data_url,
avatar_label,
path: profile_path.display().to_string(),
}
}
fn resolve_profile_avatar(
_root: &Path,
profile_path: &Path,
profile_info: Option<&Value>,
) -> Option<String> {
let picture_file = profile_info
.and_then(|value| value.get("gaia_picture_file_name"))
.and_then(Value::as_str)
.filter(|value| !value.is_empty());
if let Some(file_name) = picture_file {
let candidate = profile_path.join(file_name);
if let Some(data_url) = load_image_as_data_url(&candidate) {
return Some(data_url);
}
}
None
}
fn scan_extensions_for_profile(
profile_path: &Path,
profile_id: &str,
extensions: &mut BTreeMap<String, TempExtension>,
) {
let extensions_root = profile_path.join("Extensions");
let Ok(entries) = fs::read_dir(&extensions_root) else {
return;
};
for entry in entries.flatten() {
let Ok(file_type) = entry.file_type() else {
continue;
};
if !file_type.is_dir() {
continue;
}
let extension_id = entry.file_name().to_string_lossy().to_string();
let Some(version_path) = pick_latest_subdirectory(&entry.path()) else {
continue;
};
let manifest_path = version_path.join("manifest.json");
let Some(manifest) = read_json_file(&manifest_path) else {
continue;
};
let name = resolve_extension_name(&manifest, &version_path)
.filter(|value| !value.is_empty())
.unwrap_or_else(|| extension_id.clone());
let version = manifest
.get("version")
.and_then(Value::as_str)
.map(str::to_string);
let icon_data_url = resolve_extension_icon(&manifest, &version_path);
let entry = extensions
.entry(extension_id.clone())
.or_insert_with(|| TempExtension {
id: extension_id.clone(),
name: name.clone(),
version: version.clone(),
icon_data_url: icon_data_url.clone(),
profile_ids: BTreeSet::new(),
});
if entry.name == entry.id && name != extension_id {
entry.name = name.clone();
}
if entry.version.is_none() {
entry.version = version.clone();
}
if entry.icon_data_url.is_none() {
entry.icon_data_url = icon_data_url.clone();
}
entry.profile_ids.insert(profile_id.to_string());
}
}
fn resolve_extension_name(manifest: &Value, version_path: &Path) -> Option<String> {
let raw_name = manifest.get("name").and_then(Value::as_str)?;
if let Some(localized_name) = resolve_localized_manifest_value(raw_name, manifest, version_path)
{
return Some(localized_name);
}
manifest
.get("short_name")
.and_then(Value::as_str)
.map(str::to_string)
.or_else(|| Some(raw_name.to_string()))
}
fn resolve_localized_manifest_value(
raw_value: &str,
manifest: &Value,
version_path: &Path,
) -> Option<String> {
if !(raw_value.starts_with("__MSG_") && raw_value.ends_with("__")) {
return Some(raw_value.to_string());
}
let message_key = raw_value
.trim_start_matches("__MSG_")
.trim_end_matches("__");
let default_locale = manifest
.get("default_locale")
.and_then(Value::as_str)
.unwrap_or("en");
for locale in [default_locale, "en"] {
let messages_path = version_path
.join("_locales")
.join(locale)
.join("messages.json");
let Some(messages) = read_json_file(&messages_path) else {
continue;
};
if let Some(message) = messages
.get(message_key)
.and_then(|value| value.get("message"))
.and_then(Value::as_str)
.filter(|value| !value.is_empty())
{
return Some(message.to_string());
}
}
Some(raw_value.to_string())
}
fn resolve_extension_icon(manifest: &Value, version_path: &Path) -> Option<String> {
let mut candidates = Vec::new();
if let Some(icons) = manifest.get("icons").and_then(Value::as_object) {
candidates.extend(icon_candidates_from_object(icons));
}
for key in ["action", "browser_action", "page_action"] {
if let Some(default_icon) = manifest
.get(key)
.and_then(|value| value.get("default_icon"))
{
if let Some(icon_path) = default_icon.as_str() {
candidates.push((0, icon_path.to_string()));
} else if let Some(icon_map) = default_icon.as_object() {
candidates.extend(icon_candidates_from_object(icon_map));
}
}
}
candidates.sort_by(|left, right| right.0.cmp(&left.0));
candidates
.into_iter()
.find_map(|(_, relative_path)| load_image_as_data_url(&version_path.join(relative_path)))
}
fn icon_candidates_from_object(map: &serde_json::Map<String, Value>) -> Vec<(u32, String)> {
map.iter()
.filter_map(|(size, value)| {
value.as_str().map(|path| {
let parsed_size = size.parse::<u32>().unwrap_or(0);
(parsed_size, path.to_string())
})
})
.collect()
}
fn scan_bookmarks_for_profile(
profile_path: &Path,
profile_id: &str,
bookmarks: &mut BTreeMap<String, TempBookmark>,
) {
let bookmarks_path = profile_path.join("Bookmarks");
let Some(document) = read_json_file(&bookmarks_path) else {
return;
};
let Some(roots) = document.get("roots").and_then(Value::as_object) else {
return;
};
for root in roots.values() {
collect_bookmarks(root, profile_id, bookmarks);
}
}
fn collect_bookmarks(
node: &Value,
profile_id: &str,
bookmarks: &mut BTreeMap<String, TempBookmark>,
) {
match node.get("type").and_then(Value::as_str) {
Some("url") => {
let Some(url) = node.get("url").and_then(Value::as_str) else {
return;
};
if url.is_empty() {
return;
}
let title = node
.get("name")
.and_then(Value::as_str)
.filter(|value| !value.is_empty())
.unwrap_or(url)
.to_string();
let entry = bookmarks
.entry(url.to_string())
.or_insert_with(|| TempBookmark {
url: url.to_string(),
title: title.clone(),
profile_ids: BTreeSet::new(),
});
if entry.title == entry.url && title != url {
entry.title = title;
}
entry.profile_ids.insert(profile_id.to_string());
}
Some("folder") => {
if let Some(children) = node.get("children").and_then(Value::as_array) {
for child in children {
collect_bookmarks(child, profile_id, bookmarks);
}
}
}
_ => {}
}
}
fn pick_latest_subdirectory(root: &Path) -> Option<PathBuf> {
let entries = fs::read_dir(root).ok()?;
let mut candidates = entries
.flatten()
.filter_map(|entry| {
let file_type = entry.file_type().ok()?;
if !file_type.is_dir() {
return None;
}
let metadata = entry.metadata().ok()?;
let modified = metadata.modified().ok();
Some((
modified,
entry.file_name().to_string_lossy().to_string(),
entry.path(),
))
})
.collect::<Vec<_>>();
candidates.sort_by(|left, right| right.0.cmp(&left.0).then_with(|| right.1.cmp(&left.1)));
candidates.into_iter().next().map(|(_, _, path)| path)
}
fn load_image_as_data_url(path: &Path) -> Option<String> {
let bytes = fs::read(path).ok()?;
let extension = path
.extension()
.and_then(|value| value.to_str())
.map(|value| value.to_ascii_lowercase())?;
let mime_type = match extension.as_str() {
"png" => "image/png",
"jpg" | "jpeg" => "image/jpeg",
"webp" => "image/webp",
"gif" => "image/gif",
"svg" => "image/svg+xml",
_ => return None,
};
Some(format!(
"data:{mime_type};base64,{}",
STANDARD.encode(bytes)
))
}
fn read_json_file(path: &Path) -> Option<Value> {
let content = fs::read_to_string(path).ok()?;
serde_json::from_str(&content).ok()
}
fn first_non_empty<'a>(values: impl IntoIterator<Item = Option<&'a str>>) -> Option<&'a str> {
values
.into_iter()
.flatten()
.find(|value| !value.trim().is_empty())
}
fn sort_profiles(mut profiles: Vec<ProfileSummary>) -> Vec<ProfileSummary> {
profiles.sort_by(|left, right| profile_sort_key(&left.id).cmp(&profile_sort_key(&right.id)));
profiles
}
fn profile_sort_key(profile_id: &str) -> (u8, u32, String) {
if profile_id == "Default" {
return (0, 0, profile_id.to_string());
}
if let Some(number) = profile_id
.strip_prefix("Profile ")
.and_then(|value| value.parse::<u32>().ok())
{
return (1, number, profile_id.to_string());
}
(2, u32::MAX, profile_id.to_string())
}
fn sort_extensions(mut extensions: Vec<ExtensionSummary>) -> Vec<ExtensionSummary> {
extensions.sort_by(|left, right| {
left.name
.to_lowercase()
.cmp(&right.name.to_lowercase())
.then_with(|| left.id.cmp(&right.id))
});
extensions
}
fn sort_bookmarks(mut bookmarks: Vec<BookmarkSummary>) -> Vec<BookmarkSummary> {
bookmarks.sort_by(|left, right| {
left.title
.to_lowercase()
.cmp(&right.title.to_lowercase())
.then_with(|| left.url.cmp(&right.url))
});
bookmarks
}

78
src-tauri/src/models.rs Normal file
View File

@@ -0,0 +1,78 @@
use std::collections::BTreeSet;
use serde::Serialize;
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct ScanResponse {
pub browsers: Vec<BrowserView>,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct BrowserView {
pub browser_id: String,
pub browser_name: String,
pub data_root: String,
pub profiles: Vec<ProfileSummary>,
pub extensions: Vec<ExtensionSummary>,
pub bookmarks: Vec<BookmarkSummary>,
pub stats: BrowserStats,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct BrowserStats {
pub profile_count: usize,
pub extension_count: usize,
pub bookmark_count: usize,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct ProfileSummary {
pub id: String,
pub name: String,
pub email: Option<String>,
pub avatar_data_url: Option<String>,
pub avatar_label: String,
pub path: String,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct ExtensionSummary {
pub id: String,
pub name: String,
pub version: Option<String>,
pub icon_data_url: Option<String>,
pub profile_ids: Vec<String>,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct BookmarkSummary {
pub url: String,
pub title: String,
pub profile_ids: Vec<String>,
}
pub struct BrowserDefinition {
pub id: &'static str,
pub name: &'static str,
pub local_app_data_segments: &'static [&'static str],
}
pub struct TempExtension {
pub id: String,
pub name: String,
pub version: Option<String>,
pub icon_data_url: Option<String>,
pub profile_ids: BTreeSet<String>,
}
pub struct TempBookmark {
pub url: String,
pub title: String,
pub profile_ids: BTreeSet<String>,
}

468
src-tauri/src/scanner.rs Normal file
View File

@@ -0,0 +1,468 @@
use std::{
collections::{BTreeMap, BTreeSet},
fs,
path::Path,
};
use serde_json::Value;
use crate::{
models::{
BookmarkSummary, BrowserDefinition, BrowserStats, BrowserView, ExtensionSummary,
ProfileSummary, ScanResponse, TempBookmark, TempExtension,
},
utils::{
first_non_empty, load_image_as_data_url, local_app_data_dir, pick_latest_subdirectory,
read_json_file,
},
};
pub fn scan_browsers() -> Result<ScanResponse, String> {
let local_app_data = local_app_data_dir().ok_or_else(|| {
"Unable to resolve the LOCALAPPDATA directory for the current user.".to_string()
})?;
let browsers = browser_definitions()
.into_iter()
.filter_map(|definition| scan_browser(&local_app_data, definition))
.collect();
Ok(ScanResponse { browsers })
}
fn browser_definitions() -> Vec<BrowserDefinition> {
vec![
BrowserDefinition {
id: "chrome",
name: "Google Chrome",
local_app_data_segments: &["Google", "Chrome", "User Data"],
},
BrowserDefinition {
id: "edge",
name: "Microsoft Edge",
local_app_data_segments: &["Microsoft", "Edge", "User Data"],
},
BrowserDefinition {
id: "brave",
name: "Brave",
local_app_data_segments: &["BraveSoftware", "Brave-Browser", "User Data"],
},
]
}
fn scan_browser(local_app_data: &Path, definition: BrowserDefinition) -> Option<BrowserView> {
let root = definition
.local_app_data_segments
.iter()
.fold(local_app_data.to_path_buf(), |path, segment| {
path.join(segment)
});
if !root.is_dir() {
return None;
}
let local_state = read_json_file(&root.join("Local State")).unwrap_or(Value::Null);
let profile_cache = local_state
.get("profile")
.and_then(|value| value.get("info_cache"))
.and_then(Value::as_object);
let mut profile_ids = BTreeSet::new();
collect_profile_ids_from_fs(&root, &mut profile_ids);
if let Some(cache) = profile_cache {
for profile_id in cache.keys() {
profile_ids.insert(profile_id.to_string());
}
}
let mut profiles = Vec::new();
let mut extensions = BTreeMap::<String, TempExtension>::new();
let mut bookmarks = BTreeMap::<String, TempBookmark>::new();
for profile_id in profile_ids {
let profile_path = root.join(&profile_id);
if !profile_path.is_dir() {
continue;
}
let profile_info = profile_cache.and_then(|cache| cache.get(&profile_id));
profiles.push(build_profile_summary(
&root,
&profile_path,
&profile_id,
profile_info,
));
scan_extensions_for_profile(&profile_path, &profile_id, &mut extensions);
scan_bookmarks_for_profile(&profile_path, &profile_id, &mut bookmarks);
}
let profiles = sort_profiles(profiles);
let extensions = extensions
.into_values()
.map(|entry| ExtensionSummary {
id: entry.id,
name: entry.name,
version: entry.version,
icon_data_url: entry.icon_data_url,
profile_ids: entry.profile_ids.into_iter().collect(),
})
.collect::<Vec<_>>();
let bookmarks = bookmarks
.into_values()
.map(|entry| BookmarkSummary {
url: entry.url,
title: entry.title,
profile_ids: entry.profile_ids.into_iter().collect(),
})
.collect::<Vec<_>>();
Some(BrowserView {
browser_id: definition.id.to_string(),
browser_name: definition.name.to_string(),
data_root: root.display().to_string(),
stats: BrowserStats {
profile_count: profiles.len(),
extension_count: extensions.len(),
bookmark_count: bookmarks.len(),
},
profiles,
extensions: sort_extensions(extensions),
bookmarks: sort_bookmarks(bookmarks),
})
}
fn collect_profile_ids_from_fs(root: &Path, profile_ids: &mut BTreeSet<String>) {
let Ok(entries) = fs::read_dir(root) else {
return;
};
for entry in entries.flatten() {
let Ok(file_type) = entry.file_type() else {
continue;
};
if !file_type.is_dir() {
continue;
}
let name = entry.file_name().to_string_lossy().to_string();
if name == "Default" || name.starts_with("Profile ") {
profile_ids.insert(name);
}
}
}
fn build_profile_summary(
root: &Path,
profile_path: &Path,
profile_id: &str,
profile_info: Option<&Value>,
) -> ProfileSummary {
let name = first_non_empty([
profile_info
.and_then(|value| value.get("name"))
.and_then(Value::as_str),
profile_info
.and_then(|value| value.get("gaia_name"))
.and_then(Value::as_str),
Some(profile_id),
])
.unwrap_or(profile_id)
.to_string();
let email = first_non_empty([
profile_info
.and_then(|value| value.get("user_name"))
.and_then(Value::as_str),
None,
])
.map(str::to_string);
let avatar_data_url = resolve_profile_avatar(root, profile_path, profile_info);
let avatar_label = name
.chars()
.find(|character| !character.is_whitespace())
.map(|character| character.to_uppercase().collect::<String>())
.unwrap_or_else(|| "?".to_string());
ProfileSummary {
id: profile_id.to_string(),
name,
email,
avatar_data_url,
avatar_label,
path: profile_path.display().to_string(),
}
}
fn resolve_profile_avatar(
_root: &Path,
profile_path: &Path,
profile_info: Option<&Value>,
) -> Option<String> {
let picture_file = profile_info
.and_then(|value| value.get("gaia_picture_file_name"))
.and_then(Value::as_str)
.filter(|value| !value.is_empty());
if let Some(file_name) = picture_file {
let candidate = profile_path.join(file_name);
if let Some(data_url) = load_image_as_data_url(&candidate) {
return Some(data_url);
}
}
None
}
fn scan_extensions_for_profile(
profile_path: &Path,
profile_id: &str,
extensions: &mut BTreeMap<String, TempExtension>,
) {
let extensions_root = profile_path.join("Extensions");
let Ok(entries) = fs::read_dir(&extensions_root) else {
return;
};
for entry in entries.flatten() {
let Ok(file_type) = entry.file_type() else {
continue;
};
if !file_type.is_dir() {
continue;
}
let extension_id = entry.file_name().to_string_lossy().to_string();
let Some(version_path) = pick_latest_subdirectory(&entry.path()) else {
continue;
};
let manifest_path = version_path.join("manifest.json");
let Some(manifest) = read_json_file(&manifest_path) else {
continue;
};
let name = resolve_extension_name(&manifest, &version_path)
.filter(|value| !value.is_empty())
.unwrap_or_else(|| extension_id.clone());
let version = manifest
.get("version")
.and_then(Value::as_str)
.map(str::to_string);
let icon_data_url = resolve_extension_icon(&manifest, &version_path);
let entry = extensions
.entry(extension_id.clone())
.or_insert_with(|| TempExtension {
id: extension_id.clone(),
name: name.clone(),
version: version.clone(),
icon_data_url: icon_data_url.clone(),
profile_ids: BTreeSet::new(),
});
if entry.name == entry.id && name != extension_id {
entry.name = name.clone();
}
if entry.version.is_none() {
entry.version = version.clone();
}
if entry.icon_data_url.is_none() {
entry.icon_data_url = icon_data_url.clone();
}
entry.profile_ids.insert(profile_id.to_string());
}
}
fn resolve_extension_name(manifest: &Value, version_path: &Path) -> Option<String> {
let raw_name = manifest.get("name").and_then(Value::as_str)?;
if let Some(localized_name) = resolve_localized_manifest_value(raw_name, manifest, version_path)
{
return Some(localized_name);
}
manifest
.get("short_name")
.and_then(Value::as_str)
.map(str::to_string)
.or_else(|| Some(raw_name.to_string()))
}
fn resolve_localized_manifest_value(
raw_value: &str,
manifest: &Value,
version_path: &Path,
) -> Option<String> {
if !(raw_value.starts_with("__MSG_") && raw_value.ends_with("__")) {
return Some(raw_value.to_string());
}
let message_key = raw_value
.trim_start_matches("__MSG_")
.trim_end_matches("__");
let default_locale = manifest
.get("default_locale")
.and_then(Value::as_str)
.unwrap_or("en");
for locale in [default_locale, "en"] {
let messages_path = version_path
.join("_locales")
.join(locale)
.join("messages.json");
let Some(messages) = read_json_file(&messages_path) else {
continue;
};
if let Some(message) = messages
.get(message_key)
.and_then(|value| value.get("message"))
.and_then(Value::as_str)
.filter(|value| !value.is_empty())
{
return Some(message.to_string());
}
}
Some(raw_value.to_string())
}
fn resolve_extension_icon(manifest: &Value, version_path: &Path) -> Option<String> {
let mut candidates = Vec::new();
if let Some(icons) = manifest.get("icons").and_then(Value::as_object) {
candidates.extend(icon_candidates_from_object(icons));
}
for key in ["action", "browser_action", "page_action"] {
if let Some(default_icon) = manifest
.get(key)
.and_then(|value| value.get("default_icon"))
{
if let Some(icon_path) = default_icon.as_str() {
candidates.push((0, icon_path.to_string()));
} else if let Some(icon_map) = default_icon.as_object() {
candidates.extend(icon_candidates_from_object(icon_map));
}
}
}
candidates.sort_by(|left, right| right.0.cmp(&left.0));
candidates
.into_iter()
.find_map(|(_, relative_path)| load_image_as_data_url(&version_path.join(relative_path)))
}
fn icon_candidates_from_object(map: &serde_json::Map<String, Value>) -> Vec<(u32, String)> {
map.iter()
.filter_map(|(size, value)| {
value.as_str().map(|path| {
let parsed_size = size.parse::<u32>().unwrap_or(0);
(parsed_size, path.to_string())
})
})
.collect()
}
fn scan_bookmarks_for_profile(
profile_path: &Path,
profile_id: &str,
bookmarks: &mut BTreeMap<String, TempBookmark>,
) {
let bookmarks_path = profile_path.join("Bookmarks");
let Some(document) = read_json_file(&bookmarks_path) else {
return;
};
let Some(roots) = document.get("roots").and_then(Value::as_object) else {
return;
};
for root in roots.values() {
collect_bookmarks(root, profile_id, bookmarks);
}
}
fn collect_bookmarks(
node: &Value,
profile_id: &str,
bookmarks: &mut BTreeMap<String, TempBookmark>,
) {
match node.get("type").and_then(Value::as_str) {
Some("url") => {
let Some(url) = node.get("url").and_then(Value::as_str) else {
return;
};
if url.is_empty() {
return;
}
let title = node
.get("name")
.and_then(Value::as_str)
.filter(|value| !value.is_empty())
.unwrap_or(url)
.to_string();
let entry = bookmarks
.entry(url.to_string())
.or_insert_with(|| TempBookmark {
url: url.to_string(),
title: title.clone(),
profile_ids: BTreeSet::new(),
});
if entry.title == entry.url && title != url {
entry.title = title;
}
entry.profile_ids.insert(profile_id.to_string());
}
Some("folder") => {
if let Some(children) = node.get("children").and_then(Value::as_array) {
for child in children {
collect_bookmarks(child, profile_id, bookmarks);
}
}
}
_ => {}
}
}
fn sort_profiles(mut profiles: Vec<ProfileSummary>) -> Vec<ProfileSummary> {
profiles.sort_by(|left, right| profile_sort_key(&left.id).cmp(&profile_sort_key(&right.id)));
profiles
}
fn profile_sort_key(profile_id: &str) -> (u8, u32, String) {
if profile_id == "Default" {
return (0, 0, profile_id.to_string());
}
if let Some(number) = profile_id
.strip_prefix("Profile ")
.and_then(|value| value.parse::<u32>().ok())
{
return (1, number, profile_id.to_string());
}
(2, u32::MAX, profile_id.to_string())
}
fn sort_extensions(mut extensions: Vec<ExtensionSummary>) -> Vec<ExtensionSummary> {
extensions.sort_by(|left, right| {
left.name
.to_lowercase()
.cmp(&right.name.to_lowercase())
.then_with(|| left.id.cmp(&right.id))
});
extensions
}
fn sort_bookmarks(mut bookmarks: Vec<BookmarkSummary>) -> Vec<BookmarkSummary> {
bookmarks.sort_by(|left, right| {
left.title
.to_lowercase()
.cmp(&right.title.to_lowercase())
.then_with(|| left.url.cmp(&right.url))
});
bookmarks
}

71
src-tauri/src/utils.rs Normal file
View File

@@ -0,0 +1,71 @@
use std::{
env, fs,
path::{Path, PathBuf},
};
use base64::{engine::general_purpose::STANDARD, Engine as _};
use serde_json::Value;
pub fn local_app_data_dir() -> Option<PathBuf> {
env::var_os("LOCALAPPDATA").map(PathBuf::from).or_else(|| {
env::var_os("USERPROFILE")
.map(PathBuf::from)
.map(|path| path.join("AppData").join("Local"))
})
}
pub fn pick_latest_subdirectory(root: &Path) -> Option<PathBuf> {
let entries = fs::read_dir(root).ok()?;
let mut candidates = entries
.flatten()
.filter_map(|entry| {
let file_type = entry.file_type().ok()?;
if !file_type.is_dir() {
return None;
}
let metadata = entry.metadata().ok()?;
let modified = metadata.modified().ok();
Some((
modified,
entry.file_name().to_string_lossy().to_string(),
entry.path(),
))
})
.collect::<Vec<_>>();
candidates.sort_by(|left, right| right.0.cmp(&left.0).then_with(|| right.1.cmp(&left.1)));
candidates.into_iter().next().map(|(_, _, path)| path)
}
pub fn load_image_as_data_url(path: &Path) -> Option<String> {
let bytes = fs::read(path).ok()?;
let extension = path
.extension()
.and_then(|value| value.to_str())
.map(|value| value.to_ascii_lowercase())?;
let mime_type = match extension.as_str() {
"png" => "image/png",
"jpg" | "jpeg" => "image/jpeg",
"webp" => "image/webp",
"gif" => "image/gif",
"svg" => "image/svg+xml",
_ => return None,
};
Some(format!(
"data:{mime_type};base64,{}",
STANDARD.encode(bytes)
))
}
pub fn read_json_file(path: &Path) -> Option<Value> {
let content = fs::read_to_string(path).ok()?;
serde_json::from_str(&content).ok()
}
pub fn first_non_empty<'a>(values: impl IntoIterator<Item = Option<&'a str>>) -> Option<&'a str> {
values
.into_iter()
.flatten()
.find(|value| !value.trim().is_empty())
}

1
src/assets/vue.svg
View File

@@ -1 +0,0 @@
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="37.07" height="36" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 198"><path fill="#41B883" d="M204.8 0H256L128 220.8L0 0h97.92L128 51.2L157.44 0h47.36Z"></path><path fill="#41B883" d="m0 0l128 220.8L256 0h-51.2L128 132.48L50.56 0H0Z"></path><path fill="#35495E" d="M50.56 0L128 133.12L204.8 0h-47.36L128 51.2L97.92 0H50.56Z"></path></svg>
Side by Side

Before

Width:  |  Height:  |  Size: 496 B