polen 0.11.0-next.2 → 0.11.0-next.20

package/src/lib/ai-image-generation/$.test.ts

@@ -0,0 +1,344 @@
+import { Effect } from 'effect'
+import { buildSchema } from 'graphql'
+import { describe, expect, test } from 'vitest'
+import { AiImageGeneration } from './$.js'
+
+describe('AI Image Generation', () => {
+  describe('analyzeSchema', () => {
+    test('extracts context from e-commerce schema', () => {
+      const schema = buildSchema(`
+        type Product {
+          id: ID!
+          name: String!
+          price: Float!
+        }
+        
+        type Cart {
+          id: ID!
+          items: [Product!]!
+        }
+        
+        type Query {
+          products: [Product!]!
+          cart(id: ID!): Cart
+        }
+        
+        type Mutation {
+          addToCart(productId: ID!): Cart!
+        }
+      `)
+
+      const context = AiImageGeneration.analyzeSchema(schema)
+
+      expect(context.domain).toBe('e-commerce')
+      expect(context.suggestedTheme).toBe('digital marketplace')
+      expect(context.mainTypes).toContain('Product')
+      expect(context.mainTypes).toContain('Cart')
+      expect(context.queryNames).toContain('products')
+      expect(context.mutationNames).toContain('addToCart')
+    })
+
+    test('extracts context from social media schema', () => {
+      const schema = buildSchema(`
+        type User {
+          id: ID!
+          name: String!
+        }
+        
+        type Post {
+          id: ID!
+          author: User!
+          content: String!
+        }
+        
+        type Comment {
+          id: ID!
+          post: Post!
+          author: User!
+          text: String!
+        }
+        
+        type Query {
+          users: [User!]!
+          posts: [Post!]!
+        }
+      `)
+
+      const context = AiImageGeneration.analyzeSchema(schema)
+
+      expect(context.domain).toBe('social')
+      expect(context.suggestedTheme).toBe('connected network')
+      expect(context.mainTypes).toContain('User')
+      expect(context.mainTypes).toContain('Post')
+      expect(context.mainTypes).toContain('Comment')
+    })
+
+    test('extracts context from analytics schema', () => {
+      const schema = buildSchema(`
+        type Metric {
+          id: ID!
+          name: String!
+          value: Float!
+        }
+        
+        type Report {
+          id: ID!
+          metrics: [Metric!]!
+        }
+        
+        type Dashboard {
+          id: ID!
+          reports: [Report!]!
+        }
+        
+        type Query {
+          dashboard: Dashboard!
+          metrics: [Metric!]!
+        }
+      `)
+
+      const context = AiImageGeneration.analyzeSchema(schema)
+
+      expect(context.domain).toBe('analytics')
+      expect(context.suggestedTheme).toBe('data visualization')
+      expect(context.mainTypes).toContain('Metric')
+      expect(context.mainTypes).toContain('Report')
+      expect(context.mainTypes).toContain('Dashboard')
+    })
+
+    test('handles generic schema without specific domain', () => {
+      const schema = buildSchema(`
+        type Item {
+          id: ID!
+          name: String!
+        }
+        
+        type Query {
+          items: [Item!]!
+        }
+      `)
+
+      const context = AiImageGeneration.analyzeSchema(schema)
+
+      expect(context.domain).toBeUndefined()
+      expect(context.suggestedTheme).toBeUndefined()
+      expect(context.mainTypes).toContain('Item')
+    })
+  })
+
+  describe('buildPrompt', () => {
+    test('uses custom prompt when provided', () => {
+      const context: AiImageGeneration.SchemaContext = {
+        mainTypes: [],
+        queryNames: [],
+        mutationNames: [],
+        subscriptionNames: [],
+      }
+
+      const prompt = AiImageGeneration.buildPrompt(
+        context,
+        'modern',
+        'Beautiful landscape with mountains',
+      )
+
+      expect(prompt).toContain('Beautiful landscape with mountains')
+      expect(prompt).toContain('modern clean design')
+    })
+
+    test('generates e-commerce themed prompt', () => {
+      const context: AiImageGeneration.SchemaContext = {
+        mainTypes: ['Product', 'Cart'],
+        queryNames: ['products'],
+        mutationNames: ['addToCart'],
+        subscriptionNames: [],
+        domain: 'e-commerce',
+        suggestedTheme: 'digital marketplace',
+      }
+
+      const prompt = AiImageGeneration.buildPrompt(context, 'vibrant')
+
+      expect(prompt).toContain('digital marketplace')
+      expect(prompt).toContain('all about transactions, style, markets, trade, living')
+      expect(prompt).toContain('vibrant colors')
+      expect(prompt).toContain('no text')
+    })
+
+    test('generates social themed prompt', () => {
+      const context: AiImageGeneration.SchemaContext = {
+        mainTypes: ['User', 'Post'],
+        queryNames: ['users'],
+        mutationNames: [],
+        subscriptionNames: [],
+        domain: 'social',
+        suggestedTheme: 'connected network',
+      }
+
+      const prompt = AiImageGeneration.buildPrompt(context, 'futuristic')
+
+      expect(prompt).toContain('connected network')
+      expect(prompt).toContain('all about community, connections, sharing, communication, networks')
+      expect(prompt).toContain('futuristic sci-fi')
+      expect(prompt).toContain('no text')
+    })
+
+    test('generates analytics themed prompt', () => {
+      const context: AiImageGeneration.SchemaContext = {
+        mainTypes: ['Metric', 'Dashboard'],
+        queryNames: ['metrics'],
+        mutationNames: [],
+        subscriptionNames: [],
+        domain: 'analytics',
+        suggestedTheme: 'data visualization',
+      }
+
+      const prompt = AiImageGeneration.buildPrompt(context, 'technical')
+
+      expect(prompt).toContain('data visualization')
+      expect(prompt).toContain('all about data, insights, metrics, visualization, trends')
+      expect(prompt).toContain('technical blueprint')
+      expect(prompt).toContain('no text')
+    })
+
+    test.for([
+      { style: 'modern' as const, expected: 'modern clean design' },
+      { style: 'minimalist' as const, expected: 'minimalist flat design' },
+      { style: 'abstract' as const, expected: 'abstract artistic' },
+      { style: 'geometric' as const, expected: 'geometric patterns' },
+      { style: 'gradient' as const, expected: 'gradient mesh' },
+    ])('includes style description for "$style"', ({ style, expected }) => {
+      const context: AiImageGeneration.SchemaContext = {
+        mainTypes: [],
+        queryNames: [],
+        mutationNames: [],
+        subscriptionNames: [],
+      }
+
+      const prompt = AiImageGeneration.buildPrompt(context, style)
+      expect(prompt).toContain(expected)
+    })
+  })
+
+  describe('Pollinations Provider', () => {
+    test('generates correct URL format', async () => {
+      const { generateImage } = await import('./providers/pollinations.js')
+
+      const result = await generateImage({
+        prompt: 'Abstract API visualization',
+        width: 1200,
+        height: 400,
+        nologo: true,
+      })
+
+      expect(result.url).toContain('https://image.pollinations.ai/prompt/')
+      expect(result.url).toContain('Abstract%20API%20visualization')
+      expect(result.url).toContain('width=1200')
+      expect(result.url).toContain('height=400')
+      expect(result.url).toContain('nologo=true')
+      expect(result.cached).toBe(false)
+    })
+
+    test('includes seed when provided', async () => {
+      const { generateImage } = await import('./providers/pollinations.js')
+
+      const result = await generateImage({
+        prompt: 'Test prompt',
+        seed: 12345,
+      })
+
+      expect(result.url).toContain('seed=12345')
+    })
+  })
+
+  describe('AiImageConfig Schema', () => {
+    test('validates correct configuration', () => {
+      const config = {
+        enabled: true,
+        style: 'modern' as const,
+        width: 1200,
+        height: 400,
+        seed: 12345,
+        nologo: true,
+        cache: true,
+      }
+
+      const result = Effect.runSync(AiImageGeneration.decode(config))
+      expect(result).toEqual(config)
+    })
+
+    test('accepts optional fields', () => {
+      const config = {
+        enabled: true,
+      }
+
+      const result = Effect.runSync(AiImageGeneration.decode(config))
+      expect(result).toEqual(config)
+    })
+
+    test('validates seed field', () => {
+      const config = {
+        enabled: true,
+        seed: 42,
+      }
+
+      const result = Effect.runSync(AiImageGeneration.decode(config))
+      expect(result.seed).toBe(42)
+    })
+
+    test('validates nologo field', () => {
+      const config = {
+        enabled: true,
+        nologo: false,
+      }
+
+      const result = Effect.runSync(AiImageGeneration.decode(config))
+      expect(result.nologo).toBe(false)
+    })
+
+    test('validates art style values', () => {
+      const validStyles = [
+        'modern',
+        'minimalist',
+        'abstract',
+        'technical',
+        'vibrant',
+        'professional',
+        'futuristic',
+        'gradient',
+        'geometric',
+        'illustration',
+      ]
+
+      for (const style of validStyles) {
+        const config = {
+          enabled: true,
+          style: style as AiImageGeneration.ArtStyle,
+        }
+
+        const result = Effect.runSync(AiImageGeneration.decode(config))
+        expect(result.style).toBe(style)
+      }
+    })
+  })
+
+  describe('Service caching', () => {
+    test('generates cache key from prompt', () => {
+      const service = new AiImageGeneration.AiImageGenerationService()
+
+      // Access private method through type assertion for testing
+      const getCacheKey = (service as any).getCacheKey.bind(service)
+
+      const key1 = getCacheKey('test prompt')
+      const key2 = getCacheKey('different prompt')
+
+      // Same prompt should generate same key
+      expect(getCacheKey('test prompt')).toBe(key1)
+
+      // Different prompts should generate different keys
+      expect(key1).not.toBe(key2)
+
+      // All keys should have pollinations prefix
+      expect(key1).toContain('hero-pollinations-')
+      expect(key2).toContain('hero-pollinations-')
+    })
+  })
+})

