package com.aigeo.article.controller;

import com.aigeo.article.entity.GeneratedArticle;
import com.aigeo.article.service.GeneratedArticleService;
import com.aigeo.common.Result;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * 生成的文章控制器
 */
@Slf4j
@RestController
@RequestMapping("/api/articles")
@Tag(name = "生成文章管理", description = "AI生成文章相关接口")
public class GeneratedArticleController {
    
    @Autowired
    private GeneratedArticleService generatedArticleService;
    
    @PostMapping
    @Operation(summary = "创建生成文章", description = "创建新的生成文章")
    public Result<GeneratedArticle> createArticle(@RequestBody GeneratedArticle article) {
        try {
            GeneratedArticle savedArticle = generatedArticleService.save(article);
            return Result.success("文章创建成功", savedArticle);
        } catch (Exception e) {
            log.error("创建生成文章失败", e);
            return Result.error("文章创建失败");
        }
    }
    
    @GetMapping("/{id}")
    @Operation(summary = "获取生成文章详情", description = "根据ID获取生成文章详情")
    public Result<GeneratedArticle> getArticleById(@PathVariable Integer id) {
        try {
            return generatedArticleService.findById(id)
                    .map(article -> Result.success("查询成功", article))
                    .orElse(Result.error("文章不存在"));
        } catch (Exception e) {
            log.error("获取生成文章详情失败, id: {}", id, e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping
    @Operation(summary = "获取生成文章列表", description = "获取所有生成文章列表")
    public Result<List<GeneratedArticle>> getAllArticles() {
        try {
            List<GeneratedArticle> articles = generatedArticleService.findAll();
            return Result.success("查询成功", articles);
        } catch (Exception e) {
            log.error("获取生成文章列表失败", e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping("/task/{taskId}")
    @Operation(summary = "根据任务ID获取生成文章列表", description = "根据任务ID获取生成文章列表")
    public Result<List<GeneratedArticle>> getArticlesByTaskId(@PathVariable Integer taskId) {
        try {
            List<GeneratedArticle> articles = generatedArticleService.findByTaskId(taskId);
            return Result.success("查询成功", articles);
        } catch (Exception e) {
            log.error("根据任务ID获取生成文章列表失败, taskId: {}", taskId, e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping("/company/{companyId}")
    @Operation(summary = "根据公司ID获取生成文章列表", description = "根据公司ID获取生成文章列表")
    public Result<List<GeneratedArticle>> getArticlesByCompanyId(@PathVariable Integer companyId) {
        try {
            List<GeneratedArticle> articles = generatedArticleService.findByCompanyId(companyId);
            return Result.success("查询成功", articles);
        } catch (Exception e) {
            log.error("根据公司ID获取生成文章列表失败, companyId: {}", companyId, e);
            return Result.error("查询失败");
        }
    }
    
    @GetMapping("/status/{status}")
    @Operation(summary = "根据状态获取生成文章列表", description = "根据状态获取生成文章列表")
    public Result<List<GeneratedArticle>> getArticlesByStatus(@PathVariable String status) {
        try {
            GeneratedArticle.ArticleStatus articleStatus = GeneratedArticle.ArticleStatus.fromCode(status);
            List<GeneratedArticle> articles = generatedArticleService.findByStatus(articleStatus);
            return Result.success("查询成功", articles);
        } catch (Exception e) {
            log.error("根据状态获取生成文章列表失败, status: {}", status, e);
            return Result.error("查询失败");
        }
    }
    
    @PutMapping("/{id}")
    @Operation(summary = "更新生成文章", description = "更新生成文章信息")
    public Result<GeneratedArticle> updateArticle(@PathVariable Integer id, @RequestBody GeneratedArticle article) {
        try {
            if (!generatedArticleService.findById(id).isPresent()) {
                return Result.error("文章不存在");
            }
            article.setId(id);
            GeneratedArticle updatedArticle = generatedArticleService.save(article);
            return Result.success("文章更新成功", updatedArticle);
        } catch (Exception e) {
            log.error("更新生成文章失败, id: {}", id, e);
            return Result.error("文章更新失败");
        }
    }
    
    @DeleteMapping("/{id}")
    @Operation(summary = "删除生成文章", description = "删除指定ID的生成文章")
    public Result<String> deleteArticle(@PathVariable Integer id) {
        try {
            if (!generatedArticleService.findById(id).isPresent()) {
                return Result.error("文章不存在");
            }
            generatedArticleService.deleteById(id);
            return Result.success("文章删除成功");
        } catch (Exception e) {
            log.error("删除生成文章失败, id: {}", id, e);
            return Result.error("文章删除失败");
        }
    }
}