sam-coder-cli 1.0.69 → 2.0.1

package/bin/core/models.js

@@ -0,0 +1,290 @@
+/**
+ * Brainstorm Core - Models
+ * 
+ * Data models for project configuration, agent configuration, and session state.
+ * Ported from Python to JavaScript.
+ */
+
+const fs = require('fs').promises;
+const path = require('path');
+
+/**
+ * Agent configuration - defines an individual AI agent
+ */
+class AgentConfig {
+    constructor({
+        id,
+        name,
+        description,
+        priority = 'MEDIUM',
+        inputs = [],
+        outputs = [],
+        technologies = [],
+        validationRules = [],
+        territory = []
+    }) {
+        this.id = id;
+        this.name = name;
+        this.description = description;
+        this.priority = priority; // 'LOW', 'MEDIUM', 'HIGH'
+        this.inputs = inputs;
+        this.outputs = outputs;
+        this.technologies = technologies;
+        this.validationRules = validationRules;
+        this.territory = territory;
+    }
+
+    toJSON() {
+        return {
+            id: this.id,
+            name: this.name,
+            description: this.description,
+            priority: this.priority,
+            inputs: this.inputs,
+            outputs: this.outputs,
+            technologies: this.technologies,
+            validationRules: this.validationRules,
+            territory: this.territory
+        };
+    }
+
+    static fromJSON(data) {
+        return new AgentConfig(data);
+    }
+}
+
+/**
+ * Project configuration - defines project-level details
+ */
+class ProjectConfig {
+    constructor({
+        name,
+        description,
+        version = '1.0',
+        agents = [],
+        dataModel = {},
+        technologyStack = [],
+        governanceRules = [],
+        successCriteria = [],
+        directoryStructure = {},
+        inputFiles = []
+    }) {
+        this.name = name;
+        this.description = description;
+        this.version = version;
+        this.agents = agents.map(a => a instanceof AgentConfig ? a : new AgentConfig(a));
+        this.dataModel = dataModel;
+        this.technologyStack = technologyStack;
+        this.governanceRules = governanceRules;
+        this.successCriteria = successCriteria;
+        this.directoryStructure = directoryStructure;
+        this.inputFiles = inputFiles;
+    }
+
+    getAgent(id) {
+        return this.agents.find(a => a.id === id);
+    }
+
+    toJSON() {
+        return {
+            name: this.name,
+            description: this.description,
+            version: this.version,
+            agents: this.agents.map(a => a.toJSON()),
+            dataModel: this.dataModel,
+            technologyStack: this.technologyStack,
+            governanceRules: this.governanceRules,
+            successCriteria: this.successCriteria,
+            directoryStructure: this.directoryStructure,
+            inputFiles: this.inputFiles
+        };
+    }
+
+    static fromJSON(data) {
+        return new ProjectConfig({
+            ...data,
+            agents: (data.agents || []).map(a => AgentConfig.fromJSON(a))
+        });
+    }
+}
+
+/**
+ * Session event - logs individual events within a session
+ */
+class SessionEvent {
+    constructor({
+        timestamp = new Date().toISOString(),
+        eventType,
+        actor,
+        description,
+        affectedFiles = []
+    }) {
+        this.timestamp = timestamp;
+        this.eventType = eventType; // 'CREATED', 'UPDATED', 'EDITED', 'COMPLETED'
+        this.actor = actor;
+        this.description = description;
+        this.affectedFiles = affectedFiles;
+    }
+
+    toJSON() {
+        return {
+            timestamp: this.timestamp,
+            eventType: this.eventType,
+            actor: this.actor,
+            description: this.description,
+            affectedFiles: this.affectedFiles
+        };
+    }
+
+    static fromJSON(data) {
+        return new SessionEvent(data);
+    }
+}
+
+/**
+ * Brainstorm session - manages session state
+ */
+class BrainstormSession {
+    constructor({
+        id,
+        project,
+        createdAt = new Date().toISOString(),
+        updatedAt = new Date().toISOString(),
+        status = 'ACTIVE',
+        agentsParticipating = [],
+        outputDirectory,
+        history = [],
+        fileVersions = {},
+        mutualSummary = null
+    }) {
+        this.id = id;
+        this.project = project instanceof ProjectConfig ? project : ProjectConfig.fromJSON(project);
+        this.createdAt = createdAt;
+        this.updatedAt = updatedAt;
+        this.status = status; // 'ACTIVE', 'COMPLETED', 'ARCHIVED'
+        this.agentsParticipating = agentsParticipating;
+        this.outputDirectory = outputDirectory;
+        this.history = history.map(e => e instanceof SessionEvent ? e : SessionEvent.fromJSON(e));
+        this.fileVersions = fileVersions;
+        this.mutualSummary = mutualSummary;
+    }
+
+    static create({ project, outputDirectory, agents = ['CLAUDE-1'] }) {
+        const session = new BrainstormSession({
+            id: `session-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`,
+            project,
+            outputDirectory,
+            agentsParticipating: agents
+        });
+
+        session.addEvent({
+            eventType: 'CREATED',
+            actor: 'BRAINSTORM',
+            description: `Brainstorm session created for project '${project.name}'`
+        });
+
+        return session;
+    }
+
+    addEvent({ eventType, actor, description, affectedFiles = [] }) {
+        this.history.push(new SessionEvent({
+            eventType,
+            actor,
+            description,
+            affectedFiles
+        }));
+        this.updatedAt = new Date().toISOString();
+    }
+
+    getFileVersion(fileName) {
+        return this.fileVersions[fileName] || 0;
+    }
+
+    incrementFileVersion(fileName) {
+        this.fileVersions[fileName] = (this.fileVersions[fileName] || 0) + 1;
+        return this.fileVersions[fileName];
+    }
+
+    toJSON() {
+        return {
+            id: this.id,
+            project: this.project.toJSON(),
+            createdAt: this.createdAt,
+            updatedAt: this.updatedAt,
+            status: this.status,
+            agentsParticipating: this.agentsParticipating,
+            outputDirectory: this.outputDirectory,
+            history: this.history.map(e => e.toJSON()),
+            fileVersions: this.fileVersions,
+            mutualSummary: this.mutualSummary
+        };
+    }
+
+    static fromJSON(data) {
+        return new BrainstormSession(data);
+    }
+
+    async save(filePath = null) {
+        const savePath = filePath || path.join(this.outputDirectory, 'SESSION.json');
+        await fs.writeFile(savePath, JSON.stringify(this.toJSON(), null, 2), 'utf-8');
+        return savePath;
+    }
+
+    static async load(filePath) {
+        const content = await fs.readFile(filePath, 'utf-8');
+        return BrainstormSession.fromJSON(JSON.parse(content));
+    }
+}
+
+/**
+ * Session result - encapsulates outcome of operations
+ */
+class SessionResult {
+    constructor({
+        success,
+        sessionId,
+        status,
+        summary,
+        filesGenerated = [],
+        errors = []
+    }) {
+        this.success = success;
+        this.sessionId = sessionId;
+        this.status = status;
+        this.summary = summary;
+        this.filesGenerated = filesGenerated;
+        this.errors = errors;
+    }
+}
+
+/**
+ * Edit info - tracks edits made to completed files
+ */
+class EditInfo {
+    constructor({
+        timestamp = new Date().toISOString(),
+        filePath,
+        editor,
+        description,
+        versionBefore,
+        versionAfter,
+        backupPath = null
+    }) {
+        this.timestamp = timestamp;
+        this.filePath = filePath;
+        this.editor = editor;
+        this.description = description;
+        this.versionBefore = versionBefore;
+        this.versionAfter = versionAfter;
+        this.backupPath = backupPath;
+    }
+}
+
+module.exports = {
+    AgentConfig,
+    ProjectConfig,
+    SessionEvent,
+    BrainstormSession,
+    SessionResult,
+    EditInfo
+};