feat(CUB-56): Agent State Database Migration

This commit is contained in:
cubecraft-agents[bot]
2026-04-26 10:18:06 +00:00
parent f490098af6
commit 7b7b070dac
15 changed files with 581 additions and 0 deletions

21
backend/.gitignore vendored Normal file
View File

@@ -0,0 +1,21 @@
## .NET
bin/
obj/
*.user
*.suo
*.cache
*.dll
*.pdb
## IDE
.vs/
.idea/
*.swp
*~
## OS
.DS_Store
Thumbs.db
## Environment
.env

View File

@@ -0,0 +1,88 @@
using ControlCenter.Api.Entities;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata.Builders;
namespace ControlCenter.Api.Configurations;
/// <summary>
/// EF Core entity type configuration for the agents table.
/// Enforces snake_case naming, required fields, and index design.
/// </summary>
public class AgentConfiguration : IEntityTypeConfiguration<Agent>
{
public void Configure(EntityTypeBuilder<Agent> builder)
{
// Table name — snake_case
builder.ToTable("agents");
// Primary key
builder.HasKey(a => a.Id);
builder.Property(a => a.Id)
.HasColumnName("id")
.ValueGeneratedOnAdd();
// Status — stored as PostgreSQL enum via Npgsql
builder.Property(a => a.Status)
.HasColumnName("status")
.HasColumnType("agent_status")
.IsRequired();
// Task — nullable text
builder.Property(a => a.Task)
.HasColumnName("task")
.HasColumnType("text");
// Progress — nullable integer (0100)
builder.Property(a => a.Progress)
.HasColumnName("progress");
// Session key — required, not null
builder.Property(a => a.SessionKey)
.HasColumnName("session_key")
.HasColumnType("text")
.IsRequired();
// Channel — required, not null
builder.Property(a => a.Channel)
.HasColumnName("channel")
.HasColumnType("text")
.IsRequired();
// Last activity — required, defaults to now()
builder.Property(a => a.LastActivity)
.HasColumnName("last_activity")
.HasColumnType("timestamptz")
.IsRequired();
// Created at — auto-set on insert
builder.Property(a => a.CreatedAt)
.HasColumnName("created_at")
.HasColumnType("timestamptz")
.IsRequired()
.HasDefaultValueSql("now()");
// Updated at — auto-set on insert and update
builder.Property(a => a.UpdatedAt)
.HasColumnName("updated_at")
.HasColumnType("timestamptz")
.IsRequired()
.HasDefaultValueSql("now()");
// Indexes
// Sessions are looked up by session_key frequently
builder.HasIndex(a => a.SessionKey)
.HasDatabaseName("ix_agents_session_key")
.IsUnique();
// Agents are filtered by channel for channel-specific queries
builder.HasIndex(a => a.Channel)
.HasDatabaseName("ix_agents_channel");
// Agents are filtered by status for fleet health monitoring
builder.HasIndex(a => a.Status)
.HasDatabaseName("ix_agents_status");
// Check constraint: progress must be 0100 if present
builder.ToTable(t => t.HasCheckConstraint("ck_agents_progress_range", "progress IS NULL OR (progress >= 0 AND progress <= 100)"));
}
}

View File

@@ -0,0 +1,18 @@
<Project Sdk="Microsoft.NET.Sdk.Web">
<PropertyGroup>
<TargetFramework>net9.0</TargetFramework>
<Nullable>enable</Nullable>
<ImplicitUsings>enable</ImplicitUsings>
</PropertyGroup>
<ItemGroup>
<PackageReference Include="Microsoft.AspNetCore.OpenApi" Version="9.0.14" />
<PackageReference Include="Microsoft.EntityFrameworkCore.Design" Version="9.0.4">
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
<PrivateAssets>all</PrivateAssets>
</PackageReference>
<PackageReference Include="Npgsql.EntityFrameworkCore.PostgreSQL" Version="9.0.4" />
</ItemGroup>
</Project>

View File

@@ -0,0 +1,6 @@
@ControlCenter.Api_HostAddress = http://localhost:5178
GET {{ControlCenter.Api_HostAddress}}/weatherforecast/
Accept: application/json
###

View File

