First parse of an Excel file

1 files changed, 130 insertions, 0 deletions

diff --git a/cmd/dbasik-api/return.go b/cmd/dbasik-api/return.go
new file mode 100644
index 0000000..294c960
--- /dev/null
+++ b/cmd/dbasik-api/return.go
@@ -0,0 +1,130 @@
+package main
+
+import (
+	"fmt"
+	"github.com/tealeg/xlsx/v3"
+	"path/filepath"
+)
+
+type ReturnLine struct {
+	Sheet   string
+	CellRef string
+	Value   string
+}
+
+type Return struct {
+	Name        string
+	ReturnLines []ReturnLine
+}
+
+// NewReturnLine creates a new ReturnLine object
+func NewReturnLine(sheet, cellRef, value string) (*ReturnLine, error) {
+	if err := validateInputs(sheet, cellRef, value); err != nil {
+		return nil, err
+	}
+
+	if !validateSpreadsheetCell(cellRef) {
+		return nil, fmt.Errorf("cellRef must be A1 format")
+	}
+
+	return &ReturnLine{
+		Sheet:   sheet,
+		CellRef: cellRef,
+		Value:   value,
+	}, nil
+}
+
+func validateInputs(sheet, cellRef, value string) error {
+	if sheet == "" {
+		return fmt.Errorf("sheet parameter is required")
+	}
+	if cellRef == "" {
+		return fmt.Errorf("cellRef parameter is required")
+	}
+	if value == "" {
+		return fmt.Errorf("value parameter is required")
+	}
+	return nil
+}
+
+func NewReturn(name string, dm *Datamap, returnLines []ReturnLine) (*Return, error) {
+	if len(returnLines) == 0 {
+		return nil, fmt.Errorf("ReturnLines must contain at least one ReturnLine")
+	}
+
+	return &Return{
+		Name:        name,
+		ReturnLines: returnLines,
+	}, nil
+}
+
+//func cellVisitor(c *xlsx.Cell) error {
+//	value, err := c.FormattedValue()
+//	if err != nil {
+//		fmt.Println(err.Error())
+//	} else {
+//		fmt.Printf("Sheet: %s, Cell value: %s\n", c.Row.Sheet.Name, value)
+//	}
+//	return err
+//}
+
+//func rowVisitor(r *xlsx.Row) error {
+//	return r.ForEachCell(cellVisitor, xlsx.SkipEmptyCells)
+//}
+
+func ParseXLSX(filePath string, dm *Datamap) (*Return, error) {
+	// Use tealeg/xlsx to parse the Excel file
+	wb, err := xlsx.OpenFile(filePath)
+	if err != nil {
+		return nil, err
+	}
+
+	// Get the set of sheets from the Datamap
+	sheets := GetSheetsFromDM(*dm)
+
+	// Loop through all DatamapLines
+	returnLines := []ReturnLine{}
+	for _, dml := range dm.DMLs {
+		// Check if the sheet for this DatamapLine is in the set of sheets
+		if !contains(sheets, dml.Sheet) {
+			continue
+		}
+
+		sh, ok := wb.Sheet[dml.Sheet]
+		if !ok {
+			return nil, fmt.Errorf("sheet %s not found in Excel file", dml.Sheet)
+		}
+
+		col, row, err := xlsx.GetCoordsFromCellIDString(dml.CellRef)
+		if err != nil {
+			return nil, err
+		}
+		cell, err := sh.Cell(row, col)
+		if err != nil {
+			return nil, err
+		}
+		returnLines = append(returnLines, ReturnLine{
+			Sheet:   dml.Sheet,
+			CellRef: dml.CellRef,
+			Value:   cell.Value, // or cell.FormattedValue() if you need formatted values
+		})
+	}
+
+	// Here we create a new Return object with the name of the Excel file and the ReturnLines slice
+	// that we just populated
+	rtn, err := NewReturn(filepath.Base(filePath), dm, returnLines)
+	if err != nil {
+		return nil, err
+	}
+	return rtn, nil
+}
+
+// contains checks if a slice contains a given string
+func contains(slice []string, str string) bool {
+	for _, s := range slice {
+		if s == str {
+			return true
+		}
+	}
+	return false
+}