Initial commit: img-infinite-canvas AI design workbench MVP
Moteva-style AI design workbench replica. Home prompt creates a project; the project page provides an infinite canvas with node drag, zoom/pan, a design chat panel, history replay, image asset upload, and project save/regenerate. - Backend: go-zero, DDD layering, sqlc/pgx, optional PostgreSQL; memory or Redis cache; asynq job queue; MinIO/S3/R2/OSS object storage; sky-valley/pi agent runtime adapter. - Frontend: Next.js App Router + Vite artifact build, TypeScript, i18n, shadcn/ui components, auth (OTP/Turnstile/Google/WeChat). - Deploy: Docker Compose and k3s manifests. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,117 @@
|
||||
package realtime
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"img_infinite_canvas/internal/domain/design"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
)
|
||||
|
||||
type ProjectRepository struct {
|
||||
next design.Repository
|
||||
bus Bus
|
||||
}
|
||||
|
||||
func NewProjectRepository(next design.Repository, bus Bus) *ProjectRepository {
|
||||
return &ProjectRepository{next: next, bus: bus}
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) List(ctx context.Context) ([]design.ProjectSummary, error) {
|
||||
return r.next.List(ctx)
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) ListPage(ctx context.Context, limit int64, offset int64) ([]design.ProjectSummary, error) {
|
||||
return r.next.ListPage(ctx, limit, offset)
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) Get(ctx context.Context, id string) (design.Project, error) {
|
||||
return r.next.Get(ctx, id)
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) Save(ctx context.Context, project design.Project) error {
|
||||
if err := r.next.Save(ctx, project); err != nil {
|
||||
return err
|
||||
}
|
||||
r.publish(ctx, project, "project.updated")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) SaveCanvas(ctx context.Context, project design.Project, baseVersion string) error {
|
||||
if next, ok := r.next.(design.CanvasRepository); ok {
|
||||
if err := next.SaveCanvas(ctx, project, baseVersion); err != nil {
|
||||
return err
|
||||
}
|
||||
r.publish(ctx, project, "canvas.snapshot.created")
|
||||
return nil
|
||||
}
|
||||
return r.Save(ctx, project)
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) Delete(ctx context.Context, id string) error {
|
||||
project, err := r.next.Get(ctx, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if err := r.next.Delete(ctx, id); err != nil {
|
||||
return err
|
||||
}
|
||||
r.publish(ctx, project, "project.deleted")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) SaveMockupBlob(ctx context.Context, blob design.MockupBlobData) error {
|
||||
next, ok := r.next.(design.MockupBlobRepository)
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
return next.SaveMockupBlob(ctx, blob)
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) GetMockupBlob(ctx context.Context, dataID string) (design.MockupBlobData, error) {
|
||||
next, ok := r.next.(design.MockupBlobRepository)
|
||||
if !ok {
|
||||
return design.MockupBlobData{}, design.ErrNotFound
|
||||
}
|
||||
return next.GetMockupBlob(ctx, dataID)
|
||||
}
|
||||
|
||||
func (r *ProjectRepository) publish(ctx context.Context, project design.Project, eventType string) {
|
||||
if r.bus == nil {
|
||||
return
|
||||
}
|
||||
event := projectEvent(project, eventType)
|
||||
if err := r.bus.Publish(ctx, event); err != nil {
|
||||
logx.Errorf("publish realtime event failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func projectEvent(project design.Project, eventType string) Event {
|
||||
createdAt := project.UpdatedAt
|
||||
if createdAt.IsZero() {
|
||||
createdAt = time.Now()
|
||||
}
|
||||
eventID := uuid.NewString()
|
||||
payload, _ := json.Marshal(map[string]any{
|
||||
"id": eventID,
|
||||
"type": eventType,
|
||||
"userId": project.UserID,
|
||||
"projectId": project.ID,
|
||||
"threadId": project.LastThreadID,
|
||||
"version": project.Version,
|
||||
"snapshotId": project.SnapshotID,
|
||||
"createdAt": createdAt.UTC().Format(time.RFC3339Nano),
|
||||
})
|
||||
return Event{
|
||||
ID: eventID,
|
||||
UserID: design.NormalizeUserID(project.UserID),
|
||||
ProjectID: project.ID,
|
||||
ThreadID: project.LastThreadID,
|
||||
Type: eventType,
|
||||
Payload: payload,
|
||||
CreatedAt: createdAt,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user