90c3f6285d
- Add mock_status, mock_headers, mock_body columns to endpoints schema
- Add MockConfig struct and UpsertMock to Repository interface
- Implement UpsertMock in sqliteRepo (INSERT ON CONFLICT UPDATE)
- Update GetEndpoint to return mock configuration when present
- Add PUT /e/{endpoint_id}/mock handler with JSON body {status, headers, body}
- Add catch-all /{endpoint_id} handler returning mock or default 200
- Header sanitization: only allow Content-Type, Cache-Control, X-*
- Wire handler into cmd/hatch main with SQLite store
- Fix Dockerfile: COPY go.sum for reproducible builds
- Tests: store mock CRUD, handler mock config + header sanitization
Co-Authored-By: Paperclip <noreply@paperclip.ing>
65 lines
2.4 KiB
Go
65 lines
2.4 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type sqliteRepo struct{ db *sql.DB }
|
|
|
|
func NewSQLiteRepo(db *sql.DB) (Repository, error) {
|
|
if db == nil { return nil, fmt.Errorf("store: nil db") }
|
|
return &sqliteRepo{db: db}, nil
|
|
}
|
|
|
|
func (r *sqliteRepo) CreateEndpoint(ctx context.Context, url string) (*Endpoint, error) {
|
|
id := uuid.NewString()
|
|
now := utcNow()
|
|
e := &Endpoint{ID: id, URL: url, CreatedAt: now, UpdatedAt: now}
|
|
_, err := r.db.ExecContext(ctx, `INSERT INTO endpoints (id, url, created_at, updated_at) VALUES (?, ?, ?, ?)`, e.ID, e.URL, e.CreatedAt, e.UpdatedAt)
|
|
if err != nil { return nil, fmt.Errorf("store: create endpoint: %w", err) }
|
|
return e, nil
|
|
}
|
|
|
|
func (r *sqliteRepo) GetEndpoint(ctx context.Context, id string) (*Endpoint, error) {
|
|
row := r.db.QueryRowContext(ctx, `SELECT id, url, created_at, updated_at FROM endpoints WHERE id = ?`, id)
|
|
e := &Endpoint{}
|
|
if err := row.Scan(&e.ID, &e.URL, &e.CreatedAt, &e.UpdatedAt); err != nil {
|
|
return nil, fmt.Errorf("store: get endpoint: %w", err)
|
|
}
|
|
return e, nil
|
|
}
|
|
|
|
func (r *sqliteRepo) AppendRequest(ctx context.Context, endpointID string, req *Request) error {
|
|
req.ID = uuid.NewString()
|
|
req.EndpointID = endpointID
|
|
req.CreatedAt = utcNow()
|
|
_, err := r.db.ExecContext(ctx, `INSERT INTO requests (id, endpoint_id, method, path, headers, query, body, created_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
|
|
req.ID, req.EndpointID, req.Method, req.Path, req.Headers, req.Query, req.Body, req.CreatedAt)
|
|
if err != nil { return fmt.Errorf("store: append request: %w", err) }
|
|
return nil
|
|
}
|
|
|
|
func (r *sqliteRepo) ListRequests(ctx context.Context, endpointID string, limit int) ([]*Request, error) {
|
|
if limit <= 0 { limit = 50 }
|
|
rows, err := r.db.QueryContext(ctx, `SELECT id, endpoint_id, method, path, headers, query, body, created_at FROM requests WHERE endpoint_id = ? ORDER BY created_at DESC LIMIT ?`, endpointID, limit)
|
|
if err != nil { return nil, fmt.Errorf("store: list requests: %w", err) }
|
|
defer rows.Close()
|
|
var out []*Request
|
|
for rows.Next() {
|
|
var req Request
|
|
if err := rows.Scan(&req.ID, &req.EndpointID, &req.Method, &req.Path, &req.Headers, &req.Query, &req.Body, &req.CreatedAt); err != nil {
|
|
return nil, fmt.Errorf("store: scan request: %w", err)
|
|
}
|
|
out = append(out, &req)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (r *sqliteRepo) Close() error { return r.db.Close() }
|
|
|
|
func utcNow() string { return time.Now().UTC().Format("2006-01-02T15:04:05.000Z07:00") }
|