@2501-ai/cli/dist/utils/conf.js

[](https://www.npmjs.com/package/@2501-ai/cli) [](https://www.2501.ai/research/full-humaneval-benchmark) [![Lic

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getEligibleAgent = exports.flushAgents = exports.addAgent = exports.listAgentsFromWorkspace = exports.listAgents = exports.writeConfig = exports.setValue = exports.readConfig = void 0;
const fs_1 = __importDefault(require("fs"));
const path = __importStar(require("path"));
const logger_1 = __importDefault(require("../utils/logger"));
const workspace_1 = require("../helpers/workspace");
const constants_1 = require("../constants");
const CONFIG_FILE_PATH = path.join(constants_1.CONFIG_DIR, '2501.conf');
function readConfig() {
    try {
        if (!fs_1.default.existsSync(CONFIG_FILE_PATH)) {
            fs_1.default.mkdirSync(path.dirname(CONFIG_FILE_PATH), { recursive: true });
            fs_1.default.writeFileSync(CONFIG_FILE_PATH, JSON.stringify({ workspace_disabled: false, agents: [], stream: true }, null, 2), 'utf8');
        }
        const data = fs_1.default.readFileSync(CONFIG_FILE_PATH, 'utf8');
        return JSON.parse(data);
    }
    catch (error) {
        logger_1.default.error('Error reading configuration:', error);
        return null;
    }
}
exports.readConfig = readConfig;
function setValue(key, value) {
    try {
        const config = readConfig();
        if (config) {
            config[key] = value;
            writeConfig(config);
        }
    }
    catch (error) {
        logger_1.default.error('Error setting value:', error);
    }
}
exports.setValue = setValue;
function writeConfig(config) {
    try {
        const data = JSON.stringify(config, null, 2);
        fs_1.default.writeFileSync(CONFIG_FILE_PATH, data, 'utf8');
    }
    catch (error) {
        logger_1.default.error('Error writing configuration:', error);
    }
}
exports.writeConfig = writeConfig;
function listAgents() {
    const config = readConfig();
    if (config) {
        return config.agents;
    }
    else {
        return [];
    }
}
exports.listAgents = listAgents;
function listAgentsFromWorkspace(workspaceUrl) {
    const config = readConfig();
    if (config) {
        return config.agents.filter((agent) => agent.workspace === workspaceUrl);
    }
    else {
        return [];
    }
}
exports.listAgentsFromWorkspace = listAgentsFromWorkspace;
function addAgent(newAgent) {
    const config = readConfig();
    if (config) {
        const existingAgent = config.agents.find((agent) => agent.id === newAgent.id);
        if (existingAgent) {
            logger_1.default.error('Agent with the same ID already exists:', newAgent.id);
            return;
        }
        config.agents.push(newAgent);
        writeConfig(config);
    }
    else {
        logger_1.default.error('Invalid configuration');
    }
}
exports.addAgent = addAgent;
function flushAgents(workspacePath, all) {
    return __awaiter(this, void 0, void 0, function* () {
        try {
            const config = readConfig();
            if (!config) {
                return;
            }
            if (!all) {
                config.agents = config.agents.filter((agent) => {
                    const isInWorkspace = agent.workspace === workspacePath;
                    if (isInWorkspace) {
                        logger_1.default.debug('Flushing agent:', agent.id);
                        (0, workspace_1.clearWorkspaceState)(agent.id);
                    }
                    return !isInWorkspace;
                });
                writeConfig(config);
                return;
            }
            config.agents.forEach((agent) => (0, workspace_1.clearWorkspaceState)(agent.id));
            config.agents = [];
            writeConfig(config);
        }
        catch (error) {
            logger_1.default.error('Error flushing agents:', error);
        }
    });
}
exports.flushAgents = flushAgents;
function getEligibleAgent(workspace) {
    const agents = workspace ? listAgentsFromWorkspace(workspace) : listAgents();
    return agents[agents.length - 1];
}
exports.getEligibleAgent = getEligibleAgent;