@@ -0,0 +1,29 @@
using ControlCenter.Api.Configurations;
using ControlCenter.Api.Entities;
using Microsoft.EntityFrameworkCore;
namespace ControlCenter.Api.Data;
/// <summary>
/// EF Core DbContext for the Control Center database.
/// All table and column names use snake_case via explicit HasColumnName configuration.
/// </summary>
public class AppDbContext : DbContext
{
public AppDbContext(DbContextOptions<AppDbContext> options) : base(options) { }
public DbSet<Agent> Agents => Set<Agent>();
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
// Apply all entity type configurations from the Configurations namespace
modelBuilder.ApplyConfigurationsFromAssembly(typeof(AgentConfiguration).Assembly);
// Map the AgentStatus enum to a PostgreSQL enum type named "agent_status"
// This must be called after ApplyConfigurations to ensure the model is built
// before the enum mapping is applied.
modelBuilder.HasPostgresEnum<AgentStatus>();
base.OnModelCreating(modelBuilder);
}
}

View File

@@ -0,0 +1,27 @@
using Microsoft.EntityFrameworkCore;
using ControlCenter.Api.Entities;
namespace ControlCenter.Api.Data;
/// <summary>
/// Design-time factory for AppDbContext, used by EF Core tools (dotnet ef)
/// to create migrations without requiring a running application.
/// </summary>
public class AppDbContextFactory : Microsoft.EntityFrameworkCore.Design.IDesignTimeDbContextFactory<AppDbContext>
{
public AppDbContext CreateDbContext(string[] args)
{
var optionsBuilder = new DbContextOptionsBuilder<AppDbContext>();
// Connection string for design-time operations (migrations).
// In production, this comes from appsettings / environment variables.
var connectionString = "Host=localhost;Database=control_center;Username=postgres;Password=postgres";
optionsBuilder.UseNpgsql(connectionString, npgsqlOptions =>
{
npgsqlOptions.MigrationsAssembly(typeof(AppDbContext).Assembly.FullName);
});
return new AppDbContext(optionsBuilder.Options);
}
}

59
backend/Entities/Agent.cs Normal file
View File

@@ -0,0 +1,59 @@
namespace ControlCenter.Api.Entities;
/// <summary>
/// Represents an agent's current state in the Control Center.
/// Each row tracks one agent session's status, task, and activity.
/// </summary>
public class Agent
{
/// <summary>
/// Primary key — UUID generated on insert.
/// </summary>
public Guid Id { get; set; }
/// <summary>
/// Current operational status of the agent.
/// Stored as an enum in PostgreSQL via Npgsql.
/// </summary>
public AgentStatus Status { get; set; } = AgentStatus.Idle;
/// <summary>
/// Description of the agent's current task, if any.
/// Nullable — not all agents have an active task.
/// </summary>
public string? Task { get; set; }
/// <summary>
/// Task progress percentage (0100).
/// Nullable — progress is only meaningful when an agent has a trackable task.
/// </summary>
public int? Progress { get; set; }
/// <summary>
/// Full session key, e.g. "agent:otto:telegram:direct:8787451565".
/// Not null — every agent row must be associated with a session.
/// </summary>
public string SessionKey { get; set; } = string.Empty;
/// <summary>
/// Communication channel the agent is operating on, e.g. "telegram", "discord", "slack".
/// Not null — every agent session operates on exactly one channel.
/// </summary>
public string Channel { get; set; } = string.Empty;
/// <summary>
/// Timestamp of the agent's last activity.
/// Default: current UTC timestamp on insert.
/// </summary>
public DateTime LastActivity { get; set; } = DateTime.UtcNow;
/// <summary>
/// Row creation timestamp. Set automatically on insert.
/// </summary>
public DateTime CreatedAt { get; set; } = DateTime.UtcNow;
/// <summary>
/// Row last-update timestamp. Updated automatically on any modification.
/// </summary>
public DateTime UpdatedAt { get; set; } = DateTime.UtcNow;
}

View File

@@ -0,0 +1,13 @@
namespace ControlCenter.Api.Entities;
/// <summary>
/// Agent operational status enum.
/// Maps to the agent_status enum type in PostgreSQL.
/// </summary>
public enum AgentStatus
{
Active = 0,
Idle = 1,
Thinking = 2,
Error = 3
}

View File

