Refactor SOA generation

Signed-off-by: Bryan Frimin <bryan@getprobo.com>

pkg/soagen/header.go

@@ -0,0 +1,129 @@
+// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
+//
+// Permission to use, copy, modify, and/or distribute this software for any
+// purpose with or without fee is hereby granted, provided that the above
+// copyright notice and this permission notice appear in all copies.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+// PERFORMANCE OF THIS SOFTWARE.
+
+package soagen
+
+import (
+	"time"
+
+	"github.com/xuri/excelize/v2"
+)
+
+type HeaderCell struct {
+	Cell  string
+	Value string
+	Style string // "header" or "cell"
+}
+
+type MergedCell struct {
+	StartCell string
+	EndCell   string
+}
+
+type HeaderLayout struct {
+	InfoHeader []HeaderCell
+	InfoData   []HeaderCell
+	MainHeader []HeaderCell
+	SubHeader  []HeaderCell
+	Merges     []MergedCell
+}
+
+func getSOAHeaderLayout() HeaderLayout {
+	return HeaderLayout{
+		InfoHeader: []HeaderCell{
+			{"A2", "Version", "header"},
+			{"C2", "Date", "header"},
+			{"D2", "Comment", "header"},
+			{"E2", "Author", "header"},
+			{"F2", "Approver", "header"},
+		},
+		InfoData: []HeaderCell{
+			{"A3", "1.0", "cell"},
+			{"C3", time.Now().Format("01/02/2006"), "cell"},
+			{"D3", "Initial SoA", "cell"},
+			{"E3", "System Admin", "cell"},
+			{"F3", "Security Manager", "cell"},
+		},
+		MainHeader: []HeaderCell{
+			{"A6", "Control", "header"},
+			{"B6", "Control name", "header"},
+			{"C6", "Applicability", "header"},
+			{"D6", "Justification for exclusion", "header"},
+			{"E6", "Justification for inclusion", "header"},
+			{"F6", "Justification for inclusion", "header"},
+			{"G6", "Justification for inclusion", "header"},
+			{"H6", "Justification for inclusion", "header"},
+			{"I6", "List of security measure or policy", "header"},
+		},
+		SubHeader: []HeaderCell{
+			{"A7", "Control", "header"},
+			{"B7", "Control name", "header"},
+			{"C7", "Applicability", "header"},
+			{"D7", "Justification for exclusion", "header"},
+			{"E7", "Regulatory", "header"},
+			{"F7", "Contractual", "header"},
+			{"G7", "Best practice", "header"},
+			{"H7", "Risk assessment", "header"},
+			{"I7", "List of security measure or policy", "header"},
+		},
+		Merges: []MergedCell{
+			{"A6", "A7"},
+			{"B6", "B7"},
+			{"C6", "C7"},
+			{"D6", "D7"},
+			{"E6", "H6"},
+			{"I6", "I7"},
+		},
+	}
+}
+
+func applyHeaderLayout(f *excelize.File, sheetName string, layout HeaderLayout, headerStyle, cellStyle int) error {
+	if err := applyCells(f, sheetName, layout.InfoHeader, headerStyle, cellStyle); err != nil {
+		return err
+	}
+
+	if err := applyCells(f, sheetName, layout.InfoData, headerStyle, cellStyle); err != nil {
+		return err
+	}
+
+	if err := applyCells(f, sheetName, layout.MainHeader, headerStyle, cellStyle); err != nil {
+		return err
+	}
+
+	if err := applyCells(f, sheetName, layout.SubHeader, headerStyle, cellStyle); err != nil {
+		return err
+	}
+
+	for _, merge := range layout.Merges {
+		if err := f.MergeCell(sheetName, merge.StartCell, merge.EndCell); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func applyCells(f *excelize.File, sheetName string, cells []HeaderCell, headerStyle, cellStyle int) error {
+	for _, cell := range cells {
+		f.SetCellValue(sheetName, cell.Cell, cell.Value)
+
+		style := cellStyle
+		if cell.Style == "header" {
+			style = headerStyle
+		}
+
+		f.SetCellStyle(sheetName, cell.Cell, cell.Cell, style)
+	}
+	return nil
+}