Massive refactor and introduction of restheadspec

2026-01-06 11:54:25 +00:00 · 2025-11-06 16:15:35 +02:00
parent 399cea9335
commit 8e06736701
22 changed files with 2472 additions and 1046 deletions
--- a/pkg/restheadspec/handler.go
+++ b/pkg/restheadspec/handler.go
@@ -0,0 +1,616 @@
+package restheadspec
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"reflect"
+	"strings"
+
+	"github.com/Warky-Devs/ResolveSpec/pkg/common"
+	"github.com/Warky-Devs/ResolveSpec/pkg/logger"
+)
+
+// Handler handles API requests using database and model abstractions
+// This handler reads filters, columns, and options from HTTP headers
+type Handler struct {
+	db       common.Database
+	registry common.ModelRegistry
+}
+
+// NewHandler creates a new API handler with database and registry abstractions
+func NewHandler(db common.Database, registry common.ModelRegistry) *Handler {
+	return &Handler{
+		db:       db,
+		registry: registry,
+	}
+}
+
+// Handle processes API requests through router-agnostic interface
+// Options are read from HTTP headers instead of request body
+func (h *Handler) Handle(w common.ResponseWriter, r common.Request, params map[string]string) {
+	ctx := context.Background()
+
+	schema := params["schema"]
+	entity := params["entity"]
+	id := params["id"]
+
+	// Parse options from headers (now returns ExtendedRequestOptions)
+	options := h.parseOptionsFromHeaders(r)
+
+	// Determine operation based on HTTP method
+	method := r.Method()
+
+	logger.Info("Handling %s request for %s.%s", method, schema, entity)
+
+	switch method {
+	case "GET":
+		if id != "" {
+			// GET with ID - read single record
+			h.handleRead(ctx, w, schema, entity, id, options)
+		} else {
+			// GET without ID - read multiple records
+			h.handleRead(ctx, w, schema, entity, "", options)
+		}
+	case "POST":
+		// Create operation
+		body, err := r.Body()
+		if err != nil {
+			logger.Error("Failed to read request body: %v", err)
+			h.sendError(w, http.StatusBadRequest, "invalid_request", "Failed to read request body", err)
+			return
+		}
+		var data interface{}
+		if err := json.Unmarshal(body, &data); err != nil {
+			logger.Error("Failed to decode request body: %v", err)
+			h.sendError(w, http.StatusBadRequest, "invalid_request", "Invalid request body", err)
+			return
+		}
+		h.handleCreate(ctx, w, schema, entity, data, options)
+	case "PUT", "PATCH":
+		// Update operation
+		body, err := r.Body()
+		if err != nil {
+			logger.Error("Failed to read request body: %v", err)
+			h.sendError(w, http.StatusBadRequest, "invalid_request", "Failed to read request body", err)
+			return
+		}
+		var data interface{}
+		if err := json.Unmarshal(body, &data); err != nil {
+			logger.Error("Failed to decode request body: %v", err)
+			h.sendError(w, http.StatusBadRequest, "invalid_request", "Invalid request body", err)
+			return
+		}
+		h.handleUpdate(ctx, w, schema, entity, id, nil, data, options)
+	case "DELETE":
+		h.handleDelete(ctx, w, schema, entity, id)
+	default:
+		logger.Error("Invalid HTTP method: %s", method)
+		h.sendError(w, http.StatusMethodNotAllowed, "invalid_method", "Invalid HTTP method", nil)
+	}
+}
+
+// HandleGet processes GET requests for metadata
+func (h *Handler) HandleGet(w common.ResponseWriter, r common.Request, params map[string]string) {
+	schema := params["schema"]
+	entity := params["entity"]
+
+	logger.Info("Getting metadata for %s.%s", schema, entity)
+
+	model, err := h.registry.GetModelByEntity(schema, entity)
+	if err != nil {
+		logger.Error("Failed to get model: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_entity", "Invalid entity", err)
+		return
+	}
+
+	metadata := h.generateMetadata(schema, entity, model)
+	h.sendResponse(w, metadata, nil)
+}
+
+// parseOptionsFromHeaders is now implemented in headers.go
+
+func (h *Handler) handleRead(ctx context.Context, w common.ResponseWriter, schema, entity, id string, options ExtendedRequestOptions) {
+	logger.Info("Reading records from %s.%s", schema, entity)
+
+	model, err := h.registry.GetModelByEntity(schema, entity)
+	if err != nil {
+		logger.Error("Invalid entity: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_entity", "Invalid entity", err)
+		return
+	}
+
+	query := h.db.NewSelect().Model(model)
+
+	// Get table name
+	tableName := h.getTableName(schema, entity, model)
+	query = query.Table(tableName)
+
+	// Apply column selection
+	if len(options.Columns) > 0 {
+		logger.Debug("Selecting columns: %v", options.Columns)
+		query = query.Column(options.Columns...)
+	}
+
+	// Apply preloading
+	for _, preload := range options.Preload {
+		logger.Debug("Applying preload: %s", preload.Relation)
+		query = query.Preload(preload.Relation)
+	}
+
+	// Apply expand (LEFT JOIN)
+	for _, expand := range options.Expand {
+		logger.Debug("Applying expand: %s", expand.Relation)
+		// Note: Expand would require JOIN implementation
+		// For now, we'll use Preload as a fallback
+		query = query.Preload(expand.Relation)
+	}
+
+	// Apply DISTINCT if requested
+	if options.Distinct {
+		logger.Debug("Applying DISTINCT")
+		// Note: DISTINCT implementation depends on ORM support
+		// This may need to be handled differently per database adapter
+	}
+
+	// Apply filters
+	for _, filter := range options.Filters {
+		logger.Debug("Applying filter: %s %s %v", filter.Column, filter.Operator, filter.Value)
+		query = h.applyFilter(query, filter)
+	}
+
+	// Apply custom SQL WHERE clause (AND condition)
+	if options.CustomSQLWhere != "" {
+		logger.Debug("Applying custom SQL WHERE: %s", options.CustomSQLWhere)
+		query = query.Where(options.CustomSQLWhere)
+	}
+
+	// Apply custom SQL WHERE clause (OR condition)
+	if options.CustomSQLOr != "" {
+		logger.Debug("Applying custom SQL OR: %s", options.CustomSQLOr)
+		query = query.WhereOr(options.CustomSQLOr)
+	}
+
+	// If ID is provided, filter by ID
+	if id != "" {
+		logger.Debug("Filtering by ID: %s", id)
+		query = query.Where("id = ?", id)
+	}
+
+	// Apply sorting
+	for _, sort := range options.Sort {
+		direction := "ASC"
+		if strings.ToLower(sort.Direction) == "desc" {
+			direction = "DESC"
+		}
+		logger.Debug("Applying sort: %s %s", sort.Column, direction)
+		query = query.Order(fmt.Sprintf("%s %s", sort.Column, direction))
+	}
+
+	// Get total count before pagination (unless skip count is requested)
+	var total int
+	if !options.SkipCount {
+		count, err := query.Count(ctx)
+		if err != nil {
+			logger.Error("Error counting records: %v", err)
+			h.sendError(w, http.StatusInternalServerError, "query_error", "Error counting records", err)
+			return
+		}
+		total = count
+		logger.Debug("Total records: %d", total)
+	} else {
+		logger.Debug("Skipping count as requested")
+		total = -1 // Indicate count was skipped
+	}
+
+	// Apply pagination
+	if options.Limit != nil && *options.Limit > 0 {
+		logger.Debug("Applying limit: %d", *options.Limit)
+		query = query.Limit(*options.Limit)
+	}
+	if options.Offset != nil && *options.Offset > 0 {
+		logger.Debug("Applying offset: %d", *options.Offset)
+		query = query.Offset(*options.Offset)
+	}
+
+	// Execute query
+	resultSlice := reflect.New(reflect.SliceOf(reflect.TypeOf(model))).Interface()
+	if err := query.Scan(ctx, resultSlice); err != nil {
+		logger.Error("Error executing query: %v", err)
+		h.sendError(w, http.StatusInternalServerError, "query_error", "Error executing query", err)
+		return
+	}
+
+	limit := 0
+	if options.Limit != nil {
+		limit = *options.Limit
+	}
+	offset := 0
+	if options.Offset != nil {
+		offset = *options.Offset
+	}
+
+	metadata := &common.Metadata{
+		Total:    int64(total),
+		Filtered: int64(total),
+		Limit:    limit,
+		Offset:   offset,
+	}
+
+	h.sendFormattedResponse(w, resultSlice, metadata, options)
+}
+
+func (h *Handler) handleCreate(ctx context.Context, w common.ResponseWriter, schema, entity string, data interface{}, options ExtendedRequestOptions) {
+	logger.Info("Creating record in %s.%s", schema, entity)
+
+	model, err := h.registry.GetModelByEntity(schema, entity)
+	if err != nil {
+		logger.Error("Invalid entity: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_entity", "Invalid entity", err)
+		return
+	}
+
+	tableName := h.getTableName(schema, entity, model)
+
+	// Handle batch creation
+	dataValue := reflect.ValueOf(data)
+	if dataValue.Kind() == reflect.Slice || dataValue.Kind() == reflect.Array {
+		logger.Debug("Batch creation detected, count: %d", dataValue.Len())
+
+		// Use transaction for batch insert
+		err := h.db.RunInTransaction(ctx, func(tx common.Database) error {
+			for i := 0; i < dataValue.Len(); i++ {
+				item := dataValue.Index(i).Interface()
+
+				// Convert item to model type
+				modelValue := reflect.New(reflect.TypeOf(model).Elem()).Interface()
+				jsonData, err := json.Marshal(item)
+				if err != nil {
+					return fmt.Errorf("failed to marshal item: %w", err)
+				}
+				if err := json.Unmarshal(jsonData, modelValue); err != nil {
+					return fmt.Errorf("failed to unmarshal item: %w", err)
+				}
+
+				query := tx.NewInsert().Model(modelValue).Table(tableName)
+				if _, err := query.Exec(ctx); err != nil {
+					return fmt.Errorf("failed to insert record: %w", err)
+				}
+			}
+			return nil
+		})
+
+		if err != nil {
+			logger.Error("Error creating records: %v", err)
+			h.sendError(w, http.StatusInternalServerError, "create_error", "Error creating records", err)
+			return
+		}
+
+		h.sendResponse(w, map[string]interface{}{"created": dataValue.Len()}, nil)
+		return
+	}
+
+	// Single record creation
+	modelValue := reflect.New(reflect.TypeOf(model).Elem()).Interface()
+	jsonData, err := json.Marshal(data)
+	if err != nil {
+		logger.Error("Error marshaling data: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_data", "Invalid data format", err)
+		return
+	}
+	if err := json.Unmarshal(jsonData, modelValue); err != nil {
+		logger.Error("Error unmarshaling data: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_data", "Invalid data format", err)
+		return
+	}
+
+	query := h.db.NewInsert().Model(modelValue).Table(tableName)
+	if _, err := query.Exec(ctx); err != nil {
+		logger.Error("Error creating record: %v", err)
+		h.sendError(w, http.StatusInternalServerError, "create_error", "Error creating record", err)
+		return
+	}
+
+	h.sendResponse(w, modelValue, nil)
+}
+
+func (h *Handler) handleUpdate(ctx context.Context, w common.ResponseWriter, schema, entity, id string, idPtr *int64, data interface{}, options ExtendedRequestOptions) {
+	logger.Info("Updating record in %s.%s", schema, entity)
+
+	model, err := h.registry.GetModelByEntity(schema, entity)
+	if err != nil {
+		logger.Error("Invalid entity: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_entity", "Invalid entity", err)
+		return
+	}
+
+	tableName := h.getTableName(schema, entity, model)
+
+	// Convert data to map
+	dataMap, ok := data.(map[string]interface{})
+	if !ok {
+		jsonData, err := json.Marshal(data)
+		if err != nil {
+			logger.Error("Error marshaling data: %v", err)
+			h.sendError(w, http.StatusBadRequest, "invalid_data", "Invalid data format", err)
+			return
+		}
+		if err := json.Unmarshal(jsonData, &dataMap); err != nil {
+			logger.Error("Error unmarshaling data: %v", err)
+			h.sendError(w, http.StatusBadRequest, "invalid_data", "Invalid data format", err)
+			return
+		}
+	}
+
+	query := h.db.NewUpdate().Table(tableName).SetMap(dataMap)
+
+	// Apply ID filter
+	if id != "" {
+		query = query.Where("id = ?", id)
+	} else if idPtr != nil {
+		query = query.Where("id = ?", *idPtr)
+	} else {
+		h.sendError(w, http.StatusBadRequest, "missing_id", "ID is required for update", nil)
+		return
+	}
+
+	result, err := query.Exec(ctx)
+	if err != nil {
+		logger.Error("Error updating record: %v", err)
+		h.sendError(w, http.StatusInternalServerError, "update_error", "Error updating record", err)
+		return
+	}
+
+	h.sendResponse(w, map[string]interface{}{
+		"updated": result.RowsAffected(),
+	}, nil)
+}
+
+func (h *Handler) handleDelete(ctx context.Context, w common.ResponseWriter, schema, entity, id string) {
+	logger.Info("Deleting record from %s.%s", schema, entity)
+
+	model, err := h.registry.GetModelByEntity(schema, entity)
+	if err != nil {
+		logger.Error("Invalid entity: %v", err)
+		h.sendError(w, http.StatusBadRequest, "invalid_entity", "Invalid entity", err)
+		return
+	}
+
+	tableName := h.getTableName(schema, entity, model)
+
+	query := h.db.NewDelete().Table(tableName)
+
+	if id == "" {
+		h.sendError(w, http.StatusBadRequest, "missing_id", "ID is required for delete", nil)
+		return
+	}
+
+	query = query.Where("id = ?", id)
+
+	result, err := query.Exec(ctx)
+	if err != nil {
+		logger.Error("Error deleting record: %v", err)
+		h.sendError(w, http.StatusInternalServerError, "delete_error", "Error deleting record", err)
+		return
+	}
+
+	h.sendResponse(w, map[string]interface{}{
+		"deleted": result.RowsAffected(),
+	}, nil)
+}
+
+func (h *Handler) applyFilter(query common.SelectQuery, filter common.FilterOption) common.SelectQuery {
+	switch strings.ToLower(filter.Operator) {
+	case "eq", "equals":
+		return query.Where(fmt.Sprintf("%s = ?", filter.Column), filter.Value)
+	case "neq", "not_equals", "ne":
+		return query.Where(fmt.Sprintf("%s != ?", filter.Column), filter.Value)
+	case "gt", "greater_than":
+		return query.Where(fmt.Sprintf("%s > ?", filter.Column), filter.Value)
+	case "gte", "greater_than_equals", "ge":
+		return query.Where(fmt.Sprintf("%s >= ?", filter.Column), filter.Value)
+	case "lt", "less_than":
+		return query.Where(fmt.Sprintf("%s < ?", filter.Column), filter.Value)
+	case "lte", "less_than_equals", "le":
+		return query.Where(fmt.Sprintf("%s <= ?", filter.Column), filter.Value)
+	case "like":
+		return query.Where(fmt.Sprintf("%s LIKE ?", filter.Column), filter.Value)
+	case "ilike":
+		// Use ILIKE for case-insensitive search (PostgreSQL)
+		// For other databases, cast to citext or use LOWER()
+		return query.Where(fmt.Sprintf("CAST(%s AS TEXT) ILIKE ?", filter.Column), filter.Value)
+	case "in":
+		return query.Where(fmt.Sprintf("%s IN (?)", filter.Column), filter.Value)
+	case "between":
+		// Handle between operator - exclusive (> val1 AND < val2)
+		if values, ok := filter.Value.([]interface{}); ok && len(values) == 2 {
+			return query.Where(fmt.Sprintf("%s > ? AND %s < ?", filter.Column, filter.Column), values[0], values[1])
+		} else if values, ok := filter.Value.([]string); ok && len(values) == 2 {
+			return query.Where(fmt.Sprintf("%s > ? AND %s < ?", filter.Column, filter.Column), values[0], values[1])
+		}
+		logger.Warn("Invalid BETWEEN filter value format")
+		return query
+	case "between_inclusive":
+		// Handle between inclusive operator - inclusive (>= val1 AND <= val2)
+		if values, ok := filter.Value.([]interface{}); ok && len(values) == 2 {
+			return query.Where(fmt.Sprintf("%s >= ? AND %s <= ?", filter.Column, filter.Column), values[0], values[1])
+		} else if values, ok := filter.Value.([]string); ok && len(values) == 2 {
+			return query.Where(fmt.Sprintf("%s >= ? AND %s <= ?", filter.Column, filter.Column), values[0], values[1])
+		}
+		logger.Warn("Invalid BETWEEN INCLUSIVE filter value format")
+		return query
+	case "is_null", "isnull":
+		// Check for NULL values
+		return query.Where(fmt.Sprintf("(%s IS NULL OR %s = '')", filter.Column, filter.Column))
+	case "is_not_null", "isnotnull":
+		// Check for NOT NULL values
+		return query.Where(fmt.Sprintf("(%s IS NOT NULL AND %s != '')", filter.Column, filter.Column))
+	default:
+		logger.Warn("Unknown filter operator: %s, defaulting to equals", filter.Operator)
+		return query.Where(fmt.Sprintf("%s = ?", filter.Column), filter.Value)
+	}
+}
+
+func (h *Handler) getTableName(schema, entity string, model interface{}) string {
+	// Check if model implements TableNameProvider
+	if provider, ok := model.(common.TableNameProvider); ok {
+		tableName := provider.TableName()
+		if tableName != "" {
+			return tableName
+		}
+	}
+
+	// Default to schema.entity
+	if schema != "" {
+		return fmt.Sprintf("%s.%s", schema, entity)
+	}
+	return entity
+}
+
+func (h *Handler) generateMetadata(schema, entity string, model interface{}) *common.TableMetadata {
+	modelType := reflect.TypeOf(model)
+	if modelType.Kind() == reflect.Ptr {
+		modelType = modelType.Elem()
+	}
+
+	tableName := h.getTableName(schema, entity, model)
+
+	metadata := &common.TableMetadata{
+		Schema:  schema,
+		Table:   tableName,
+		Columns: []common.Column{},
+	}
+
+	for i := 0; i < modelType.NumField(); i++ {
+		field := modelType.Field(i)
+
+		// Get column name from gorm tag or json tag
+		columnName := field.Tag.Get("gorm")
+		if strings.Contains(columnName, "column:") {
+			parts := strings.Split(columnName, ";")
+			for _, part := range parts {
+				if strings.HasPrefix(part, "column:") {
+					columnName = strings.TrimPrefix(part, "column:")
+					break
+				}
+			}
+		} else {
+			columnName = field.Tag.Get("json")
+			if columnName == "" || columnName == "-" {
+				columnName = strings.ToLower(field.Name)
+			}
+		}
+
+		// Check for primary key and unique constraint
+		gormTag := field.Tag.Get("gorm")
+
+		column := common.Column{
+			Name:       columnName,
+			Type:       h.getColumnType(field.Type),
+			IsNullable: h.isNullable(field),
+			IsPrimary:  strings.Contains(gormTag, "primaryKey") || strings.Contains(gormTag, "primary_key"),
+			IsUnique:   strings.Contains(gormTag, "unique"),
+			HasIndex:   strings.Contains(gormTag, "index"),
+		}
+
+		metadata.Columns = append(metadata.Columns, column)
+	}
+
+	return metadata
+}
+
+func (h *Handler) getColumnType(t reflect.Type) string {
+	switch t.Kind() {
+	case reflect.String:
+		return "string"
+	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
+		return "integer"
+	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
+		return "integer"
+	case reflect.Float32, reflect.Float64:
+		return "float"
+	case reflect.Bool:
+		return "boolean"
+	case reflect.Ptr:
+		return h.getColumnType(t.Elem())
+	default:
+		return "unknown"
+	}
+}
+
+func (h *Handler) isNullable(field reflect.StructField) bool {
+	return field.Type.Kind() == reflect.Ptr
+}
+
+func (h *Handler) sendResponse(w common.ResponseWriter, data interface{}, metadata *common.Metadata) {
+	response := common.Response{
+		Success:  true,
+		Data:     data,
+		Metadata: metadata,
+	}
+	w.WriteHeader(http.StatusOK)
+	w.WriteJSON(response)
+}
+
+// sendFormattedResponse sends response with formatting options
+func (h *Handler) sendFormattedResponse(w common.ResponseWriter, data interface{}, metadata *common.Metadata, options ExtendedRequestOptions) {
+	// Clean JSON if requested (remove null/empty fields)
+	if options.CleanJSON {
+		data = h.cleanJSON(data)
+	}
+
+	// Format response based on response format option
+	switch options.ResponseFormat {
+	case "simple":
+		// Simple format: just return the data array
+		w.WriteHeader(http.StatusOK)
+		w.WriteJSON(data)
+	case "syncfusion":
+		// Syncfusion format: { result: data, count: total }
+		response := map[string]interface{}{
+			"result": data,
+		}
+		if metadata != nil {
+			response["count"] = metadata.Total
+		}
+		w.WriteHeader(http.StatusOK)
+		w.WriteJSON(response)
+	default:
+		// Default/detail format: standard response with metadata
+		response := common.Response{
+			Success:  true,
+			Data:     data,
+			Metadata: metadata,
+		}
+		w.WriteHeader(http.StatusOK)
+		w.WriteJSON(response)
+	}
+}
+
+// cleanJSON removes null and empty fields from the response
+func (h *Handler) cleanJSON(data interface{}) interface{} {
+	// This is a simplified implementation
+	// A full implementation would recursively clean nested structures
+	// For now, we'll return the data as-is
+	// TODO: Implement recursive cleaning
+	return data
+}
+
+func (h *Handler) sendError(w common.ResponseWriter, statusCode int, code, message string, err error) {
+	var details string
+	if err != nil {
+		details = err.Error()
+	}
+
+	response := common.Response{
+		Success: false,
+		Error: &common.APIError{
+			Code:    code,
+			Message: message,
+			Details: details,
+		},
+	}
+	w.WriteHeader(statusCode)
+	w.WriteJSON(response)
+}