package com.xuqm.tenant.controller;

import com.xuqm.tenant.config.PrivateDeploymentProperties;
import com.xuqm.tenant.service.SystemUpdateService;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;

@RestController
@RequestMapping("/api/system")
public class SystemUpdateController {

    private final PrivateDeploymentProperties deployProps;
    private final SystemUpdateService updateService;

    public SystemUpdateController(PrivateDeploymentProperties deployProps,
                                  SystemUpdateService updateService) {
        this.deployProps = deployProps;
        this.updateService = updateService;
    }

    /**
     * 触发私有化部署一键升级，以流式文本返回进度日志。
     * 仅在 PRIVATE 模式下可用；需要 JWT 认证（租户账号即可）。
     */
    @PostMapping(value = "/update", produces = MediaType.TEXT_PLAIN_VALUE)
    public ResponseEntity<StreamingResponseBody> update() {
        if (!deployProps.isPrivate()) {
            return ResponseEntity.status(403)
                    .contentType(MediaType.TEXT_PLAIN)
                    .body(out -> out.write("此接口仅在私有化部署可用\n".getBytes()));
        }

        StreamingResponseBody body = outputStream -> {
            updateService.runUpdate(line -> {
                try {
                    outputStream.write((line + "\n").getBytes());
                    outputStream.flush();
                } catch (Exception ignored) {}
            });
        };

        return ResponseEntity.ok()
                .contentType(MediaType.TEXT_PLAIN)
                .body(body);
    }
}