✨ OneDrive 支持

2025-04-19 05:34:52 +00:00 · 2020-01-19 21:58:02 +08:00
parent ad0ad12c08
commit dcc4cb19ad
13 changed files with 383 additions and 58 deletions
--- a/src/main/java/im/zhaojun/onedrive/service/OneDriveService.java
+++ b/src/main/java/im/zhaojun/onedrive/service/OneDriveService.java
@@ -0,0 +1,124 @@
+package im.zhaojun.onedrive.service;
+
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import cn.hutool.http.HttpUtil;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import im.zhaojun.common.model.StorageConfig;
+import im.zhaojun.common.model.constant.StorageConfigConstant;
+import im.zhaojun.common.model.dto.FileItemDTO;
+import im.zhaojun.common.model.enums.FileTypeEnum;
+import im.zhaojun.common.model.enums.StorageTypeEnum;
+import im.zhaojun.common.repository.StorageConfigRepository;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Service;
+import org.springframework.web.client.RestTemplate;
+
+import javax.annotation.Resource;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author zhaojun
+ * @date 2020/1/12 12:17
+ */
+@Service
+public class OneDriveService {
+
+    public static final String DRIVER_INFO_URL = "https://graph.microsoft.com/v1.0/drive";
+
+    public static final String DRIVER_ROOT_URL = "https://graph.microsoft.com/v1.0/drive/root/children";
+
+    public static final String DRIVER_ITEMS_URL = "https://graph.microsoft.com/v1.0/drive/root:{path}:/children";
+
+    public static final String AUTHENTICATE_URL = "https://login.microsoftonline.com/common/oauth2/v2.0/token";
+
+    @Value("${zfile.onedirve.clientId}")
+    private String clientId;
+
+    @Value("${zfile.onedirve.redirectUri}")
+    private String redirectUri;
+
+    @Value("${zfile.onedirve.clientSecret}")
+    private String clientSecret;
+
+    @Value("${zfile.onedirve.scope}")
+    private String scope;
+
+    @Resource
+    private RestTemplate oneDriveRestTemplate;
+
+    @Resource
+    private StorageConfigRepository storageConfigRepository;
+
+    public OneDriveToken getToken(String code) {
+        String param = "client_id=" + clientId +
+                "&redirect_uri=" + redirectUri +
+                "&client_secret=" + clientSecret +
+                "&code=" + code +
+                "&scope=" + scope +
+                "&grant_type=authorization_code";
+
+        HttpRequest post = HttpUtil.createPost(AUTHENTICATE_URL);
+
+        post.body(param, "application/x-www-form-urlencoded");
+        HttpResponse response = post.execute();
+        return JSONObject.parseObject(response.body(), OneDriveToken.class);
+    }
+
+    public OneDriveToken getRefreshToken() {
+        StorageConfig refreshStorageConfig =
+                storageConfigRepository.findByTypeAndKey(StorageTypeEnum.ONE_DRIVE, StorageConfigConstant.REFRESH_TOKEN_KEY);
+
+        String param = "client_id=" + clientId +
+                "&redirect_uri=" + redirectUri +
+                "&client_secret=" + clientSecret +
+                "&refresh_token=" + refreshStorageConfig.getValue() +
+                "&grant_type=refresh_token";
+
+        HttpRequest post = HttpUtil.createPost(AUTHENTICATE_URL);
+
+        post.body(param, "application/x-www-form-urlencoded");
+        HttpResponse response = post.execute();
+        return JSONObject.parseObject(response.body(), OneDriveToken.class);
+    }
+
+    public String getUserInfo() {
+        return oneDriveRestTemplate.getForObject(DRIVER_INFO_URL, String.class);
+    }
+
+    public List<FileItemDTO> list(String path) {
+        ResponseEntity<String> responseEntity = oneDriveRestTemplate.getForEntity("/".equalsIgnoreCase(path) ? DRIVER_ROOT_URL : DRIVER_ITEMS_URL, String.class, path);
+        String body = responseEntity.getBody();
+
+        JSONObject root = JSON.parseObject(body);
+
+        JSONArray fileList = root.getJSONArray("value");
+        List<FileItemDTO> result = new ArrayList<>();
+
+        for (int i = 0; i < fileList.size(); i++) {
+
+            FileItemDTO fileItemDTO = new FileItemDTO();
+            JSONObject fileItem = fileList.getJSONObject(i);
+            fileItemDTO.setName(fileItem.getString("name"));
+            fileItemDTO.setSize(fileItem.getLong("size"));
+            fileItemDTO.setTime(fileItem.getDate("lastModifiedDateTime"));
+
+            if (fileItem.containsKey("file")) {
+                fileItemDTO.setUrl(fileItem.getString("@microsoft.graph.downloadUrl"));
+                fileItemDTO.setType(FileTypeEnum.FILE);
+            } else {
+                fileItemDTO.setType(FileTypeEnum.FOLDER);
+            }
+
+            fileItemDTO.setPath(path);
+            result.add(fileItemDTO);
+        }
+
+        return result;
+    }
+
+}