Add method priceLines in ISeriesAPI

tradingview · Nov 21, 2024 · 580ae8e · 580ae8e
1 parent 40c8cc5
commit 580ae8e
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 2 deletions.
diff --git a/src/api/iseries-api.ts b/src/api/iseries-api.ts
@@ -259,6 +259,11 @@ export interface ISeriesApi<
 	 */
 	removePriceLine(line: IPriceLine): void;
 
+	/**
+	 * Returns an array of price lines.
+	 */
+	priceLines(): IPriceLine[];
+
 	/**
 	 * Return current series type.
 	 *

diff --git a/src/api/series-api.ts b/src/api/series-api.ts
@@ -210,6 +210,10 @@ export class SeriesApi<
 		this._series.removePriceLine((line as PriceLine).priceLine());
 	}
 
+	public priceLines(): IPriceLine[] {
+		return this._series.priceLines();
+	}
+
 	public seriesType(): TSeriesType {
 		return this._series.seriesType();
 	}

diff --git a/src/model/series.ts b/src/model/series.ts
@@ -272,6 +272,10 @@ export class Series<T extends SeriesType> extends PriceDataSource implements IDe
 		this.model().updateSource(this);
 	}
 
+	public priceLines(): CustomPriceLine[] {
+		return this._customPriceLines.slice();
+	}
+
 	public seriesType(): T {
 		return this._seriesType;
 	}

diff --git a/tests/e2e/graphics/test-cases/series/price-lines-remove.js b/tests/e2e/graphics/test-cases/series/price-lines-remove.js
@@ -18,11 +18,12 @@ function runTestCase(container) {
 	const series = chart.addSeries(LightweightCharts.LineSeries);
 	series.setData(generateData());
 
-	const line1 = series.createPriceLine({ price: 10 });
-	const line2 = series.createPriceLine({ price: 20 });
+	series.createPriceLine({ price: 10 });
+	series.createPriceLine({ price: 20 });
 
 	return new Promise(resolve => {
 		setTimeout(() => {
+			const [line1, line2] = series.priceLines();
 			series.removePriceLine(line2);
 			series.removePriceLine(line1);
 			resolve();