import {
  table,
  text,
  integer,
  now,
  ownableColumns,
  createSharesTable,
} from "@agent-native/core/db/schema";

export const assetLibraries = table("image_libraries", {
  id: text("id").primaryKey(),
  title: text("title").notNull(),
  description: text("description"),
  customInstructions: text("custom_instructions").notNull().default(""),
  styleBrief: text("style_brief").notNull().default("{}"),
  settings: text("settings").notNull().default("{}"),
  canonicalLogoAssetId: text("canonical_logo_asset_id"),
  coverAssetId: text("cover_asset_id"),
  archivedAt: text("archived_at"),
  createdAt: text("created_at").notNull().default(now()),
  updatedAt: text("updated_at").notNull().default(now()),
  ...ownableColumns(),
});

export const assetLibraryShares = createSharesTable("image_library_shares");

export const assetCollections = table("image_collections", {
  id: text("id").primaryKey(),
  libraryId: text("library_id").notNull(),
  title: text("title").notNull(),
  description: text("description"),
  category: text("category").notNull().default("style-only"),
  styleBrief: text("style_brief").notNull().default("{}"),
  promptTemplate: text("prompt_template"),
  defaultAspectRatio: text("default_aspect_ratio").notNull().default("16:9"),
  defaultImageSize: text("default_image_size").notNull().default("2K"),
  sortOrder: integer("sort_order").notNull().default(0),
  createdAt: text("created_at").notNull().default(now()),
  updatedAt: text("updated_at").notNull().default(now()),
});

export const assetFolders = table("asset_folders", {
  id: text("id").primaryKey(),
  libraryId: text("library_id").notNull(),
  parentId: text("parent_id"),
  title: text("title").notNull(),
  description: text("description"),
  sortOrder: integer("sort_order").notNull().default(0),
  createdAt: text("created_at").notNull().default(now()),
  updatedAt: text("updated_at").notNull().default(now()),
});

export const assetGenerationPresets = table("image_generation_presets", {
  id: text("id").primaryKey(),
  libraryId: text("library_id").notNull(),
  collectionId: text("collection_id"),
  title: text("title").notNull(),
  description: text("description"),
  category: text("category").notNull().default("style-only"),
  mediaType: text("media_type").notNull().default("image"),
  promptTemplate: text("prompt_template"),
  aspectRatio: text("aspect_ratio").notNull().default("16:9"),
  imageSize: text("image_size").notNull().default("2K"),
  model: text("model").notNull().default("gemini-3.1-flash-image"),
  textPolicy: text("text_policy").notNull().default(""),
  referencePolicy: text("reference_policy").notNull().default("auto"),
  settings: text("settings").notNull().default("{}"),
  sortOrder: integer("sort_order").notNull().default(0),
  createdAt: text("created_at").notNull().default(now()),
  updatedAt: text("updated_at").notNull().default(now()),
});

export const assetGenerationSessions = table("image_generation_sessions", {
  id: text("id").primaryKey(),
  libraryId: text("library_id").notNull(),
  collectionId: text("collection_id"),
  presetId: text("preset_id"),
  title: text("title").notNull(),
  brief: text("brief"),
  status: text("status").notNull().default("open"),
  activeAssetId: text("active_asset_id"),
  feedbackSummary: text("feedback_summary").notNull().default(""),
  metadata: text("metadata").notNull().default("{}"),
  createdBy: text("created_by"),
  createdAt: text("created_at").notNull().default(now()),
  updatedAt: text("updated_at").notNull().default(now()),
});

export const assetGenerationSessionItems = table(
  "image_generation_session_items",
  {
    id: text("id").primaryKey(),
    sessionId: text("session_id").notNull(),
    assetId: text("asset_id"),
    generationRunId: text("generation_run_id"),
    role: text("role").notNull().default("candidate"),
    note: text("note"),
    sortOrder: integer("sort_order").notNull().default(0),
    createdAt: text("created_at").notNull().default(now()),
  },
);

export const assets = table("image_assets", {
  id: text("id").primaryKey(),
  libraryId: text("library_id").notNull(),
  collectionId: text("collection_id"),
  folderId: text("folder_id"),
  mediaType: text("media_type").notNull().default("image"),
  role: text("role").notNull().default("generated"),
  status: text("status").notNull().default("candidate"),
  title: text("title"),
  description: text("description"),
  altText: text("alt_text"),
  prompt: text("prompt"),
  model: text("model"),
  aspectRatio: text("aspect_ratio"),
  imageSize: text("image_size"),
  mimeType: text("mime_type").notNull(),
  width: integer("width"),
  height: integer("height"),
  durationSeconds: integer("duration_seconds"),
  sizeBytes: integer("size_bytes"),
  objectKey: text("object_key").notNull(),
  thumbnailObjectKey: text("thumbnail_object_key"),
  sourceUrl: text("source_url"),
  generationRunId: text("generation_run_id"),
  metadata: text("metadata").notNull().default("{}"),
  createdAt: text("created_at").notNull().default(now()),
  updatedAt: text("updated_at").notNull().default(now()),
});

export const assetGenerationRuns = table("image_generation_runs", {
  id: text("id").primaryKey(),
  libraryId: text("library_id").notNull(),
  collectionId: text("collection_id"),
  presetId: text("preset_id"),
  sessionId: text("session_id"),
  prompt: text("prompt").notNull(),
  compiledPrompt: text("compiled_prompt").notNull(),
  mediaType: text("media_type").notNull().default("image"),
  model: text("model").notNull(),
  aspectRatio: text("aspect_ratio").notNull().default("16:9"),
  imageSize: text("image_size").notNull().default("2K"),
  durationSeconds: integer("duration_seconds"),
  resolution: text("resolution"),
  groundingMode: text("grounding_mode").notNull().default("auto"),
  referenceAssetIds: text("reference_asset_ids").notNull().default("[]"),
  status: text("status").notNull().default("pending"),
  error: text("error"),
  metadata: text("metadata").notNull().default("{}"),
  createdAt: text("created_at").notNull().default(now()),
  completedAt: text("completed_at"),
  // ── audit-log columns (v6-v9 migrations) ──
  // `source`: who triggered the generation ("chat" | "ui" | "a2a"). Defaulted
  // to "chat" because that's the historical path; UI button popovers and A2A
  // callers update this on insert.
  source: text("source").notNull().default("chat"),
  // `callerAppId`: only set for `source = "a2a"` — the calling app's id
  // (e.g. "slides", "design"). Lets the audit log filter "all generations
  // triggered by slides".
  callerAppId: text("caller_app_id"),
  // Identity columns for org-admin audit. Captured at insert time from the
  // request context so audit reads don't need to re-resolve who owned the run.
  ownerEmail: text("owner_email"),
  orgId: text("org_id"),
});

// Legacy export aliases keep existing generated action code and external
// imports working while the app slug/resource name moves from Images to Assets.
export const imageLibraries = assetLibraries;
export const imageLibraryShares = assetLibraryShares;
export const imageCollections = assetCollections;
export const imageGenerationPresets = assetGenerationPresets;
export const imageGenerationSessions = assetGenerationSessions;
export const imageGenerationSessionItems = assetGenerationSessionItems;
export const imageAssets = assets;
export const imageGenerationRuns = assetGenerationRuns;