vodorod/parcer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add browser state

Browse Source
This commit is contained in:
Rohit
2025-03-12 16:57:23 +05:30
parent 9a2b14b938
commit db4669ab8c
1 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
server/src/browser-management/controller.ts
View File

@@ -21,12 +21,12 @@ import logger from "../logger";
* @category BrowserManagement-Controller * @category BrowserManagement-Controller
*/ */
export const initializeRemoteBrowserForRecording = (userId: string): string => { export const initializeRemoteBrowserForRecording = (userId: string): string => {
const id = getActiveBrowserId(userId) || uuid(); const id = getActiveBrowserIdByState(userId, "recording") || uuid();
createSocketConnection( createSocketConnection(
io.of(id), io.of(id),
async (socket: Socket) => { async (socket: Socket) => {
// browser is already active // browser is already active
const activeId = getActiveBrowserId(userId); const activeId = getActiveBrowserIdByState(userId, "recording");
if (activeId) { if (activeId) {
const remoteBrowser = browserPool.getRemoteBrowser(activeId); const remoteBrowser = browserPool.getRemoteBrowser(activeId);
remoteBrowser?.updateSocket(socket); remoteBrowser?.updateSocket(socket);
@@ -37,7 +37,7 @@ export const initializeRemoteBrowserForRecording = (userId: string): string => {
await browserSession.initialize(userId); await browserSession.initialize(userId);
await browserSession.registerEditorEvents(); await browserSession.registerEditorEvents();
await browserSession.subscribeToScreencast(); await browserSession.subscribeToScreencast();
browserPool.addRemoteBrowser(id, browserSession, userId); browserPool.addRemoteBrowser(id, browserSession, userId, false, "recording");
} }
socket.emit('loaded'); socket.emit('loaded');
}); });
@@ -59,7 +59,7 @@ export const createRemoteBrowserForRun = (userId: string): string => {
async (socket: Socket) => { async (socket: Socket) => {
const browserSession = new RemoteBrowser(socket, userId); const browserSession = new RemoteBrowser(socket, userId);
await browserSession.initialize(userId); await browserSession.initialize(userId);
browserPool.addRemoteBrowser(id, browserSession, userId); browserPool.addRemoteBrowser(id, browserSession, userId, false, "run");
socket.emit('ready-for-run'); socket.emit('ready-for-run');
}); });
return id; return id;
@@ -92,6 +92,17 @@ export const getActiveBrowserId = (userId: string): string | null => {
return browserPool.getActiveBrowserId(userId); return browserPool.getActiveBrowserId(userId);
}; };
/**
* Returns the id of an active browser with the specified state or null.
* @param userId the user ID to find the browser for
* @param state the browser state to filter by ("recording" or "run")
* @returns {string | null}
* @category BrowserManagement-Controller
*/
export const getActiveBrowserIdByState = (userId: string, state: "recording" | "run"): string | null => {
return browserPool.getActiveBrowserId(userId, state);
};
/** /**
* Returns the url string from a remote browser if exists in the browser pool. * Returns the url string from a remote browser if exists in the browser pool.
* @param id instance id of the remote browser * @param id instance id of the remote browser
@@ -127,7 +138,7 @@ export const getRemoteBrowserCurrentTabs = (id: string, userId: string): string[
* @category BrowserManagement-Controller * @category BrowserManagement-Controller
*/ */
export const interpretWholeWorkflow = async (userId: string) => { export const interpretWholeWorkflow = async (userId: string) => {
const id = getActiveBrowserId(userId); const id = getActiveBrowserIdByState(userId, "recording");
if (id) { if (id) {
const browser = browserPool.getRemoteBrowser(id); const browser = browserPool.getRemoteBrowser(id);
if (browser) { if (browser) {
@@ -147,7 +158,7 @@ export const interpretWholeWorkflow = async (userId: string) => {
* @category BrowserManagement-Controller * @category BrowserManagement-Controller
*/ */
export const stopRunningInterpretation = async (userId: string) => { export const stopRunningInterpretation = async (userId: string) => {
const id = getActiveBrowserId(userId); const id = getActiveBrowserIdByState(userId, "recording");
if (id) { if (id) {
const browser = browserPool.getRemoteBrowser(id); const browser = browserPool.getRemoteBrowser(id);
await browser?.stopCurrentInterpretation(); await browser?.stopCurrentInterpretation();