Source/BladeX-Tool/blade-core-boot/src/main/java/org/springblade/core/boot/ctrl/BladeController.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,279 @@
/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill åº„骞 (smallchill@163.com)
 */
package org.springblade.core.boot.ctrl;

import org.springblade.core.boot.file.LocalFile;
import org.springblade.core.boot.file.BladeFileUtil;
import org.springblade.core.secure.BladeUser;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.Charsets;
import org.springblade.core.tool.utils.StringPool;
import org.springblade.core.tool.utils.WebUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.ResourceRegion;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.util.UriUtils;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.IOException;
import java.util.List;

/**
 * Blade控制器封装类
 *
 * @author Chill
 */
public class BladeController {

   /**
    * ============================     REQUEST    =================================================
    */

   @Autowired
   private HttpServletRequest request;

   /**
    * èŽ·å–request
    */
   public HttpServletRequest getRequest() {
      return this.request;
   }

   /**
    * èŽ·å–当前用户
    *
    * @return
    */
   public BladeUser getUser() {
      return AuthUtil.getUser();
   }

   /** ============================     API_RESULT    =================================================  */

   /**
    * è¿”回ApiResult
    *
    * @param data
    * @return R
    */
   public <T> R<T> data(T data) {
      return R.data(data);
   }

   /**
    * è¿”回ApiResult
    *
    * @param data
    * @param message
    * @return R
    */
   public <T> R<T> data(T data, String message) {
      return R.data(data, message);
   }

   /**
    * è¿”回ApiResult
    *
    * @param data
    * @param message
    * @param code
    * @return R
    */
   public <T> R<T> data(T data, String message, int code) {
      return R.data(code, data, message);
   }

   /**
    * è¿”回ApiResult
    *
    * @param message
    * @return R
    */
   public R success(String message) {
      return R.success(message);
   }

   /**
    * è¿”回ApiResult
    *
    * @param message
    * @return R
    */
   public R fail(String message) {
      return R.fail(message);
   }

   /**
    * è¿”回ApiResult
    *
    * @param flag
    * @return R
    */
   public R status(boolean flag) {
      return R.status(flag);
   }


   /**============================     FILE    =================================================  */

   /**
    * èŽ·å–BladeFile封装类
    *
    * @param file
    * @return
    */
   public LocalFile getFile(MultipartFile file) {
      return BladeFileUtil.getFile(file);
   }

   /**
    * èŽ·å–BladeFile封装类
    *
    * @param file
    * @param dir
    * @return
    */
   public LocalFile getFile(MultipartFile file, String dir) {
      return BladeFileUtil.getFile(file, dir);
   }

   /**
    * èŽ·å–BladeFile封装类
    *
    * @param file
    * @param dir
    * @param path
    * @param virtualPath
    * @return
    */
   public LocalFile getFile(MultipartFile file, String dir, String path, String virtualPath) {
      return BladeFileUtil.getFile(file, dir, path, virtualPath);
   }

   /**
    * èŽ·å–BladeFile封装类
    *
    * @param files
    * @return
    */
   public List<LocalFile> getFiles(List<MultipartFile> files) {
      return BladeFileUtil.getFiles(files);
   }

   /**
    * èŽ·å–BladeFile封装类
    *
    * @param files
    * @param dir
    * @return
    */
   public List<LocalFile> getFiles(List<MultipartFile> files, String dir) {
      return BladeFileUtil.getFiles(files, dir);
   }

   /**
    * èŽ·å–BladeFile封装类
    *
    * @param files
    * @param path
    * @param virtualPath
    * @return
    */
   public List<LocalFile> getFiles(List<MultipartFile> files, String dir, String path, String virtualPath) {
      return BladeFileUtil.getFiles(files, dir, path, virtualPath);
   }
   /**
    * ä¸‹è½½æ–‡ä»¶
    *
    * @param file æ–‡ä»¶
    * @return {ResponseEntity}
    * @throws IOException io异常
    */
   protected ResponseEntity<ResourceRegion> download(File file) throws IOException {
      String fileName = file.getName();
      return download(file, fileName);
   }

   /**
    * ä¸‹è½½
    *
    * @param file     æ–‡ä»¶
    * @param fileName ç”Ÿæˆçš„文件名
    * @return {ResponseEntity}
    * @throws IOException io异常
    */
   protected ResponseEntity<ResourceRegion> download(File file, String fileName) throws IOException {
      Resource resource = new FileSystemResource(file);
      return download(resource, fileName);
   }

   /**
    * ä¸‹è½½
    *
    * @param resource èµ„源
    * @param fileName ç”Ÿæˆçš„文件名
    * @return {ResponseEntity}
    * @throws IOException io异常
    */
   protected ResponseEntity<ResourceRegion> download(Resource resource, String fileName) throws IOException {
      HttpServletRequest request = WebUtil.getRequest();
      String header = request.getHeader(HttpHeaders.USER_AGENT);
      // é¿å…ç©ºæŒ‡é’ˆ
      header = header == null ? StringPool.EMPTY : header.toUpperCase();
      HttpStatus status;
      String msie= "MSIE";
      String trident= "TRIDENT";
      String edge= "EDGE";
      if (header.contains(msie) || header.contains(trident) || header.contains(edge)) {
         status = HttpStatus.OK;
      } else {
         status = HttpStatus.CREATED;
      }
      // æ–­ç‚¹ç»­ä¼ 
      long position = 0;
      long count = resource.contentLength();
      String range = request.getHeader(HttpHeaders.RANGE);
      if (null != range) {
         status = HttpStatus.PARTIAL_CONTENT;
         String[] rangeRange = range.replace("bytes=", StringPool.EMPTY).split(StringPool.DASH);
         position = Long.parseLong(rangeRange[0]);
         if (rangeRange.length > 1) {
            long end = Long.parseLong(rangeRange[1]);
            count = end - position + 1;
         }
      }
      HttpHeaders headers = new HttpHeaders();
      headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
      String encodeFileName = UriUtils.encode(fileName, Charsets.UTF_8);
      // å…¼å®¹å„种浏览器下载：
      // https://blog.robotshell.org/2012/deal-with-http-header-encoding-for-file-download/
      String disposition = "attachment;" +
         "filename=\"" + encodeFileName + "\";" +
         "filename*=utf-8''" + encodeFileName;
      headers.set(HttpHeaders.CONTENT_DISPOSITION, disposition);
      return new ResponseEntity<>(new ResourceRegion(resource, position, count), headers, status);
   }

}