mirror of
https://github.com/LightZirconite/Microsoft-Rewards-Bot.git
synced 2026-01-10 17:26:17 +00:00
refactor: simplify bot control logic and improve error handling in routes
This commit is contained in:
@@ -1,15 +1,12 @@
|
||||
import { dashboardState } from './state'
|
||||
|
||||
// We'll import and run the bot directly in the same process
|
||||
let botRunning = false
|
||||
let botPromise: Promise<void> | null = null
|
||||
import type { MicrosoftRewardsBot } from '../index'
|
||||
|
||||
export class BotController {
|
||||
private isRunning: boolean = false
|
||||
private botInstance: MicrosoftRewardsBot | null = null
|
||||
private botPromise: Promise<void> | null = null
|
||||
private startTime?: Date
|
||||
|
||||
constructor() {
|
||||
// Cleanup on exit
|
||||
process.on('exit', () => this.stop())
|
||||
}
|
||||
|
||||
@@ -26,30 +23,26 @@ export class BotController {
|
||||
}
|
||||
|
||||
public async start(): Promise<{ success: boolean; error?: string; pid?: number }> {
|
||||
if (this.isRunning || botRunning) {
|
||||
if (this.botInstance) {
|
||||
return { success: false, error: 'Bot is already running' }
|
||||
}
|
||||
|
||||
try {
|
||||
this.log('🚀 Starting bot...', 'log')
|
||||
|
||||
// Import the bot main logic
|
||||
const { MicrosoftRewardsBot } = await import('../index')
|
||||
|
||||
this.isRunning = true
|
||||
botRunning = true
|
||||
this.botInstance = new MicrosoftRewardsBot(false)
|
||||
this.startTime = new Date()
|
||||
dashboardState.setRunning(true)
|
||||
dashboardState.setBotInstance(this.botInstance)
|
||||
|
||||
// Run the bot in the same process using the exact same logic as npm start
|
||||
botPromise = (async () => {
|
||||
this.botPromise = (async () => {
|
||||
try {
|
||||
const rewardsBot = new MicrosoftRewardsBot(false)
|
||||
|
||||
this.log('✓ Bot initialized, starting execution...', 'log')
|
||||
|
||||
await rewardsBot.initialize()
|
||||
await rewardsBot.run()
|
||||
await this.botInstance!.initialize()
|
||||
await this.botInstance!.run()
|
||||
|
||||
this.log('✓ Bot completed successfully', 'log')
|
||||
} catch (error) {
|
||||
@@ -61,8 +54,7 @@ export class BotController {
|
||||
}
|
||||
})()
|
||||
|
||||
// Don't await - let it run in background
|
||||
botPromise.catch(error => {
|
||||
this.botPromise.catch(error => {
|
||||
this.log(`Bot execution failed: ${error instanceof Error ? error.message : String(error)}`, 'error')
|
||||
})
|
||||
|
||||
@@ -77,15 +69,12 @@ export class BotController {
|
||||
}
|
||||
|
||||
public stop(): { success: boolean; error?: string } {
|
||||
if (!this.isRunning && !botRunning) {
|
||||
if (!this.botInstance) {
|
||||
return { success: false, error: 'Bot is not running' }
|
||||
}
|
||||
|
||||
try {
|
||||
this.log('🛑 Stopping bot...', 'warn')
|
||||
|
||||
// For now, we can't gracefully stop a running bot in the same process
|
||||
// This would require refactoring the bot to support cancellation
|
||||
this.log('⚠ Note: Bot will complete current task before stopping', 'warn')
|
||||
|
||||
this.cleanup()
|
||||
@@ -103,7 +92,6 @@ export class BotController {
|
||||
this.log('🔄 Restarting bot...', 'log')
|
||||
this.stop()
|
||||
|
||||
// Wait a bit before restarting
|
||||
return new Promise((resolve) => {
|
||||
setTimeout(async () => {
|
||||
const result = await this.start()
|
||||
@@ -119,7 +107,7 @@ export class BotController {
|
||||
startTime?: string
|
||||
} {
|
||||
return {
|
||||
running: this.isRunning || botRunning,
|
||||
running: !!this.botInstance,
|
||||
pid: process.pid,
|
||||
uptime: this.startTime ? Date.now() - this.startTime.getTime() : undefined,
|
||||
startTime: this.startTime?.toISOString()
|
||||
@@ -127,13 +115,12 @@ export class BotController {
|
||||
}
|
||||
|
||||
private cleanup(): void {
|
||||
this.isRunning = false
|
||||
botRunning = false
|
||||
botPromise = null
|
||||
this.botInstance = null
|
||||
this.botPromise = null
|
||||
this.startTime = undefined
|
||||
dashboardState.setRunning(false)
|
||||
dashboardState.setBotInstance(undefined)
|
||||
}
|
||||
}
|
||||
|
||||
// Singleton instance
|
||||
export const botController = new BotController()
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
import { Router, Request, Response } from 'express'
|
||||
import fs from 'fs'
|
||||
import path from 'path'
|
||||
import { spawn } from 'child_process'
|
||||
import { dashboardState } from './state'
|
||||
import { loadAccounts, loadConfig, getConfigPath } from '../util/Load'
|
||||
import { botController } from './BotController'
|
||||
@@ -209,7 +208,7 @@ apiRouter.post('/restart', async (_req: Request, res: Response): Promise<void> =
|
||||
}
|
||||
})
|
||||
|
||||
// POST /api/sync/:email - Force sync single account
|
||||
// POST /api/sync/:email - Force sync single account (deprecated - use full bot restart)
|
||||
apiRouter.post('/sync/:email', async (req: Request, res: Response): Promise<void> => {
|
||||
try {
|
||||
const { email } = req.params
|
||||
@@ -226,18 +225,10 @@ apiRouter.post('/sync/:email', async (req: Request, res: Response): Promise<void
|
||||
return
|
||||
}
|
||||
|
||||
dashboardState.updateAccount(email, { status: 'running', lastSync: new Date().toISOString() })
|
||||
|
||||
// Spawn single account run
|
||||
const child = spawn(process.execPath, [
|
||||
path.join(process.cwd(), 'dist', 'index.js'),
|
||||
'-account',
|
||||
email
|
||||
], { detached: true, stdio: 'ignore' })
|
||||
|
||||
if (child.unref) child.unref()
|
||||
|
||||
res.json({ success: true, pid: child.pid || undefined })
|
||||
res.status(501).json({
|
||||
error: 'Single account sync not implemented. Please use restart bot instead.',
|
||||
suggestion: 'Use /api/restart endpoint to run all accounts'
|
||||
})
|
||||
} catch (error) {
|
||||
res.status(500).json({ error: error instanceof Error ? error.message : 'Unknown error' })
|
||||
}
|
||||
@@ -318,7 +309,13 @@ apiRouter.post('/account/:email/reset', (req: Request, res: Response): void => {
|
||||
function maskUrl(url: string): string {
|
||||
try {
|
||||
const parsed = new URL(url)
|
||||
return `${parsed.protocol}//${parsed.hostname.slice(0, 3)}***${parsed.pathname.slice(0, 5)}***`
|
||||
const maskedHost = parsed.hostname.length > 6
|
||||
? `${parsed.hostname.slice(0, 3)}***${parsed.hostname.slice(-3)}`
|
||||
: '***'
|
||||
const maskedPath = parsed.pathname.length > 5
|
||||
? `${parsed.pathname.slice(0, 3)}***`
|
||||
: '***'
|
||||
return `${parsed.protocol}//${maskedHost}${maskedPath}`
|
||||
} catch {
|
||||
return '***'
|
||||
}
|
||||
|
||||
@@ -108,17 +108,14 @@ export class DashboardServer {
|
||||
}
|
||||
|
||||
private interceptBotLogs(): void {
|
||||
// Store reference to this.clients for closure
|
||||
const clients = this.clients
|
||||
|
||||
// Intercept bot logs and forward to dashboard
|
||||
const originalLog = botLog
|
||||
;(global as Record<string, unknown>).botLog = function(
|
||||
|
||||
;(global as Record<string, unknown>).botLog = (
|
||||
isMobile: boolean | 'main',
|
||||
title: string,
|
||||
message: string,
|
||||
type: 'log' | 'warn' | 'error' = 'log'
|
||||
) {
|
||||
) => {
|
||||
const result = originalLog(isMobile, title, message, type)
|
||||
|
||||
const logEntry: DashboardLog = {
|
||||
@@ -130,18 +127,7 @@ export class DashboardServer {
|
||||
}
|
||||
|
||||
dashboardState.addLog(logEntry)
|
||||
|
||||
// Broadcast to WebSocket clients
|
||||
const payload = JSON.stringify({ type: 'log', log: logEntry })
|
||||
for (const client of clients) {
|
||||
if (client.readyState === WebSocket.OPEN) {
|
||||
try {
|
||||
client.send(payload)
|
||||
} catch (error) {
|
||||
console.error('[Dashboard] Error sending to WebSocket client:', error)
|
||||
}
|
||||
}
|
||||
}
|
||||
this.broadcastUpdate('log', { log: logEntry })
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
14
src/index.ts
14
src/index.ts
@@ -48,16 +48,12 @@ export class MicrosoftRewardsBot {
|
||||
public queryEngine?: QueryDiversityEngine
|
||||
public compromisedModeActive: boolean = false
|
||||
public compromisedReason?: string
|
||||
// Mutex-like flag to prevent parallel execution when config.parallel is accidentally misconfigured
|
||||
private isDesktopRunning: boolean = false
|
||||
private isMobileRunning: boolean = false
|
||||
|
||||
private activeWorkers: number
|
||||
private browserFactory: Browser = new Browser(this)
|
||||
private accounts: Account[]
|
||||
private workers: Workers
|
||||
private login = new Login(this)
|
||||
// Buy mode (manual spending) tracking
|
||||
private buyMode: { enabled: boolean; email?: string } = { enabled: false }
|
||||
|
||||
// Summary collection (per process)
|
||||
@@ -632,12 +628,7 @@ export class MicrosoftRewardsBot {
|
||||
}
|
||||
} else {
|
||||
// Sequential execution with safety checks
|
||||
if (this.isDesktopRunning || this.isMobileRunning) {
|
||||
log('main', 'TASK', `Race condition detected: Desktop=${this.isDesktopRunning}, Mobile=${this.isMobileRunning}. Skipping to prevent conflicts.`, 'error')
|
||||
errors.push('race-condition-detected')
|
||||
} else {
|
||||
this.isMobile = false
|
||||
this.isDesktopRunning = true
|
||||
const desktopResult = await this.Desktop(account).catch(e => {
|
||||
const msg = e instanceof Error ? e.message : String(e)
|
||||
this.recordRiskEvent('error', 6, `desktop:${msg}`)
|
||||
@@ -654,12 +645,9 @@ export class MicrosoftRewardsBot {
|
||||
desktopInitial = desktopResult.initialPoints
|
||||
desktopCollected = desktopResult.collectedPoints
|
||||
}
|
||||
this.isDesktopRunning = false
|
||||
|
||||
// If banned or compromised detected, skip mobile to save time
|
||||
if (!banned.status && !this.compromisedModeActive) {
|
||||
this.isMobile = true
|
||||
this.isMobileRunning = true
|
||||
const mobileResult = await this.Mobile(account).catch((e: unknown) => {
|
||||
const msg = e instanceof Error ? e.message : String(e)
|
||||
this.recordRiskEvent('error', 6, `mobile:${msg}`)
|
||||
@@ -676,13 +664,11 @@ export class MicrosoftRewardsBot {
|
||||
mobileInitial = mobileResult.initialPoints
|
||||
mobileCollected = mobileResult.collectedPoints
|
||||
}
|
||||
this.isMobileRunning = false
|
||||
} else {
|
||||
const why = banned.status ? 'banned status' : 'compromised status'
|
||||
log(true, 'TASK', `Skipping mobile flow for ${account.email} due to ${why}`, 'warn')
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const accountEnd = Date.now()
|
||||
const durationMs = accountEnd - accountStart
|
||||
|
||||
Reference in New Issue
Block a user