package/src/lib/ai-image-generation/$.ts

@@ -0,0 +1 @@
+export * as AiImageGeneration from './$$.js'

package/src/lib/ai-image-generation/ai-image-generation.ts

@@ -0,0 +1,231 @@
+import * as S from 'effect/Schema'
+import type { GraphQLSchema } from 'graphql'
+
+// ============================================================================
+// Schemas & Types
+// ============================================================================
+
+/**
+ * Art style preferences for generated images
+ */
+export const ArtStyle = S.Literal(
+  'modern',
+  'minimalist',
+  'abstract',
+  'technical',
+  'vibrant',
+  'professional',
+  'futuristic',
+  'gradient',
+  'geometric',
+  'illustration',
+)
+export type ArtStyle = typeof ArtStyle.Type
+
+/**
+ * Configuration for AI image generation using Pollinations
+ */
+export const AiImageConfig = S.Struct({
+  /**
+   * Enable AI hero image generation
+   * @default false
+   */
+  enabled: S.Boolean,
+  /**
+   * Custom prompt override (if not provided, will generate from schema)
+   */
+  prompt: S.optional(S.String),
+  /**
+   * Art style for the generated image
+   * @default 'modern'
+   */
+  style: S.optional(ArtStyle),
+  /**
+   * Image dimensions
+   */
+  width: S.optional(S.Number),
+  height: S.optional(S.Number),
+  /**
+   * Random seed for reproducible generation
+   */
+  seed: S.optional(S.Number),
+  /**
+   * Whether to include Pollinations logo
+   * @default false
+   */
+  nologo: S.optional(S.Boolean),
+  /**
+   * Whether to cache generated images
+   * @default true
+   */
+  cache: S.optional(S.Boolean),
+})
+
+export type AiImageConfig = typeof AiImageConfig.Type
+
+/**
+ * Result of image generation
+ */
+export const GeneratedImage = S.Struct({
+  url: S.String,
+  prompt: S.String,
+  cached: S.Boolean,
+})
+
+export type GeneratedImage = typeof GeneratedImage.Type
+
+// ============================================================================
+// Schema Context Extraction
+// ============================================================================
+
+export interface SchemaContext {
+  name?: string | undefined
+  description?: string | undefined
+  mainTypes: string[]
+  queryNames: string[]
+  mutationNames: string[]
+  subscriptionNames: string[]
+  domain?: string | undefined // e.g., 'e-commerce', 'social', 'analytics'
+  suggestedTheme?: string | undefined // e.g., 'data visualization', 'connectivity', 'growth'
+  topics?: string[] | undefined // Custom or inferred topics
+}
+
+/**
+ * Analyzes a GraphQL schema to extract contextual information for image generation
+ */
+export const analyzeSchema = (schema: GraphQLSchema): SchemaContext => {
+  const typeMap = schema.getTypeMap()
+  const queryType = schema.getQueryType()
+  const mutationType = schema.getMutationType()
+  const subscriptionType = schema.getSubscriptionType()
+
+  // Extract main types (excluding built-ins)
+  const mainTypes = Object.keys(typeMap)
+    .filter(name => !name.startsWith('__') && !['Query', 'Mutation', 'Subscription'].includes(name))
+    .slice(0, 10) // Limit to most important types
+
+  // Extract operation names
+  const queryNames = queryType
+    ? Object.keys(queryType.getFields()).slice(0, 5)
+    : []
+
+  const mutationNames = mutationType
+    ? Object.keys(mutationType.getFields()).slice(0, 5)
+    : []
+
+  const subscriptionNames = subscriptionType
+    ? Object.keys(subscriptionType.getFields()).slice(0, 5)
+    : []
+
+  // Infer domain from type/operation names
+  const allNames = [...mainTypes, ...queryNames, ...mutationNames].join(' ').toLowerCase()
+
+  let domain: string | undefined
+  let suggestedTheme: string | undefined
+
+  if (allNames.includes('product') || allNames.includes('cart') || allNames.includes('order')) {
+    domain = 'e-commerce'
+    suggestedTheme = 'digital marketplace'
+  } else if (allNames.includes('user') || allNames.includes('post') || allNames.includes('comment')) {
+    domain = 'social'
+    suggestedTheme = 'connected network'
+  } else if (allNames.includes('metric') || allNames.includes('report') || allNames.includes('dashboard')) {
+    domain = 'analytics'
+    suggestedTheme = 'data visualization'
+  } else if (allNames.includes('pokemon') || allNames.includes('game')) {
+    domain = 'gaming'
+    suggestedTheme = 'digital adventure'
+  }
+
+  return {
+    mainTypes,
+    queryNames,
+    mutationNames,
+    subscriptionNames,
+    domain,
+    suggestedTheme,
+  }
+}
+
+// ============================================================================
+// Prompt Building
+// ============================================================================
+
+/**
+ * Default topics for each domain
+ */
+const DEFAULT_TOPICS_BY_DOMAIN: Record<string, string[]> = {
+  'e-commerce': ['transactions', 'style', 'markets', 'trade', 'living'],
+  'social': ['community', 'connections', 'sharing', 'communication', 'networks'],
+  'analytics': ['data', 'insights', 'metrics', 'visualization', 'trends'],
+  'gaming': ['adventure', 'competition', 'strategy', 'fantasy', 'achievement'],
+  'technology': ['innovation', 'connectivity', 'automation', 'digital', 'future'],
+}
+
+/**
+ * Builds an image generation prompt from schema context and configuration
+ */
+export const buildPrompt = (
+  context: SchemaContext,
+  style: ArtStyle = 'modern',
+  customPrompt?: string,
+  layout?: 'asymmetric' | 'cinematic' | 'auto',
+): string => {
+  // If custom prompt provided, use it with style suffix and ensure no text
+  if (customPrompt) {
+    return `${customPrompt}, ${getStyleDescription(style)}, no text, no words, no letters, no typography`
+  }
+
+  // Build automatic prompt from context
+  const theme = context.suggestedTheme || 'modern API'
+  const domain = context.domain || 'technology'
+
+  // Create a descriptive prompt based on the schema analysis
+  let prompt = layout === 'asymmetric'
+    ? `Floating ${theme} illustration for a GraphQL API, no text, no words, no letters, isolated graphic element, clean edges, no background, white space around subject, perfect for placing alongside typography`
+    : `Abstract ${theme} hero banner for a GraphQL API, no text, no words, no letters`
+
+  // Use topics if provided, otherwise fall back to domain defaults
+  const topics = context.topics || DEFAULT_TOPICS_BY_DOMAIN[domain] || DEFAULT_TOPICS_BY_DOMAIN['technology']
+
+  // Add topics to the prompt
+  if (topics && topics.length > 0) {
+    prompt += `, all about ${topics.join(', ')}`
+  }
+
+  // Add style description
+  prompt += `, ${getStyleDescription(style)}`
+
+  // Add technical specifications and reinforce no text
+  prompt += ', high quality, absolutely no text or typography or letters or words'
+
+  return prompt
+}
+
+/**
+ * Gets a description for the art style
+ */
+const getStyleDescription = (style: ArtStyle): string => {
+  const styleMap: Record<ArtStyle, string> = {
+    modern: 'modern clean design with subtle gradients',
+    minimalist: 'minimalist flat design with simple shapes',
+    abstract: 'abstract artistic representation',
+    technical: 'technical blueprint style with grid patterns',
+    vibrant: 'vibrant colors and dynamic composition',
+    professional: 'professional corporate style',
+    futuristic: 'futuristic sci-fi aesthetic with neon accents',
+    gradient: 'smooth gradient mesh with flowing colors',
+    geometric: 'geometric patterns and angular shapes',
+    illustration: 'vector illustration style with clean lines, suitable for placement on backgrounds',
+  }
+
+  return styleMap[style]
+}
+
+// ============================================================================
+// Exports
+// ============================================================================
+
+export const make = AiImageConfig.make
+export const decode = S.decode(AiImageConfig)
+export const encode = S.encode(AiImageConfig)

