DeepSeek AI Book Writer: 基于大型语言模型的WordPress图书写作插件技术详解

deepseek-ai-book-writer/
├── admin/                      # 管理界面相关文件
│   ├── css/                    # 样式文件
│   ├── js/                     # JavaScript交互脚本
│   ├── partials/               # 页面模板
│   └── class-*.php             # 管理类文件
├── includes/                   # 核心功能文件
│   ├── class-*.php             # 核心类文件
├── languages/                  # 国际化文件
└── deepseek-ai-book-writer.php # 插件主入口文件

public function generate_book_outline($book_title, $book_genre, $book_description, $target_audience) {
    $prompt = $this->create_outline_prompt($book_title, $book_genre, $book_description, $target_audience);
    
    $response = $this->call_deepseek_api($prompt);
    
    if (is_wp_error($response)) {
        return array(
            'success' => false,
            'message' => $response->get_error_message()
        );
    }
    
    // 处理响应，提取大纲
    $outline = $this->parse_outline_response($response);
    
    return array(
        'success' => true,
        'outline' => $outline
    );
}

// 开始批量生成
function startBatchGeneration() {
    batchGenerationInProgress = true;
    
    // 从配置中获取并发限制数量
    var concurrentLimit = parseInt(deepseek_book_writer_vars.concurrent_requests) || 2; 
    var activeRequests = 0;
    var completedCount = 0;
    
    // 递归函数处理队列
    function processQueue() {
        // 如果已取消，则停止处理
        if (batchGenerationCancelled) {
            finishBatchGeneration();
            return;
        }
        
        // 当队列为空且没有活动请求时，完成批量生成
        if (batchGenerationQueue.length === 0 && activeRequests === 0) {
            finishBatchGeneration();
            return;
        }
        
        // 同时发送多个请求，不超过并发限制
        while (batchGenerationQueue.length > 0 && activeRequests < concurrentLimit) {
            var chapterIndex = batchGenerationQueue.shift();
            activeRequests++;
            
            // 生成章节内容
            generateSingleChapterContent(chapterIndex).always(function() {
                activeRequests--;
                completedCount++;
                
                // 更新进度条
                var progressPercent = (completedCount / bookInfo.chapters.length) * 100;
                $('.progress-bar').width(progressPercent + '%');
                $('.completed-count').text(completedCount);
                
                // 继续处理队列
                processQueue();
            });
        }
    }
    
    // 开始处理队列
    processQueue();
}

private function call_deepseek_api($prompt) {
    if (empty($this->config['api_key'])) {
        return new WP_Error('api_key_missing', '未设置DeepSeek API密钥。请在设置中配置API密钥。');
    }
    
    $args = array(
        'headers' => array(
            'Authorization' => 'Bearer ' . $this->config['api_key'],
            'Content-Type' => 'application/json'
        ),
        'body' => json_encode(array(
            'model' => $this->config['model'],
            'messages' => array(
                array(
                    'role' => 'system',
                    'content' => '你是专业的内容创作者'
                ),
                array(
                    'role' => 'user',
                    'content' => $prompt
                )
            ),
        )),
        'timeout' => 120
    );
    
    $response = wp_remote_post($this->config['api_url'], $args);
    
    if (is_wp_error($response)) {
        return $response;
    }
    
    $body = json_decode(wp_remote_retrieve_body($response), true);
    
    if (!isset($body['choices'][0]['message']['content'])) {
        return new WP_Error('api_error', '从DeepSeek API接收到无效的响应: ' . wp_json_encode($body));
    }
    
    return $body['choices'][0]['message']['content'];
}

