Merge pull request #145 from willdale/bugfix/stop-extraline-legends-from-duplicating

Ensure that extra line legends can't be added multiple times.

Author: willdale

Sources/SwiftUICharts/SharedLineAndBar/ViewModifiers/ExtraLine.swift

@@ -67,34 +67,55 @@ internal struct ExtraLine<T>: ViewModifier where T: CTLineBarChartDataProtocol {
         if self.chartData.extraLineData.style.lineColour.colourType == .colour,
            let colour = self.chartData.extraLineData.style.lineColour.colour
         {
-            chartData.legends.append(LegendData(id: self.chartData.extraLineData.id,
-                                                legend: self.chartData.extraLineData.legendTitle,
-                                                colour: ColourStyle(colour: colour),
-                                                strokeStyle: self.chartData.extraLineData.style.strokeStyle,
-                                                prioity: 3,
-                                                chartType: .line))
+            if !chartData.legends.contains(where: {
+                $0.legend == chartData.extraLineData.legendTitle &&
+                $0.colour == ColourStyle(colour: colour) &&
+                $0.prioity == 3 &&
+                $0.chartType == .line
+            }) {
+                chartData.legends.append(LegendData(id: self.chartData.extraLineData.id,
+                                                    legend: self.chartData.extraLineData.legendTitle,
+                                                    colour: ColourStyle(colour: colour),
+                                                    strokeStyle: self.chartData.extraLineData.style.strokeStyle,
+                                                    prioity: 3,
+                                                    chartType: .line))
+            }
         } else if self.chartData.extraLineData.style.lineColour.colourType == .gradientColour,
                   let colours = self.chartData.extraLineData.style.lineColour.colours
         {
-            chartData.legends.append(LegendData(id: self.chartData.extraLineData.id,
-                                                legend: self.chartData.extraLineData.legendTitle,
-                                                colour: ColourStyle(colours: colours,
-                                                                    startPoint: .leading,
-                                                                    endPoint: .trailing),
-                                                strokeStyle: self.chartData.extraLineData.style.strokeStyle,
-                                                prioity: 3,
-                                                chartType: .line))
+            if !chartData.legends.contains(where: {
+                $0.legend == chartData.extraLineData.legendTitle &&
+                $0.colour == ColourStyle(colours: colours, startPoint: .leading, endPoint: .trailing) &&
+                $0.prioity == 3 &&
+                $0.chartType == .line
+            }) {
+                chartData.legends.append(LegendData(id: self.chartData.extraLineData.id,
+                                                    legend: self.chartData.extraLineData.legendTitle,
+                                                    colour: ColourStyle(colours: colours,
+                                                                        startPoint: .leading,
+                                                                        endPoint: .trailing),
+                                                    strokeStyle: self.chartData.extraLineData.style.strokeStyle,
+                                                    prioity: 3,
+                                                    chartType: .line))
+            }
         } else if self.chartData.extraLineData.style.lineColour.colourType == .gradientStops,
                   let stops = self.chartData.extraLineData.style.lineColour.stops
         {
-            chartData.legends.append(LegendData(id: self.chartData.extraLineData.id,
-                                                legend: self.chartData.extraLineData.legendTitle,
-                                                colour: ColourStyle(stops: stops,
-                                                                    startPoint: .leading,
-                                                                    endPoint: .trailing),
-                                                strokeStyle: self.chartData.extraLineData.style.strokeStyle,
-                                                prioity: 3,
-                                                chartType: .line))
+            if !chartData.legends.contains(where: {
+                $0.legend == chartData.extraLineData.legendTitle &&
+                $0.colour == ColourStyle(stops: stops, startPoint: .leading, endPoint: .trailing) &&
+                $0.prioity == 3 &&
+                $0.chartType == .line
+            }) {
+                chartData.legends.append(LegendData(id: self.chartData.extraLineData.id,
+                                                    legend: self.chartData.extraLineData.legendTitle,
+                                                    colour: ColourStyle(stops: stops,
+                                                                        startPoint: .leading,
+                                                                        endPoint: .trailing),
+                                                    strokeStyle: self.chartData.extraLineData.style.strokeStyle,
+                                                    prioity: 3,
+                                                    chartType: .line))
+            }
         }
     }
 }