XuqmGroup-Server/tenant-service/src/main/java/com/xuqm/tenant/controller/AppController.java

package com.xuqm.tenant.controller;

import com.xuqm.common.model.ApiResponse;
import com.xuqm.common.exception.BusinessException;
import com.xuqm.tenant.dto.CreateAppRequest;
import com.xuqm.tenant.entity.AppEntity;
import com.xuqm.tenant.entity.FeatureServiceEntity;
import com.xuqm.tenant.entity.TenantEntity;
import com.xuqm.tenant.repository.TenantRepository;
import com.xuqm.tenant.service.AppService;
import com.xuqm.tenant.service.AppUserClient;
import com.xuqm.tenant.service.EmailService;
import com.xuqm.tenant.service.FeatureServiceManager;
import com.xuqm.tenant.service.LicenseFileCrypto;
import com.xuqm.tenant.service.OperationLogService;
import jakarta.validation.Valid;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.ContentDisposition;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.security.core.annotation.AuthenticationPrincipal;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;
import java.util.Map;

@RestController
@RequestMapping("/api/apps")
public class AppController {

    private final AppService appService;
    private final EmailService emailService;
    private final OperationLogService operationLogService;
    private final TenantRepository tenantRepository;
    private final FeatureServiceManager featureServiceManager;
    private final AppUserClient appUserClient;

    @Value("${license.public-base-url:https://auto.dev.xuqinmin.com/}")
    private String licensePublicBaseUrl;

    public AppController(AppService appService, EmailService emailService,
                         OperationLogService operationLogService,
                         TenantRepository tenantRepository,
                         FeatureServiceManager featureServiceManager,
                         AppUserClient appUserClient) {
        this.appService = appService;
        this.emailService = emailService;
        this.operationLogService = operationLogService;
        this.tenantRepository = tenantRepository;
        this.featureServiceManager = featureServiceManager;
        this.appUserClient = appUserClient;
    }

    @GetMapping
    public ResponseEntity<ApiResponse<List<AppEntity>>> list(@AuthenticationPrincipal String tenantId) {
        return ResponseEntity.ok(ApiResponse.success(appService.listByTenant(tenantId)));
    }

    @GetMapping("/{appKey}")
    public ResponseEntity<ApiResponse<AppEntity>> get(@PathVariable String appKey,
                                                       @AuthenticationPrincipal String tenantId) {
        return ResponseEntity.ok(ApiResponse.success(appService.getByAppKey(appKey, tenantId)));
    }

    @PostMapping
    public ResponseEntity<ApiResponse<AppEntity>> create(@Valid @RequestBody CreateAppRequest req,
                                                          @AuthenticationPrincipal String tenantId) {
        return ResponseEntity.ok(ApiResponse.success(appService.create(tenantId, req)));
    }

    @PutMapping("/{appKey}")
    public ResponseEntity<ApiResponse<AppEntity>> update(@PathVariable String appKey,
                                                          @Valid @RequestBody CreateAppRequest req,
                                                          @AuthenticationPrincipal String tenantId) {
        return ResponseEntity.ok(ApiResponse.success(appService.update(appKey, tenantId, req)));
    }

    @DeleteMapping("/{appKey}")
    public ResponseEntity<ApiResponse<Void>> delete(@PathVariable String appKey,
                                                     @AuthenticationPrincipal String tenantId) {
        appService.delete(appKey, tenantId);
        return ResponseEntity.ok(ApiResponse.ok());
    }

    /** Step 1: send email verification code for secret reveal or reset. */
    @PostMapping("/{appKey}/request-secret-verify")
    public ResponseEntity<ApiResponse<Void>> requestSecretVerify(
            @PathVariable String appKey,
            @RequestParam String purpose,
            @AuthenticationPrincipal String tenantId) {
        appService.getByAppKey(appKey, tenantId);
        TenantEntity tenant = tenantRepository.findById(tenantId)
                .orElseThrow(() -> new RuntimeException("Tenant not found"));
        emailService.sendVerificationCode(tenant.getEmail(), purpose);
        operationLogService.record(tenantId, "APP", "APP_SECRET", appKey, "REQUEST_SECRET_VERIFY", Map.of(
                "purpose", purpose
        ));
        return ResponseEntity.ok(ApiResponse.ok());
    }

    /** Step 2a: verify code and return the full appSecret. */
    @PostMapping("/{appKey}/reveal-secret")
    public ResponseEntity<ApiResponse<Map<String, String>>> revealSecret(
            @PathVariable String appKey,
            @RequestBody Map<String, String> body,
            @AuthenticationPrincipal String tenantId) {
        AppEntity app = appService.getByAppKey(appKey, tenantId);
        TenantEntity tenant = tenantRepository.findById(tenantId)
                .orElseThrow(() -> new RuntimeException("Tenant not found"));
        emailService.verify(tenant.getEmail(), body.get("code"), "REVEAL_SECRET");
        operationLogService.record(tenantId, "APP", "APP_SECRET", appKey, "REVEAL_APP_SECRET", Map.of(
                "appKey", app.getAppKey()
        ));
        return ResponseEntity.ok(ApiResponse.success(Map.of("appSecret", app.getAppSecret())));
    }

