From ebbc8cd85980f6a92b31a7cde9c5f4fd6633e6e8 Mon Sep 17 00:00:00 2001
From: Augustin Husson <augustin.husson@amadeus.com>
Date: Mon, 6 Jan 2025 16:35:41 +0100
Subject: [PATCH] add missing schema for prometheus timeseries query

Signed-off-by: Augustin Husson <augustin.husson@amadeus.com>
---
 .../migrate/migrate.cue                       | 59 +++++++++++++++++++
 .../prometheus-time-series-query/query.cue    | 30 ++++++++++
 2 files changed, 89 insertions(+)
 create mode 100644 Prometheus/schemas/prometheus-time-series-query/migrate/migrate.cue
 create mode 100644 Prometheus/schemas/prometheus-time-series-query/query.cue

diff --git a/Prometheus/schemas/prometheus-time-series-query/migrate/migrate.cue b/Prometheus/schemas/prometheus-time-series-query/migrate/migrate.cue
new file mode 100644
index 0000000..3f267a2
--- /dev/null
+++ b/Prometheus/schemas/prometheus-time-series-query/migrate/migrate.cue
@@ -0,0 +1,59 @@
+// Copyright 2025 The Perses Authors
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package migrate
+
+#target: _
+
+// NB we would need `if` to support short-circuit in order to avoid code duplication here.
+//    See https://github.com/cue-lang/cue/issues/2232
+if (*#target.datasource.type | null) == "prometheus" && #target.expr != _|_ {
+	kind: "PrometheusTimeSeriesQuery"
+	spec: {
+		datasource: {
+			kind: "PrometheusDatasource"
+			name: #target.datasource.uid
+		}
+		query:         #target.expr
+		#legendFormat: *#target.legendFormat | "__auto"
+		if #legendFormat != "__auto" {
+			seriesNameFormat: #legendFormat
+		}
+		if #target.interval != _|_ {
+			minStep: #target.interval
+		}
+	}
+},
+
+// The datasource.type may not be present while we are dealing with a prometheus query.
+// In such case, rely on the "expr" field, whose presence likely indicates that this is a prometheus query.
+// /!\ This is a best-effort conversion logic and may wrongly convert not-prometheus queries to PrometheusTimeSeriesQuery
+if #target.expr != _|_ {
+	kind: "PrometheusTimeSeriesQuery"
+	spec: {
+		if #target.datasource != _|_ {
+			datasource: {
+				kind: "PrometheusDatasource"
+				name: #target.datasource.uid
+			}
+		}
+		query:         #target.expr
+		#legendFormat: *#target.legendFormat | "__auto"
+		if #legendFormat != "__auto" {
+			seriesNameFormat: #legendFormat
+		}
+		if #target.interval != _|_ {
+			minStep: #target.interval
+		}
+	}
+},
diff --git a/Prometheus/schemas/prometheus-time-series-query/query.cue b/Prometheus/schemas/prometheus-time-series-query/query.cue
new file mode 100644
index 0000000..0d23ecb
--- /dev/null
+++ b/Prometheus/schemas/prometheus-time-series-query/query.cue
@@ -0,0 +1,30 @@
+// Copyright 2025 The Perses Authors
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package model
+
+import (
+	"strings"
+	prometheus "github.com/perses/perses/cue/schemas/datasources/prometheus:model"
+)
+
+kind: "PrometheusTimeSeriesQuery"
+spec: close({
+	datasource?: {
+		kind: prometheus.kind
+	}
+	query:             strings.MinRunes(1)
+	seriesNameFormat?: string
+	minStep?:          =~"^(?:(\\d+)y)?(?:(\\d+)w)?(?:(\\d+)d)?(?:(\\d+)h)?(?:(\\d+)m)?(?:(\\d+)s)?(?:(\\d+)ms)?(?:\\$\\w+)?$"
+	resolution?:       number
+})