303 lines
10 KiB
JavaScript
303 lines
10 KiB
JavaScript
const { app, BrowserWindow, session, desktopCapturer, ipcMain, clipboard } = require('electron');
|
|
const { autoUpdater } = require('electron-updater');
|
|
const path = require('path');
|
|
const fs = require('fs');
|
|
const url = require('url');
|
|
|
|
const updateState = {
|
|
status: 'idle', // idle | checking | available | downloading | downloaded | not-available | error
|
|
info: null,
|
|
error: null,
|
|
};
|
|
|
|
function broadcastUpdateState() {
|
|
const wins = BrowserWindow.getAllWindows();
|
|
for (const win of wins) {
|
|
if (!win.isDestroyed()) {
|
|
win.webContents.send('update-state', updateState);
|
|
}
|
|
}
|
|
}
|
|
|
|
function getStorageFilePath() {
|
|
return path.join(app.getPath('userData'), 'renderer-storage.json');
|
|
}
|
|
|
|
function readPersistentStore() {
|
|
const filePath = getStorageFilePath();
|
|
try {
|
|
if (!fs.existsSync(filePath)) return {};
|
|
const raw = fs.readFileSync(filePath, 'utf8');
|
|
const parsed = JSON.parse(raw);
|
|
return parsed && typeof parsed === 'object' ? parsed : {};
|
|
} catch (err) {
|
|
console.error('Failed to read persistent store', err);
|
|
return {};
|
|
}
|
|
}
|
|
|
|
function writePersistentStore(store) {
|
|
const filePath = getStorageFilePath();
|
|
try {
|
|
fs.mkdirSync(path.dirname(filePath), { recursive: true });
|
|
fs.writeFileSync(filePath, JSON.stringify(store), 'utf8');
|
|
} catch (err) {
|
|
console.error('Failed to write persistent store', err);
|
|
}
|
|
}
|
|
|
|
function isNewerVersionAvailable(info) {
|
|
const next = info && typeof info.version === 'string' ? info.version : '';
|
|
return Boolean(next) && next !== app.getVersion();
|
|
}
|
|
|
|
async function runUpdateCheck(reason = 'manual') {
|
|
updateState.status = 'checking';
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
try {
|
|
const result = await autoUpdater.checkForUpdates();
|
|
const info = result && result.updateInfo ? result.updateInfo : null;
|
|
if (isNewerVersionAvailable(info)) {
|
|
updateState.status = 'available';
|
|
updateState.info = info;
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
} else {
|
|
updateState.status = 'not-available';
|
|
updateState.info = info;
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
}
|
|
} catch (err) {
|
|
updateState.status = 'error';
|
|
updateState.error = err && err.message ? err.message : String(err);
|
|
broadcastUpdateState();
|
|
console.error(`Update check failed (${reason})`, err);
|
|
}
|
|
}
|
|
|
|
function createWindow() {
|
|
// Create a persistent session for chattz to keep the user logged in
|
|
const sess = session.fromPartition('persist:chattz');
|
|
|
|
const win = new BrowserWindow({
|
|
width: 1200,
|
|
height: 800,
|
|
title: "Chattz Desktop",
|
|
webPreferences: {
|
|
nodeIntegration: false,
|
|
contextIsolation: true,
|
|
webSecurity: false, // Required for cross-origin fetch/ws from file:// with cookies
|
|
session: sess,
|
|
preload: path.join(__dirname, 'desktop', 'preload.js')
|
|
}
|
|
});
|
|
|
|
win.setAutoHideMenuBar(true);
|
|
win.setMenuBarVisibility(false);
|
|
|
|
ipcMain.handle('get-config', () => {
|
|
return {
|
|
backendUrl: (process.env.CHATTZ_URL || 'https://discord.flegr.me').replace(/\/$/, '')
|
|
};
|
|
});
|
|
|
|
ipcMain.handle('clipboard-write', (event, text) => {
|
|
clipboard.writeText(text);
|
|
return true;
|
|
});
|
|
|
|
ipcMain.handle('storage-get', (event, key) => {
|
|
if (typeof key !== 'string' || key.length === 0) return null;
|
|
const store = readPersistentStore();
|
|
return Object.prototype.hasOwnProperty.call(store, key) ? store[key] : null;
|
|
});
|
|
|
|
ipcMain.handle('storage-set', (event, key, value) => {
|
|
if (typeof key !== 'string' || key.length === 0) return false;
|
|
const store = readPersistentStore();
|
|
store[key] = value;
|
|
writePersistentStore(store);
|
|
return true;
|
|
});
|
|
|
|
ipcMain.handle('storage-remove', (event, key) => {
|
|
if (typeof key !== 'string' || key.length === 0) return false;
|
|
const store = readPersistentStore();
|
|
delete store[key];
|
|
writePersistentStore(store);
|
|
return true;
|
|
});
|
|
|
|
ipcMain.handle('get-update-state', () => ({ ...updateState }));
|
|
ipcMain.handle('check-for-updates-now', async () => {
|
|
await runUpdateCheck('renderer-direct');
|
|
return { ...updateState };
|
|
});
|
|
|
|
// Auto-approve media permissions (camera, microphone)
|
|
sess.setPermissionCheckHandler((webContents, permission) => {
|
|
if (permission === 'media' || permission === 'clipboard-read' || permission === 'clipboard-write') {
|
|
return true;
|
|
}
|
|
return false;
|
|
});
|
|
|
|
sess.setPermissionRequestHandler((webContents, permission, callback) => {
|
|
if (permission === 'media' || permission === 'clipboard-read' || permission === 'clipboard-write') {
|
|
callback(true);
|
|
} else {
|
|
callback(false);
|
|
}
|
|
});
|
|
|
|
// Handle screen share requests
|
|
sess.setDisplayMediaRequestHandler((request, callback) => {
|
|
desktopCapturer.getSources({ types: ['screen', 'window'] }).then((sources) => {
|
|
// Provide the first screen source by default, or implement a picker window here
|
|
if (sources && sources.length > 0) {
|
|
// We prefer a screen over a window if available, simple heuristic
|
|
const screenSource = sources.find(s => s.id.startsWith('screen')) || sources[0];
|
|
callback({ video: screenSource, audio: 'loopback' });
|
|
} else {
|
|
callback(null); // Reject safely
|
|
}
|
|
}).catch(err => {
|
|
console.error("Failed to get desktop sources for screen share", err);
|
|
callback(null);
|
|
});
|
|
});
|
|
|
|
const backendUrl = (process.env.CHATTZ_URL || 'https://discord.flegr.me').replace(/\/$/, '');
|
|
const indexPath = path.join(__dirname, 'desktop', 'index.html');
|
|
|
|
const loadDesktopApp = (queryParams = '') => {
|
|
const options = {};
|
|
if (queryParams) {
|
|
try {
|
|
options.query = Object.fromEntries(new URLSearchParams(queryParams));
|
|
} catch (e) {
|
|
console.error("Failed to parse query params", e);
|
|
}
|
|
}
|
|
|
|
// Use setImmediate to ensure the current navigation tick is cleared,
|
|
// which prevents ERR_ABORTED (-3) on some platforms when interrupting a redirect.
|
|
setImmediate(() => {
|
|
if (win.isDestroyed()) return;
|
|
win.loadFile(indexPath, options).catch((err) => {
|
|
// Ignore aborted errors as they often happen during fast redirects
|
|
if (err.toString().includes('-3') || err.code === -3) return;
|
|
console.error(`Failed to load desktop file: ${err}`);
|
|
});
|
|
});
|
|
};
|
|
|
|
// Use a navigation listener to detect when the remote login is complete
|
|
win.webContents.on('will-navigate', (event, navigatedUrl) => {
|
|
try {
|
|
const urlObj = new URL(navigatedUrl);
|
|
const backendObj = new URL(backendUrl);
|
|
// Detect the redirect back to the home page with a token
|
|
if (urlObj.origin === backendObj.origin && urlObj.pathname === '/') {
|
|
if (urlObj.searchParams.has('token')) {
|
|
console.log("Detected login success redirect, returning to desktop UI...");
|
|
event.preventDefault();
|
|
loadDesktopApp(urlObj.search.slice(1));
|
|
}
|
|
}
|
|
} catch (e) {
|
|
console.error(e);
|
|
}
|
|
});
|
|
|
|
loadDesktopApp();
|
|
|
|
// Ensure renderer receives latest updater state after any (re)load.
|
|
win.webContents.on('did-finish-load', () => {
|
|
broadcastUpdateState();
|
|
// Login flow swaps pages; force a check after desktop UI reloads so
|
|
// renderer always gets fresh updater state.
|
|
void runUpdateCheck('did-finish-load');
|
|
});
|
|
|
|
// Uncomment to debug
|
|
// win.webContents.openDevTools();
|
|
}
|
|
|
|
app.commandLine.appendSwitch('disable-webrtc-hw-encoding'); // Sometime helps resolve codec mismatch behavior
|
|
app.commandLine.appendSwitch('log-level', '3'); // Silences standard Chromium warning logs (like SRTP transport unprotect logs which are benign timing issues)
|
|
|
|
app.whenReady().then(() => {
|
|
createWindow();
|
|
|
|
// Configure Auto-Updater
|
|
autoUpdater.autoDownload = false;
|
|
autoUpdater.logger = console;
|
|
|
|
autoUpdater.on('update-available', (info) => {
|
|
updateState.status = 'available';
|
|
updateState.info = info;
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
const wins = BrowserWindow.getAllWindows();
|
|
if (wins.length > 0) wins[0].webContents.send('update-available', info);
|
|
});
|
|
|
|
autoUpdater.on('update-not-available', (info) => {
|
|
updateState.status = 'not-available';
|
|
updateState.info = info || null;
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
});
|
|
|
|
autoUpdater.on('update-downloaded', (info) => {
|
|
updateState.status = 'downloaded';
|
|
updateState.info = info;
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
const wins = BrowserWindow.getAllWindows();
|
|
if (wins.length > 0) wins[0].webContents.send('update-downloaded', info);
|
|
});
|
|
|
|
autoUpdater.on('error', (err) => {
|
|
updateState.status = 'error';
|
|
updateState.error = err.message;
|
|
broadcastUpdateState();
|
|
const wins = BrowserWindow.getAllWindows();
|
|
if (wins.length > 0) wins[0].webContents.send('update-error', err.message);
|
|
});
|
|
|
|
ipcMain.on('check-for-updates', () => {
|
|
void runUpdateCheck('manual');
|
|
});
|
|
|
|
ipcMain.on('download-update', () => {
|
|
updateState.status = 'downloading';
|
|
updateState.error = null;
|
|
broadcastUpdateState();
|
|
autoUpdater.downloadUpdate();
|
|
});
|
|
|
|
ipcMain.on('quit-and-install', () => {
|
|
autoUpdater.quitAndInstall();
|
|
});
|
|
|
|
// Check once on startup
|
|
setTimeout(() => {
|
|
void runUpdateCheck('startup');
|
|
}, 5000);
|
|
|
|
app.on('activate', () => {
|
|
if (BrowserWindow.getAllWindows().length === 0) {
|
|
createWindow();
|
|
}
|
|
});
|
|
});
|
|
|
|
app.on('window-all-closed', () => {
|
|
if (process.platform !== 'darwin') {
|
|
app.quit();
|
|
}
|
|
});
|