import * as shotPool from './lib/shot-pool'; import * as Logger from 'log4js'; import * as express from 'express'; import * as internalIp from 'internal-ip'; process.env.loglevel = process.env.loglevel || 'INFO'; process.env.webshotDebugPort = process.env.webshotDebugPort || '3030'; Logger.setGlobalLogLevel(process.env.loglevel); let app = express(); let ip; let _logger = Logger.getLogger("index"); internalIp.v4().then(_ip => ip = _ip); export interface Config extends shotPool.PoolConfig { webshotDebugPort?: string; } let defaultConfig: Config = { concurrency: 10, callbackName: '', warmerUrl: '', width: 800, height: 600, timeout: 60000, webshotDebugPort: process.env.webshotDebugPort }; let passedConfig : Config; export async function init(options: Config) { options = Object.assign({}, defaultConfig, options); passedConfig = options; await shotPool.create( options ); app.listen(parseInt(passedConfig.webshotDebugPort), ip); _logger.info('Listening on debug port', passedConfig.webshotDebugPort, ip); return shotPool; } export function getShot(url: string): PromiseLike { return shotPool.getShot(url); } app.get('/status', (req, res) => { let status = shotPool.getStatus(); let workerDetails = status.allWorkers.map(worker => { let workerStatus = worker.getStatus(); let link = `http://${ip}:${workerStatus.debugPort}`; return `

Worker Id: #${workerStatus.id}

ConnectWS: ${workerStatus.browser.wsEndpoint()}

DebugLink: ${link}

`; }).join(''); res.end(` Webshot Factory Status

Webshot Factory Status

Job Queue

Number of Jobs in queue: ${status.jobQueue.getStatus().jobs.length}

Total Jobs processed: ${status.jobQueue.getStatus().total}

Workers

Total Workers: ${status.allWorkers.length}

Idle Workers: ${status.idleWorkers.length}

Worker details

${workerDetails} `) });