wip:三级也匹配修改

internal/model/scenario.go

@@ -22,40 +22,17 @@ var (
 	}
 )
 
-// GetScenario 获取场景
-func GetScenario(refer string) int {
-	u, err := url.Parse(refer)
-	if err == nil {
-		if strings.HasPrefix(u.Path, "/page_workDesktop") {
-			if strings.HasPrefix(u.Path, "/page_workDesktop/work-bench/page") {
-				uu, err := url.Parse(u.Query().Get("link"))
-				if err == nil && regExpDetail.MatchString(uu.Path) {
-					return DetailPage
-				}
-			}
-			return WorkDesk
-		}
-	}
-	return IndexPage
-}
-
 // GetScenarioAndInfoId 获取场景和信息id
 func GetScenarioAndInfoId(refer string) (scenario int, sid string) {
 	u, err := url.Parse(refer)
-	if err == nil {
+	if err == nil && u.Path != "" {
 		if strings.HasPrefix(u.Path, "/page_workDesktop") {
-			if strings.HasPrefix(u.Path, "/page_workDesktop/work-bench/page") {
-				uu, err := url.Parse(u.Query().Get("link"))
-				if err == nil && uu.Path != "" {
-					if arr := regExpDetail.FindStringSubmatch(uu.Path); len(arr) == 3 {
-						if idArr := encrypt.CommonDecodeArticle(arr[1], arr[2]); len(idArr) > 0 {
-							sid = idArr[0]
-						}
-						return DetailPage, sid
-					}
-				}
-			}
 			return WorkDesk, ""
+		} else if arr := regExpDetail.FindStringSubmatch(u.Path); len(arr) == 3 {
+			if idArr := encrypt.CommonDecodeArticle(arr[1], arr[2]); len(idArr) > 0 {
+				sid = idArr[0]
+			}
+			return DetailPage, sid
 		}
 	}
 	return IndexPage, ""