Files
ultimate-ban-tracker/frontend/electron/main.ts
Nils Pukropp 64fe49e58e
Some checks failed
Build and Release / build (push) Has been cancelled
init
2026-02-21 01:48:48 +01:00

601 lines
22 KiB
TypeScript

import { app, BrowserWindow, ipcMain, shell, session, protocol, net } from 'electron';
import path from 'path';
import Store from 'electron-store';
import { exec } from 'child_process';
import dotenv from 'dotenv';
import cron from 'node-cron';
import axios from 'axios';
import fs from 'fs';
import { pathToFileURL } from 'url';
import { fetchProfileData, scrapeBanStatus } from './services/steam-web';
import { scrapeCooldown } from './services/scraper';
import { steamClient, LocalSteamAccount } from './services/steam-client';
import { BackendService } from './services/backend';
// Reliable isDev check
const isDev = !app.isPackaged;
app.name = "Ultimate Ban Tracker";
// Load environment variables
dotenv.config({ path: path.join(app.getAppPath(), '..', '.env') });
// --- Server Configuration ---
let backend: BackendService | null = null;
const initBackend = () => {
const config = store.get('serverConfig');
if (config && config.enabled && config.url) {
console.log(`[Backend] Initializing with URL: ${config.url}`);
backend = new BackendService(config.url, config.token);
} else {
backend = null;
}
};
// --- Local Data Store ---
interface Account {
_id: string;
steamId: string;
personaName: string;
loginName: string;
steamLoginSecure?: string;
loginConfig?: any;
sessionUpdatedAt?: string;
autoCheckCooldown: boolean;
avatar: string;
localAvatar?: string;
profileUrl: string;
status: string;
vacBanned: boolean;
gameBans: number;
lastBanCheck: string;
lastScrapeTime?: string;
cooldownExpiresAt?: string;
authError?: boolean;
notes?: string;
}
interface ServerConfig {
url: string;
token?: string;
serverSteamId?: string;
enabled: boolean;
}
const store = new Store<{ accounts: Account[], serverConfig: ServerConfig }>({
defaults: {
accounts: [],
serverConfig: { url: 'https://ultimate-ban-tracker.narl.io', enabled: false }
}
}) as any;
// --- Avatar Cache Logic ---
const AVATAR_DIR = path.join(app.getPath('userData'), 'avatars');
if (!fs.existsSync(AVATAR_DIR)) fs.mkdirSync(AVATAR_DIR, { recursive: true });
const downloadAvatar = async (steamId: string, url: string): Promise<string | undefined> => {
if (!url) return undefined;
const localPath = path.join(AVATAR_DIR, `${steamId}.jpg`);
try {
const response = await axios.get(url, { responseType: 'arraybuffer', timeout: 5000 });
fs.writeFileSync(localPath, Buffer.from(response.data));
return localPath;
} catch (e) {
return undefined;
}
};
protocol.registerSchemesAsPrivileged([
{ scheme: 'steam-resource', privileges: { secure: true, standard: true, supportFetchAPI: true } }
]);
// --- Main Window ---
let mainWindow: BrowserWindow | null = null;
function createWindow() {
mainWindow = new BrowserWindow({
width: 1280,
height: 800,
title: "Ultimate Ban Tracker Desktop",
backgroundColor: '#171a21',
autoHideMenuBar: true,
webPreferences: {
preload: path.join(__dirname, 'preload.js'),
nodeIntegration: false,
contextIsolation: true,
},
});
mainWindow.setMenu(null);
if (isDev) {
mainWindow.loadURL('http://localhost:5173');
} else {
mainWindow.loadFile(path.join(__dirname, '..', 'dist', 'index.html'));
}
}
// --- Sync Logic ---
const syncAccounts = async () => {
initBackend();
let accounts = store.get('accounts') as Account[];
let hasChanges = false;
// 1. PULL SHARED ACCOUNTS FROM SERVER
if (backend) {
console.log('[Sync] Phase 1: Pulling from server...');
try {
const shared = await backend.getSharedAccounts();
for (const s of shared) {
const exists = accounts.find(a => a.steamId === s.steamId);
if (!exists) {
console.log(`[Sync] Discovered new account on server: ${s.personaName}`);
accounts.push({
_id: `shared_${s.steamId}`,
steamId: s.steamId,
personaName: s.personaName,
avatar: s.avatar,
profileUrl: s.profileUrl,
vacBanned: s.vacBanned,
gameBans: s.gameBans,
cooldownExpiresAt: s.cooldownExpiresAt,
loginName: s.loginName || '',
steamLoginSecure: s.steamLoginSecure,
loginConfig: s.loginConfig,
sessionUpdatedAt: s.sessionUpdatedAt,
autoCheckCooldown: s.steamLoginSecure ? true : false,
status: (s.vacBanned || s.gameBans > 0) ? 'banned' : 'none',
lastBanCheck: new Date().toISOString()
});
hasChanges = true;
} else {
const sDate = s.sessionUpdatedAt ? new Date(s.sessionUpdatedAt) : new Date(0);
const lDate = exists.sessionUpdatedAt ? new Date(exists.sessionUpdatedAt) : new Date(0);
if (sDate > lDate) {
console.log(`[Sync] Updating session for ${exists.personaName} (Server is newer)`);
if (s.loginName) exists.loginName = s.loginName;
if (s.loginConfig) exists.loginConfig = s.loginConfig;
if (s.steamLoginSecure) {
exists.steamLoginSecure = s.steamLoginSecure;
exists.autoCheckCooldown = true;
exists.authError = false;
}
exists.sessionUpdatedAt = s.sessionUpdatedAt;
hasChanges = true;
}
if (s.cooldownExpiresAt && (!exists.cooldownExpiresAt || new Date(s.cooldownExpiresAt) > new Date(exists.cooldownExpiresAt))) {
exists.cooldownExpiresAt = s.cooldownExpiresAt;
hasChanges = true;
}
}
}
} catch (e) {
console.error('[Sync] Pull failed');
}
}
// BROADCAST PULL RESULTS IMMEDIATELY
if (hasChanges) {
store.set('accounts', accounts);
if (mainWindow) mainWindow.webContents.send('accounts-updated', accounts);
}
if (accounts.length === 0) return;
// 2. BACKGROUND STEALTH CHECKS
console.log(`[Sync] Phase 2: Starting background checks for ${accounts.length} accounts...`);
const updatedAccounts = [...accounts];
let scrapeChanges = false;
for (const account of updatedAccounts) {
try {
const now = new Date();
const lastCheck = account.lastBanCheck ? new Date(account.lastBanCheck) : new Date(0);
const hoursSinceCheck = (now.getTime() - lastCheck.getTime()) / 3600000;
if (hoursSinceCheck > 6 || !account.personaName) {
const profile = await fetchProfileData(account.steamId, account.steamLoginSecure);
const bans = await scrapeBanStatus(profile.profileUrl, account.steamLoginSecure);
account.personaName = profile.personaName;
account.profileUrl = profile.profileUrl;
account.vacBanned = bans.vacBanned;
account.gameBans = bans.gameBans;
account.status = (bans.vacBanned || bans.gameBans > 0) ? 'banned' : 'none';
account.lastBanCheck = now.toISOString();
if (profile.avatar && (!account.localAvatar || profile.avatar !== account.avatar)) {
account.avatar = profile.avatar;
const localPath = await downloadAvatar(account.steamId, profile.avatar);
if (localPath) account.localAvatar = localPath;
}
if (account.loginName) {
const config = steamClient.extractAccountConfig(account.loginName);
if (config) {
account.loginConfig = config;
account.sessionUpdatedAt = new Date().toISOString();
}
}
if (backend) await backend.shareAccount(account);
scrapeChanges = true;
}
if (account.autoCheckCooldown && account.steamLoginSecure) {
if (account.cooldownExpiresAt && new Date(account.cooldownExpiresAt) > now) continue;
const lastScrape = account.lastScrapeTime ? new Date(account.lastScrapeTime) : new Date(0);
const hoursSinceScrape = (now.getTime() - lastScrape.getTime()) / 3600000;
if (hoursSinceScrape > 8) {
const jitter = Math.floor(Math.random() * 60000) + 5000;
await new Promise(r => setTimeout(r, jitter));
try {
const result = await scrapeCooldown(account.steamId, account.steamLoginSecure);
account.authError = false;
account.lastScrapeTime = new Date().toISOString();
if (result.isActive) {
if (result.expiresAt) {
account.cooldownExpiresAt = result.expiresAt.toISOString();
} else if (!account.cooldownExpiresAt) {
const placeholder = new Date();
placeholder.setHours(placeholder.getHours() + 24);
account.cooldownExpiresAt = placeholder.toISOString();
}
if (backend) await backend.pushCooldown(account.steamId, account.cooldownExpiresAt);
} else if (account.cooldownExpiresAt) {
account.cooldownExpiresAt = undefined;
if (backend) await backend.pushCooldown(account.steamId, undefined);
}
scrapeChanges = true;
} catch (e: any) {
if (e.message.includes('cookie') || e.message.includes('Sign In')) {
account.authError = true;
scrapeChanges = true;
}
}
}
}
} catch (error) { }
}
if (scrapeChanges) {
store.set('accounts', updatedAccounts);
if (mainWindow) mainWindow.webContents.send('accounts-updated', updatedAccounts);
}
console.log('[Sync] Sync cycle finished.');
};
const scheduleNextSync = () => {
const delay = isDev ? 120000 : (Math.random() * 30 * 60000) + 30 * 60000;
setTimeout(async () => { await syncAccounts(); scheduleNextSync(); }, delay);
};
// --- Steam Auto-Discovery ---
const addingAccounts = new Set<string>();
const handleLocalAccountsFound = async (localAccounts: LocalSteamAccount[]) => {
const currentAccounts = store.get('accounts') as Account[];
let hasChanges = false;
for (const local of localAccounts) {
if (addingAccounts.has(local.steamId)) continue;
const exists = currentAccounts.find(a => a.steamId === local.steamId);
if (exists) {
if (!exists.loginName && local.accountName) { exists.loginName = local.accountName; hasChanges = true; }
} else {
addingAccounts.add(local.steamId);
try {
const profile = await fetchProfileData(local.steamId);
const bans = await scrapeBanStatus(profile.profileUrl);
const localPath = await downloadAvatar(profile.steamId, profile.avatar);
currentAccounts.push({
_id: Date.now().toString() + Math.random().toString().slice(2, 5),
steamId: local.steamId,
personaName: profile.personaName || local.personaName || local.accountName,
loginName: local.accountName,
autoCheckCooldown: false,
avatar: profile.avatar,
localAvatar: localPath,
profileUrl: profile.profileUrl,
status: (bans.vacBanned || bans.gameBans > 0) ? 'banned' : 'none',
vacBanned: bans.vacBanned,
gameBans: bans.gameBans,
lastBanCheck: new Date().toISOString()
});
hasChanges = true;
} catch (e) { }
addingAccounts.delete(local.steamId);
}
}
if (hasChanges) {
store.set('accounts', currentAccounts);
if (mainWindow) mainWindow.webContents.send('accounts-updated', currentAccounts);
}
};
app.whenReady().then(() => {
protocol.handle('steam-resource', (request) => {
let rawPath = decodeURIComponent(request.url.replace('steam-resource://', ''));
if (process.platform !== 'win32' && !rawPath.startsWith('/')) rawPath = '/' + rawPath;
const absolutePath = path.isAbsolute(rawPath) ? rawPath : path.resolve(rawPath);
if (!fs.existsSync(absolutePath)) return new Response('Not Found', { status: 404 });
try { return net.fetch(pathToFileURL(absolutePath).toString()); } catch (e) { return new Response('Error', { status: 500 }); }
});
createWindow();
initBackend();
setTimeout(syncAccounts, 5000);
scheduleNextSync();
steamClient.startWatching(handleLocalAccountsFound);
});
// --- IPC Handlers ---
console.log('[Main] Registering IPC Handlers...');
ipcMain.handle('get-accounts', () => store.get('accounts'));
ipcMain.handle('get-server-config', () => store.get('serverConfig'));
ipcMain.handle('update-server-config', (event, config: Partial<ServerConfig>) => {
const current = store.get('serverConfig');
const updated = { ...current, ...config };
store.set('serverConfig', updated);
initBackend();
return updated;
});
ipcMain.handle('login-to-server', async () => {
initBackend();
const config = store.get('serverConfig') as ServerConfig;
if (!config.url) return false;
return new Promise<boolean>((resolve) => {
const authWindow = new BrowserWindow({
width: 800, height: 700, parent: mainWindow || undefined, modal: true, title: 'Login to Ban Tracker Server',
webPreferences: { nodeIntegration: false, contextIsolation: true }
});
authWindow.loadURL(`${config.url}/auth/steam`);
let captured = false;
const saveServerAuth = (token: string) => {
if (captured) return;
captured = true;
console.log('[ServerAuth] Securely captured token');
let serverSteamId = undefined;
try {
const payload = JSON.parse(Buffer.from(token.split('.')[1]!, 'base64').toString());
serverSteamId = payload.steamId;
} catch (e) {}
const current = store.get('serverConfig');
store.set('serverConfig', { ...current, token, serverSteamId, enabled: true });
initBackend();
authWindow.close();
resolve(true);
};
// METHOD 1: Sniff HTTP Headers
const filter = { urls: [`${config.url}/*`] };
authWindow.webContents.session.webRequest.onHeadersReceived(filter, (details, callback) => {
const headers = details.responseHeaders || {};
const authToken = headers['x-ubt-auth-token']?.[0] || headers['X-UBT-Auth-Token']?.[0];
if (authToken) saveServerAuth(authToken);
callback({ cancel: false });
});
// METHOD 2: Watch Window Title (Fallback)
authWindow.on('page-title-updated', (event, title) => {
if (title.includes('AUTH_TOKEN:')) {
const token = title.split('AUTH_TOKEN:')[1];
if (token) saveServerAuth(token);
}
});
authWindow.on('closed', () => { resolve(false); });
});
});
ipcMain.handle('get-server-user-info', () => ({ steamId: store.get('serverConfig').serverSteamId }));
ipcMain.handle('sync-now', async () => { await syncAccounts(); return true; });
ipcMain.handle('add-account', async (event, { identifier }) => {
try {
initBackend();
// OPTIMIZATION: Check community server first
if (backend) {
const shared = await backend.getCommunityAccounts();
const existing = shared.find((s: any) => s.steamId === identifier || s.profileUrl.includes(identifier));
if (existing) {
const accounts = store.get('accounts') as Account[];
if (accounts.find(a => a.steamId === existing.steamId)) throw new Error('Account already tracked');
const newAccount: Account = {
_id: `shared_${existing.steamId}`,
steamId: existing.steamId,
personaName: existing.personaName,
avatar: existing.avatar,
profileUrl: existing.profileUrl,
vacBanned: existing.vacBanned,
gameBans: existing.gameBans,
cooldownExpiresAt: existing.cooldownExpiresAt,
loginName: existing.loginName || '',
steamLoginSecure: existing.steamLoginSecure,
loginConfig: existing.loginConfig,
sessionUpdatedAt: existing.sessionUpdatedAt,
autoCheckCooldown: existing.steamLoginSecure ? true : false,
status: (existing.vacBanned || existing.gameBans > 0) ? 'banned' : 'none',
lastBanCheck: new Date().toISOString()
};
store.set('accounts', [...accounts, newAccount]);
return newAccount;
}
}
const profile = await fetchProfileData(identifier);
const bans = await scrapeBanStatus(profile.profileUrl);
const localAvatar = await downloadAvatar(profile.steamId, profile.avatar);
const accounts = store.get('accounts') as Account[];
const newAccount: Account = {
_id: Date.now().toString(),
steamId: profile.steamId, personaName: profile.personaName, loginName: '',
avatar: profile.avatar, localAvatar: localAvatar, profileUrl: profile.profileUrl,
autoCheckCooldown: false, vacBanned: bans.vacBanned, gameBans: bans.gameBans,
status: (bans.vacBanned || bans.gameBans > 0) ? 'banned' : 'none', lastBanCheck: new Date().toISOString()
};
store.set('accounts', [...accounts, newAccount]);
return newAccount;
} catch (error: any) { throw error; }
});
ipcMain.handle('update-account', (event, id: string, data: Partial<Account>) => {
const accounts = store.get('accounts') as Account[];
const index = accounts.findIndex((a: Account) => a._id === id);
if (index !== -1) { accounts[index] = { ...accounts[index], ...data } as Account; store.set('accounts', accounts); return accounts[index]; }
return null;
});
ipcMain.handle('delete-account', (event, id: string) => {
const accounts = store.get('accounts') as Account[];
store.set('accounts', accounts.filter((a: Account) => a._id !== id));
return true;
});
ipcMain.handle('share-account-with-user', async (event, steamId: string, targetSteamId: string) => {
initBackend();
if (backend) {
const accounts = store.get('accounts') as Account[];
const account = accounts.find(a => a.steamId === steamId);
if (account) await backend.shareAccount(account);
return await backend.shareWithUser(steamId, targetSteamId);
}
throw new Error('Backend not configured');
});
ipcMain.handle('get-community-accounts', async () => { initBackend(); return backend ? await backend.getCommunityAccounts() : []; });
ipcMain.handle('get-server-users', async () => { initBackend(); return backend ? await backend.getServerUsers() : []; });
const killSteam = async () => {
return new Promise<void>((resolve) => {
const command = process.platform === 'win32' ? 'taskkill /f /im steam.exe' : 'pkill -9 steam';
exec(command, () => setTimeout(resolve, 1000));
});
};
const startSteam = () => {
const command = process.platform === 'win32' ? 'start steam://open/main' : 'steam &';
exec(command);
};
ipcMain.handle('switch-account', async (event, loginName: string) => {
if (!loginName) return false;
try {
await killSteam();
const accounts = store.get('accounts') as Account[];
const account = accounts.find(a => a.loginName === loginName);
if (process.platform === 'win32') {
const regCommand = `reg add "HKCU\\Software\\Valve\\Steam" /v AutoLoginUser /t REG_SZ /d "${loginName}" /f`;
const rememberCommand = `reg add "HKCU\\Software\\Valve\\Steam" /v RememberPassword /t REG_DWORD /d 1 /f`;
await new Promise<void>((res, rej) => exec(`${regCommand} && ${rememberCommand}`, (e) => e ? rej(e) : res()));
if (account && account.loginConfig) steamClient.injectAccountConfig(loginName, account.loginConfig);
} else if (process.platform === 'linux') {
await steamClient.setAutoLoginUser(loginName, account?.loginConfig, account?.steamId);
}
startSteam();
return true;
} catch (e) { return false; }
});
ipcMain.handle('open-external', (event, url: string) => shell.openExternal(url));
ipcMain.handle('open-steam-login', async (event, expectedSteamId: string) => {
const loginSession = session.fromPartition('persist:steam-login');
await loginSession.clearStorageData({ storages: ['cookies', 'localstorage', 'indexdb'] });
return new Promise<boolean>((resolve) => {
const loginWindow = new BrowserWindow({
width: 800,
height: 700,
parent: mainWindow || undefined,
modal: true,
title: 'Login to Steam (Ensure "Remember Me" is checked!)',
webPreferences: {
nodeIntegration: false,
contextIsolation: true,
partition: 'persist:steam-login'
}
});
loginWindow.loadURL('https://steamcommunity.com/login/home/?goto=my/gcpd/730');
const checkCookie = setInterval(async () => {
try {
const cookies = await loginSession.cookies.get({ domain: 'steamcommunity.com' });
const secureCookie = cookies.find(c => c.name === 'steamLoginSecure');
if (secureCookie) {
const steamId = decodeURIComponent(secureCookie.value).split('|')[0];
if (steamId) {
if (expectedSteamId && steamId !== expectedSteamId) {
console.error(`[Auth] ID Mismatch! Expected ${expectedSteamId}, got ${steamId}`);
return;
}
clearInterval(checkCookie);
const cookieString = cookies.map(c => `${c.name}=${c.value}`).join('; ');
console.log(`[Auth] Captured session for SteamID: ${steamId}`);
const accounts = store.get('accounts') as Account[];
const accountIndex = accounts.findIndex(a => a.steamId === steamId);
if (accountIndex !== -1) {
const account = accounts[accountIndex]!;
account.steamLoginSecure = cookieString;
account.autoCheckCooldown = true;
account.authError = false;
account.sessionUpdatedAt = new Date().toISOString();
if (account.loginName) {
const config = steamClient.extractAccountConfig(account.loginName);
if (config) account.loginConfig = config;
}
try {
console.log(`[Auth] Performing initial scrape for ${account.personaName}...`);
const result = await scrapeCooldown(account.steamId, cookieString);
account.lastScrapeTime = new Date().toISOString();
if (result.isActive && result.expiresAt) {
account.cooldownExpiresAt = result.expiresAt.toISOString();
} else if (!result.isActive) {
account.cooldownExpiresAt = undefined;
}
} catch (e) {
console.error('[Auth] Initial scrape failed:', e);
}
initBackend();
if (backend) await backend.shareAccount(account);
store.set('accounts', accounts);
if (mainWindow) mainWindow.webContents.send('accounts-updated', accounts);
loginWindow.close();
resolve(true);
}
}
}
} catch (error) { }
}, 1000);
loginWindow.on('closed', () => {
clearInterval(checkCookie);
resolve(false);
});
});
});
app.on('window-all-closed', () => { if (process.platform !== 'darwin') app.quit(); });