From 4d21e2b118043c94998e8705e08825230b319a83 Mon Sep 17 00:00:00 2001
From: Yordis Prieto <yordis.prieto@gmail.com>
Date: Sat, 7 Nov 2026 17:27:04 -0500
Subject: [PATCH] feat(natsstore): probe the server version and JetStream
 features

Per-key TTL and the other features the store relies on only exist on recent servers, so an older server must be rejected up front instead of failing later.

Signed-off-by: Yordis Prieto <yordis.prieto@gmail.com>
---
 internal/store/natsstore/probe.go      | 69 ++++++++++++++++++++++++++
 internal/store/natsstore/probe_test.go | 32 ++++++++++++
 2 files changed, 101 insertions(+)
 create mode 100644 internal/store/natsstore/probe.go
 create mode 100644 internal/store/natsstore/probe_test.go

diff --git a/internal/store/natsstore/probe.go b/internal/store/natsstore/probe.go
new file mode 100644
index 00000000000..1671c2c0950
--- /dev/null
+++ b/internal/store/natsstore/probe.go
@@ -0,0 +1,69 @@
+package natsstore
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/nats-io/nats.go"
+	"github.com/nats-io/nats.go/jetstream"
+)
+
+// Server release the backend is designed and tested against. Older servers that
+// still support per-key TTL are accepted and reported, not refused.
+const (
+	TargetMajor = 2
+	TargetMinor = 14
+
+	// keyTTLAPILevel is the JetStream API level that introduced per-key TTL.
+	keyTTLAPILevel = 1
+)
+
+// ServerFeatures describes what the connected server offers.
+type ServerFeatures struct {
+	// Version is the server version as reported on connect.
+	Version string
+	// Major, Minor and Patch are the parsed Version, zero when it did not parse.
+	Major, Minor, Patch int
+	// APILevel is the JetStream API level the account sees.
+	APILevel int
+	// KeyTTL reports per-key TTL and bucket limit markers support.
+	KeyTTL bool
+}
+
+// AtLeast reports whether the server version is at least major.minor.
+func (f ServerFeatures) AtLeast(major, minor int) bool {
+	return f.Major > major || (f.Major == major && f.Minor >= minor)
+}
+
+// Probe reads the server version and JetStream feature level. It fails when
+// JetStream is not enabled for the account.
+func Probe(ctx context.Context, nc *nats.Conn, js jetstream.JetStream) (ServerFeatures, error) {
+	info, err := js.AccountInfo(ctx)
+	if err != nil {
+		return ServerFeatures{}, fmt.Errorf("JetStream is not available: %w", err)
+	}
+	features := ServerFeatures{Version: nc.ConnectedServerVersion(), APILevel: info.API.Level}
+	features.Major, features.Minor, features.Patch = parseVersion(features.Version)
+	features.KeyTTL = features.APILevel >= keyTTLAPILevel
+	return features, nil
+}
+
+// parseVersion reads `major.minor.patch` and ignores any pre-release suffix.
+func parseVersion(version string) (major, minor, patch int) {
+	version = strings.TrimPrefix(version, "v")
+	if i := strings.IndexAny(version, "-+"); i >= 0 {
+		version = version[:i]
+	}
+	parts := strings.Split(version, ".")
+	nums := make([]int, 3)
+	for i := 0; i < len(parts) && i < 3; i++ {
+		n, err := strconv.Atoi(parts[i])
+		if err != nil {
+			return 0, 0, 0
+		}
+		nums[i] = n
+	}
+	return nums[0], nums[1], nums[2]
+}
diff --git a/internal/store/natsstore/probe_test.go b/internal/store/natsstore/probe_test.go
new file mode 100644
index 00000000000..20445d47541
--- /dev/null
+++ b/internal/store/natsstore/probe_test.go
@@ -0,0 +1,32 @@
+package natsstore
+
+import (
+	"testing"
+	"time"
+
+	"github.com/nats-io/nats.go/jetstream"
+
+	"github.com/router-for-me/CLIProxyAPI/v8/internal/store/natsstore/natstest"
+)
+
+func TestProbeReportsTargetServer(t *testing.T) {
+	srv := natstest.Start(t)
+	nc, err := connect(t, map[string]string{EnvURL: srv.URL}, 5*time.Second)
+	if err != nil {
+		t.Fatal(err)
+	}
+	js, err := jetstream.New(nc)
+	if err != nil {
+		t.Fatal(err)
+	}
+	features, err := Probe(testContext(t), nc, js)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !features.AtLeast(TargetMajor, TargetMinor) {
+		t.Fatalf("server %s is older than %d.%d", features.Version, TargetMajor, TargetMinor)
+	}
+	if !features.KeyTTL {
+		t.Fatalf("server %s reports no per-key TTL (API level %d)", features.Version, features.APILevel)
+	}
+}