summaryrefslogtreecommitdiff
path: root/src/utils/file.js
blob: 2914762b0966b38c60217baff5a0d3bfe7c7334b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import { ElMessage } from 'element-plus';

// 上传图片不携带baseUrl 本地预览时显示
export const imageBaseUrl = (image) => {
  if (isDev() && image && image.startsWith('/')) {
    let baseURL = axiosInstance.defaults.baseURL;
    if (baseURL.startsWith('/')) {
      baseURL = window.location.host;
    }
    baseURL = baseURL?.endsWith('/') ? baseURL.slice(0, -1) : baseURL;
    return baseURL && baseURL + image;
  } else {
    return image;
  }
};

// 下载文件
export function downloadFile(res, filename) {
  if (res.status == 200) {
    if (res.headers['content-disposition']) {
      filename = decodeURIComponent(
        res.headers['content-disposition'].split(';')[1].split('filename=')[1]
      );
    }

    const blobObject = new Blob([res.data], { type: res.data.type });
    if (window.navigator.msSaveBlob) {
      // 兼容ie11
      window.navigator.msSaveBlob(blobObject, filename);
    } else {
      const url = URL.createObjectURL(blobObject);
      const a = document.createElement('a');
      document.body.appendChild(a); // 此处增加了将创建的添加到body当中
      a.href = url;
      a.download = filename;
      a.target = '_blank';
      a.click();
      a.remove(); // 将a标签移除
    }
  } else {
    const reader = new FileReader();
    reader.onload = function () {
      const responseText = reader.result;
      if (reader.result) {
        const exception = JSON.parse(responseText);
        ElMessage.error(exception.msg || exception.error);
      }
    };
    reader.readAsText(res.data);
  }
}

// 获取文件后缀
export const getSuffix = (name) => {
  const suffix = name.includes('.')
    ? name.substring(name.lastIndexOf('.') + 1)
    : '';
  return suffix;
};

// 获取文件类型
export const getLanguage = (name) => {
  const suffix = getSuffix(name);
  let language = '';
  switch (suffix) {
    case 'md': {
      language = 'markdown';
      break;
    }
    case 'js': {
      language = 'javascript';
      break;
    }
    case 'txt': {
      language = 'plaintext';
      break;
    }
    case '': {
      language = 'plaintext';
      break;
    }
    default:
      language = suffix;
      break;
  }
  return language;
};

// 判断是否为图片
export const isImage = (name) => {
  const suffix = getSuffix(name);
  const imageTypes = ['jpg', 'jpeg', 'png', 'gif', 'bmp', 'webp', 'svg'];
  return imageTypes.includes(suffix);
};

// 图片base64增加前缀
export const imageBase64Prefix = (name, content) => {
  const suffix = getSuffix(name);
  let prefix = '';
  switch (suffix) {
    case 'jpg': {
      prefix = 'data:image/jpeg;base64,';
      break;
    }
    case 'svg': {
      prefix = 'data:image/svg+xml;base64,';
      break;
    }
    default:
      prefix = `data:image/${suffix};base64,`;
      break;
  }
  return prefix + content;
};