diff --git a/src/main/java/im/zhaojun/zfile/controller/home/FileController.java b/src/main/java/im/zhaojun/zfile/controller/home/FileController.java index 20c98df..78efb08 100644 --- a/src/main/java/im/zhaojun/zfile/controller/home/FileController.java +++ b/src/main/java/im/zhaojun/zfile/controller/home/FileController.java @@ -68,7 +68,10 @@ public class FileController { */ @GetMapping("/drive/list") public ResultBean drives() { - return ResultBean.success(driveConfigService.listOnlyEnable()); + List driveList = driveConfigService.listOnlyEnable(); + boolean isInstall = systemConfigService.getIsInstall(); + DriveListDTO driveListDTO = new DriveListDTO(driveList, isInstall); + return ResultBean.success(driveListDTO); } /** diff --git a/src/main/java/im/zhaojun/zfile/model/dto/DriveListDTO.java b/src/main/java/im/zhaojun/zfile/model/dto/DriveListDTO.java new file mode 100644 index 0000000..9bac8b5 --- /dev/null +++ b/src/main/java/im/zhaojun/zfile/model/dto/DriveListDTO.java @@ -0,0 +1,21 @@ +package im.zhaojun.zfile.model.dto; + +import im.zhaojun.zfile.model.entity.DriveConfig; +import lombok.AllArgsConstructor; +import lombok.Data; + +import java.util.List; + +/** + * @author Zhao Jun + * 2021/5/26 15:17 + */ +@Data +@AllArgsConstructor +public class DriveListDTO { + + private List driveList; + + private Boolean isInstall; + +} \ No newline at end of file diff --git a/src/main/java/im/zhaojun/zfile/service/SystemConfigService.java b/src/main/java/im/zhaojun/zfile/service/SystemConfigService.java index 09b5790..bbd0d65 100644 --- a/src/main/java/im/zhaojun/zfile/service/SystemConfigService.java +++ b/src/main/java/im/zhaojun/zfile/service/SystemConfigService.java @@ -1,6 +1,7 @@ package im.zhaojun.zfile.service; import cn.hutool.core.convert.Convert; +import cn.hutool.core.util.StrUtil; import cn.hutool.crypto.SecureUtil; import im.zhaojun.zfile.cache.ZFileCache; import im.zhaojun.zfile.exception.InvalidDriveException; @@ -173,4 +174,14 @@ public class SystemConfigService { return systemConfigDTO.getDomain(); } + + /** + * 获取是否已安装初始化 + * + * @return 是否已安装初始化 + */ + public boolean getIsInstall() { + SystemConfigDTO systemConfigDTO = getSystemConfig(); + return StrUtil.isNotEmpty(systemConfigDTO.getUsername()); + } } \ No newline at end of file