From f06765412364026e062a4ce9a124955a3b298a37 Mon Sep 17 00:00:00 2001
From: pwh-pwh <plmoknpwh@163.com>
Date: Thu, 1 Aug 2024 15:58:41 +0800
Subject: [PATCH] =?UTF-8?q?=E9=80=82=E9=85=8D=E6=98=9F=E7=81=AB=E6=9C=80?=
 =?UTF-8?q?=E6=96=B0websocket=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config/spark.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/config/spark.go b/config/spark.go
index cc55041..159516d 100644
--- a/config/spark.go
+++ b/config/spark.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"os"
 	"regexp"
+	"strings"
 )
 
 const (
@@ -23,6 +24,9 @@ type SparkConfig struct {
 }
 
 func extractVersion(url string) string {
+	if strings.Contains(url, "pro-128k") {
+		return "pro-128k"
+	}
 	// 使用正则表达式匹配版本号
 	regex := regexp.MustCompile(`v(\d+)\.(\d+)`)
 	matches := regex.FindStringSubmatch(url)
@@ -43,16 +47,20 @@ func GetSparkConfig() (cfg *SparkConfig, err error) {
 	var sparkDomainVersion = ""
 
 	switch version {
+	case "pro-128k":
+		sparkDomainVersion = "pro-128k"
+	case "4.0":
+		sparkDomainVersion = "4.0Ultra"
 	case "3.5":
 		sparkDomainVersion = "generalv3.5"
 	case "3.1":
 		sparkDomainVersion = "generalv3"
 	case "2.1":
 		sparkDomainVersion = "generalv2"
-	case "1.5":
+	case "1.1":
 		sparkDomainVersion = "general"
 	default:
-		sparkDomainVersion = ""
+		sparkDomainVersion = "general"
 	}
 
 	cfg = &SparkConfig{