templates.go

package main

import (
	"fmt"
	"os"
	"strings"
)

var builtInTemplates = map[string]string{
	"e-commerce": templateECommerce,
	"saas":       templateSaaS,
	"social":     templateSocial,
}

func listTemplates() []string {
	names := make([]string, 0, len(builtInTemplates))
	for name := range builtInTemplates {
		names = append(names, name)
	}
	return names
}

func getTemplate(name string) (string, error) {
	tmpl, ok := builtInTemplates[strings.ToLower(name)]
	if !ok {
		return "", fmt.Errorf("unknown template %q (available: %s)", name, strings.Join(listTemplates(), ", "))
	}
	return tmpl, nil
}

func RunTemplate(name, dbType string, preview int) error {
	tmpl, err := getTemplate(name)
	if err != nil {
		return err
	}
	return runSchemaBytes([]byte(tmpl), dbType, preview, fmt.Sprintf("template:%s", name))
}

func SaveTemplate(name, outputPath string) error {
	tmpl, err := getTemplate(name)
	if err != nil {
		return err
	}
	return os.WriteFile(outputPath, []byte(tmpl), 0644)
}

const templateECommerce = `version: "1.0"
tables:
  - name: customers
    count: 1000
    columns:
      - name: id
        type: int
        primary: true
      - name: first_name
        type: first_name
      - name: last_name
        type: last_name
      - name: email
        type: email
      - name: phone
        type: phone
      - name: city
        type: city
      - name: country
        type: country
      - name: created_at
        type: timestamp

  - name: products
    count: 500
    columns:
      - name: id
        type: int
        primary: true
      - name: name
        type: sentence
      - name: description
        type: paragraph
      - name: category
        type: enum
        values: [Electronics, Clothing, Home, Sports, Books, Food]
      - name: price
        type: float
        min: 9.99
        max: 999.99
      - name: stock
        type: int
        min: 0
        max: 1000
      - name: created_at
        type: timestamp

  - name: orders
    count: 5000
    columns:
      - name: id
        type: int
        primary: true
      - name: customer_id
        type: fk
        ref: customers.id
      - name: total
        type: float
        min: 10.0
        max: 2000.0
      - name: status
        type: enum
        values: [pending, paid, shipped, delivered, cancelled]
      - name: created_at
        type: timestamp

  - name: order_items
    count: 10000
    columns:
      - name: id
        type: int
        primary: true
      - name: order_id
        type: fk
        ref: orders.id
      - name: product_id
        type: fk
        ref: products.id
      - name: quantity
        type: int
        min: 1
        max: 10
      - name: unit_price
        type: float
        min: 9.99
        max: 999.99
      - name: created_at
        type: timestamp
`

const templateSaaS = `version: "1.0"
tables:
  - name: tenants
    count: 50
    columns:
      - name: id
        type: int
        primary: true
      - name: name
        type: company
      - name: domain
        type: email
      - name: plan
        type: enum
        values: [free, starter, pro, enterprise]
      - name: created_at
        type: timestamp

  - name: users
    count: 500
    columns:
      - name: id
        type: int
        primary: true
      - name: tenant_id
        type: fk
        ref: tenants.id
      - name: first_name
        type: first_name
      - name: last_name
        type: last_name
      - name: email
        type: email
      - name: role
        type: enum
        values: [admin, editor, viewer]
      - name: created_at
        type: timestamp

  - name: subscriptions
    count: 400
    columns:
      - name: id
        type: int
        primary: true
      - name: user_id
        type: fk
        ref: users.id
      - name: status
        type: enum
        values: [active, paused, cancelled, expired]
      - name: price
        type: float
        min: 9.99
        max: 299.99
      - name: created_at
        type: timestamp

  - name: invoices
    count: 1000
    columns:
      - name: id
        type: int
        primary: true
      - name: subscription_id
        type: fk
        ref: subscriptions.id
      - name: amount
        type: float
        min: 9.99
        max: 299.99
      - name: status
        type: enum
        values: [draft, sent, paid, overdue]
      - name: created_at
        type: timestamp
`

const templateSocial = `version: "1.0"
tables:
  - name: users
    count: 1000
    columns:
      - name: id
        type: int
        primary: true
      - name: username
        type: email
      - name: full_name
        type: full_name
      - name: bio
        type: sentence
      - name: created_at
        type: timestamp

  - name: posts
    count: 5000
    columns:
      - name: id
        type: int
        primary: true
      - name: user_id
        type: fk
        ref: users.id
      - name: content
        type: paragraph
      - name: likes_count
        type: int
        min: 0
        max: 10000
      - name: created_at
        type: timestamp

  - name: comments
    count: 10000
    columns:
      - name: id
        type: int
        primary: true
      - name: post_id
        type: fk
        ref: posts.id
      - name: user_id
        type: fk
        ref: users.id
      - name: content
        type: sentence
      - name: created_at
        type: timestamp
`