private function create_chapter_prompt($book_title, $chapter_title, $chapter_number, $previous_chapter_summary, $chapter_description) {
    $prompt = "请为以下图书章节创建详细内容。确保内容有吸引力，符合章节描述，并保持与之前章节的连贯性。\n\n";
    $prompt .= "图书标题: {$book_title}\n";
    $prompt .= "章节编号: {$chapter_number}\n";
    $prompt .= "章节标题: {$chapter_title}\n";
    $prompt .= "章节描述: {$chapter_description}\n";
    
    if (!empty($previous_chapter_summary)) {
        $prompt .= "\n前一章节摘要:\n{$previous_chapter_summary}\n";
    }
    
    $prompt .= "\n生成一个详细、生动且有吸引力的章节内容。包含对话、描述和情节发展。长度应在1500-2000字左右。";
    $prompt .= "\n直接输出章节内容，不需要包含标题。确保内容格式良好，可以直接用于HTML显示（包含段落等）。";
    
    return $prompt;
}

private function parse_outline_response($response) {
    // 尝试从响应中提取JSON
    $json_start = strpos($response, '[');
    $json_end = strrpos($response, ']');
    
    if ($json_start !== false && $json_end !== false) {
        $json = substr($response, $json_start, $json_end - $json_start + 1);
        $decoded = json_decode($json, true);
        
        if (is_array($decoded)) {
            return $decoded;
        }
    }
    
    // 如果无法提取JSON，则尝试用正则表达式解析
    $chapters = array();
    $regex = '/"title":\s*"([^"]+)"[^}]+"description":\s*"([^"]+)"/';
    
    if (preg_match_all($regex, $response, $matches, PREG_SET_ORDER)) {
        foreach ($matches as $match) {
            $chapters[] = array(
                'title' => $match[1],
                'description' => $match[2]
            );
        }
        return $chapters;
    }
    
    // 如果所有方法都失败，则返回一个章节的示例
    return array(
        array(
            'title' => '解析失败 - 请手动编辑',
            'description' => '无法从AI响应中解析章节。请手动创建您的章节或重试。'
        )
    );
}

public function add_plugin_admin_menu() {
    // 主菜单
    add_menu_page(
        'DeepSeek AI Book Writer', 
        'AI图书写作', 
        'manage_options', 
        $this->plugin_name, 
        array($this, 'display_plugin_setup_page'),
        'dashicons-book-alt'
    );
    
    // 设置子菜单
    add_submenu_page(
        $this->plugin_name,
        'DeepSeek AI Book Writer 设置',
        '设置',
        'manage_options',
        $this->plugin_name . '-settings',
        array($this, 'display_plugin_settings_page')
    );
    
    // 重命名默认子菜单
    global $submenu;
    if (isset($submenu[$this->plugin_name])) {
        $submenu[$this->plugin_name][0][0] = '创建图书';
    }
}

// 章节内容编辑更新预览
$('#chapter-content').on('input', function() {
    var content = $(this).val();
    $('#chapter-preview').html(formatChapterContent(content));
    
    // 保存当前内容到章节
    if (bookInfo.chapters.length > 0) {
        bookInfo.chapters[currentChapterIndex].content = content;
        updateChapterStatusIndicator(currentChapterIndex, 'completed');
    }
});

.chapter-item .status-indicator {
    display: inline-block;
    width: 10px;
    height: 10px;
    border-radius: 50%;
    margin-right: 5px;
    background-color: #ccc;
}

.chapter-item .status-pending {
    background-color: #ccc;
}

.chapter-item .status-generating {
    background-color: #ffcc00;
    animation: blink 1s infinite;
}

.chapter-item .status-completed {
    background-color: #4CAF50;
}

.chapter-item .status-error {
    background-color: #ff0000;
}

@keyframes blink {
    0% { opacity: 0.4; }
    50% { opacity: 1; }
    100% { opacity: 0.4; }
}

// 获取前一章节摘要（如果有的话）
var previousChapterSummary = '';
if (chapterIndex > 0 && bookInfo.chapters[chapterIndex - 1].content) {
    previousChapterSummary = truncateText(bookInfo.chapters[chapterIndex - 1].content, 300);
}