@@ -0,0 +1,94 @@
// <auto-generated />
using System;
using ControlCenter.Api.Data;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.EntityFrameworkCore.Migrations;
using Microsoft.EntityFrameworkCore.Storage.ValueConversion;
using Npgsql.EntityFrameworkCore.PostgreSQL.Metadata;
#nullable disable
namespace ControlCenter.Api.Migrations
{
[DbContext(typeof(AppDbContext))]
[Migration("20260426101703_CreateAgentsTable")]
partial class CreateAgentsTable
{
/// <inheritdoc />
protected override void BuildTargetModel(ModelBuilder modelBuilder)
{
#pragma warning disable 612, 618
modelBuilder
.HasAnnotation("ProductVersion", "9.0.4")
.HasAnnotation("Relational:MaxIdentifierLength", 63);
NpgsqlModelBuilderExtensions.HasPostgresEnum(modelBuilder, "agent_status", new[] { "active", "idle", "thinking", "error" });
NpgsqlModelBuilderExtensions.UseIdentityByDefaultColumns(modelBuilder);
modelBuilder.Entity("ControlCenter.Api.Entities.Agent", b =>
{
b.Property<Guid>("Id")
.ValueGeneratedOnAdd()
.HasColumnType("uuid")
.HasColumnName("id");
b.Property<string>("Channel")
.IsRequired()
.HasColumnType("text")
.HasColumnName("channel");
b.Property<DateTime>("CreatedAt")
.ValueGeneratedOnAdd()
.HasColumnType("timestamptz")
.HasColumnName("created_at")
.HasDefaultValueSql("now()");
b.Property<DateTime>("LastActivity")
.HasColumnType("timestamptz")
.HasColumnName("last_activity");
b.Property<int?>("Progress")
.HasColumnType("integer")
.HasColumnName("progress");
b.Property<string>("SessionKey")
.IsRequired()
.HasColumnType("text")
.HasColumnName("session_key");
b.Property<int>("Status")
.HasColumnType("agent_status")
.HasColumnName("status");
b.Property<string>("Task")
.HasColumnType("text")
.HasColumnName("task");
b.Property<DateTime>("UpdatedAt")
.ValueGeneratedOnAdd()
.HasColumnType("timestamptz")
.HasColumnName("updated_at")
.HasDefaultValueSql("now()");
b.HasKey("Id");
b.HasIndex("Channel")
.HasDatabaseName("ix_agents_channel");
b.HasIndex("SessionKey")
.IsUnique()
.HasDatabaseName("ix_agents_session_key");
b.HasIndex("Status")
.HasDatabaseName("ix_agents_status");
b.ToTable("agents", null, t =>
{
t.HasCheckConstraint("ck_agents_progress_range", "progress IS NULL OR (progress >= 0 AND progress <= 100)");
});
});
#pragma warning restore 612, 618
}
}
}

View File

@@ -0,0 +1,61 @@
using System;
using Microsoft.EntityFrameworkCore.Migrations;
#nullable disable
namespace ControlCenter.Api.Migrations
{
/// <inheritdoc />
public partial class CreateAgentsTable : Migration
{
/// <inheritdoc />
protected override void Up(MigrationBuilder migrationBuilder)
{
migrationBuilder.AlterDatabase()
.Annotation("Npgsql:Enum:agent_status", "active,idle,thinking,error");
migrationBuilder.CreateTable(
name: "agents",
columns: table => new
{
id = table.Column<Guid>(type: "uuid", nullable: false),
status = table.Column<int>(type: "agent_status", nullable: false),
task = table.Column<string>(type: "text", nullable: true),
progress = table.Column<int>(type: "integer", nullable: true),
session_key = table.Column<string>(type: "text", nullable: false),
channel = table.Column<string>(type: "text", nullable: false),
last_activity = table.Column<DateTime>(type: "timestamptz", nullable: false),
created_at = table.Column<DateTime>(type: "timestamptz", nullable: false, defaultValueSql: "now()"),
updated_at = table.Column<DateTime>(type: "timestamptz", nullable: false, defaultValueSql: "now()")
},
constraints: table =>
{
table.PrimaryKey("PK_agents", x => x.id);
table.CheckConstraint("ck_agents_progress_range", "progress IS NULL OR (progress >= 0 AND progress <= 100)");
});
migrationBuilder.CreateIndex(
name: "ix_agents_channel",
table: "agents",
column: "channel");
migrationBuilder.CreateIndex(
name: "ix_agents_session_key",
table: "agents",
column: "session_key",
unique: true);
migrationBuilder.CreateIndex(
name: "ix_agents_status",
table: "agents",
column: "status");
}
/// <inheritdoc />
protected override void Down(MigrationBuilder migrationBuilder)
{
migrationBuilder.DropTable(
name: "agents");
}
}
}

View File

