wip: mating EPs with Operations

2 files changed, 36 insertions, 2 deletions

diff --git a/internal/models/engagement.go b/internal/models/engagement.go
new file mode 100644
index 0000000..db1c2f5
--- /dev/null
+++ b/internal/models/engagement.go
@@ -0,0 +1,36 @@
+package models
+
+import (
+	"database/sql"
+	"errors"
+	"time"
+)
+
+type EngagementStrategy struct {
+	ID        int
+	ValidFrom time.Time
+	ValidTo   time.Time
+	Operation Operation
+}
+
+type EngagementStrategyModel struct {
+	DB *sql.DB
+}
+
+func (m *EngagementStrategyModel) GetForOperation(id int) (EngagementStrategy, error) {
+	stmt := `SELECT id, valid_from, valid_to FROM engagement_strategies
+WHERE operation_id = ?`
+	row := m.DB.QueryRow(stmt, id)
+
+	var es EngagementStrategy
+
+	err := row.Scan(&es.ID, &es.ValidFrom, &es.ValidTo)
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return EngagementStrategy{}, ErrNoRecord
+		} else {
+			return EngagementStrategy{}, err
+		}
+	}
+	return es, nil
+}
diff --git a/internal/models/organisations.go b/internal/models/organisations.go
index b765f91..88f7bc9 100644
--- a/internal/models/organisations.go
+++ b/internal/models/organisations.go
@@ -82,6 +82,4 @@ func (m *OrganisationModel) Latest() ([]Organisation, error) {
 	}
 
 	return organisations, err
-
-	return nil, nil
 }