XuqmGroup-Server/update-service/src/main/java/com/xuqm/update/controller/AppVersionController.java

package com.xuqm.update.controller;

import com.xuqm.common.model.ApiResponse;
import com.xuqm.update.entity.AppVersionEntity;
import com.xuqm.update.repository.AppVersionRepository;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.UUID;
import com.xuqm.update.service.UpdateAssetService;

@RestController
@RequestMapping("/api/v1/updates")
public class AppVersionController {

    private final AppVersionRepository versionRepository;
    private final UpdateAssetService updateAssetService;

    public AppVersionController(AppVersionRepository versionRepository, UpdateAssetService updateAssetService) {
        this.versionRepository = versionRepository;
        this.updateAssetService = updateAssetService;
    }

    @GetMapping("/app/check")
    public ResponseEntity<ApiResponse<Map<String, Object>>> checkUpdate(
            @RequestParam String appId,
            @RequestParam AppVersionEntity.Platform platform,
            @RequestParam int currentVersionCode) {

        Optional<AppVersionEntity> latest = versionRepository
                .findTopByAppIdAndPlatformAndPublishStatusOrderByVersionCodeDesc(
                        appId, platform, AppVersionEntity.PublishStatus.PUBLISHED);

        if (latest.isEmpty() || latest.get().getVersionCode() <= currentVersionCode) {
            return ResponseEntity.ok(ApiResponse.success(Map.of("needsUpdate", false)));
        }

        AppVersionEntity v = latest.get();
        return ResponseEntity.ok(ApiResponse.success(Map.of(
                "needsUpdate", true,
                "versionName", v.getVersionName(),
                "versionCode", v.getVersionCode(),
                "downloadUrl", v.getDownloadUrl() != null ? v.getDownloadUrl() : "",
                "changeLog", v.getChangeLog() != null ? v.getChangeLog() : "",
                "forceUpdate", v.isForceUpdate(),
                "appStoreUrl", v.getAppStoreUrl() != null ? v.getAppStoreUrl() : "",
                "marketUrl", v.getMarketUrl() != null ? v.getMarketUrl() : ""
        )));
    }

    @PostMapping("/app/upload")
    public ResponseEntity<ApiResponse<AppVersionEntity>> upload(
            @RequestParam String appId,
            @RequestParam AppVersionEntity.Platform platform,
            @RequestParam String versionName,
            @RequestParam int versionCode,
            @RequestParam(required = false) String changeLog,
            @RequestParam(defaultValue = "false") boolean forceUpdate,
            @RequestParam(required = false) MultipartFile apkFile) throws Exception {

        AppVersionEntity entity = new AppVersionEntity();
        entity.setId(UUID.randomUUID().toString());
        entity.setAppId(appId);
        entity.setPlatform(platform);
        entity.setVersionName(versionName);
        entity.setVersionCode(versionCode);
        entity.setDownloadUrl(updateAssetService.storeAppPackage(apkFile));
        entity.setChangeLog(changeLog);
        entity.setForceUpdate(forceUpdate);
        entity.setPublishStatus(AppVersionEntity.PublishStatus.DRAFT);
        entity.setCreatedAt(LocalDateTime.now());
        return ResponseEntity.ok(ApiResponse.success(versionRepository.save(entity)));
    }

    @PostMapping("/app/{id}/publish")
    public ResponseEntity<ApiResponse<AppVersionEntity>> publish(@PathVariable String id) {
        AppVersionEntity entity = versionRepository.findById(id).orElseThrow();
        entity.setPublishStatus(AppVersionEntity.PublishStatus.PUBLISHED);
        entity.setGrayEnabled(false);
        entity.setGrayPercent(0);
        return ResponseEntity.ok(ApiResponse.success(versionRepository.save(entity)));
    }

    @PostMapping("/app/{id}/unpublish")
    public ResponseEntity<ApiResponse<AppVersionEntity>> unpublish(@PathVariable String id) {
        AppVersionEntity entity = versionRepository.findById(id).orElseThrow();
        entity.setPublishStatus(AppVersionEntity.PublishStatus.DEPRECATED);
        return ResponseEntity.ok(ApiResponse.success(versionRepository.save(entity)));
    }

    @PostMapping("/app/{id}/gray")
    public ResponseEntity<ApiResponse<AppVersionEntity>> gray(
            @PathVariable String id,
            @RequestBody Map<String, Object> body) {
        AppVersionEntity entity = versionRepository.findById(id).orElseThrow();
        entity.setGrayEnabled(Boolean.TRUE.equals(body.get("enabled")));
        entity.setGrayPercent(body.get("percent") instanceof Number n ? n.intValue() : 0);
        entity.setPublishStatus(AppVersionEntity.PublishStatus.PUBLISHED);
        return ResponseEntity.ok(ApiResponse.success(versionRepository.save(entity)));
    }

    @GetMapping("/app/list")
    public ResponseEntity<ApiResponse<List<AppVersionEntity>>> list(
            @RequestParam String appId, @RequestParam AppVersionEntity.Platform platform) {
        return ResponseEntity.ok(ApiResponse.success(
                versionRepository.findByAppIdAndPlatformOrderByVersionCodeDesc(appId, platform)));
    }
}