package/src/lib/ai-image-generation/providers/pollinations.ts

@@ -0,0 +1,65 @@
+/**
+ * Pollinations.ai provider for AI image generation
+ * Free service, no API key required
+ */
+
+import type { GeneratedImage } from '../ai-image-generation.js'
+
+export interface PollinationsOptions {
+  prompt: string
+  width?: number
+  height?: number
+  seed?: number
+  nologo?: boolean
+}
+
+/**
+ * Generate an image using Pollinations.ai
+ * This is a free service that doesn't require authentication
+ */
+export const generateImage = async (options: PollinationsOptions): Promise<GeneratedImage> => {
+  const {
+    prompt,
+    width = 1200,
+    height = 400,
+    seed,
+    nologo = true,
+  } = options
+
+  // Build the URL with query parameters
+  const params = new URLSearchParams({
+    prompt,
+    width: width.toString(),
+    height: height.toString(),
+    nologo: nologo.toString(),
+  })
+
+  if (seed !== undefined) {
+    params.append('seed', seed.toString())
+  }
+
+  // Pollinations.ai URL format
+  const url = `https://image.pollinations.ai/prompt/${encodeURIComponent(prompt)}?${params.toString()}`
+
+  // The URL itself is the image endpoint - no need to fetch
+  // Pollinations generates the image on-demand when the URL is accessed
+  return {
+    url,
+    prompt,
+    cached: false,
+  }
+}
+
+/**
+ * Check if the Pollinations service is available
+ */
+export const isAvailable = async (): Promise<boolean> => {
+  try {
+    const response = await fetch('https://image.pollinations.ai/prompt/test', {
+      method: 'HEAD',
+    })
+    return response.ok || response.status === 200
+  } catch {
+    return false
+  }
+}