fanqie-go/internal/handlers/content.go

package handlers

import (
	"fanqie/internal/config"
	"fanqie/internal/fqapi"
	"fmt"
	"net/http"
)

// HandleContent 内容处理
func HandleContent(w http.ResponseWriter, r *http.Request) {
	incrementTotal()
	incrementContent()
	w.Header().Set("Content-Type", "application/json; charset=utf-8")

	// 从查询参数获取item_id
	itemID := r.URL.Query().Get("item_id")
	
	// 获取URL参数（支持动态覆盖配置）
	textMode := r.URL.Query().Get("text_mode")     // 1=纯文本模式
	imageMode := r.URL.Query().Get("image_mode")   // 0=移除图片, 1=保留图片
	reviewParam := r.URL.Query().Get("review")     // 1=启用段评, 0=禁用段评

	result, err := fqapi.HandleGetContent(itemID, textMode, imageMode)
	if err != nil {
		fmt.Fprintf(w, `{"code": -1, "msg": "获取内容失败: %s"}`, err.Error())
		return
	}

	// 判断是否启用段评功能
	// 优先使用URL参数，如果没有则使用配置文件
	enableReview := false
	if reviewParam == "1" {
		enableReview = true
	} else if reviewParam == "0" {
		enableReview = false
	} else {
		// 没有URL参数时，使用配置文件
		cfg := config.GetConfig()
		if cfg != nil {
			enableReview = cfg.Fanqie.Content.EnableReview
		}
	}

	// 如果启用了段评功能，处理正文内容添加段评标记
	if enableReview {
		result = addReviewToContent(result, itemID, r)
	}

	fmt.Fprint(w, result)
}

// HandleMultiContent 批量内容处理
func HandleMultiContent(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")

	bookID := r.URL.Query().Get("book_id")
	itemIDs := r.URL.Query().Get("item_ids")
	textMode := r.URL.Query().Get("text_mode")
	imageMode := r.URL.Query().Get("image_mode")
	ignoreFailed := r.URL.Query().Get("ignore_failed")

	result, err := fqapi.HandleMultiContent(itemIDs, bookID, textMode, imageMode, ignoreFailed)
	if err != nil {
		fmt.Fprintf(w, `{"code": -1, "msg": "批量获取内容失败: %s"}`, err.Error())
		return
	}

	fmt.Fprint(w, result)
}

// HandleItemSummary 章节摘要处理
func HandleItemSummary(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")

	bookID := r.URL.Query().Get("book_id")
	itemIDs := r.URL.Query().Get("item_ids")

	result, err := fqapi.HandleItemSummary(itemIDs, bookID)
	if err != nil {
		fmt.Fprintf(w, `{"code": -1, "msg": "获取章节摘要失败: %s"}`, err.Error())
		return
	}

	fmt.Fprint(w, result)
}

// addReviewToContent 为正文内容添加段评标记
func addReviewToContent(jsonResult, itemID string, r *http.Request) string {
	// 解析 JSON 结果
	var resultData map[string]interface{}
	if err := fqapi.ParseJSON([]byte(jsonResult), &resultData); err != nil {
		return jsonResult
	}
	
	// 获取 data 字段
	data, ok := resultData["data"].(map[string]interface{})
	if !ok {
		return jsonResult
	}
	
	// 获取 content 字段
	content, ok := data["content"].(string)
	if !ok || content == "" {
		return jsonResult
	}
	
	// 从响应数据中提取 book_id (data.novel_data.book_id)
	bookID := ""
	if novelData, ok := data["novel_data"].(map[string]interface{}); ok {
		if bookIDVal, ok := novelData["book_id"].(string); ok {
			bookID = bookIDVal
		} else if bookIDVal, ok := novelData["book_id"].(float64); ok {
			bookID = fmt.Sprintf("%.0f", bookIDVal)
		}
	}
	
	// 获取基础 URL
	cfg := config.GetConfig()
	baseURL := ""
	if cfg != nil {
		host := cfg.Server.Host
		if host == "0.0.0.0" {
			host = "localhost"
		}
		port := cfg.Server.Port
		scheme := "http"
		
		// 检查是否来自外部请求
		if r.Host != "" {
			baseURL = scheme + "://" + r.Host
		} else {
			baseURL = scheme + "://" + host + ":" + port
		}
	}
	
	// 生成带段评标记的内容
	newContent, err := fqapi.GenerateParagraphsWithComments(itemID, bookID, content, baseURL)
	if err != nil {
		// 如果失败，返回原始内容
		return jsonResult
	}
	
	// 替换 content 字段
	data["content"] = newContent
	resultData["data"] = data
	
	// 重新序列化为 JSON
	newJSON, err := fqapi.ToJSON(resultData)
	if err != nil {
		return jsonResult
	}
	
	return newJSON
}