mirror of
https://github.com/AutoMaker-Org/automaker.git
synced 2026-02-04 21:23:07 +00:00
adding a queue system to the agent runner
This commit is contained in:
@@ -48,6 +48,7 @@ import { createClaudeRoutes } from './routes/claude/index.js';
|
|||||||
import { ClaudeUsageService } from './services/claude-usage-service.js';
|
import { ClaudeUsageService } from './services/claude-usage-service.js';
|
||||||
import { createGitHubRoutes } from './routes/github/index.js';
|
import { createGitHubRoutes } from './routes/github/index.js';
|
||||||
import { createContextRoutes } from './routes/context/index.js';
|
import { createContextRoutes } from './routes/context/index.js';
|
||||||
|
import { createBacklogPlanRoutes } from './routes/backlog-plan/index.js';
|
||||||
import { cleanupStaleValidations } from './routes/github/routes/validation-common.js';
|
import { cleanupStaleValidations } from './routes/github/routes/validation-common.js';
|
||||||
|
|
||||||
// Load environment variables
|
// Load environment variables
|
||||||
@@ -160,6 +161,7 @@ app.use('/api/settings', createSettingsRoutes(settingsService));
|
|||||||
app.use('/api/claude', createClaudeRoutes(claudeUsageService));
|
app.use('/api/claude', createClaudeRoutes(claudeUsageService));
|
||||||
app.use('/api/github', createGitHubRoutes(events, settingsService));
|
app.use('/api/github', createGitHubRoutes(events, settingsService));
|
||||||
app.use('/api/context', createContextRoutes(settingsService));
|
app.use('/api/context', createContextRoutes(settingsService));
|
||||||
|
app.use('/api/backlog-plan', createBacklogPlanRoutes(events, settingsService));
|
||||||
|
|
||||||
// Create HTTP server
|
// Create HTTP server
|
||||||
const server = createServer(app);
|
const server = createServer(app);
|
||||||
|
|||||||
@@ -12,6 +12,10 @@ import { createHistoryHandler } from './routes/history.js';
|
|||||||
import { createStopHandler } from './routes/stop.js';
|
import { createStopHandler } from './routes/stop.js';
|
||||||
import { createClearHandler } from './routes/clear.js';
|
import { createClearHandler } from './routes/clear.js';
|
||||||
import { createModelHandler } from './routes/model.js';
|
import { createModelHandler } from './routes/model.js';
|
||||||
|
import { createQueueAddHandler } from './routes/queue-add.js';
|
||||||
|
import { createQueueListHandler } from './routes/queue-list.js';
|
||||||
|
import { createQueueRemoveHandler } from './routes/queue-remove.js';
|
||||||
|
import { createQueueClearHandler } from './routes/queue-clear.js';
|
||||||
|
|
||||||
export function createAgentRoutes(agentService: AgentService, _events: EventEmitter): Router {
|
export function createAgentRoutes(agentService: AgentService, _events: EventEmitter): Router {
|
||||||
const router = Router();
|
const router = Router();
|
||||||
@@ -27,5 +31,15 @@ export function createAgentRoutes(agentService: AgentService, _events: EventEmit
|
|||||||
router.post('/clear', createClearHandler(agentService));
|
router.post('/clear', createClearHandler(agentService));
|
||||||
router.post('/model', createModelHandler(agentService));
|
router.post('/model', createModelHandler(agentService));
|
||||||
|
|
||||||
|
// Queue routes
|
||||||
|
router.post(
|
||||||
|
'/queue/add',
|
||||||
|
validatePathParams('imagePaths[]'),
|
||||||
|
createQueueAddHandler(agentService)
|
||||||
|
);
|
||||||
|
router.post('/queue/list', createQueueListHandler(agentService));
|
||||||
|
router.post('/queue/remove', createQueueRemoveHandler(agentService));
|
||||||
|
router.post('/queue/clear', createQueueClearHandler(agentService));
|
||||||
|
|
||||||
return router;
|
return router;
|
||||||
}
|
}
|
||||||
|
|||||||
34
apps/server/src/routes/agent/routes/queue-add.ts
Normal file
34
apps/server/src/routes/agent/routes/queue-add.ts
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
/**
|
||||||
|
* POST /queue/add endpoint - Add a prompt to the queue
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import { AgentService } from '../../../services/agent-service.js';
|
||||||
|
import { getErrorMessage, logError } from '../common.js';
|
||||||
|
|
||||||
|
export function createQueueAddHandler(agentService: AgentService) {
|
||||||
|
return async (req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const { sessionId, message, imagePaths, model } = req.body as {
|
||||||
|
sessionId: string;
|
||||||
|
message: string;
|
||||||
|
imagePaths?: string[];
|
||||||
|
model?: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!sessionId || !message) {
|
||||||
|
res.status(400).json({
|
||||||
|
success: false,
|
||||||
|
error: 'sessionId and message are required',
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await agentService.addToQueue(sessionId, { message, imagePaths, model });
|
||||||
|
res.json(result);
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Add to queue failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
29
apps/server/src/routes/agent/routes/queue-clear.ts
Normal file
29
apps/server/src/routes/agent/routes/queue-clear.ts
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
/**
|
||||||
|
* POST /queue/clear endpoint - Clear all prompts from the queue
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import { AgentService } from '../../../services/agent-service.js';
|
||||||
|
import { getErrorMessage, logError } from '../common.js';
|
||||||
|
|
||||||
|
export function createQueueClearHandler(agentService: AgentService) {
|
||||||
|
return async (req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const { sessionId } = req.body as { sessionId: string };
|
||||||
|
|
||||||
|
if (!sessionId) {
|
||||||
|
res.status(400).json({
|
||||||
|
success: false,
|
||||||
|
error: 'sessionId is required',
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await agentService.clearQueue(sessionId);
|
||||||
|
res.json(result);
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Clear queue failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
29
apps/server/src/routes/agent/routes/queue-list.ts
Normal file
29
apps/server/src/routes/agent/routes/queue-list.ts
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
/**
|
||||||
|
* POST /queue/list endpoint - List queued prompts
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import { AgentService } from '../../../services/agent-service.js';
|
||||||
|
import { getErrorMessage, logError } from '../common.js';
|
||||||
|
|
||||||
|
export function createQueueListHandler(agentService: AgentService) {
|
||||||
|
return async (req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const { sessionId } = req.body as { sessionId: string };
|
||||||
|
|
||||||
|
if (!sessionId) {
|
||||||
|
res.status(400).json({
|
||||||
|
success: false,
|
||||||
|
error: 'sessionId is required',
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = agentService.getQueue(sessionId);
|
||||||
|
res.json(result);
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'List queue failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
32
apps/server/src/routes/agent/routes/queue-remove.ts
Normal file
32
apps/server/src/routes/agent/routes/queue-remove.ts
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
/**
|
||||||
|
* POST /queue/remove endpoint - Remove a prompt from the queue
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import { AgentService } from '../../../services/agent-service.js';
|
||||||
|
import { getErrorMessage, logError } from '../common.js';
|
||||||
|
|
||||||
|
export function createQueueRemoveHandler(agentService: AgentService) {
|
||||||
|
return async (req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const { sessionId, promptId } = req.body as {
|
||||||
|
sessionId: string;
|
||||||
|
promptId: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!sessionId || !promptId) {
|
||||||
|
res.status(400).json({
|
||||||
|
success: false,
|
||||||
|
error: 'sessionId and promptId are required',
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await agentService.removeFromQueue(sessionId, promptId);
|
||||||
|
res.json(result);
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Remove from queue failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
39
apps/server/src/routes/backlog-plan/common.ts
Normal file
39
apps/server/src/routes/backlog-plan/common.ts
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
/**
|
||||||
|
* Common utilities for backlog plan routes
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { createLogger } from '@automaker/utils';
|
||||||
|
|
||||||
|
const logger = createLogger('BacklogPlan');
|
||||||
|
|
||||||
|
// State for tracking running generation
|
||||||
|
let isRunning = false;
|
||||||
|
let currentAbortController: AbortController | null = null;
|
||||||
|
|
||||||
|
export function getBacklogPlanStatus(): { isRunning: boolean } {
|
||||||
|
return { isRunning };
|
||||||
|
}
|
||||||
|
|
||||||
|
export function setRunningState(running: boolean, abortController?: AbortController | null): void {
|
||||||
|
isRunning = running;
|
||||||
|
if (abortController !== undefined) {
|
||||||
|
currentAbortController = abortController;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function getAbortController(): AbortController | null {
|
||||||
|
return currentAbortController;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function getErrorMessage(error: unknown): string {
|
||||||
|
if (error instanceof Error) {
|
||||||
|
return error.message;
|
||||||
|
}
|
||||||
|
return String(error);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function logError(error: unknown, context: string): void {
|
||||||
|
logger.error(`[BacklogPlan] ${context}:`, getErrorMessage(error));
|
||||||
|
}
|
||||||
|
|
||||||
|
export { logger };
|
||||||
217
apps/server/src/routes/backlog-plan/generate-plan.ts
Normal file
217
apps/server/src/routes/backlog-plan/generate-plan.ts
Normal file
@@ -0,0 +1,217 @@
|
|||||||
|
/**
|
||||||
|
* Generate backlog plan using Claude AI
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { EventEmitter } from '../../lib/events.js';
|
||||||
|
import type { Feature, BacklogPlanResult, BacklogChange, DependencyUpdate } from '@automaker/types';
|
||||||
|
import { FeatureLoader } from '../../services/feature-loader.js';
|
||||||
|
import { ProviderFactory } from '../../providers/provider-factory.js';
|
||||||
|
import { logger, setRunningState, getErrorMessage } from './common.js';
|
||||||
|
import type { SettingsService } from '../../services/settings-service.js';
|
||||||
|
import { getAutoLoadClaudeMdSetting } from '../../lib/settings-helpers.js';
|
||||||
|
|
||||||
|
const featureLoader = new FeatureLoader();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Format features for the AI prompt
|
||||||
|
*/
|
||||||
|
function formatFeaturesForPrompt(features: Feature[]): string {
|
||||||
|
if (features.length === 0) {
|
||||||
|
return 'No features in backlog yet.';
|
||||||
|
}
|
||||||
|
|
||||||
|
return features
|
||||||
|
.map((f) => {
|
||||||
|
const deps = f.dependencies?.length ? `Dependencies: [${f.dependencies.join(', ')}]` : '';
|
||||||
|
const priority = f.priority !== undefined ? `Priority: ${f.priority}` : '';
|
||||||
|
return `- ID: ${f.id}
|
||||||
|
Title: ${f.title || 'Untitled'}
|
||||||
|
Description: ${f.description}
|
||||||
|
Category: ${f.category}
|
||||||
|
Status: ${f.status || 'backlog'}
|
||||||
|
${priority}
|
||||||
|
${deps}`.trim();
|
||||||
|
})
|
||||||
|
.join('\n\n');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Parse the AI response into a BacklogPlanResult
|
||||||
|
*/
|
||||||
|
function parsePlanResponse(response: string): BacklogPlanResult {
|
||||||
|
try {
|
||||||
|
// Try to extract JSON from the response
|
||||||
|
const jsonMatch = response.match(/```json\n?([\s\S]*?)\n?```/);
|
||||||
|
if (jsonMatch) {
|
||||||
|
return JSON.parse(jsonMatch[1]);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Try to parse the whole response as JSON
|
||||||
|
return JSON.parse(response);
|
||||||
|
} catch {
|
||||||
|
// If parsing fails, return an empty result
|
||||||
|
logger.warn('[BacklogPlan] Failed to parse AI response as JSON');
|
||||||
|
return {
|
||||||
|
changes: [],
|
||||||
|
summary: 'Failed to parse AI response',
|
||||||
|
dependencyUpdates: [],
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Generate a backlog modification plan based on user prompt
|
||||||
|
*/
|
||||||
|
export async function generateBacklogPlan(
|
||||||
|
projectPath: string,
|
||||||
|
prompt: string,
|
||||||
|
events: EventEmitter,
|
||||||
|
abortController: AbortController,
|
||||||
|
settingsService?: SettingsService,
|
||||||
|
model?: string
|
||||||
|
): Promise<BacklogPlanResult> {
|
||||||
|
try {
|
||||||
|
// Load current features
|
||||||
|
const features = await featureLoader.getAll(projectPath);
|
||||||
|
|
||||||
|
events.emit('backlog-plan:event', {
|
||||||
|
type: 'backlog_plan_progress',
|
||||||
|
content: `Loaded ${features.length} features from backlog`,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Build the system prompt
|
||||||
|
const systemPrompt = `You are an AI assistant helping to modify a software project's feature backlog.
|
||||||
|
You will be given the current list of features and a user request to modify the backlog.
|
||||||
|
|
||||||
|
IMPORTANT CONTEXT (automatically injected):
|
||||||
|
- Remember to update the dependency graph if deleting existing features
|
||||||
|
- Remember to define dependencies on new features hooked into relevant existing ones
|
||||||
|
- Maintain dependency graph integrity (no orphaned dependencies)
|
||||||
|
- When deleting a feature, identify which other features depend on it
|
||||||
|
|
||||||
|
Your task is to analyze the request and produce a structured JSON plan with:
|
||||||
|
1. Features to ADD (include title, description, category, and dependencies)
|
||||||
|
2. Features to UPDATE (specify featureId and the updates)
|
||||||
|
3. Features to DELETE (specify featureId)
|
||||||
|
4. A summary of the changes
|
||||||
|
5. Any dependency updates needed (removed dependencies due to deletions, new dependencies for new features)
|
||||||
|
|
||||||
|
Respond with ONLY a JSON object in this exact format:
|
||||||
|
\`\`\`json
|
||||||
|
{
|
||||||
|
"changes": [
|
||||||
|
{
|
||||||
|
"type": "add",
|
||||||
|
"feature": {
|
||||||
|
"title": "Feature title",
|
||||||
|
"description": "Feature description",
|
||||||
|
"category": "Category name",
|
||||||
|
"dependencies": ["existing-feature-id"],
|
||||||
|
"priority": 1
|
||||||
|
},
|
||||||
|
"reason": "Why this feature should be added"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "update",
|
||||||
|
"featureId": "existing-feature-id",
|
||||||
|
"feature": {
|
||||||
|
"title": "Updated title"
|
||||||
|
},
|
||||||
|
"reason": "Why this feature should be updated"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "delete",
|
||||||
|
"featureId": "feature-id-to-delete",
|
||||||
|
"reason": "Why this feature should be deleted"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"summary": "Brief overview of all proposed changes",
|
||||||
|
"dependencyUpdates": [
|
||||||
|
{
|
||||||
|
"featureId": "feature-that-depended-on-deleted",
|
||||||
|
"removedDependencies": ["deleted-feature-id"],
|
||||||
|
"addedDependencies": []
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
\`\`\``;
|
||||||
|
|
||||||
|
// Build the user prompt
|
||||||
|
const userPrompt = `Current Features in Backlog:
|
||||||
|
${formatFeaturesForPrompt(features)}
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
User Request: ${prompt}
|
||||||
|
|
||||||
|
Please analyze the current backlog and the user's request, then provide a JSON plan for the modifications.`;
|
||||||
|
|
||||||
|
events.emit('backlog-plan:event', {
|
||||||
|
type: 'backlog_plan_progress',
|
||||||
|
content: 'Generating plan with AI...',
|
||||||
|
});
|
||||||
|
|
||||||
|
// Get the model to use
|
||||||
|
const effectiveModel = model || 'sonnet';
|
||||||
|
const provider = ProviderFactory.getProviderForModel(effectiveModel);
|
||||||
|
|
||||||
|
// Get autoLoadClaudeMd setting
|
||||||
|
const autoLoadClaudeMd = await getAutoLoadClaudeMdSetting(
|
||||||
|
projectPath,
|
||||||
|
settingsService,
|
||||||
|
'[BacklogPlan]'
|
||||||
|
);
|
||||||
|
|
||||||
|
// Execute the query
|
||||||
|
const stream = provider.executeQuery({
|
||||||
|
prompt: userPrompt,
|
||||||
|
model: effectiveModel,
|
||||||
|
cwd: projectPath,
|
||||||
|
systemPrompt,
|
||||||
|
maxTurns: 1,
|
||||||
|
allowedTools: [], // No tools needed for this
|
||||||
|
abortController,
|
||||||
|
settingSources: autoLoadClaudeMd ? ['user', 'project'] : undefined,
|
||||||
|
});
|
||||||
|
|
||||||
|
let responseText = '';
|
||||||
|
|
||||||
|
for await (const msg of stream) {
|
||||||
|
if (abortController.signal.aborted) {
|
||||||
|
throw new Error('Generation aborted');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msg.type === 'assistant') {
|
||||||
|
if (msg.message?.content) {
|
||||||
|
for (const block of msg.message.content) {
|
||||||
|
if (block.type === 'text') {
|
||||||
|
responseText += block.text;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the response
|
||||||
|
const result = parsePlanResponse(responseText);
|
||||||
|
|
||||||
|
events.emit('backlog-plan:event', {
|
||||||
|
type: 'backlog_plan_complete',
|
||||||
|
result,
|
||||||
|
});
|
||||||
|
|
||||||
|
return result;
|
||||||
|
} catch (error) {
|
||||||
|
const errorMessage = getErrorMessage(error);
|
||||||
|
logger.error('[BacklogPlan] Generation failed:', errorMessage);
|
||||||
|
|
||||||
|
events.emit('backlog-plan:event', {
|
||||||
|
type: 'backlog_plan_error',
|
||||||
|
error: errorMessage,
|
||||||
|
});
|
||||||
|
|
||||||
|
throw error;
|
||||||
|
} finally {
|
||||||
|
setRunningState(false, null);
|
||||||
|
}
|
||||||
|
}
|
||||||
30
apps/server/src/routes/backlog-plan/index.ts
Normal file
30
apps/server/src/routes/backlog-plan/index.ts
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
/**
|
||||||
|
* Backlog Plan routes - HTTP API for AI-assisted backlog modification
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { Router } from 'express';
|
||||||
|
import type { EventEmitter } from '../../lib/events.js';
|
||||||
|
import { validatePathParams } from '../../middleware/validate-paths.js';
|
||||||
|
import { createGenerateHandler } from './routes/generate.js';
|
||||||
|
import { createStopHandler } from './routes/stop.js';
|
||||||
|
import { createStatusHandler } from './routes/status.js';
|
||||||
|
import { createApplyHandler } from './routes/apply.js';
|
||||||
|
import type { SettingsService } from '../../services/settings-service.js';
|
||||||
|
|
||||||
|
export function createBacklogPlanRoutes(
|
||||||
|
events: EventEmitter,
|
||||||
|
settingsService?: SettingsService
|
||||||
|
): Router {
|
||||||
|
const router = Router();
|
||||||
|
|
||||||
|
router.post(
|
||||||
|
'/generate',
|
||||||
|
validatePathParams('projectPath'),
|
||||||
|
createGenerateHandler(events, settingsService)
|
||||||
|
);
|
||||||
|
router.post('/stop', createStopHandler());
|
||||||
|
router.get('/status', createStatusHandler());
|
||||||
|
router.post('/apply', validatePathParams('projectPath'), createApplyHandler());
|
||||||
|
|
||||||
|
return router;
|
||||||
|
}
|
||||||
147
apps/server/src/routes/backlog-plan/routes/apply.ts
Normal file
147
apps/server/src/routes/backlog-plan/routes/apply.ts
Normal file
@@ -0,0 +1,147 @@
|
|||||||
|
/**
|
||||||
|
* POST /apply endpoint - Apply a backlog plan
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import type { BacklogPlanResult, BacklogChange, Feature } from '@automaker/types';
|
||||||
|
import { FeatureLoader } from '../../../services/feature-loader.js';
|
||||||
|
import { getErrorMessage, logError, logger } from '../common.js';
|
||||||
|
|
||||||
|
const featureLoader = new FeatureLoader();
|
||||||
|
|
||||||
|
export function createApplyHandler() {
|
||||||
|
return async (req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const { projectPath, plan } = req.body as {
|
||||||
|
projectPath: string;
|
||||||
|
plan: BacklogPlanResult;
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!projectPath) {
|
||||||
|
res.status(400).json({ success: false, error: 'projectPath required' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!plan || !plan.changes) {
|
||||||
|
res.status(400).json({ success: false, error: 'plan with changes required' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const appliedChanges: string[] = [];
|
||||||
|
|
||||||
|
// Load current features for dependency validation
|
||||||
|
const allFeatures = await featureLoader.getAll(projectPath);
|
||||||
|
const featureMap = new Map(allFeatures.map((f) => [f.id, f]));
|
||||||
|
|
||||||
|
// Process changes in order: deletes first, then adds, then updates
|
||||||
|
// This ensures we can remove dependencies before they cause issues
|
||||||
|
|
||||||
|
// 1. First pass: Handle deletes
|
||||||
|
const deletions = plan.changes.filter((c) => c.type === 'delete');
|
||||||
|
for (const change of deletions) {
|
||||||
|
if (!change.featureId) continue;
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Before deleting, update any features that depend on this one
|
||||||
|
for (const feature of allFeatures) {
|
||||||
|
if (feature.dependencies?.includes(change.featureId)) {
|
||||||
|
const newDeps = feature.dependencies.filter((d) => d !== change.featureId);
|
||||||
|
await featureLoader.update(projectPath, feature.id, { dependencies: newDeps });
|
||||||
|
logger.info(
|
||||||
|
`[BacklogPlan] Removed dependency ${change.featureId} from ${feature.id}`
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now delete the feature
|
||||||
|
const deleted = await featureLoader.delete(projectPath, change.featureId);
|
||||||
|
if (deleted) {
|
||||||
|
appliedChanges.push(`deleted:${change.featureId}`);
|
||||||
|
featureMap.delete(change.featureId);
|
||||||
|
logger.info(`[BacklogPlan] Deleted feature ${change.featureId}`);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(
|
||||||
|
`[BacklogPlan] Failed to delete ${change.featureId}:`,
|
||||||
|
getErrorMessage(error)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Second pass: Handle adds
|
||||||
|
const additions = plan.changes.filter((c) => c.type === 'add');
|
||||||
|
for (const change of additions) {
|
||||||
|
if (!change.feature) continue;
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Create the new feature
|
||||||
|
const newFeature = await featureLoader.create(projectPath, {
|
||||||
|
title: change.feature.title,
|
||||||
|
description: change.feature.description || '',
|
||||||
|
category: change.feature.category || 'Uncategorized',
|
||||||
|
dependencies: change.feature.dependencies,
|
||||||
|
priority: change.feature.priority,
|
||||||
|
status: 'backlog',
|
||||||
|
});
|
||||||
|
|
||||||
|
appliedChanges.push(`added:${newFeature.id}`);
|
||||||
|
featureMap.set(newFeature.id, newFeature);
|
||||||
|
logger.info(`[BacklogPlan] Created feature ${newFeature.id}: ${newFeature.title}`);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(`[BacklogPlan] Failed to add feature:`, getErrorMessage(error));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. Third pass: Handle updates
|
||||||
|
const updates = plan.changes.filter((c) => c.type === 'update');
|
||||||
|
for (const change of updates) {
|
||||||
|
if (!change.featureId || !change.feature) continue;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const updated = await featureLoader.update(projectPath, change.featureId, change.feature);
|
||||||
|
appliedChanges.push(`updated:${change.featureId}`);
|
||||||
|
featureMap.set(change.featureId, updated);
|
||||||
|
logger.info(`[BacklogPlan] Updated feature ${change.featureId}`);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(
|
||||||
|
`[BacklogPlan] Failed to update ${change.featureId}:`,
|
||||||
|
getErrorMessage(error)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. Apply dependency updates from the plan
|
||||||
|
if (plan.dependencyUpdates) {
|
||||||
|
for (const depUpdate of plan.dependencyUpdates) {
|
||||||
|
try {
|
||||||
|
const feature = featureMap.get(depUpdate.featureId);
|
||||||
|
if (feature) {
|
||||||
|
const currentDeps = feature.dependencies || [];
|
||||||
|
const newDeps = currentDeps
|
||||||
|
.filter((d) => !depUpdate.removedDependencies.includes(d))
|
||||||
|
.concat(depUpdate.addedDependencies.filter((d) => !currentDeps.includes(d)));
|
||||||
|
|
||||||
|
await featureLoader.update(projectPath, depUpdate.featureId, {
|
||||||
|
dependencies: newDeps,
|
||||||
|
});
|
||||||
|
logger.info(`[BacklogPlan] Updated dependencies for ${depUpdate.featureId}`);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(
|
||||||
|
`[BacklogPlan] Failed to update dependencies for ${depUpdate.featureId}:`,
|
||||||
|
getErrorMessage(error)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
appliedChanges,
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Apply backlog plan failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
62
apps/server/src/routes/backlog-plan/routes/generate.ts
Normal file
62
apps/server/src/routes/backlog-plan/routes/generate.ts
Normal file
@@ -0,0 +1,62 @@
|
|||||||
|
/**
|
||||||
|
* POST /generate endpoint - Generate a backlog plan
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import type { EventEmitter } from '../../../lib/events.js';
|
||||||
|
import { getBacklogPlanStatus, setRunningState, getErrorMessage, logError } from '../common.js';
|
||||||
|
import { generateBacklogPlan } from '../generate-plan.js';
|
||||||
|
import type { SettingsService } from '../../../services/settings-service.js';
|
||||||
|
|
||||||
|
export function createGenerateHandler(events: EventEmitter, settingsService?: SettingsService) {
|
||||||
|
return async (req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const { projectPath, prompt, model } = req.body as {
|
||||||
|
projectPath: string;
|
||||||
|
prompt: string;
|
||||||
|
model?: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!projectPath) {
|
||||||
|
res.status(400).json({ success: false, error: 'projectPath required' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!prompt) {
|
||||||
|
res.status(400).json({ success: false, error: 'prompt required' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { isRunning } = getBacklogPlanStatus();
|
||||||
|
if (isRunning) {
|
||||||
|
res.json({
|
||||||
|
success: false,
|
||||||
|
error: 'Backlog plan generation is already running',
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
setRunningState(true);
|
||||||
|
const abortController = new AbortController();
|
||||||
|
setRunningState(true, abortController);
|
||||||
|
|
||||||
|
// Start generation in background
|
||||||
|
generateBacklogPlan(projectPath, prompt, events, abortController, settingsService, model)
|
||||||
|
.catch((error) => {
|
||||||
|
logError(error, 'Generate backlog plan failed (background)');
|
||||||
|
events.emit('backlog-plan:event', {
|
||||||
|
type: 'backlog_plan_error',
|
||||||
|
error: getErrorMessage(error),
|
||||||
|
});
|
||||||
|
})
|
||||||
|
.finally(() => {
|
||||||
|
setRunningState(false, null);
|
||||||
|
});
|
||||||
|
|
||||||
|
res.json({ success: true });
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Generate backlog plan failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
18
apps/server/src/routes/backlog-plan/routes/status.ts
Normal file
18
apps/server/src/routes/backlog-plan/routes/status.ts
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
/**
|
||||||
|
* GET /status endpoint - Get backlog plan generation status
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import { getBacklogPlanStatus, getErrorMessage, logError } from '../common.js';
|
||||||
|
|
||||||
|
export function createStatusHandler() {
|
||||||
|
return async (_req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const status = getBacklogPlanStatus();
|
||||||
|
res.json({ success: true, ...status });
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Get backlog plan status failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
22
apps/server/src/routes/backlog-plan/routes/stop.ts
Normal file
22
apps/server/src/routes/backlog-plan/routes/stop.ts
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
/**
|
||||||
|
* POST /stop endpoint - Stop the current backlog plan generation
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Request, Response } from 'express';
|
||||||
|
import { getAbortController, setRunningState, getErrorMessage, logError } from '../common.js';
|
||||||
|
|
||||||
|
export function createStopHandler() {
|
||||||
|
return async (_req: Request, res: Response): Promise<void> => {
|
||||||
|
try {
|
||||||
|
const abortController = getAbortController();
|
||||||
|
if (abortController) {
|
||||||
|
abortController.abort();
|
||||||
|
setRunningState(false, null);
|
||||||
|
}
|
||||||
|
res.json({ success: true });
|
||||||
|
} catch (error) {
|
||||||
|
logError(error, 'Stop backlog plan failed');
|
||||||
|
res.status(500).json({ success: false, error: getErrorMessage(error) });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
@@ -32,6 +32,14 @@ interface Message {
|
|||||||
isError?: boolean;
|
isError?: boolean;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
interface QueuedPrompt {
|
||||||
|
id: string;
|
||||||
|
message: string;
|
||||||
|
imagePaths?: string[];
|
||||||
|
model?: string;
|
||||||
|
addedAt: string;
|
||||||
|
}
|
||||||
|
|
||||||
interface Session {
|
interface Session {
|
||||||
messages: Message[];
|
messages: Message[];
|
||||||
isRunning: boolean;
|
isRunning: boolean;
|
||||||
@@ -39,6 +47,7 @@ interface Session {
|
|||||||
workingDirectory: string;
|
workingDirectory: string;
|
||||||
model?: string;
|
model?: string;
|
||||||
sdkSessionId?: string; // Claude SDK session ID for conversation continuity
|
sdkSessionId?: string; // Claude SDK session ID for conversation continuity
|
||||||
|
promptQueue: QueuedPrompt[]; // Queue of prompts to auto-run after current task
|
||||||
}
|
}
|
||||||
|
|
||||||
interface SessionMetadata {
|
interface SessionMetadata {
|
||||||
@@ -94,12 +103,16 @@ export class AgentService {
|
|||||||
// Validate that the working directory is allowed using centralized validation
|
// Validate that the working directory is allowed using centralized validation
|
||||||
validateWorkingDirectory(resolvedWorkingDirectory);
|
validateWorkingDirectory(resolvedWorkingDirectory);
|
||||||
|
|
||||||
|
// Load persisted queue
|
||||||
|
const promptQueue = await this.loadQueueState(sessionId);
|
||||||
|
|
||||||
this.sessions.set(sessionId, {
|
this.sessions.set(sessionId, {
|
||||||
messages,
|
messages,
|
||||||
isRunning: false,
|
isRunning: false,
|
||||||
abortController: null,
|
abortController: null,
|
||||||
workingDirectory: resolvedWorkingDirectory,
|
workingDirectory: resolvedWorkingDirectory,
|
||||||
sdkSessionId: sessionMetadata?.sdkSessionId, // Load persisted SDK session ID
|
sdkSessionId: sessionMetadata?.sdkSessionId, // Load persisted SDK session ID
|
||||||
|
promptQueue,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -178,6 +191,11 @@ export class AgentService {
|
|||||||
session.isRunning = true;
|
session.isRunning = true;
|
||||||
session.abortController = new AbortController();
|
session.abortController = new AbortController();
|
||||||
|
|
||||||
|
// Emit started event so UI can show thinking indicator
|
||||||
|
this.emitAgentEvent(sessionId, {
|
||||||
|
type: 'started',
|
||||||
|
});
|
||||||
|
|
||||||
// Emit user message event
|
// Emit user message event
|
||||||
this.emitAgentEvent(sessionId, {
|
this.emitAgentEvent(sessionId, {
|
||||||
type: 'message',
|
type: 'message',
|
||||||
@@ -336,6 +354,9 @@ export class AgentService {
|
|||||||
session.isRunning = false;
|
session.isRunning = false;
|
||||||
session.abortController = null;
|
session.abortController = null;
|
||||||
|
|
||||||
|
// Process next item in queue after completion
|
||||||
|
setImmediate(() => this.processNextInQueue(sessionId));
|
||||||
|
|
||||||
return {
|
return {
|
||||||
success: true,
|
success: true,
|
||||||
message: currentAssistantMessage,
|
message: currentAssistantMessage,
|
||||||
@@ -574,6 +595,167 @@ export class AgentService {
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Queue management methods
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Add a prompt to the queue for later execution
|
||||||
|
*/
|
||||||
|
async addToQueue(
|
||||||
|
sessionId: string,
|
||||||
|
prompt: { message: string; imagePaths?: string[]; model?: string }
|
||||||
|
): Promise<{ success: boolean; queuedPrompt?: QueuedPrompt; error?: string }> {
|
||||||
|
const session = this.sessions.get(sessionId);
|
||||||
|
if (!session) {
|
||||||
|
return { success: false, error: 'Session not found' };
|
||||||
|
}
|
||||||
|
|
||||||
|
const queuedPrompt: QueuedPrompt = {
|
||||||
|
id: this.generateId(),
|
||||||
|
message: prompt.message,
|
||||||
|
imagePaths: prompt.imagePaths,
|
||||||
|
model: prompt.model,
|
||||||
|
addedAt: new Date().toISOString(),
|
||||||
|
};
|
||||||
|
|
||||||
|
session.promptQueue.push(queuedPrompt);
|
||||||
|
await this.saveQueueState(sessionId, session.promptQueue);
|
||||||
|
|
||||||
|
// Emit queue update event
|
||||||
|
this.emitAgentEvent(sessionId, {
|
||||||
|
type: 'queue_updated',
|
||||||
|
queue: session.promptQueue,
|
||||||
|
});
|
||||||
|
|
||||||
|
return { success: true, queuedPrompt };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get the current queue for a session
|
||||||
|
*/
|
||||||
|
getQueue(sessionId: string): { success: boolean; queue?: QueuedPrompt[]; error?: string } {
|
||||||
|
const session = this.sessions.get(sessionId);
|
||||||
|
if (!session) {
|
||||||
|
return { success: false, error: 'Session not found' };
|
||||||
|
}
|
||||||
|
return { success: true, queue: session.promptQueue };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Remove a specific prompt from the queue
|
||||||
|
*/
|
||||||
|
async removeFromQueue(
|
||||||
|
sessionId: string,
|
||||||
|
promptId: string
|
||||||
|
): Promise<{ success: boolean; error?: string }> {
|
||||||
|
const session = this.sessions.get(sessionId);
|
||||||
|
if (!session) {
|
||||||
|
return { success: false, error: 'Session not found' };
|
||||||
|
}
|
||||||
|
|
||||||
|
const index = session.promptQueue.findIndex((p) => p.id === promptId);
|
||||||
|
if (index === -1) {
|
||||||
|
return { success: false, error: 'Prompt not found in queue' };
|
||||||
|
}
|
||||||
|
|
||||||
|
session.promptQueue.splice(index, 1);
|
||||||
|
await this.saveQueueState(sessionId, session.promptQueue);
|
||||||
|
|
||||||
|
this.emitAgentEvent(sessionId, {
|
||||||
|
type: 'queue_updated',
|
||||||
|
queue: session.promptQueue,
|
||||||
|
});
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Clear all prompts from the queue
|
||||||
|
*/
|
||||||
|
async clearQueue(sessionId: string): Promise<{ success: boolean; error?: string }> {
|
||||||
|
const session = this.sessions.get(sessionId);
|
||||||
|
if (!session) {
|
||||||
|
return { success: false, error: 'Session not found' };
|
||||||
|
}
|
||||||
|
|
||||||
|
session.promptQueue = [];
|
||||||
|
await this.saveQueueState(sessionId, []);
|
||||||
|
|
||||||
|
this.emitAgentEvent(sessionId, {
|
||||||
|
type: 'queue_updated',
|
||||||
|
queue: [],
|
||||||
|
});
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Save queue state to disk for persistence
|
||||||
|
*/
|
||||||
|
private async saveQueueState(sessionId: string, queue: QueuedPrompt[]): Promise<void> {
|
||||||
|
const queueFile = path.join(this.stateDir, `${sessionId}-queue.json`);
|
||||||
|
try {
|
||||||
|
await secureFs.writeFile(queueFile, JSON.stringify(queue, null, 2), 'utf-8');
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[AgentService] Failed to save queue state:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Load queue state from disk
|
||||||
|
*/
|
||||||
|
private async loadQueueState(sessionId: string): Promise<QueuedPrompt[]> {
|
||||||
|
const queueFile = path.join(this.stateDir, `${sessionId}-queue.json`);
|
||||||
|
try {
|
||||||
|
const data = (await secureFs.readFile(queueFile, 'utf-8')) as string;
|
||||||
|
return JSON.parse(data);
|
||||||
|
} catch {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Process the next item in the queue (called after task completion)
|
||||||
|
*/
|
||||||
|
private async processNextInQueue(sessionId: string): Promise<void> {
|
||||||
|
const session = this.sessions.get(sessionId);
|
||||||
|
if (!session || session.promptQueue.length === 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Don't process if already running
|
||||||
|
if (session.isRunning) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const nextPrompt = session.promptQueue.shift();
|
||||||
|
if (!nextPrompt) return;
|
||||||
|
|
||||||
|
await this.saveQueueState(sessionId, session.promptQueue);
|
||||||
|
|
||||||
|
this.emitAgentEvent(sessionId, {
|
||||||
|
type: 'queue_updated',
|
||||||
|
queue: session.promptQueue,
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log(`[AgentService] Processing next queued prompt for session ${sessionId}`);
|
||||||
|
|
||||||
|
try {
|
||||||
|
await this.sendMessage({
|
||||||
|
sessionId,
|
||||||
|
message: nextPrompt.message,
|
||||||
|
imagePaths: nextPrompt.imagePaths,
|
||||||
|
model: nextPrompt.model,
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[AgentService] Failed to process queued prompt:', error);
|
||||||
|
this.emitAgentEvent(sessionId, {
|
||||||
|
type: 'queue_error',
|
||||||
|
error: (error as Error).message,
|
||||||
|
promptId: nextPrompt.id,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private emitAgentEvent(sessionId: string, data: Record<string, unknown>): void {
|
private emitAgentEvent(sessionId: string, data: Record<string, unknown>): void {
|
||||||
this.events.emit('agent:stream', { sessionId, ...data });
|
this.events.emit('agent:stream', { sessionId, ...data });
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,6 +18,7 @@ import {
|
|||||||
ChevronDown,
|
ChevronDown,
|
||||||
FileText,
|
FileText,
|
||||||
Square,
|
Square,
|
||||||
|
ListOrdered,
|
||||||
} from 'lucide-react';
|
} from 'lucide-react';
|
||||||
import { cn } from '@/lib/utils';
|
import { cn } from '@/lib/utils';
|
||||||
import { useElectronAgent } from '@/hooks/use-electron-agent';
|
import { useElectronAgent } from '@/hooks/use-electron-agent';
|
||||||
@@ -86,6 +87,10 @@ export function AgentView() {
|
|||||||
clearHistory,
|
clearHistory,
|
||||||
stopExecution,
|
stopExecution,
|
||||||
error: agentError,
|
error: agentError,
|
||||||
|
serverQueue,
|
||||||
|
addToServerQueue,
|
||||||
|
removeFromServerQueue,
|
||||||
|
clearServerQueue,
|
||||||
} = useElectronAgent({
|
} = useElectronAgent({
|
||||||
sessionId: currentSessionId || '',
|
sessionId: currentSessionId || '',
|
||||||
workingDirectory: currentProject?.path,
|
workingDirectory: currentProject?.path,
|
||||||
@@ -134,11 +139,7 @@ export function AgentView() {
|
|||||||
}, [currentProject?.path]);
|
}, [currentProject?.path]);
|
||||||
|
|
||||||
const handleSend = useCallback(async () => {
|
const handleSend = useCallback(async () => {
|
||||||
if (
|
if (!input.trim() && selectedImages.length === 0 && selectedTextFiles.length === 0) return;
|
||||||
(!input.trim() && selectedImages.length === 0 && selectedTextFiles.length === 0) ||
|
|
||||||
isProcessing
|
|
||||||
)
|
|
||||||
return;
|
|
||||||
|
|
||||||
const messageContent = input;
|
const messageContent = input;
|
||||||
const messageImages = selectedImages;
|
const messageImages = selectedImages;
|
||||||
@@ -149,8 +150,13 @@ export function AgentView() {
|
|||||||
setSelectedTextFiles([]);
|
setSelectedTextFiles([]);
|
||||||
setShowImageDropZone(false);
|
setShowImageDropZone(false);
|
||||||
|
|
||||||
|
// If already processing, add to server queue instead
|
||||||
|
if (isProcessing) {
|
||||||
|
await addToServerQueue(messageContent, messageImages, messageTextFiles);
|
||||||
|
} else {
|
||||||
await sendMessage(messageContent, messageImages, messageTextFiles);
|
await sendMessage(messageContent, messageImages, messageTextFiles);
|
||||||
}, [input, selectedImages, selectedTextFiles, isProcessing, sendMessage]);
|
}
|
||||||
|
}, [input, selectedImages, selectedTextFiles, isProcessing, sendMessage, addToServerQueue]);
|
||||||
|
|
||||||
const handleImagesSelected = useCallback((images: ImageAttachment[]) => {
|
const handleImagesSelected = useCallback((images: ImageAttachment[]) => {
|
||||||
setSelectedImages(images);
|
setSelectedImages(images);
|
||||||
@@ -536,41 +542,6 @@ export function AgentView() {
|
|||||||
|
|
||||||
{/* Status indicators & actions */}
|
{/* Status indicators & actions */}
|
||||||
<div className="flex items-center gap-3">
|
<div className="flex items-center gap-3">
|
||||||
{/* Model Selector */}
|
|
||||||
<DropdownMenu>
|
|
||||||
<DropdownMenuTrigger asChild>
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
size="sm"
|
|
||||||
className="h-8 gap-1.5 text-xs font-medium"
|
|
||||||
disabled={isProcessing}
|
|
||||||
data-testid="model-selector"
|
|
||||||
>
|
|
||||||
<Bot className="w-3.5 h-3.5" />
|
|
||||||
{CLAUDE_MODELS.find((m) => m.id === selectedModel)?.label.replace(
|
|
||||||
'Claude ',
|
|
||||||
''
|
|
||||||
) || 'Sonnet'}
|
|
||||||
<ChevronDown className="w-3 h-3 opacity-50" />
|
|
||||||
</Button>
|
|
||||||
</DropdownMenuTrigger>
|
|
||||||
<DropdownMenuContent align="end" className="w-48">
|
|
||||||
{CLAUDE_MODELS.map((model) => (
|
|
||||||
<DropdownMenuItem
|
|
||||||
key={model.id}
|
|
||||||
onClick={() => setSelectedModel(model.id)}
|
|
||||||
className={cn('cursor-pointer', selectedModel === model.id && 'bg-accent')}
|
|
||||||
data-testid={`model-option-${model.id}`}
|
|
||||||
>
|
|
||||||
<div className="flex flex-col">
|
|
||||||
<span className="font-medium">{model.label}</span>
|
|
||||||
<span className="text-xs text-muted-foreground">{model.description}</span>
|
|
||||||
</div>
|
|
||||||
</DropdownMenuItem>
|
|
||||||
))}
|
|
||||||
</DropdownMenuContent>
|
|
||||||
</DropdownMenu>
|
|
||||||
|
|
||||||
{currentTool && (
|
{currentTool && (
|
||||||
<div className="flex items-center gap-2 text-xs text-muted-foreground bg-muted/50 px-3 py-1.5 rounded-full border border-border">
|
<div className="flex items-center gap-2 text-xs text-muted-foreground bg-muted/50 px-3 py-1.5 rounded-full border border-border">
|
||||||
<Wrench className="w-3 h-3 text-primary" />
|
<Wrench className="w-3 h-3 text-primary" />
|
||||||
@@ -760,10 +731,52 @@ export function AgentView() {
|
|||||||
images={selectedImages}
|
images={selectedImages}
|
||||||
maxFiles={5}
|
maxFiles={5}
|
||||||
className="mb-4"
|
className="mb-4"
|
||||||
disabled={isProcessing || !isConnected}
|
disabled={!isConnected}
|
||||||
/>
|
/>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
{/* Queued Prompts List */}
|
||||||
|
{serverQueue.length > 0 && (
|
||||||
|
<div className="mb-4 space-y-2">
|
||||||
|
<div className="flex items-center justify-between">
|
||||||
|
<p className="text-xs font-medium text-muted-foreground">
|
||||||
|
{serverQueue.length} prompt{serverQueue.length > 1 ? 's' : ''} queued
|
||||||
|
</p>
|
||||||
|
<button
|
||||||
|
onClick={clearServerQueue}
|
||||||
|
className="text-xs text-muted-foreground hover:text-foreground transition-colors"
|
||||||
|
>
|
||||||
|
Clear all
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
{serverQueue.map((item, index) => (
|
||||||
|
<div
|
||||||
|
key={item.id}
|
||||||
|
className="group flex items-center gap-2 text-sm bg-muted/50 rounded-lg px-3 py-2 border border-border"
|
||||||
|
>
|
||||||
|
<span className="text-xs text-muted-foreground font-medium min-w-[1.5rem]">
|
||||||
|
{index + 1}.
|
||||||
|
</span>
|
||||||
|
<span className="flex-1 truncate text-foreground">{item.message}</span>
|
||||||
|
{item.imagePaths && item.imagePaths.length > 0 && (
|
||||||
|
<span className="text-xs text-muted-foreground">
|
||||||
|
+{item.imagePaths.length} file{item.imagePaths.length > 1 ? 's' : ''}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
<button
|
||||||
|
onClick={() => removeFromServerQueue(item.id)}
|
||||||
|
className="opacity-0 group-hover:opacity-100 p-1 hover:bg-destructive/10 hover:text-destructive rounded transition-all"
|
||||||
|
title="Remove from queue"
|
||||||
|
>
|
||||||
|
<X className="w-3 h-3" />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* Selected Files Preview - only show when ImageDropZone is hidden to avoid duplicate display */}
|
{/* Selected Files Preview - only show when ImageDropZone is hidden to avoid duplicate display */}
|
||||||
{(selectedImages.length > 0 || selectedTextFiles.length > 0) && !showImageDropZone && (
|
{(selectedImages.length > 0 || selectedTextFiles.length > 0) && !showImageDropZone && (
|
||||||
<div className="mb-4 space-y-2">
|
<div className="mb-4 space-y-2">
|
||||||
@@ -778,7 +791,6 @@ export function AgentView() {
|
|||||||
setSelectedTextFiles([]);
|
setSelectedTextFiles([]);
|
||||||
}}
|
}}
|
||||||
className="text-xs text-muted-foreground hover:text-foreground transition-colors"
|
className="text-xs text-muted-foreground hover:text-foreground transition-colors"
|
||||||
disabled={isProcessing}
|
|
||||||
>
|
>
|
||||||
Clear all
|
Clear all
|
||||||
</button>
|
</button>
|
||||||
@@ -869,13 +881,17 @@ export function AgentView() {
|
|||||||
<Input
|
<Input
|
||||||
ref={inputRef}
|
ref={inputRef}
|
||||||
placeholder={
|
placeholder={
|
||||||
isDragOver ? 'Drop your files here...' : 'Describe what you want to build...'
|
isDragOver
|
||||||
|
? 'Drop your files here...'
|
||||||
|
: isProcessing
|
||||||
|
? 'Type to queue another prompt...'
|
||||||
|
: 'Describe what you want to build...'
|
||||||
}
|
}
|
||||||
value={input}
|
value={input}
|
||||||
onChange={(e) => setInput(e.target.value)}
|
onChange={(e) => setInput(e.target.value)}
|
||||||
onKeyPress={handleKeyPress}
|
onKeyPress={handleKeyPress}
|
||||||
onPaste={handlePaste}
|
onPaste={handlePaste}
|
||||||
disabled={isProcessing || !isConnected}
|
disabled={!isConnected}
|
||||||
data-testid="agent-input"
|
data-testid="agent-input"
|
||||||
className={cn(
|
className={cn(
|
||||||
'h-11 bg-background border-border rounded-xl pl-4 pr-20 text-sm transition-all',
|
'h-11 bg-background border-border rounded-xl pl-4 pr-20 text-sm transition-all',
|
||||||
@@ -899,12 +915,44 @@ export function AgentView() {
|
|||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{/* Model Selector */}
|
||||||
|
<DropdownMenu>
|
||||||
|
<DropdownMenuTrigger asChild>
|
||||||
|
<Button
|
||||||
|
variant="outline"
|
||||||
|
className="h-11 gap-1 text-xs font-medium rounded-xl border-border px-2.5"
|
||||||
|
data-testid="model-selector"
|
||||||
|
>
|
||||||
|
{CLAUDE_MODELS.find((m) => m.id === selectedModel)?.label.replace(
|
||||||
|
'Claude ',
|
||||||
|
''
|
||||||
|
) || 'Sonnet'}
|
||||||
|
<ChevronDown className="w-3 h-3 opacity-50" />
|
||||||
|
</Button>
|
||||||
|
</DropdownMenuTrigger>
|
||||||
|
<DropdownMenuContent align="end" className="w-48">
|
||||||
|
{CLAUDE_MODELS.map((model) => (
|
||||||
|
<DropdownMenuItem
|
||||||
|
key={model.id}
|
||||||
|
onClick={() => setSelectedModel(model.id)}
|
||||||
|
className={cn('cursor-pointer', selectedModel === model.id && 'bg-accent')}
|
||||||
|
data-testid={`model-option-${model.id}`}
|
||||||
|
>
|
||||||
|
<div className="flex flex-col">
|
||||||
|
<span className="font-medium">{model.label}</span>
|
||||||
|
<span className="text-xs text-muted-foreground">{model.description}</span>
|
||||||
|
</div>
|
||||||
|
</DropdownMenuItem>
|
||||||
|
))}
|
||||||
|
</DropdownMenuContent>
|
||||||
|
</DropdownMenu>
|
||||||
|
|
||||||
{/* File Attachment Button */}
|
{/* File Attachment Button */}
|
||||||
<Button
|
<Button
|
||||||
variant="outline"
|
variant="outline"
|
||||||
size="icon"
|
size="icon"
|
||||||
onClick={toggleImageDropZone}
|
onClick={toggleImageDropZone}
|
||||||
disabled={isProcessing || !isConnected}
|
disabled={!isConnected}
|
||||||
className={cn(
|
className={cn(
|
||||||
'h-11 w-11 rounded-xl border-border',
|
'h-11 w-11 rounded-xl border-border',
|
||||||
showImageDropZone && 'bg-primary/10 text-primary border-primary/30',
|
showImageDropZone && 'bg-primary/10 text-primary border-primary/30',
|
||||||
@@ -916,8 +964,8 @@ export function AgentView() {
|
|||||||
<Paperclip className="w-4 h-4" />
|
<Paperclip className="w-4 h-4" />
|
||||||
</Button>
|
</Button>
|
||||||
|
|
||||||
{/* Send / Stop Button */}
|
{/* Stop Button (only when processing) */}
|
||||||
{isProcessing ? (
|
{isProcessing && (
|
||||||
<Button
|
<Button
|
||||||
onClick={stopExecution}
|
onClick={stopExecution}
|
||||||
disabled={!isConnected}
|
disabled={!isConnected}
|
||||||
@@ -928,7 +976,9 @@ export function AgentView() {
|
|||||||
>
|
>
|
||||||
<Square className="w-4 h-4 fill-current" />
|
<Square className="w-4 h-4 fill-current" />
|
||||||
</Button>
|
</Button>
|
||||||
) : (
|
)}
|
||||||
|
|
||||||
|
{/* Send / Queue Button */}
|
||||||
<Button
|
<Button
|
||||||
onClick={handleSend}
|
onClick={handleSend}
|
||||||
disabled={
|
disabled={
|
||||||
@@ -938,11 +988,12 @@ export function AgentView() {
|
|||||||
!isConnected
|
!isConnected
|
||||||
}
|
}
|
||||||
className="h-11 px-4 rounded-xl"
|
className="h-11 px-4 rounded-xl"
|
||||||
|
variant={isProcessing ? 'outline' : 'default'}
|
||||||
data-testid="send-message"
|
data-testid="send-message"
|
||||||
|
title={isProcessing ? 'Add to queue' : 'Send message'}
|
||||||
>
|
>
|
||||||
<Send className="w-4 h-4" />
|
{isProcessing ? <ListOrdered className="w-4 h-4" /> : <Send className="w-4 h-4" />}
|
||||||
</Button>
|
</Button>
|
||||||
)}
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{/* Keyboard hint */}
|
{/* Keyboard hint */}
|
||||||
|
|||||||
@@ -9,7 +9,9 @@ import {
|
|||||||
import { useAppStore, Feature } from '@/store/app-store';
|
import { useAppStore, Feature } from '@/store/app-store';
|
||||||
import { getElectronAPI } from '@/lib/electron';
|
import { getElectronAPI } from '@/lib/electron';
|
||||||
import type { AutoModeEvent } from '@/types/electron';
|
import type { AutoModeEvent } from '@/types/electron';
|
||||||
|
import type { BacklogPlanResult } from '@automaker/types';
|
||||||
import { pathsEqual } from '@/lib/utils';
|
import { pathsEqual } from '@/lib/utils';
|
||||||
|
import { toast } from 'sonner';
|
||||||
import { getBlockingDependencies } from '@automaker/dependency-resolver';
|
import { getBlockingDependencies } from '@automaker/dependency-resolver';
|
||||||
import { BoardBackgroundModal } from '@/components/dialogs/board-background-modal';
|
import { BoardBackgroundModal } from '@/components/dialogs/board-background-modal';
|
||||||
import { RefreshCw } from 'lucide-react';
|
import { RefreshCw } from 'lucide-react';
|
||||||
@@ -25,6 +27,7 @@ import { GraphView } from './graph-view';
|
|||||||
import {
|
import {
|
||||||
AddFeatureDialog,
|
AddFeatureDialog,
|
||||||
AgentOutputModal,
|
AgentOutputModal,
|
||||||
|
BacklogPlanDialog,
|
||||||
CompletedFeaturesModal,
|
CompletedFeaturesModal,
|
||||||
ArchiveAllVerifiedDialog,
|
ArchiveAllVerifiedDialog,
|
||||||
DeleteCompletedFeatureDialog,
|
DeleteCompletedFeatureDialog,
|
||||||
@@ -125,6 +128,11 @@ export function BoardView() {
|
|||||||
} | null>(null);
|
} | null>(null);
|
||||||
const [worktreeRefreshKey, setWorktreeRefreshKey] = useState(0);
|
const [worktreeRefreshKey, setWorktreeRefreshKey] = useState(0);
|
||||||
|
|
||||||
|
// Backlog plan dialog state
|
||||||
|
const [showPlanDialog, setShowPlanDialog] = useState(false);
|
||||||
|
const [pendingBacklogPlan, setPendingBacklogPlan] = useState<BacklogPlanResult | null>(null);
|
||||||
|
const [isGeneratingPlan, setIsGeneratingPlan] = useState(false);
|
||||||
|
|
||||||
// Follow-up state hook
|
// Follow-up state hook
|
||||||
const {
|
const {
|
||||||
showFollowUpDialog,
|
showFollowUpDialog,
|
||||||
@@ -578,6 +586,37 @@ export function BoardView() {
|
|||||||
return unsubscribe;
|
return unsubscribe;
|
||||||
}, [currentProject]);
|
}, [currentProject]);
|
||||||
|
|
||||||
|
// Listen for backlog plan events (for background generation)
|
||||||
|
useEffect(() => {
|
||||||
|
const api = getElectronAPI();
|
||||||
|
if (!api?.backlogPlan) return;
|
||||||
|
|
||||||
|
const unsubscribe = api.backlogPlan.onEvent(
|
||||||
|
(event: { type: string; result?: BacklogPlanResult; error?: string }) => {
|
||||||
|
if (event.type === 'backlog_plan_complete') {
|
||||||
|
setIsGeneratingPlan(false);
|
||||||
|
if (event.result && event.result.changes?.length > 0) {
|
||||||
|
setPendingBacklogPlan(event.result);
|
||||||
|
toast.success('Plan ready! Click to review.', {
|
||||||
|
duration: 10000,
|
||||||
|
action: {
|
||||||
|
label: 'Review',
|
||||||
|
onClick: () => setShowPlanDialog(true),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
toast.info('No changes generated. Try again with a different prompt.');
|
||||||
|
}
|
||||||
|
} else if (event.type === 'backlog_plan_error') {
|
||||||
|
setIsGeneratingPlan(false);
|
||||||
|
toast.error(`Plan generation failed: ${event.error}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
return unsubscribe;
|
||||||
|
}, []);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!autoMode.isRunning || !currentProject) {
|
if (!autoMode.isRunning || !currentProject) {
|
||||||
return;
|
return;
|
||||||
@@ -935,6 +974,7 @@ export function BoardView() {
|
|||||||
}
|
}
|
||||||
}}
|
}}
|
||||||
onAddFeature={() => setShowAddDialog(true)}
|
onAddFeature={() => setShowAddDialog(true)}
|
||||||
|
onOpenPlanDialog={() => setShowPlanDialog(true)}
|
||||||
addFeatureShortcut={{
|
addFeatureShortcut={{
|
||||||
key: shortcuts.addFeature,
|
key: shortcuts.addFeature,
|
||||||
action: () => setShowAddDialog(true),
|
action: () => setShowAddDialog(true),
|
||||||
@@ -1172,6 +1212,18 @@ export function BoardView() {
|
|||||||
setIsGenerating={setIsGeneratingSuggestions}
|
setIsGenerating={setIsGeneratingSuggestions}
|
||||||
/>
|
/>
|
||||||
|
|
||||||
|
{/* Backlog Plan Dialog */}
|
||||||
|
<BacklogPlanDialog
|
||||||
|
open={showPlanDialog}
|
||||||
|
onClose={() => setShowPlanDialog(false)}
|
||||||
|
projectPath={currentProject.path}
|
||||||
|
onPlanApplied={loadFeatures}
|
||||||
|
pendingPlanResult={pendingBacklogPlan}
|
||||||
|
setPendingPlanResult={setPendingBacklogPlan}
|
||||||
|
isGeneratingPlan={isGeneratingPlan}
|
||||||
|
setIsGeneratingPlan={setIsGeneratingPlan}
|
||||||
|
/>
|
||||||
|
|
||||||
{/* Plan Approval Dialog */}
|
{/* Plan Approval Dialog */}
|
||||||
<PlanApprovalDialog
|
<PlanApprovalDialog
|
||||||
open={pendingPlanApproval !== null}
|
open={pendingPlanApproval !== null}
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
import { HotkeyButton } from '@/components/ui/hotkey-button';
|
import { HotkeyButton } from '@/components/ui/hotkey-button';
|
||||||
|
import { Button } from '@/components/ui/button';
|
||||||
import { Slider } from '@/components/ui/slider';
|
import { Slider } from '@/components/ui/slider';
|
||||||
import { Switch } from '@/components/ui/switch';
|
import { Switch } from '@/components/ui/switch';
|
||||||
import { Label } from '@/components/ui/label';
|
import { Label } from '@/components/ui/label';
|
||||||
import { Plus, Bot } from 'lucide-react';
|
import { Plus, Bot, Wand2 } from 'lucide-react';
|
||||||
import { KeyboardShortcut } from '@/hooks/use-keyboard-shortcuts';
|
import { KeyboardShortcut } from '@/hooks/use-keyboard-shortcuts';
|
||||||
import { ClaudeUsagePopover } from '@/components/claude-usage-popover';
|
import { ClaudeUsagePopover } from '@/components/claude-usage-popover';
|
||||||
import { useAppStore } from '@/store/app-store';
|
import { useAppStore } from '@/store/app-store';
|
||||||
@@ -15,6 +16,7 @@ interface BoardHeaderProps {
|
|||||||
isAutoModeRunning: boolean;
|
isAutoModeRunning: boolean;
|
||||||
onAutoModeToggle: (enabled: boolean) => void;
|
onAutoModeToggle: (enabled: boolean) => void;
|
||||||
onAddFeature: () => void;
|
onAddFeature: () => void;
|
||||||
|
onOpenPlanDialog: () => void;
|
||||||
addFeatureShortcut: KeyboardShortcut;
|
addFeatureShortcut: KeyboardShortcut;
|
||||||
isMounted: boolean;
|
isMounted: boolean;
|
||||||
}
|
}
|
||||||
@@ -27,6 +29,7 @@ export function BoardHeader({
|
|||||||
isAutoModeRunning,
|
isAutoModeRunning,
|
||||||
onAutoModeToggle,
|
onAutoModeToggle,
|
||||||
onAddFeature,
|
onAddFeature,
|
||||||
|
onOpenPlanDialog,
|
||||||
addFeatureShortcut,
|
addFeatureShortcut,
|
||||||
isMounted,
|
isMounted,
|
||||||
}: BoardHeaderProps) {
|
}: BoardHeaderProps) {
|
||||||
@@ -89,6 +92,16 @@ export function BoardHeader({
|
|||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
<Button
|
||||||
|
size="sm"
|
||||||
|
variant="outline"
|
||||||
|
onClick={onOpenPlanDialog}
|
||||||
|
data-testid="plan-backlog-button"
|
||||||
|
>
|
||||||
|
<Wand2 className="w-4 h-4 mr-2" />
|
||||||
|
Plan
|
||||||
|
</Button>
|
||||||
|
|
||||||
<HotkeyButton
|
<HotkeyButton
|
||||||
size="sm"
|
size="sm"
|
||||||
onClick={onAddFeature}
|
onClick={onAddFeature}
|
||||||
|
|||||||
@@ -0,0 +1,418 @@
|
|||||||
|
import { useEffect, useState, useCallback } from 'react';
|
||||||
|
import {
|
||||||
|
Dialog,
|
||||||
|
DialogContent,
|
||||||
|
DialogDescription,
|
||||||
|
DialogFooter,
|
||||||
|
DialogHeader,
|
||||||
|
DialogTitle,
|
||||||
|
} from '@/components/ui/dialog';
|
||||||
|
import { Button } from '@/components/ui/button';
|
||||||
|
import { Textarea } from '@/components/ui/textarea';
|
||||||
|
import { Checkbox } from '@/components/ui/checkbox';
|
||||||
|
import {
|
||||||
|
Loader2,
|
||||||
|
Wand2,
|
||||||
|
Check,
|
||||||
|
Plus,
|
||||||
|
Pencil,
|
||||||
|
Trash2,
|
||||||
|
ChevronDown,
|
||||||
|
ChevronRight,
|
||||||
|
} from 'lucide-react';
|
||||||
|
import { getElectronAPI } from '@/lib/electron';
|
||||||
|
import { toast } from 'sonner';
|
||||||
|
import { cn } from '@/lib/utils';
|
||||||
|
import type { BacklogPlanResult, BacklogChange } from '@automaker/types';
|
||||||
|
|
||||||
|
interface BacklogPlanDialogProps {
|
||||||
|
open: boolean;
|
||||||
|
onClose: () => void;
|
||||||
|
projectPath: string;
|
||||||
|
onPlanApplied?: () => void;
|
||||||
|
// Props for background generation
|
||||||
|
pendingPlanResult: BacklogPlanResult | null;
|
||||||
|
setPendingPlanResult: (result: BacklogPlanResult | null) => void;
|
||||||
|
isGeneratingPlan: boolean;
|
||||||
|
setIsGeneratingPlan: (generating: boolean) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
type DialogMode = 'input' | 'review' | 'applying';
|
||||||
|
|
||||||
|
export function BacklogPlanDialog({
|
||||||
|
open,
|
||||||
|
onClose,
|
||||||
|
projectPath,
|
||||||
|
onPlanApplied,
|
||||||
|
pendingPlanResult,
|
||||||
|
setPendingPlanResult,
|
||||||
|
isGeneratingPlan,
|
||||||
|
setIsGeneratingPlan,
|
||||||
|
}: BacklogPlanDialogProps) {
|
||||||
|
const [mode, setMode] = useState<DialogMode>('input');
|
||||||
|
const [prompt, setPrompt] = useState('');
|
||||||
|
const [expandedChanges, setExpandedChanges] = useState<Set<number>>(new Set());
|
||||||
|
const [selectedChanges, setSelectedChanges] = useState<Set<number>>(new Set());
|
||||||
|
|
||||||
|
// Set mode based on whether we have a pending result
|
||||||
|
useEffect(() => {
|
||||||
|
if (open) {
|
||||||
|
if (pendingPlanResult) {
|
||||||
|
setMode('review');
|
||||||
|
// Select all changes by default
|
||||||
|
setSelectedChanges(new Set(pendingPlanResult.changes.map((_, i) => i)));
|
||||||
|
setExpandedChanges(new Set());
|
||||||
|
} else {
|
||||||
|
setMode('input');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}, [open, pendingPlanResult]);
|
||||||
|
|
||||||
|
const handleGenerate = useCallback(async () => {
|
||||||
|
if (!prompt.trim()) {
|
||||||
|
toast.error('Please enter a prompt describing the changes you want');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const api = getElectronAPI();
|
||||||
|
if (!api?.backlogPlan) {
|
||||||
|
toast.error('API not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start generation in background
|
||||||
|
setIsGeneratingPlan(true);
|
||||||
|
|
||||||
|
const result = await api.backlogPlan.generate(projectPath, prompt);
|
||||||
|
if (!result.success) {
|
||||||
|
setIsGeneratingPlan(false);
|
||||||
|
toast.error(result.error || 'Failed to start plan generation');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Show toast and close dialog - generation runs in background
|
||||||
|
toast.info('Generating plan... This will be ready soon!', {
|
||||||
|
duration: 3000,
|
||||||
|
});
|
||||||
|
setPrompt('');
|
||||||
|
onClose();
|
||||||
|
}, [projectPath, prompt, setIsGeneratingPlan, onClose]);
|
||||||
|
|
||||||
|
const handleApply = useCallback(async () => {
|
||||||
|
if (!pendingPlanResult) return;
|
||||||
|
|
||||||
|
// Filter to only selected changes
|
||||||
|
const selectedChangesList = pendingPlanResult.changes.filter((_, index) =>
|
||||||
|
selectedChanges.has(index)
|
||||||
|
);
|
||||||
|
|
||||||
|
if (selectedChangesList.length === 0) {
|
||||||
|
toast.error('Please select at least one change to apply');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const api = getElectronAPI();
|
||||||
|
if (!api?.backlogPlan) {
|
||||||
|
toast.error('API not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
setMode('applying');
|
||||||
|
|
||||||
|
// Create a filtered plan result with only selected changes
|
||||||
|
const filteredPlanResult: BacklogPlanResult = {
|
||||||
|
...pendingPlanResult,
|
||||||
|
changes: selectedChangesList,
|
||||||
|
// Filter dependency updates to only include those for selected features
|
||||||
|
dependencyUpdates:
|
||||||
|
pendingPlanResult.dependencyUpdates?.filter((update) => {
|
||||||
|
const isDeleting = selectedChangesList.some(
|
||||||
|
(c) => c.type === 'delete' && c.featureId === update.featureId
|
||||||
|
);
|
||||||
|
return !isDeleting;
|
||||||
|
}) || [],
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = await api.backlogPlan.apply(projectPath, filteredPlanResult);
|
||||||
|
if (result.success) {
|
||||||
|
toast.success(`Applied ${result.appliedChanges?.length || 0} changes`);
|
||||||
|
setPendingPlanResult(null);
|
||||||
|
onPlanApplied?.();
|
||||||
|
onClose();
|
||||||
|
} else {
|
||||||
|
toast.error(result.error || 'Failed to apply plan');
|
||||||
|
setMode('review');
|
||||||
|
}
|
||||||
|
}, [
|
||||||
|
projectPath,
|
||||||
|
pendingPlanResult,
|
||||||
|
selectedChanges,
|
||||||
|
setPendingPlanResult,
|
||||||
|
onPlanApplied,
|
||||||
|
onClose,
|
||||||
|
]);
|
||||||
|
|
||||||
|
const handleDiscard = useCallback(() => {
|
||||||
|
setPendingPlanResult(null);
|
||||||
|
setMode('input');
|
||||||
|
}, [setPendingPlanResult]);
|
||||||
|
|
||||||
|
const toggleChangeExpanded = (index: number) => {
|
||||||
|
setExpandedChanges((prev) => {
|
||||||
|
const next = new Set(prev);
|
||||||
|
if (next.has(index)) {
|
||||||
|
next.delete(index);
|
||||||
|
} else {
|
||||||
|
next.add(index);
|
||||||
|
}
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const toggleChangeSelected = (index: number) => {
|
||||||
|
setSelectedChanges((prev) => {
|
||||||
|
const next = new Set(prev);
|
||||||
|
if (next.has(index)) {
|
||||||
|
next.delete(index);
|
||||||
|
} else {
|
||||||
|
next.add(index);
|
||||||
|
}
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const toggleAllChanges = () => {
|
||||||
|
if (!pendingPlanResult) return;
|
||||||
|
if (selectedChanges.size === pendingPlanResult.changes.length) {
|
||||||
|
setSelectedChanges(new Set());
|
||||||
|
} else {
|
||||||
|
setSelectedChanges(new Set(pendingPlanResult.changes.map((_, i) => i)));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const getChangeIcon = (type: BacklogChange['type']) => {
|
||||||
|
switch (type) {
|
||||||
|
case 'add':
|
||||||
|
return <Plus className="w-4 h-4 text-green-500" />;
|
||||||
|
case 'update':
|
||||||
|
return <Pencil className="w-4 h-4 text-yellow-500" />;
|
||||||
|
case 'delete':
|
||||||
|
return <Trash2 className="w-4 h-4 text-red-500" />;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const getChangeLabel = (change: BacklogChange) => {
|
||||||
|
switch (change.type) {
|
||||||
|
case 'add':
|
||||||
|
return change.feature?.title || 'New Feature';
|
||||||
|
case 'update':
|
||||||
|
return `Update: ${change.featureId}`;
|
||||||
|
case 'delete':
|
||||||
|
return `Delete: ${change.featureId}`;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const renderContent = () => {
|
||||||
|
switch (mode) {
|
||||||
|
case 'input':
|
||||||
|
return (
|
||||||
|
<div className="space-y-4">
|
||||||
|
<div className="text-sm text-muted-foreground">
|
||||||
|
Describe the changes you want to make to your backlog. The AI will analyze your
|
||||||
|
current features and propose additions, updates, or deletions.
|
||||||
|
</div>
|
||||||
|
<Textarea
|
||||||
|
placeholder="e.g., Add authentication features with login, signup, and password reset. Also add a dashboard feature that depends on authentication."
|
||||||
|
value={prompt}
|
||||||
|
onChange={(e) => setPrompt(e.target.value)}
|
||||||
|
className="min-h-[150px] resize-none"
|
||||||
|
autoFocus
|
||||||
|
/>
|
||||||
|
<div className="text-xs text-muted-foreground">
|
||||||
|
The AI will automatically handle dependency graph updates when adding or removing
|
||||||
|
features.
|
||||||
|
</div>
|
||||||
|
{isGeneratingPlan && (
|
||||||
|
<div className="flex items-center gap-2 text-sm text-muted-foreground bg-muted/50 rounded-lg p-3">
|
||||||
|
<Loader2 className="w-4 h-4 animate-spin" />A plan is currently being generated in
|
||||||
|
the background...
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
|
||||||
|
case 'review':
|
||||||
|
if (!pendingPlanResult) return null;
|
||||||
|
|
||||||
|
const additions = pendingPlanResult.changes.filter((c) => c.type === 'add');
|
||||||
|
const updates = pendingPlanResult.changes.filter((c) => c.type === 'update');
|
||||||
|
const deletions = pendingPlanResult.changes.filter((c) => c.type === 'delete');
|
||||||
|
const allSelected = selectedChanges.size === pendingPlanResult.changes.length;
|
||||||
|
const someSelected = selectedChanges.size > 0 && !allSelected;
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="space-y-4">
|
||||||
|
{/* Summary */}
|
||||||
|
<div className="rounded-lg border bg-muted/30 p-4">
|
||||||
|
<h4 className="font-medium mb-2">Summary</h4>
|
||||||
|
<p className="text-sm text-muted-foreground">{pendingPlanResult.summary}</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Stats */}
|
||||||
|
<div className="flex gap-4 text-sm">
|
||||||
|
{additions.length > 0 && (
|
||||||
|
<span className="flex items-center gap-1 text-green-600">
|
||||||
|
<Plus className="w-4 h-4" /> {additions.length} additions
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
{updates.length > 0 && (
|
||||||
|
<span className="flex items-center gap-1 text-yellow-600">
|
||||||
|
<Pencil className="w-4 h-4" /> {updates.length} updates
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
{deletions.length > 0 && (
|
||||||
|
<span className="flex items-center gap-1 text-red-600">
|
||||||
|
<Trash2 className="w-4 h-4" /> {deletions.length} deletions
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Select all */}
|
||||||
|
<div className="flex items-center gap-2 pb-2 border-b">
|
||||||
|
<Checkbox
|
||||||
|
id="select-all"
|
||||||
|
checked={allSelected}
|
||||||
|
// @ts-expect-error - indeterminate is valid but not in types
|
||||||
|
indeterminate={someSelected}
|
||||||
|
onCheckedChange={toggleAllChanges}
|
||||||
|
/>
|
||||||
|
<label htmlFor="select-all" className="text-sm font-medium cursor-pointer">
|
||||||
|
{allSelected ? 'Deselect all' : 'Select all'} ({selectedChanges.size}/
|
||||||
|
{pendingPlanResult.changes.length})
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Changes list */}
|
||||||
|
<div className="max-h-[300px] overflow-y-auto space-y-2">
|
||||||
|
{pendingPlanResult.changes.map((change, index) => (
|
||||||
|
<div
|
||||||
|
key={index}
|
||||||
|
className={cn(
|
||||||
|
'rounded-lg border p-3',
|
||||||
|
change.type === 'add' && 'border-green-500/30 bg-green-500/5',
|
||||||
|
change.type === 'update' && 'border-yellow-500/30 bg-yellow-500/5',
|
||||||
|
change.type === 'delete' && 'border-red-500/30 bg-red-500/5',
|
||||||
|
!selectedChanges.has(index) && 'opacity-50'
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<Checkbox
|
||||||
|
checked={selectedChanges.has(index)}
|
||||||
|
onCheckedChange={() => toggleChangeSelected(index)}
|
||||||
|
/>
|
||||||
|
<button
|
||||||
|
className="flex-1 flex items-center gap-2 text-left"
|
||||||
|
onClick={() => toggleChangeExpanded(index)}
|
||||||
|
>
|
||||||
|
{expandedChanges.has(index) ? (
|
||||||
|
<ChevronDown className="w-4 h-4 text-muted-foreground" />
|
||||||
|
) : (
|
||||||
|
<ChevronRight className="w-4 h-4 text-muted-foreground" />
|
||||||
|
)}
|
||||||
|
{getChangeIcon(change.type)}
|
||||||
|
<span className="font-medium text-sm">{getChangeLabel(change)}</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{expandedChanges.has(index) && (
|
||||||
|
<div className="mt-3 pl-10 space-y-2 text-sm">
|
||||||
|
<p className="text-muted-foreground">{change.reason}</p>
|
||||||
|
{change.feature && (
|
||||||
|
<div className="rounded bg-background/50 p-2 text-xs font-mono">
|
||||||
|
{change.feature.description && (
|
||||||
|
<p className="text-foreground">{change.feature.description}</p>
|
||||||
|
)}
|
||||||
|
{change.feature.dependencies &&
|
||||||
|
change.feature.dependencies.length > 0 && (
|
||||||
|
<p className="text-muted-foreground mt-1">
|
||||||
|
Dependencies: {change.feature.dependencies.join(', ')}
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
|
||||||
|
case 'applying':
|
||||||
|
return (
|
||||||
|
<div className="flex flex-col items-center justify-center py-12">
|
||||||
|
<Loader2 className="w-8 h-8 animate-spin text-primary mb-4" />
|
||||||
|
<p className="text-muted-foreground">Applying changes...</p>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Dialog open={open} onOpenChange={(isOpen) => !isOpen && onClose()}>
|
||||||
|
<DialogContent className="max-w-2xl">
|
||||||
|
<DialogHeader>
|
||||||
|
<DialogTitle className="flex items-center gap-2">
|
||||||
|
<Wand2 className="w-5 h-5 text-primary" />
|
||||||
|
{mode === 'review' ? 'Review Plan' : 'Plan Backlog Changes'}
|
||||||
|
</DialogTitle>
|
||||||
|
<DialogDescription>
|
||||||
|
{mode === 'review'
|
||||||
|
? 'Select which changes to apply to your backlog'
|
||||||
|
: 'Use AI to add, update, or remove features from your backlog'}
|
||||||
|
</DialogDescription>
|
||||||
|
</DialogHeader>
|
||||||
|
|
||||||
|
<div className="py-4">{renderContent()}</div>
|
||||||
|
|
||||||
|
<DialogFooter>
|
||||||
|
{mode === 'input' && (
|
||||||
|
<>
|
||||||
|
<Button variant="outline" onClick={onClose}>
|
||||||
|
Cancel
|
||||||
|
</Button>
|
||||||
|
<Button onClick={handleGenerate} disabled={!prompt.trim() || isGeneratingPlan}>
|
||||||
|
{isGeneratingPlan ? (
|
||||||
|
<>
|
||||||
|
<Loader2 className="w-4 h-4 mr-2 animate-spin" />
|
||||||
|
Generating...
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<Wand2 className="w-4 h-4 mr-2" />
|
||||||
|
Generate Plan
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</Button>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{mode === 'review' && (
|
||||||
|
<>
|
||||||
|
<Button variant="outline" onClick={handleDiscard}>
|
||||||
|
Discard
|
||||||
|
</Button>
|
||||||
|
<Button variant="outline" onClick={onClose}>
|
||||||
|
Review Later
|
||||||
|
</Button>
|
||||||
|
<Button onClick={handleApply} disabled={selectedChanges.size === 0}>
|
||||||
|
<Check className="w-4 h-4 mr-2" />
|
||||||
|
Apply {selectedChanges.size} Change{selectedChanges.size !== 1 ? 's' : ''}
|
||||||
|
</Button>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</DialogFooter>
|
||||||
|
</DialogContent>
|
||||||
|
</Dialog>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -1,5 +1,6 @@
|
|||||||
export { AddFeatureDialog } from './add-feature-dialog';
|
export { AddFeatureDialog } from './add-feature-dialog';
|
||||||
export { AgentOutputModal } from './agent-output-modal';
|
export { AgentOutputModal } from './agent-output-modal';
|
||||||
|
export { BacklogPlanDialog } from './backlog-plan-dialog';
|
||||||
export { CompletedFeaturesModal } from './completed-features-modal';
|
export { CompletedFeaturesModal } from './completed-features-modal';
|
||||||
export { ArchiveAllVerifiedDialog } from './archive-all-verified-dialog';
|
export { ArchiveAllVerifiedDialog } from './archive-all-verified-dialog';
|
||||||
export { DeleteCompletedFeatureDialog } from './delete-completed-feature-dialog';
|
export { DeleteCompletedFeatureDialog } from './delete-completed-feature-dialog';
|
||||||
|
|||||||
@@ -12,6 +12,15 @@ interface UseElectronAgentOptions {
|
|||||||
onToolUse?: (toolName: string, toolInput: unknown) => void;
|
onToolUse?: (toolName: string, toolInput: unknown) => void;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Server-side queued prompt type
|
||||||
|
interface QueuedPrompt {
|
||||||
|
id: string;
|
||||||
|
message: string;
|
||||||
|
imagePaths?: string[];
|
||||||
|
model?: string;
|
||||||
|
addedAt: string;
|
||||||
|
}
|
||||||
|
|
||||||
interface UseElectronAgentResult {
|
interface UseElectronAgentResult {
|
||||||
messages: Message[];
|
messages: Message[];
|
||||||
isProcessing: boolean;
|
isProcessing: boolean;
|
||||||
@@ -24,7 +33,7 @@ interface UseElectronAgentResult {
|
|||||||
stopExecution: () => Promise<void>;
|
stopExecution: () => Promise<void>;
|
||||||
clearHistory: () => Promise<void>;
|
clearHistory: () => Promise<void>;
|
||||||
error: string | null;
|
error: string | null;
|
||||||
// Queue-related state
|
// Client-side queue (local)
|
||||||
queuedMessages: {
|
queuedMessages: {
|
||||||
id: string;
|
id: string;
|
||||||
content: string;
|
content: string;
|
||||||
@@ -34,6 +43,15 @@ interface UseElectronAgentResult {
|
|||||||
}[];
|
}[];
|
||||||
isQueueProcessing: boolean;
|
isQueueProcessing: boolean;
|
||||||
clearMessageQueue: () => void;
|
clearMessageQueue: () => void;
|
||||||
|
// Server-side queue (persistent, auto-runs)
|
||||||
|
serverQueue: QueuedPrompt[];
|
||||||
|
addToServerQueue: (
|
||||||
|
message: string,
|
||||||
|
images?: ImageAttachment[],
|
||||||
|
textFiles?: TextFileAttachment[]
|
||||||
|
) => Promise<void>;
|
||||||
|
removeFromServerQueue: (promptId: string) => Promise<void>;
|
||||||
|
clearServerQueue: () => Promise<void>;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -52,6 +70,7 @@ export function useElectronAgent({
|
|||||||
const [isProcessing, setIsProcessing] = useState(false);
|
const [isProcessing, setIsProcessing] = useState(false);
|
||||||
const [isConnected, setIsConnected] = useState(false);
|
const [isConnected, setIsConnected] = useState(false);
|
||||||
const [error, setError] = useState<string | null>(null);
|
const [error, setError] = useState<string | null>(null);
|
||||||
|
const [serverQueue, setServerQueue] = useState<QueuedPrompt[]>([]);
|
||||||
const unsubscribeRef = useRef<(() => void) | null>(null);
|
const unsubscribeRef = useRef<(() => void) | null>(null);
|
||||||
const currentMessageRef = useRef<Message | null>(null);
|
const currentMessageRef = useRef<Message | null>(null);
|
||||||
|
|
||||||
@@ -231,6 +250,12 @@ export function useElectronAgent({
|
|||||||
console.log('[useElectronAgent] Stream event for', sessionId, ':', event.type);
|
console.log('[useElectronAgent] Stream event for', sessionId, ':', event.type);
|
||||||
|
|
||||||
switch (event.type) {
|
switch (event.type) {
|
||||||
|
case 'started':
|
||||||
|
// Agent started processing (including from queue)
|
||||||
|
console.log('[useElectronAgent] Agent started processing for session:', sessionId);
|
||||||
|
setIsProcessing(true);
|
||||||
|
break;
|
||||||
|
|
||||||
case 'message':
|
case 'message':
|
||||||
// User message added
|
// User message added
|
||||||
setMessages((prev) => [...prev, event.message]);
|
setMessages((prev) => [...prev, event.message]);
|
||||||
@@ -299,6 +324,18 @@ export function useElectronAgent({
|
|||||||
setMessages((prev) => [...prev, errorMessage]);
|
setMessages((prev) => [...prev, errorMessage]);
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
|
|
||||||
|
case 'queue_updated':
|
||||||
|
// Server queue was updated
|
||||||
|
console.log('[useElectronAgent] Queue updated:', event.queue);
|
||||||
|
setServerQueue(event.queue || []);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'queue_error':
|
||||||
|
// Error processing a queued prompt
|
||||||
|
console.error('[useElectronAgent] Queue error:', event.error);
|
||||||
|
setError(event.error);
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -438,6 +475,102 @@ export function useElectronAgent({
|
|||||||
}
|
}
|
||||||
}, [sessionId]);
|
}, [sessionId]);
|
||||||
|
|
||||||
|
// Add a prompt to the server queue (will auto-run when current task finishes)
|
||||||
|
const addToServerQueue = useCallback(
|
||||||
|
async (message: string, images?: ImageAttachment[], textFiles?: TextFileAttachment[]) => {
|
||||||
|
const api = getElectronAPI();
|
||||||
|
if (!api?.agent?.queueAdd) {
|
||||||
|
setError('Queue API not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Build message content with text file context
|
||||||
|
let messageContent = message;
|
||||||
|
if (textFiles && textFiles.length > 0) {
|
||||||
|
const contextParts = textFiles.map((file) => {
|
||||||
|
return `<file name="${file.filename}">\n${file.content}\n</file>`;
|
||||||
|
});
|
||||||
|
const contextBlock = `Here are some files for context:\n\n${contextParts.join('\n\n')}\n\n`;
|
||||||
|
messageContent = contextBlock + message;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save images and get paths
|
||||||
|
let imagePaths: string[] | undefined;
|
||||||
|
if (images && images.length > 0 && api.saveImageToTemp) {
|
||||||
|
imagePaths = [];
|
||||||
|
for (const image of images) {
|
||||||
|
const result = await api.saveImageToTemp(
|
||||||
|
image.data,
|
||||||
|
sanitizeFilename(image.filename),
|
||||||
|
image.mimeType,
|
||||||
|
workingDirectory
|
||||||
|
);
|
||||||
|
if (result.success && result.path) {
|
||||||
|
imagePaths.push(result.path);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('[useElectronAgent] Adding to server queue');
|
||||||
|
const result = await api.agent.queueAdd(sessionId, messageContent, imagePaths, model);
|
||||||
|
|
||||||
|
if (!result.success) {
|
||||||
|
setError(result.error || 'Failed to add to queue');
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error('[useElectronAgent] Failed to add to queue:', err);
|
||||||
|
setError(err instanceof Error ? err.message : 'Failed to add to queue');
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[sessionId, workingDirectory, model]
|
||||||
|
);
|
||||||
|
|
||||||
|
// Remove a prompt from the server queue
|
||||||
|
const removeFromServerQueue = useCallback(
|
||||||
|
async (promptId: string) => {
|
||||||
|
const api = getElectronAPI();
|
||||||
|
if (!api?.agent?.queueRemove) {
|
||||||
|
setError('Queue API not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
console.log('[useElectronAgent] Removing from server queue:', promptId);
|
||||||
|
const result = await api.agent.queueRemove(sessionId, promptId);
|
||||||
|
|
||||||
|
if (!result.success) {
|
||||||
|
setError(result.error || 'Failed to remove from queue');
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error('[useElectronAgent] Failed to remove from queue:', err);
|
||||||
|
setError(err instanceof Error ? err.message : 'Failed to remove from queue');
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[sessionId]
|
||||||
|
);
|
||||||
|
|
||||||
|
// Clear the entire server queue
|
||||||
|
const clearServerQueue = useCallback(async () => {
|
||||||
|
const api = getElectronAPI();
|
||||||
|
if (!api?.agent?.queueClear) {
|
||||||
|
setError('Queue API not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
console.log('[useElectronAgent] Clearing server queue');
|
||||||
|
const result = await api.agent.queueClear(sessionId);
|
||||||
|
|
||||||
|
if (!result.success) {
|
||||||
|
setError(result.error || 'Failed to clear queue');
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error('[useElectronAgent] Failed to clear queue:', err);
|
||||||
|
setError(err instanceof Error ? err.message : 'Failed to clear queue');
|
||||||
|
}
|
||||||
|
}, [sessionId]);
|
||||||
|
|
||||||
return {
|
return {
|
||||||
messages,
|
messages,
|
||||||
isProcessing,
|
isProcessing,
|
||||||
@@ -449,5 +582,10 @@ export function useElectronAgent({
|
|||||||
queuedMessages,
|
queuedMessages,
|
||||||
isQueueProcessing: isProcessingQueue,
|
isQueueProcessing: isProcessingQueue,
|
||||||
clearMessageQueue: clearQueue,
|
clearMessageQueue: clearQueue,
|
||||||
|
// Server-side queue
|
||||||
|
serverQueue,
|
||||||
|
addToServerQueue,
|
||||||
|
removeFromServerQueue,
|
||||||
|
clearServerQueue,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -829,6 +829,47 @@ export class HttpApiClient implements ElectronAPI {
|
|||||||
onStream: (callback: (data: unknown) => void): (() => void) => {
|
onStream: (callback: (data: unknown) => void): (() => void) => {
|
||||||
return this.subscribeToEvent('agent:stream', callback as EventCallback);
|
return this.subscribeToEvent('agent:stream', callback as EventCallback);
|
||||||
},
|
},
|
||||||
|
|
||||||
|
// Queue management
|
||||||
|
queueAdd: (
|
||||||
|
sessionId: string,
|
||||||
|
message: string,
|
||||||
|
imagePaths?: string[],
|
||||||
|
model?: string
|
||||||
|
): Promise<{
|
||||||
|
success: boolean;
|
||||||
|
queuedPrompt?: {
|
||||||
|
id: string;
|
||||||
|
message: string;
|
||||||
|
imagePaths?: string[];
|
||||||
|
model?: string;
|
||||||
|
addedAt: string;
|
||||||
|
};
|
||||||
|
error?: string;
|
||||||
|
}> => this.post('/api/agent/queue/add', { sessionId, message, imagePaths, model }),
|
||||||
|
|
||||||
|
queueList: (
|
||||||
|
sessionId: string
|
||||||
|
): Promise<{
|
||||||
|
success: boolean;
|
||||||
|
queue?: Array<{
|
||||||
|
id: string;
|
||||||
|
message: string;
|
||||||
|
imagePaths?: string[];
|
||||||
|
model?: string;
|
||||||
|
addedAt: string;
|
||||||
|
}>;
|
||||||
|
error?: string;
|
||||||
|
}> => this.post('/api/agent/queue/list', { sessionId }),
|
||||||
|
|
||||||
|
queueRemove: (
|
||||||
|
sessionId: string,
|
||||||
|
promptId: string
|
||||||
|
): Promise<{ success: boolean; error?: string }> =>
|
||||||
|
this.post('/api/agent/queue/remove', { sessionId, promptId }),
|
||||||
|
|
||||||
|
queueClear: (sessionId: string): Promise<{ success: boolean; error?: string }> =>
|
||||||
|
this.post('/api/agent/queue/clear', { sessionId }),
|
||||||
};
|
};
|
||||||
|
|
||||||
// Templates API
|
// Templates API
|
||||||
@@ -1045,6 +1086,45 @@ export class HttpApiClient implements ElectronAPI {
|
|||||||
error?: string;
|
error?: string;
|
||||||
}> => this.post('/api/context/describe-file', { filePath }),
|
}> => this.post('/api/context/describe-file', { filePath }),
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Backlog Plan API
|
||||||
|
backlogPlan = {
|
||||||
|
generate: (
|
||||||
|
projectPath: string,
|
||||||
|
prompt: string,
|
||||||
|
model?: string
|
||||||
|
): Promise<{ success: boolean; error?: string }> =>
|
||||||
|
this.post('/api/backlog-plan/generate', { projectPath, prompt, model }),
|
||||||
|
|
||||||
|
stop: (): Promise<{ success: boolean; error?: string }> =>
|
||||||
|
this.post('/api/backlog-plan/stop', {}),
|
||||||
|
|
||||||
|
status: (): Promise<{ success: boolean; isRunning?: boolean; error?: string }> =>
|
||||||
|
this.get('/api/backlog-plan/status'),
|
||||||
|
|
||||||
|
apply: (
|
||||||
|
projectPath: string,
|
||||||
|
plan: {
|
||||||
|
changes: Array<{
|
||||||
|
type: 'add' | 'update' | 'delete';
|
||||||
|
featureId?: string;
|
||||||
|
feature?: Record<string, unknown>;
|
||||||
|
reason: string;
|
||||||
|
}>;
|
||||||
|
summary: string;
|
||||||
|
dependencyUpdates: Array<{
|
||||||
|
featureId: string;
|
||||||
|
removedDependencies: string[];
|
||||||
|
addedDependencies: string[];
|
||||||
|
}>;
|
||||||
|
}
|
||||||
|
): Promise<{ success: boolean; appliedChanges?: string[]; error?: string }> =>
|
||||||
|
this.post('/api/backlog-plan/apply', { projectPath, plan }),
|
||||||
|
|
||||||
|
onEvent: (callback: (data: unknown) => void): (() => void) => {
|
||||||
|
return this.subscribeToEvent('backlog-plan:event', callback as EventCallback);
|
||||||
|
},
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
// Singleton instance
|
// Singleton instance
|
||||||
|
|||||||
67
libs/types/src/backlog-plan.ts
Normal file
67
libs/types/src/backlog-plan.ts
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
/**
|
||||||
|
* Backlog Plan types for AI-assisted backlog modification
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { Feature } from './feature.js';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A single proposed change to the backlog
|
||||||
|
*/
|
||||||
|
export interface BacklogChange {
|
||||||
|
type: 'add' | 'update' | 'delete';
|
||||||
|
featureId?: string; // For update/delete operations
|
||||||
|
feature?: Partial<Feature>; // For add/update (includes title, description, category, dependencies, priority)
|
||||||
|
reason: string; // AI explanation of why this change is proposed
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Dependency updates that need to happen as a result of the plan
|
||||||
|
*/
|
||||||
|
export interface DependencyUpdate {
|
||||||
|
featureId: string;
|
||||||
|
removedDependencies: string[]; // Dependencies removed due to deleted features
|
||||||
|
addedDependencies: string[]; // New dependencies based on AI analysis
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Result from the AI when generating a backlog plan
|
||||||
|
*/
|
||||||
|
export interface BacklogPlanResult {
|
||||||
|
changes: BacklogChange[];
|
||||||
|
summary: string; // Overview of proposed changes
|
||||||
|
dependencyUpdates: DependencyUpdate[];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Events emitted during backlog plan generation
|
||||||
|
*/
|
||||||
|
export interface BacklogPlanEvent {
|
||||||
|
type:
|
||||||
|
| 'backlog_plan_progress'
|
||||||
|
| 'backlog_plan_tool'
|
||||||
|
| 'backlog_plan_complete'
|
||||||
|
| 'backlog_plan_error';
|
||||||
|
content?: string;
|
||||||
|
tool?: string;
|
||||||
|
input?: unknown;
|
||||||
|
result?: BacklogPlanResult;
|
||||||
|
error?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Request to generate a backlog plan
|
||||||
|
*/
|
||||||
|
export interface BacklogPlanRequest {
|
||||||
|
projectPath: string;
|
||||||
|
prompt: string;
|
||||||
|
model?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Response from apply operation
|
||||||
|
*/
|
||||||
|
export interface BacklogPlanApplyResult {
|
||||||
|
success: boolean;
|
||||||
|
appliedChanges: string[]; // IDs of features affected
|
||||||
|
error?: string;
|
||||||
|
}
|
||||||
@@ -9,6 +9,7 @@ export type EventType =
|
|||||||
| 'auto-mode:stopped'
|
| 'auto-mode:stopped'
|
||||||
| 'auto-mode:idle'
|
| 'auto-mode:idle'
|
||||||
| 'auto-mode:error'
|
| 'auto-mode:error'
|
||||||
|
| 'backlog-plan:event'
|
||||||
| 'feature:started'
|
| 'feature:started'
|
||||||
| 'feature:completed'
|
| 'feature:completed'
|
||||||
| 'feature:stopped'
|
| 'feature:stopped'
|
||||||
|
|||||||
@@ -95,3 +95,13 @@ export type {
|
|||||||
IssueValidationEvent,
|
IssueValidationEvent,
|
||||||
StoredValidation,
|
StoredValidation,
|
||||||
} from './issue-validation.js';
|
} from './issue-validation.js';
|
||||||
|
|
||||||
|
// Backlog plan types
|
||||||
|
export type {
|
||||||
|
BacklogChange,
|
||||||
|
DependencyUpdate,
|
||||||
|
BacklogPlanResult,
|
||||||
|
BacklogPlanEvent,
|
||||||
|
BacklogPlanRequest,
|
||||||
|
BacklogPlanApplyResult,
|
||||||
|
} from './backlog-plan.js';
|
||||||
|
|||||||
Reference in New Issue
Block a user