init dataleak sample feature

Signed-off-by: Hadi <112569860+anotherhadi@users.noreply.github.com>

back/api/api.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/anotherhadi/eleakxir/backend/search"
+	"github.com/anotherhadi/eleakxir/backend/search/dataleak"
 	"github.com/anotherhadi/eleakxir/backend/server"
 	"github.com/gin-gonic/gin"
 )
@@ -116,6 +117,20 @@ func routes(s *server.Server, cache *map[string]*search.Result, searchQueue chan
 		}
 		c.JSON(http.StatusOK, r)
 	})
+
+	s.Router.GET("/dataleak/sample", func(c *gin.Context) {
+		path := c.Query("path")
+		if path == "" {
+			c.JSON(http.StatusBadRequest, gin.H{"Error": "path is required"})
+			return
+		}
+		sample, err := dataleak.GetDataleakSample(*s, path)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"Error": err.Error()})
+			return
+		}
+		c.JSON(http.StatusOK, gin.H{"Sample": sample})
+	})
 }
 
 func Init(s *server.Server) {

back/search/dataleak/dataleak.go

@@ -217,3 +217,49 @@ func getFromClause(s *server.Server) string {
 	}
 	return fmt.Sprintf("read_parquet([%s], union_by_name=true, filename=true)", strings.Join(parquets, ", "))
 }
+
+func GetDataleakSample(s server.Server, path string) ([][]string, error) {
+	rowsData := [][]string{}
+
+	query := fmt.Sprintf("SELECT * FROM read_parquet('%s') LIMIT 5", path)
+	rows, err := s.Duckdb.Query(query)
+	if err != nil {
+		return rowsData, err
+	}
+	defer rows.Close()
+
+	cols, err := rows.Columns()
+	if err != nil {
+		return rowsData, err
+	}
+
+	rowsData = append(rowsData, cols)
+
+	rawResult := make([][]byte, len(cols))
+	dest := make([]any, len(cols))
+	for i := range rawResult {
+		dest[i] = &rawResult[i]
+	}
+
+	for rows.Next() {
+		if err := rows.Scan(dest...); err != nil {
+			return rowsData, err
+		}
+
+		row := make([]string, len(cols))
+		for i := range cols {
+			if rawResult[i] == nil {
+				row[i] = ""
+			} else {
+				row[i] = string(rawResult[i])
+			}
+		}
+		rowsData = append(rowsData, row)
+	}
+
+	if err = rows.Err(); err != nil {
+		return rowsData, err
+	}
+
+	return rowsData, nil
+}