vy 0321

src/model/akm_init.go

@@ -62,15 +62,21 @@ func (akm *AkmClient) init() error {
 		return err
 	}
 	scripts := doc.Find("script")
-	sensorurl, ok := scripts.Last().Attr("src")
-	if ok {
-		akm.SensorUrl = fmt.Sprintf("%s://%s%s", uri.Scheme, uri.Host, sensorurl)
-		//reg, _ := regexp.Compile("</noscript><script type=\"text/javascript\"  src=\"(.+?)\"></s")
-		//matchs := reg.FindStringSubmatch(string(readBytes))
-		//
-		//if len(matchs) == 2 {
+	lastScript := scripts.Last()
+	for {
+		src, ok := scripts.Last().Attr("src")
+		types, ok2 := scripts.Last().Attr("type")
+
+		if ok == false {
+			lastScript = lastScript.Prev()
+		}
+		if ok2 && types == "text/javascript" {
+			akm.SensorUrl = fmt.Sprintf("%s://%s%s", uri.Scheme, uri.Host, src)
+			break
+		}
+	}
 	//	akm.SensorUrl = fmt.Sprintf("%s://%s%s", uri.Scheme, uri.Host, matchs[1])
-	} else {
+	if akm.SensorUrl == "" {
 		akm.logger.Errorln("提取不到sensorUrl失败")
 		return errors.New("提取不到sensorUrl 失败")
 	}