@@ -0,0 +1,91 @@
// <auto-generated />
using System;
using ControlCenter.Api.Data;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.EntityFrameworkCore.Storage.ValueConversion;
using Npgsql.EntityFrameworkCore.PostgreSQL.Metadata;
#nullable disable
namespace ControlCenter.Api.Migrations
{
[DbContext(typeof(AppDbContext))]
partial class AppDbContextModelSnapshot : ModelSnapshot
{
protected override void BuildModel(ModelBuilder modelBuilder)
{
#pragma warning disable 612, 618
modelBuilder
.HasAnnotation("ProductVersion", "9.0.4")
.HasAnnotation("Relational:MaxIdentifierLength", 63);
NpgsqlModelBuilderExtensions.HasPostgresEnum(modelBuilder, "agent_status", new[] { "active", "idle", "thinking", "error" });
NpgsqlModelBuilderExtensions.UseIdentityByDefaultColumns(modelBuilder);
modelBuilder.Entity("ControlCenter.Api.Entities.Agent", b =>
{
b.Property<Guid>("Id")
.ValueGeneratedOnAdd()
.HasColumnType("uuid")
.HasColumnName("id");
b.Property<string>("Channel")
.IsRequired()
.HasColumnType("text")
.HasColumnName("channel");
b.Property<DateTime>("CreatedAt")
.ValueGeneratedOnAdd()
.HasColumnType("timestamptz")
.HasColumnName("created_at")
.HasDefaultValueSql("now()");
b.Property<DateTime>("LastActivity")
.HasColumnType("timestamptz")
.HasColumnName("last_activity");
b.Property<int?>("Progress")
.HasColumnType("integer")
.HasColumnName("progress");
b.Property<string>("SessionKey")
.IsRequired()
.HasColumnType("text")
.HasColumnName("session_key");
b.Property<int>("Status")
.HasColumnType("agent_status")
.HasColumnName("status");
b.Property<string>("Task")
.HasColumnType("text")
.HasColumnName("task");
b.Property<DateTime>("UpdatedAt")
.ValueGeneratedOnAdd()
.HasColumnType("timestamptz")
.HasColumnName("updated_at")
.HasDefaultValueSql("now()");
b.HasKey("Id");
b.HasIndex("Channel")
.HasDatabaseName("ix_agents_channel");
b.HasIndex("SessionKey")
.IsUnique()
.HasDatabaseName("ix_agents_session_key");
b.HasIndex("Status")
.HasDatabaseName("ix_agents_status");
b.ToTable("agents", null, t =>
{
t.HasCheckConstraint("ck_agents_progress_range", "progress IS NULL OR (progress >= 0 AND progress <= 100)");
});
});
#pragma warning restore 612, 618
}
}
}

31
backend/Program.cs Normal file
View File

@@ -0,0 +1,31 @@
using ControlCenter.Api.Data;
using Microsoft.EntityFrameworkCore;
var builder = WebApplication.CreateBuilder(args);
// Add services to the container.
builder.Services.AddOpenApi();
// Register DbContext with PostgreSQL
builder.Services.AddDbContext<AppDbContext>(options =>
{
var connectionString = builder.Configuration.GetConnectionString("DefaultConnection")
?? "Host=localhost;Database=control_center;Username=postgres;Password=postgres";
options.UseNpgsql(connectionString, npgsqlOptions =>
{
npgsqlOptions.MigrationsAssembly(typeof(AppDbContext).Assembly.FullName);
});
});
var app = builder.Build();
// Configure the HTTP request pipeline.
if (app.Environment.IsDevelopment())
{
app.MapOpenApi();
}
app.UseHttpsRedirection();
app.Run();

View File

@@ -0,0 +1,23 @@
{
"$schema": "https://json.schemastore.org/launchsettings.json",
"profiles": {
"http": {
"commandName": "Project",
"dotnetRunMessages": true,
"launchBrowser": false,
"applicationUrl": "http://localhost:5178",
"environmentVariables": {
"ASPNETCORE_ENVIRONMENT": "Development"
}
},
"https": {
"commandName": "Project",
"dotnetRunMessages": true,
"launchBrowser": false,
"applicationUrl": "https://localhost:7041;http://localhost:5178",
"environmentVariables": {
"ASPNETCORE_ENVIRONMENT": "Development"
}
}
}
}

View File

@@ -0,0 +1,8 @@
{
"Logging": {
"LogLevel": {
"Default": "Information",
"Microsoft.AspNetCore": "Warning"
}
}
}

12
backend/appsettings.json Normal file
View File

@@ -0,0 +1,12 @@
{
"Logging": {
"LogLevel": {
"Default": "Information",
"Microsoft.AspNetCore": "Warning"
}
},
"AllowedHosts": "*",
"ConnectionStrings": {
"DefaultConnection": "Host=localhost;Database=control_center;Username=postgres;Password=postgres"
}
}