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) {
	incrementStat("total", "content")
	w.Header().Set("Content-Type", "application/json; charset=utf-8")

	params, err := validateParams(r, "item_id")
	if err != nil {
		respondError(w, err.Error(), true)
		return
	}
	itemID := params["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 {
		respondError(w, "获取内容失败: "+err.Error(), false)
		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) {
	incrementStat("total", "content")
	w.Header().Set("Content-Type", "application/json; charset=utf-8")

	params, err := validateParams(r, "book_id", "item_ids")
	if err != nil {
		respondError(w, err.Error(), true)
		return
	}

	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(params["item_ids"], params["book_id"], textMode, imageMode, ignoreFailed)
	if err != nil {
		respondError(w, "批量获取内容失败: "+err.Error(), false)
		return
	}

	fmt.Fprint(w, result)
}

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

	params, err := validateParams(r, "book_id", "item_ids")
	if err != nil {
		respondError(w, err.Error(), true)
		return
	}

	result, err := fqapi.HandleItemSummary(params["item_ids"], params["book_id"])
	if err != nil {
		respondError(w, "获取章节摘要失败: "+err.Error(), false)
		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
}