go-duck-cli 1.1.22 → 1.1.23

package/generators/security.js

@@ -342,6 +342,7 @@ func CORSMiddleware(cfg *config.Config) gin.HandlerFunc {
 		}
 		
 		c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
+		c.Writer.Header().Set("Access-Control-Expose-Headers", "X-Total-Count")
 		
 		headerString := ""
 		for i, h := range headers {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "go-duck-cli",
-  "version": "1.1.22",
+  "version": "1.1.23",
   "description": "The Ultimate Evolutionary Go Microservice Scaffolder.",
   "main": "index.js",
   "type": "module",

package/templates/go/controller.go.hbs

@@ -181,8 +181,6 @@ func (ctrl *{{capitalize name}}Controller) GetAll(c *gin.Context) {
 
 	db, _ := c.Get("tenantMongoDB")
 	tenantDB := db.(*mongo.Database)
-	var entities []models.{{capitalize name}}
-	opts := options.Find().SetSkip(int64(page * size)).SetLimit(int64(size))
 	
 	// PostgREST-lite Filter Translation (Simple eq check for now)
 	filter := bson.M{}
@@ -192,6 +190,11 @@ func (ctrl *{{capitalize name}}Controller) GetAll(c *gin.Context) {
 		}
 	}
 
+	totalCount, _ := tenantDB.Collection("{{toLowerCase name}}s").CountDocuments(ctx, filter)
+	c.Header("X-Total-Count", strconv.FormatInt(totalCount, 10))
+
+	var entities []models.{{capitalize name}}
+	opts := options.Find().SetSkip(int64(page * size)).SetLimit(int64(size))
 	cursor, err := tenantDB.Collection("{{toLowerCase name}}s").Find(ctx, filter, opts)
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
@@ -232,6 +235,13 @@ func (ctrl *{{capitalize name}}Controller) GetAll(c *gin.Context) {
 	}
 	{{/if}}
 
+	var totalCount int64
+	if err := tenantDB.WithContext(ctx).Model(&models.{{capitalize name}}{}).Count(&totalCount).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	c.Header("X-Total-Count", strconv.FormatInt(totalCount, 10))
+
 	var entities []models.{{capitalize name}}
 	if err := tenantDB.WithContext(ctx).Offset(page * size).Limit(size).Find(&entities).Error; err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})