🔖 版本更新

src/main/java/im/zhaojun/common/controller/InstallController.java

@@ -0,0 +1,103 @@
+package im.zhaojun.common.controller;
+
+import im.zhaojun.common.config.WebMvcConfig;
+import im.zhaojun.common.model.StorageConfig;
+import im.zhaojun.common.model.SystemConfigDTO;
+import im.zhaojun.common.model.dto.InstallModelDTO;
+import im.zhaojun.common.model.dto.ResultBean;
+import im.zhaojun.common.model.enums.StorageTypeEnum;
+import im.zhaojun.common.service.FileService;
+import im.zhaojun.common.service.StorageConfigService;
+import im.zhaojun.common.service.SystemConfigService;
+import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
+import org.springframework.stereotype.Controller;
+import org.springframework.util.AntPathMatcher;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.ResponseBody;
+import org.springframework.web.servlet.HandlerMapping;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletRequest;
+import java.io.File;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+/**
+ * 系统安装初始化
+ */
+@Controller
+public class InstallController {
+
+    @Resource
+    private SystemConfigService systemConfigService;
+
+    @Resource
+    private StorageConfigService storageConfigService;
+
+    @Resource
+    private AdminController adminController;
+
+    @GetMapping("/is-installed")
+    @ResponseBody
+    public ResultBean isInstall() {
+        if (systemConfigService.getCurrentStorageStrategy() == null) {
+            return ResultBean.success();
+        }
+        return ResultBean.error("请勿重复初始化");
+    }
+
+
+    @PostMapping("/install")
+    @ResponseBody
+    public ResultBean install(InstallModelDTO installModelDTO) {
+        SystemConfigDTO systemConfigDTO = systemConfigService.getSystemConfig();
+
+        if (systemConfigDTO.getStorageStrategy() != null) {
+            return ResultBean.error("请勿重复初始化.");
+        }
+
+        systemConfigDTO.setSiteName(installModelDTO.getSiteName());
+        StorageTypeEnum storageTypeEnum = installModelDTO.getStorageStrategy();
+        systemConfigDTO.setStorageStrategy(storageTypeEnum);
+        systemConfigDTO.setUsername(installModelDTO.getUsername());
+        systemConfigDTO.setPassword(new BCryptPasswordEncoder().encode(installModelDTO.getPassword()));
+        systemConfigService.updateSystemConfig(systemConfigDTO);
+
+        Map<String, String> storageStrategyConfig = installModelDTO.getStorageStrategyConfig();
+
+        List<StorageConfig> storageConfigList = storageConfigService.selectStorageConfigByType(storageTypeEnum);
+        for (StorageConfig storageConfig : storageConfigList) {
+            String key = storageConfig.getKey();
+            String value = storageStrategyConfig.get(key);
+            storageConfig.setValue(value);
+        }
+
+        storageConfigService.updateStorageConfig(storageConfigList);
+        adminController.refreshStorageStrategy();
+        return ResultBean.success();
+    }
+
+    @PostMapping("/storage-strategy")
+    @ResponseBody
+    public ResultBean save(@RequestParam Map<String, String> storageStrategyConfig, StorageTypeEnum storageStrategy) {
+        List<StorageConfig> storageConfigList = storageConfigService.selectStorageConfigByType(storageStrategy);
+        for (StorageConfig storageConfig : storageConfigList) {
+            String key = storageConfig.getKey();
+            String value = storageStrategyConfig.get(key);
+            storageConfig.setValue(value);
+        }
+        storageConfigService.updateStorageConfig(storageConfigList);
+
+        StorageTypeEnum currentStorageStrategy = systemConfigService.getCurrentStorageStrategy();
+        if (Objects.equals(storageStrategy, currentStorageStrategy)) {
+            FileService fileService = systemConfigService.getCurrentFileService();
+            fileService.clearCache();
+            fileService.init();
+        }
+
+        return ResultBean.success();
+    }
+}