aiChat/internal/model/scenario.go

package model

import (
	"net/url"
	"regexp"
	"strings"
)

const (
	IndexPage = iota //b=0
	DetailPage
	WorkDesk //c=1   相当于c=iota
)

var (
	scenarioName = map[int]string{
		IndexPage:  "index",    //网站首页
		DetailPage: "detail",   //三级页
		WorkDesk:   "workDesk", //工作桌面
	}
	regExpDetail = regexp.MustCompile("^/article/(\\w+)/(.*).html")
)

// 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
}