package com.example.webfluxreactivedemo1.controller;

import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * HTTP请求头IfMatch和ETag处理
 *
 * @author 单红宇
 * @date 2023/11/2 10:09
 */

@RestController
@RequestMapping
public class IfMatchController {

    /**
     * 获取资源接口
     *
     * @param id         资源ID
     * @param clientETag 请求头中的 If-None-Match
     * @return ResponseEntity
     */
    @GetMapping("/resource/{id}")
    public ResponseEntity<String> getResource(@PathVariable String id,
                                              @RequestHeader("If-None-Match") String clientETag) {
        // 检查资源是否存在以及资源最新的ETag是否与请求头中的If-None-Match匹配
        boolean resourceExists = checkResourceExists(id);
        boolean etagMatch = checkETagMatch(id, clientETag);

        if (!resourceExists) {
            // 如果资源不存在，返回404 Not Found
            return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
        } else if (!etagMatch) {
            // 如果资源存在且ETag不匹配（即资源已经发生了变更），则返回资源内容
            return ResponseEntity.ok().header(HttpHeaders.ETAG, this.generateETag(id)).body("Resource content");
        } else {
            // 如果资源存在且ETag匹配（即资源没有发生变更），返回304和空响应体
            return ResponseEntity.status(HttpStatus.NOT_MODIFIED).build();
        }
    }

    /**
     * 修改文章内容
     *
     * @param id          文章ID
     * @param clientETag 请求头中的 If-Match
     * @return ResponseEntity
     */
    @GetMapping("/updateArticle/{id}")
    public ResponseEntity<String> updateArticle(@PathVariable String id,
                                              @RequestHeader("If-Match") String clientETag) {
        // 检查资源是否存在以及资源最新的ETag是否与请求头中的If-Match匹配
        boolean etagMatch = checkETagMatch(id, clientETag);

        if (etagMatch) {
            // 如果资源存在且ETag匹配，即文章没有被其他人修改过，执行更新操作
            String newETag = "返回文章最新的ETag";
            // articleService.update(id);
            return ResponseEntity.ok().header(HttpHeaders.ETAG, newETag).body("修改成功");
        } else {
            // 如果ETag不匹配，说明文章被其他人修改过，用户需要获取最新内容后再基于最新内容修改提交，防止多人同时修改文章内容出现覆盖问题
            // 返回412 Precondition Failed
            return ResponseEntity.status(HttpStatus.PRECONDITION_FAILED).build();
        }
    }


    // 如果资源存在但ETag不匹配，返回412 Precondition Failed

    /**
     * 检查资源是否存在
     *
     * @param id 资源ID
     * @return true=存在
     */
    private boolean checkResourceExists(String id) {
        // 在这里实现检查资源是否存在的逻辑
        return true;
    }

    /**
     * 检查资源ETag是否匹配，即判定资源的ETag是否发生了变动
     *
     * @param id         资源ID
     * @param clientETag 浏览器客户端传过来的ETag
     * @return 当资源已经被更新时返回false，资源未更新返回true
     */
    private boolean checkETagMatch(String id, String clientETag) {
        // 在这里实现检查资源ETag是否与请求头中的If-Match/If-None-Match匹配的逻辑
        return true;
    }

    /**
     * 生成一个ETag
     *
     * @param resourceId 资源ID
     * @return ETag
     */
    public String generateETag(String resourceId) {
        // 在这里实现检查资源是否存在的逻辑
        String eTag = "根据resourceId按照自己的逻辑生成etag，比如你可以使用md5";

        // 注意ETag必须使用双引号包起来返回，这是HTTP协议规范要求
        return "\"" + eTag + "\"";
    }
}