    /** Step 2b: verify code and regenerate appSecret (old one invalidated immediately). */
    @PostMapping("/{appKey}/reset-secret")
    public ResponseEntity<ApiResponse<Map<String, String>>> resetSecret(
            @PathVariable String appKey,
            @RequestBody Map<String, String> body,
            @AuthenticationPrincipal String tenantId) {
        AppEntity app = appService.getByAppKey(appKey, tenantId);
        TenantEntity tenant = tenantRepository.findById(tenantId)
                .orElseThrow(() -> new RuntimeException("Tenant not found"));
        emailService.verify(tenant.getEmail(), body.get("code"), "RESET_SECRET");
        String newSecret = appService.resetSecret(appKey, tenantId);
        return ResponseEntity.ok(ApiResponse.success(Map.of("appSecret", newSecret)));
    }

    /**
     * List users of an app. Queries IM accounts when IM is enabled, Push accounts otherwise.
     * The ?source=IM|PUSH param overrides auto-detection.
     */
    @GetMapping("/{appKey}/users")
    public ResponseEntity<ApiResponse<Map<String, Object>>> listUsers(
            @PathVariable String appKey,
            @RequestParam(required = false, defaultValue = "") String keyword,
            @RequestParam(defaultValue = "0") int page,
            @RequestParam(defaultValue = "20") int size,
            @RequestParam(required = false) String source,
            @AuthenticationPrincipal String tenantId) {
        appService.getByAppKey(appKey, tenantId);
        List<FeatureServiceEntity> services = featureServiceManager.listByApp(appKey);
        boolean imEnabled = services.stream().anyMatch(s ->
                s.getServiceType() == FeatureServiceEntity.ServiceType.IM && s.isEnabled());
        boolean pushEnabled = services.stream().anyMatch(s ->
                s.getServiceType() == FeatureServiceEntity.ServiceType.PUSH && s.isEnabled());
        String effectiveSource = (source != null && !source.isBlank())
                ? source.trim().toUpperCase()
                : (imEnabled ? "IM" : "PUSH");
        Map<String, Object> result = "PUSH".equals(effectiveSource)
                ? appUserClient.listPushUsers(appKey, keyword, page, size)
                : appUserClient.listImUsers(appKey, keyword, page, size);
        return ResponseEntity.ok(ApiResponse.success(result));
    }

    @GetMapping("/{appKey}/license-file")
    public ResponseEntity<byte[]> downloadLicenseFile(@PathVariable String appKey,
                                                       @AuthenticationPrincipal String tenantId) {
        AppEntity app = appService.getByAppKey(appKey, tenantId);
        boolean licenseEnabled = featureServiceManager.listByApp(appKey).stream()
                .anyMatch(service -> service.getServiceType() == FeatureServiceEntity.ServiceType.LICENSE && service.isEnabled());
        if (!licenseEnabled) {
            throw new BusinessException(400, "License 服务未开通");
        }
        Map<String, Object> payload = new java.util.LinkedHashMap<>();
        payload.put("appKey", app.getAppKey());
        payload.put("appName", app.getName());
        payload.put("packageName", app.getPackageName());
        payload.put("baseUrl", normalizeBaseUrl(licensePublicBaseUrl));
        payload.put("issuedAt", java.time.Instant.now().toString());
        String encrypted = LicenseFileCrypto.encrypt(new com.fasterxml.jackson.databind.ObjectMapper().valueToTree(payload).toString());
        String filename = sanitizeFileName(app.getName()) + ".xuqmlicense";
        return ResponseEntity.ok()
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .header(HttpHeaders.CONTENT_DISPOSITION, ContentDisposition.attachment().filename(filename, java.nio.charset.StandardCharsets.UTF_8).build().toString())
                .body(encrypted.getBytes(java.nio.charset.StandardCharsets.UTF_8));
    }

    private static String normalizeBaseUrl(String value) {
        String baseUrl = value == null || value.isBlank() ? "https://auto.dev.xuqinmin.com/" : value.trim();
        return baseUrl.endsWith("/") ? baseUrl : baseUrl + "/";
    }

    private static String sanitizeFileName(String value) {
        String name = value == null || value.isBlank() ? "license" : value.trim();
        return name.replaceAll("[\\\\/:*?\"<>|\\s]+", "_");
    }
}