multipleof4/KalBot
1
0
Fork 0
mirror of https://github.com/multipleof4/KalBot.git synced 2026-03-16 21:41:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Auto-load strategies dynamically from folder

Browse Source
This commit is contained in:
multipleof4
2026-03-16 00:33:56 -07:00
parent 29fd889acb
commit fc8565e562
1 changed files with 26 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
worker.js
View File

@@ -1,17 +1,13 @@
import { MarketTracker } from './lib/market/tracker.js'; import { MarketTracker } from './lib/market/tracker.js';
import { PaperEngine } from './lib/paper/engine.js'; import { PaperEngine } from './lib/paper/engine.js';
import { MartingaleStrategy } from './lib/strategies/martingale.js';
import { MartingaleAlphaStrategy } from './lib/strategies/martingale-alpha.js';
import { ThresholdStrategy } from './lib/strategies/threshold.js';
import { BullDipBuyer } from './lib/strategies/bull-dip-buyer.js';
import { SniperReversalStrategy } from './lib/strategies/sniper-reversal.js';
import { MomentumRiderStrategy } from './lib/strategies/momentum-rider.js';
import { DontDoubtBullStrategy } from './lib/strategies/dont-doubt-bull.js';
import { getMarket } from './lib/kalshi/rest.js'; import { getMarket } from './lib/kalshi/rest.js';
import { db } from './lib/db.js'; import { db } from './lib/db.js';
import { notify } from './lib/notify.js'; import { notify } from './lib/notify.js';
import fs from 'fs'; import fs from 'fs';
import path from 'path';
import { fileURLToPath, pathToFileURL } from 'url';
const __dirname = path.dirname(fileURLToPath(import.meta.url));
const STATE_FILE = '/tmp/kalbot-state.json'; const STATE_FILE = '/tmp/kalbot-state.json';
const HEARTBEAT_MS = 2000; const HEARTBEAT_MS = 2000;
@@ -35,22 +31,33 @@ async function main() {
const paper = new PaperEngine(1000); const paper = new PaperEngine(1000);
await paper.init(); await paper.init();
// Load all 7 strategies! // Dynamically load all strategies from the strategies directory!
const strategies = [ const strategies = [];
new MartingaleStrategy({ threshold: 70, baseBet: 1, maxDoublings: 5 }), const strategiesDir = path.join(__dirname, 'lib', 'strategies');
new MartingaleAlphaStrategy({ minPct: 40, maxPct: 60, baseBet: 1, maxRounds: 3 }), const stratFiles = fs.readdirSync(strategiesDir).filter(f => f.endsWith('.js') && f !== 'base.js');
new ThresholdStrategy({ triggerPct: 65, betSize: 1 }),
new BullDipBuyer({ maxYesPrice: 45, minYesPrice: 15, betSize: 2 }), for (const file of stratFiles) {
new SniperReversalStrategy({ triggerPct: 95, minsLeft: 3, betSize: 1 }), try {
new MomentumRiderStrategy({ triggerPct: 75, betSize: 2 }), const fileUrl = pathToFileURL(path.join(strategiesDir, file)).href;
new DontDoubtBullStrategy({ minYesPct: 30, maxYesPct: 40, betSize: 2 }) const mod = await import(fileUrl);
];
// Grab the first exported class/function
for (const key in mod) {
if (typeof mod[key] === 'function') {
strategies.push(new mod[key]()); // Instances use their own internal default configs!
break;
}
}
} catch (err) {
console.error(`[Worker] Failed to load strategy ${file}:`, err.message);
}
}
for (const s of strategies) { for (const s of strategies) {
paper._getAccount(s.name); paper._getAccount(s.name);
} }
console.log(`[Worker] Loaded ${strategies.length} strategies: ${strategies.map((s) => s.name).join(', ')}`); console.log(`[Worker] Dynamically loaded ${strategies.length} strategies: ${strategies.map((s) => s.name).join(', ')}`);
let latestMarketState = null; let latestMarketState = null;
@@ -146,7 +153,7 @@ async function main() {
}); });
await tracker.start(); await tracker.start();
await notify('🤖 Kalbot Worker started with 7 strats!', 'Kalbot Online', 'low', 'robot,green_circle'); await notify(`🤖 Kalbot Worker started with ${strategies.length} strats!`, 'Kalbot Online', 'low', 'robot,green_circle');
heartbeatTimer = setInterval(() => { heartbeatTimer = setInterval(() => {
writeState(latestMarketState, paper, strategies); writeState(latestMarketState, paper, strategies);