From afd464369608a9fcd7ecc3f0e31ef52819c4b2dc Mon Sep 17 00:00:00 2001
From: windyboy <windyboy@gmail.com>
Date: Sat, 20 Jul 2024 00:46:31 +0800
Subject: [PATCH] feat: Add FPL message parsing functionality

This commit adds functionality to parse FPL messages in the `aftn_parser.go` file. The `Parse` method now correctly parses the FPL message type and extracts the relevant data fields. The `FPL` struct is introduced in the `domain` package to represent FPL messages. The parsing logic is implemented to populate the `FPL` struct with the parsed data. This change enables the application to handle FPL messages and process them accordingly.
---
 internal/config/config.go            | 10 ++++++++++
 internal/parsers/aftn_parser.go      | 16 ++++++++++++++++
 internal/parsers/aftn_parser_test.go | 28 ++++++++++++++++++++++++++++
 3 files changed, 54 insertions(+)

diff --git a/internal/config/config.go b/internal/config/config.go
index e78f8e9..80e04f1 100644
--- a/internal/config/config.go
+++ b/internal/config/config.go
@@ -49,6 +49,7 @@ type PatternConfig struct {
 const (
 	arrPatternString = `^\((?P<type>[A-Z]{3})-(?P<number>[A-Z0-9]+)-(?P<ssr>[A-Z0-9]+)-(?P<departure>[A-Z]{4})-(?P<arrival>[A-Z]{4})\)$`
 	depPatternString = `^\((?P<type>[A-Z]{3})-(?P<number>[A-Z0-9]+)-(?P<ssr>[A-Z0-9]+)-(?P<departure>[A-Z]{4})-(?P<departure_time>\d{4})-(?P<arrival>[A-Z]{4})\)$`
+	fplPatternString = `^\((?P<type>[A-Z]{3})\-(?P<number>[A-Z]+\d+)\-(?P<indicator>[A-Z]{2})(?:.*\s*)?\-(?P<aircraft>[A-Z]+\d+\/?[A-Z]?)\s*\-(?P<surve>.*)\s*\-(?P<departure>[A-Z]{4})(?P<departure_time>\d{4})\s*\-(?P<speed>[A-Z]+\d+)(?P<level>[A-Z0-9]+)\s(?P<route>.*)\s*\-(?P<destination>[A-Z]{4})(?P<estt>\d{4})\s(?P<alter>[A-Z]{4})\s*\-(?P<pbn>PBN\/[A-Z0-9]+)\s(?P<nav>NAV\/\w+)\sREG\/(?P<reg>[A-Z0-9]+)\sEET\/(?P<eet>\w{4}\d{4})\sSEL\/(?P<sel>\w+)\sPER\/(?P<performance>\w)\sRIF\/(?P<rif>\w+\s[A-Z0-9]+\s[A-Z]+)\s*RMK\/(?P<remark>.*)\)$`
 )
 
 // Initialize the bodyPatterns map
@@ -71,6 +72,15 @@ var bodyPatterns = map[string]BodyConfig{
 			},
 		},
 	},
+	"FPL": {
+		Patterns: []PatternConfig{
+			{
+				Pattern:    fplPatternString,
+				Comments:   "Pattern for FPL message",
+				Expression: regexp.MustCompile(fplPatternString),
+			},
+		},
+	},
 }
 
 func GetBodyPatterns() map[string]BodyConfig {
diff --git a/internal/parsers/aftn_parser.go b/internal/parsers/aftn_parser.go
index 14cfcd5..edff235 100644
--- a/internal/parsers/aftn_parser.go
+++ b/internal/parsers/aftn_parser.go
@@ -41,6 +41,22 @@ func (p AFTNParser) Parse(text string) (interface{}, error) {
 			DepartureTime:    data["departure_time"],
 			Destination:      data["arrival"],
 		}, nil
+	case "FPL":
+		return &domain.FPL{
+			Category:                data["type"],
+			FlightNumber:            data["number"],
+			FlightRulesAndType:      data["indicator"],
+			AircraftID:              data["aircraft"],
+			SSRModeAndCode:          data["surve"],
+			DepartureAirport:        data["departure"],
+			DepartureTime:           data["departure_time"],
+			CruisingSpeedAndLevel:   data["speed"] + data["level"],
+			Route:                   data["route"],
+			DestinationAndTotalTime: data["destination"] + data["estt"],
+			AlternateAirport:        data["alter"],
+			OtherInfo:               data["pbn"] + " " + data["nav"] + " " + "REG/" + data["reg"] + " " + "EET/" + data["eet"] + " " + "SEL/" + data["sel"] + " " + "PER/" + data["performance"] + " " + "RIF/" + data["rif"],
+			SupplementaryInfo:       "RMK/" + data["remark"],
+		}, nil
 	default:
 		return nil, fmt.Errorf("invalid message type")
 	}
diff --git a/internal/parsers/aftn_parser_test.go b/internal/parsers/aftn_parser_test.go
index f770a96..4a56571 100644
--- a/internal/parsers/aftn_parser_test.go
+++ b/internal/parsers/aftn_parser_test.go
@@ -37,6 +37,34 @@ var _ = Describe("AFTN Parser", func() {
 			Expect(depMessage.Destination).To(Equal("KLAX"))
 		})
 
+		It("should parse FPL messages correctly", func() {
+			message := `(FPL-CCA1532-IS
+-A332/H
+-SDE3FGHIJ4J5M1RWY/LB101
+-ZSSS2035
+-K0859S1040 PIAKS G330 PIMOL A539 BTO W82 DOGAR
+-ZBAA0153 ZBYN
+-PBN/A1B2B3B4B5D1L1 NAV/ABAS REG/B6513 EET/ZBPE0112 SEL/KMAL PER/C RIF/FRT N640 ZBYN RMK/TCAS EQUIPPED)`
+			parser := AFTNParser{}
+			parsedMessage, err := parser.Parse(message)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(parsedMessage).To(BeAssignableToTypeOf(&domain.FPL{}))
+			fplMessage := parsedMessage.(*domain.FPL)
+			Expect(fplMessage.FlightNumber).To(Equal("CCA1532"))
+			Expect(fplMessage.FlightRulesAndType).To(Equal("IS"))
+			Expect(fplMessage.AircraftID).To(Equal("A332/H"))
+			Expect(fplMessage.SSRModeAndCode).To(Equal("SDE3FGHIJ4J5M1RWY/LB101"))
+			Expect(fplMessage.DepartureAirport).To(Equal("ZSSS"))
+			Expect(fplMessage.DepartureTime).To(Equal("2035"))
+			Expect(fplMessage.CruisingSpeedAndLevel).To(Equal("K0859S1040"))
+			Expect(fplMessage.Route).To(Equal("PIAKS G330 PIMOL A539 BTO W82 DOGAR"))
+			Expect(fplMessage.DestinationAndTotalTime).To(Equal("ZBAA0153"))
+			Expect(fplMessage.AlternateAirport).To(Equal("ZBYN"))
+			Expect(fplMessage.OtherInfo).To(Equal("PBN/A1B2B3B4B5D1L1 NAV/ABAS REG/B6513 EET/ZBPE0112 SEL/KMAL PER/C RIF/FRT N640 ZBYN"))
+			Expect(fplMessage.SupplementaryInfo).To(Equal("RMK/TCAS EQUIPPED"))
+
+		})
+
 		It("should return an error for invalid message types", func() {
 			message := "(XYZ-AB123-SSR1234-KJFK-KLAX)"
 			parser := AFTNParser{}