From 44adfea363c4654129b288259d1928a83ecdc288 Mon Sep 17 00:00:00 2001
From: Fred Boniface <fred@fjla.uk>
Date: Wed, 24 Apr 2024 23:19:34 +0100
Subject: [PATCH] Add operator code to VSTP schedules

---
 vstp/convert.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/vstp/convert.go b/vstp/convert.go
index bb97acf..e07e58b 100644
--- a/vstp/convert.go
+++ b/vstp/convert.go
@@ -30,6 +30,11 @@ func convertCifType(in *upstreamApi.VSTPSchedule) (error, *upstreamApi.JsonSched
 			log.Warn("More than one element in schedule segment")
 		}
 		out.ScheduleSegment = convertSchedule(&in.ScheduleSegment[0])
+		if trim(in.ScheduleSegment[0].AtocCode) != "" {
+			out.AtocCode = in.ScheduleSegment[0].AtocCode
+		} else {
+			out.AtocCode = "UK"
+		}
 		return nil, out
 	} else {
 		log.Warn("VSTP Schedule Segment empty")