fix: find plan (#48)

Author: klxiaoniu

plan.go

@@ -31,30 +31,141 @@ func (s *Student) GetCultivatePlan() (string, error) {
 	if err != nil {
 		return "", err
 	}
+
+	// 获取初始页面状态
 	viewStateMap, err := s.getState(constants.CultivatePlanURL)
 	if err != nil {
 		return "", err
 	}
+
+	// 尝试精确匹配学院和专业代码
+	url, err := s.getCultivatePlanWithPreciseMatch(info, viewStateMap)
+	if err == nil {
+		return url, nil
+	}
+
+	// 如果精确匹配失败，使用fallback逻辑
+	return s.getCultivatePlanWithFallback(info, viewStateMap)
+}
+
+// 精确匹配学院和专业代码获取培养方案
+func (s *Student) getCultivatePlanWithPreciseMatch(info *StudentDetail, viewStateMap map[string]string) (string, error) {
+	// 获取学院选择页面
+	initialDoc, err := s.GetWithIdentifier(constants.CultivatePlanURL)
+	if err != nil {
+		return "", err
+	}
+
+	// 查找学院代码
+	collegeSelect := htmlquery.FindOne(initialDoc, `//select[@id="xymcdpl"]`)
+	if collegeSelect == nil {
+		return "", fmt.Errorf("college select not found")
+	}
+
+	collegeCode := ""
+	collegeOptions := htmlquery.Find(collegeSelect, ".//option")
+	for _, option := range collegeOptions {
+		optionText := htmlquery.InnerText(option)
+		optionValue := htmlquery.SelectAttr(option, "value")
+
+		// 直接匹配学院名称
+		if optionText == info.College {
+			collegeCode = optionValue
+			break
+		}
+
+		// 处理学院改名的情况
+		if (strings.Contains(optionText, "计算机与大数据") || strings.Contains(optionText, "数学与统计")) &&
+			strings.Contains(info.College, "数学与计算机") {
+			collegeCode = optionValue
+			break
+		}
+	}
+
+	if collegeCode == "" {
+		return "", fmt.Errorf("college code not found for %s", info.College)
+	}
+
+	viewStateGenerator := htmlquery.SelectAttr(htmlquery.FindOne(initialDoc, `//*[@id="__VIEWSTATEGENERATOR"]`), "value")
+
+	// 选择年级和学院后获取专业列表
+	majorListResp, err := s.PostWithIdentifier(constants.CultivatePlanURL, map[string]string{
+		"__VIEWSTATE":                         viewStateMap["VIEWSTATE"],
+		"__EVENTVALIDATION":                   viewStateMap["EVENTVALIDATION"],
+		"__EVENTTARGET":                       "ctl00$njdpl",
+		"__EVENTARGUMENT":                     "",
+		"__VIEWSTATEGENERATOR":                viewStateGenerator,
+		"ctl00$njdpl":                         info.Grade,  // 年级
+		"ctl00$xymcdpl":                       collegeCode, // 学院名称
+		"ctl00$dldpl":                         "<-全部->",    // 大类
+		"ctl00$zymcdpl":                       "<-全部->",    // 专业代码
+		"ctl00$zylbdpl":                       "本专业",       // 修读类别：本专业/辅修
+		"ctl00$ContentPlaceHolder1$DDL_syxw":  "<-全部->",    // 授予学位
+		"ctl00$ContentPlaceHolder1$BT_submit": "确定",
+	})
+	if err != nil {
+		return "", err
+	}
+
+	// 查找专业代码
+	majorSelect := htmlquery.FindOne(majorListResp, `//select[@id="zymcdpl"]`)
+	if majorSelect == nil {
+		return "", fmt.Errorf("major select not found")
+	}
+
+	majorCode := ""
+	majorOptions := htmlquery.Find(majorSelect, ".//option")
+	for _, option := range majorOptions {
+		optionText := htmlquery.InnerText(option)
+		if optionText == info.Major {
+			majorCode = htmlquery.SelectAttr(option, "value")
+			break
+		}
+	}
+
+	if majorCode == "" {
+		return "", fmt.Errorf("major code not found for %s", info.Major)
+	}
+
+	// 构造最终URL
+	finalURL := fmt.Sprintf("/pyfa/pyjh/pyfa_bzy.aspx?nj=%s&xyh=%s&zyh=%s&zylb=本专业&id=%s",
+		info.Grade, collegeCode, majorCode, s.Identifier)
+
+	return constants.JwchPrefix + finalURL, nil
+}
+
+// fallback逻辑：当精确匹配失败时使用
+func (s *Student) getCultivatePlanWithFallback(info *StudentDetail, viewStateMap map[string]string) (string, error) {
+	// 获取初始页面状态
+	initialDoc, err := s.GetWithIdentifier(constants.CultivatePlanURL)
+	if err != nil {
+		return "", err
+	}
+
+	viewStateGenerator := htmlquery.SelectAttr(htmlquery.FindOne(initialDoc, `//*[@id="__VIEWSTATEGENERATOR"]`), "value")
+
+	// 只选择年级，提交查询
 	res, err := s.PostWithIdentifier(constants.CultivatePlanURL,
 		map[string]string{
-			"__VIEWSTATE":       viewStateMap["VIEWSTATE"],
-			"__EVENTVALIDATION": viewStateMap["EVENTVALIDATION"],
-			"ctl00$njdpl":       info.Grade, // 年级
-			// "ctl00$xymcdpl":                       "010",// 学院名称
-			"ctl00$dldpl":                         "<-全部->", // 大类
-			"ctl00$zymcdpl":                       "<-全部->", // 专业代码(无法获取)
-			"ctl00$zylbdpl":                       "本专业",    // 本专业、辅修
+			"__VIEWSTATE":                         viewStateMap["VIEWSTATE"],
+			"__EVENTVALIDATION":                   viewStateMap["EVENTVALIDATION"],
+			"__EVENTTARGET":                       "",
+			"__EVENTARGUMENT":                     "",
+			"__VIEWSTATEGENERATOR":                viewStateGenerator,
+			"ctl00$njdpl":                         info.Grade,
+			"ctl00$dldpl":                         "<-全部->",
+			"ctl00$zymcdpl":                       "<-全部->",
+			"ctl00$zylbdpl":                       "本专业",
 			"ctl00$ContentPlaceHolder1$DDL_syxw":  "<-全部->",
 			"ctl00$ContentPlaceHolder1$BT_submit": "确定",
 		})
 	if err != nil {
 		return "", err
 	}
 	xpathExpr := fmt.Sprintf("//tr[td[matches(string(.), '^（.*?）%s$')]]/td/a[contains(@href, 'pyfa')]/@href", regexp.QuoteMeta(info.Major))
-
 	node := htmlquery.FindOne(res, xpathExpr)
 	if node == nil {
-		return "", fmt.Errorf("%s", "cultivate plan not found")
+		return "", fmt.Errorf("cultivate plan not found for major: %s", info.Major)
 	}
 
 	url := htmlquery.SelectAttr(node, "href")