boku7-Loki/agent/agent.js

const { generateUUID, generateAESKey } = require('./crypt.js');

class Window {
    constructor() {
        this.BrowserWindow = null;
    }

    setBrowserWindow(BrowserWindow) {
        this.BrowserWindow = BrowserWindow;
    }
}

class Container {
    constructor() {
        this.name = 's' + generateUUID(10);
        this.key = generateAESKey();
        this.blobs = {
            'key': 'k-' + generateUUID(10),
            'checkin': 'c-' + generateUUID(10),
            'in': 'i-' + generateUUID(10),
            'out': 'o-' + generateUUID(10)
        };
    }

    setName(name) {
        this.name = name;
    }

    setKey(key) {
        this.key = {
            'key': key.key,
            'iv': key.iv
        };
    }
}

class Agent {
    constructor(windowid, status = null) {
        this.window = new Window(windowid, status);
        this.agentid = 'a' + generateUUID(16);
        this.container = new Container();
        this.checkin = Date.now();
        this.sleepinterval = 5;
        this.sleepjitter = 15;
        this.thissleep = 5000;
    }

    setAgentId(agentid) {
        this.agentid = agentid;
    }

    setContainer(container) {
        this.container = container;
    }

    // Method to replace the window
    setWindow(window) {
        this.window = window;
    }
}

module.exports = {
    Agent,
    Window,
    Container
};