synthra 0.1.0

data/tech_example/10_custom_types.rb

@@ -0,0 +1,387 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# ============================================================================
+# FakeDataDSL - Custom Types Examples
+# ============================================================================
+# This file demonstrates how to create and register custom types.
+# Run with: cd fake_data_dsl && ruby tech_example/10_custom_types.rb
+# ============================================================================
+
+# Load the local gem
+$LOAD_PATH.unshift(File.expand_path('../lib', __dir__))
+require 'fake_data_dsl'
+require 'json'
+
+puts "=" * 70
+puts "FakeDataDSL Custom Types Examples"
+puts "=" * 70
+puts
+
+# ============================================================================
+# Example 1: Simple Block-Based Custom Type
+# ============================================================================
+puts "1. SIMPLE BLOCK-BASED CUSTOM TYPE"
+puts "-" * 40
+
+# Register a custom SSN type
+FakeDataDSL.register_type(:ssn) do |rng, _context, _args, mode|
+  case mode
+  when :random
+    "#{rng.int(100, 999)}-#{rng.int(10, 99)}-#{rng.int(1000, 9999)}"
+  when :edge
+    ["000-00-0000", "999-99-9999", "123-45-6789"].sample
+  when :invalid
+    [nil, "invalid-ssn", "12345", 12345].sample
+  else
+    "#{rng.int(100, 999)}-#{rng.int(10, 99)}-#{rng.int(1000, 9999)}"
+  end
+end
+
+schema = FakeDataDSL.parse(<<~DSL)
+  Person:
+    name: name
+    ssn: ssn
+DSL
+
+puts "Generated Person with SSN:"
+person = schema.generate
+puts JSON.pretty_generate(person)
+puts
+puts "Edge case SSN:"
+edge_person = schema.generate(mode: :edge)
+puts "  SSN: #{edge_person['ssn']}"
+puts
+
+# ============================================================================
+# Example 2: Custom Type with Arguments
+# ============================================================================
+puts "2. CUSTOM TYPE WITH ARGUMENTS"
+puts "-" * 40
+
+# Product code with configurable prefix
+FakeDataDSL.register_type(:product_code) do |rng, _context, args, _mode|
+  prefix = args[:prefix] || "PRD"
+  length = args[:length] || 8
+  
+  suffix = (1..length).map { rng.int(0, 9) }.join
+  "#{prefix}-#{suffix}"
+end
+
+schema2 = FakeDataDSL.parse(<<~DSL)
+  Product:
+    id: uuid
+    code_default: product_code
+    code_custom: product_code(prefix: SKU, length: 6)
+    code_long: product_code(prefix: ITEM, length: 12)
+DSL
+
+puts "Generated Product Codes:"
+product = schema2.generate
+puts JSON.pretty_generate(product)
+puts
+
+# ============================================================================
+# Example 3: Class-Based Custom Type
+# ============================================================================
+puts "3. CLASS-BASED CUSTOM TYPE"
+puts "-" * 40
+
+# Define a custom type class
+class VinType < FakeDataDSL::Types::Base
+  VIN_CHARS = "ABCDEFGHJKLMNPRSTUVWXYZ0123456789".chars.freeze
+  
+  def generate_random(rng, _args)
+    # VIN is 17 characters
+    17.times.map { VIN_CHARS[rng.int(0, VIN_CHARS.length - 1)] }.join
+  end
+  
+  def generate_edge(rng, _args)
+    # Edge cases
+    [
+      "11111111111111111",  # All 1s
+      "00000000000000000",  # All 0s
+      "WVWZZZ3CZWE123456",  # Real-ish format
+      "A" * 17,             # All same letter
+    ].sample
+  end
+  
+  def generate_invalid(rng, _args)
+    [
+      nil,
+      "INVALID",
+      "TOO-SHORT",
+      "THIS-IS-WAY-TOO-LONG-FOR-A-VIN",
+      12345,
+      ""
+    ].sample
+  end
+end
+
+FakeDataDSL.register_type(:vin, VinType)
+
+schema3 = FakeDataDSL.parse(<<~DSL)
+  Vehicle:
+    id: uuid
+    make: enum(Toyota, Honda, Ford, BMW, Tesla)
+    model: text(5..15)
+    year: number(1990..2024)
+    vin: vin
+DSL
+
+puts "Generated Vehicle with VIN:"
+vehicle = schema3.generate
+puts JSON.pretty_generate(vehicle)
+puts
+puts "Edge case VIN: #{schema3.generate(mode: :edge)['vin']}"
+puts
+
+# ============================================================================
+# Example 4: Custom Type Using Faker
+# ============================================================================
+puts "4. CUSTOM TYPE USING FAKER"
+puts "-" * 40
+
+require 'faker'
+
+FakeDataDSL.register_type(:pokemon_name) do |_rng, _context, _args, mode|
+  case mode
+  when :random
+    Faker::Games::Pokemon.name
+  when :edge
+    ["Pikachu", "MissingNo", "", "A" * 100].sample
+  when :invalid
+    [nil, 123, [], {}].sample
+  else
+    Faker::Games::Pokemon.name
+  end
+end
+
+FakeDataDSL.register_type(:lorem_ipsum) do |_rng, _context, args, _mode|
+  sentences = args[:sentences] || 3
+  Faker::Lorem.sentences(number: sentences).join(" ")
+end
+
+schema4 = FakeDataDSL.parse(<<~DSL)
+  GameCharacter:
+    id: uuid
+    name: pokemon_name
+    bio: lorem_ipsum(sentences: 5)
+    level: number(1..100)
+DSL
+
+puts "Generated Game Character:"
+character = schema4.generate
+puts JSON.pretty_generate(character)
+puts
+
+# ============================================================================
+# Example 5: Complex Custom Type
+# ============================================================================
+puts "5. COMPLEX CUSTOM TYPE (Credit Card)"
+puts "-" * 40
+
+class CreditCardType < FakeDataDSL::Types::Base
+  PREFIXES = {
+    visa: ["4"],
+    mastercard: ["51", "52", "53", "54", "55"],
+    amex: ["34", "37"],
+    discover: ["6011"]
+  }.freeze
+  
+  def generate_random(rng, args)
+    card_type = args[:type]&.to_sym || PREFIXES.keys.sample
+    generate_card(rng, card_type)
+  end
+  
+  def generate_edge(rng, args)
+    # Famous test card numbers
+    [
+      "4111111111111111",  # Visa test
+      "5500000000000004",  # Mastercard test
+      "340000000000009",   # Amex test
+      "0000000000000000",  # All zeros
+    ].sample
+  end
+  
+  def generate_invalid(rng, args)
+    [
+      nil,
+      "invalid",
+      "1234",
+      "1234-5678-9012-3456-7890",  # Too long
+      12345678901234,              # Number not string
+    ].sample
+  end
+  
+  private
+  
+  def generate_card(rng, card_type)
+    prefix = PREFIXES[card_type]&.sample || "4"
+    length = card_type == :amex ? 15 : 16
+    
+    # Generate number with prefix
+    remaining = length - prefix.length - 1
+    number = prefix + remaining.times.map { rng.int(0, 9) }.join
+    
+    # Add Luhn check digit
+    number + luhn_check_digit(number)
+  end
+  
+  def luhn_check_digit(number)
+    digits = number.chars.map(&:to_i).reverse
+    sum = digits.each_with_index.sum do |digit, i|
+      if i.odd?
+        doubled = digit * 2
+        doubled > 9 ? doubled - 9 : doubled
+      else
+        digit
+      end
+    end
+    ((10 - (sum % 10)) % 10).to_s
+  end
+end
+
+FakeDataDSL.register_type(:credit_card_number, CreditCardType)
+
+schema5 = FakeDataDSL.parse(<<~DSL)
+  PaymentMethod:
+    id: uuid
+    card_number: credit_card_number
+    visa_card: credit_card_number(type: visa)
+    amex_card: credit_card_number(type: amex)
+    expiry_month: number(1..12)
+    expiry_year: number(2024..2030)
+    cvv: text(3)
+DSL
+
+puts "Generated Payment Method:"
+payment = schema5.generate
+puts JSON.pretty_generate(payment)
+puts
+
+# ============================================================================
+# Example 6: Custom Type with Context
+# ============================================================================
+puts "6. CUSTOM TYPE WITH CONTEXT"
+puts "-" * 40
+
+FakeDataDSL.register_type(:username_from_email) do |rng, context, _args, _mode|
+  email = context['email']
+  if email
+    # Extract username from email
+    base = email.split('@').first.gsub('.', '_')
+    "#{base}#{rng.int(1, 999)}"
+  else
+    "user#{rng.int(10000, 99999)}"
+  end
+end
+
+schema6 = FakeDataDSL.parse(<<~DSL)
+  UserAccount:
+    email: email
+    username: username_from_email
+DSL
+
+puts "Generated User (username derived from email):"
+3.times do
+  account = schema6.generate
+  puts "  #{account['email']} -> #{account['username']}"
+end
+puts
+
+# ============================================================================
+# Example 7: Locale-Aware Custom Type
+# ============================================================================
+puts "7. LOCALE-AWARE CUSTOM TYPE"
+puts "-" * 40
+
+FakeDataDSL.register_type(:local_phone) do |rng, _context, args, _mode|
+  country = args[:country] || "US"
+  
+  case country.to_s.upcase
+  when "US"
+    area = rng.int(200, 999)
+    prefix = rng.int(200, 999)
+    line = rng.int(1000, 9999)
+    "+1-#{area}-#{prefix}-#{line}"
+  when "UK"
+    "+44-#{rng.int(1000, 9999)}-#{rng.int(100000, 999999)}"
+  when "DE"
+    "+49-#{rng.int(100, 999)}-#{rng.int(1000000, 9999999)}"
+  when "JP"
+    "+81-#{rng.int(10, 99)}-#{rng.int(1000, 9999)}-#{rng.int(1000, 9999)}"
+  else
+    "+#{rng.int(1, 999)}-#{rng.int(100000000, 999999999)}"
+  end
+end
+
+schema7 = FakeDataDSL.parse(<<~DSL)
+  Contact:
+    name: name
+    phone_us: local_phone(country: US)
+    phone_uk: local_phone(country: UK)
+    phone_de: local_phone(country: DE)
+    phone_jp: local_phone(country: JP)
+DSL
+
+puts "Generated Contact with Localized Phones:"
+contact = schema7.generate
+puts JSON.pretty_generate(contact)
+puts
+
+# ============================================================================
+# Example 8: Stateful Custom Type
+# ============================================================================
+puts "8. STATEFUL CUSTOM TYPE (Auto-increment)"
+puts "-" * 40
+
+class TicketNumberType < FakeDataDSL::Types::Base
+  def initialize
+    super
+    @mutex = Mutex.new
+    @counter = 0
+  end
+  
+  def generate_random(_rng, args)
+    prefix = args[:prefix] || "TKT"
+    year = Time.now.year
+    
+    number = @mutex.synchronize do
+      @counter += 1
+      @counter
+    end
+    
+    "#{prefix}-#{year}-#{number.to_s.rjust(6, '0')}"
+  end
+  
+  def generate_edge(_rng, _args)
+    generate_random(nil, {})  # Use same logic for consistency
+  end
+  
+  def generate_invalid(_rng, _args)
+    [nil, "", "INVALID", 12345].sample
+  end
+end
+
+FakeDataDSL.register_type(:ticket_number, TicketNumberType.new)
+
+schema8 = FakeDataDSL.parse(<<~DSL)
+  SupportTicket:
+    ticket_number: ticket_number
+    subject: text(20..100)
+    priority: enum(low, medium, high, critical)
+    status: enum(open, in_progress, resolved, closed)
+DSL
+
+puts "Generated Support Tickets with Auto-Increment:"
+5.times do
+  ticket = schema8.generate
+  puts "  #{ticket['ticket_number']}: #{ticket['subject'][0..30]}..."
+end
+puts
+
+puts "=" * 70
+puts "Custom types examples completed!"
+puts "=" * 70
+