|
1 | | -import { extension as getExtensionFromMime } from 'mime-types'; |
| 1 | +import { extension as getExtensionFromMimeType, extension as getMimeTypeFromExtension } from 'mime-types'; |
2 | 2 | import type { Ref } from 'vue'; |
3 | 3 | import _ from 'lodash'; |
4 | 4 |
|
5 | | -export { getMimeTypeFromBase64, useDownloadFileFromBase64 }; |
| 5 | +export { |
| 6 | + getMimeTypeFromBase64, |
| 7 | + getMimeTypeFromExtension, getExtensionFromMimeType, |
| 8 | + useDownloadFileFromBase64, useDownloadFileFromBase64Refs, |
| 9 | +}; |
6 | 10 |
|
7 | 11 | const commonMimeTypesSignatures = { |
8 | 12 | 'JVBERi0': 'application/pdf', |
@@ -36,30 +40,55 @@ function getFileExtensionFromMimeType({ |
36 | 40 | defaultExtension?: string |
37 | 41 | }) { |
38 | 42 | if (mimeType) { |
39 | | - return getExtensionFromMime(mimeType) ?? defaultExtension; |
| 43 | + return getExtensionFromMimeType(mimeType) ?? defaultExtension; |
40 | 44 | } |
41 | 45 |
|
42 | 46 | return defaultExtension; |
43 | 47 | } |
44 | 48 |
|
45 | | -function useDownloadFileFromBase64({ source, filename }: { source: Ref<string>; filename?: string }) { |
46 | | - return { |
47 | | - download() { |
48 | | - if (source.value === '') { |
49 | | - throw new Error('Base64 string is empty'); |
50 | | - } |
| 49 | +function downloadFromBase64({ sourceValue, filename, extension, fileMimeType }: |
| 50 | +{ sourceValue: string; filename?: string; extension?: string; fileMimeType?: string }) { |
| 51 | + if (sourceValue === '') { |
| 52 | + throw new Error('Base64 string is empty'); |
| 53 | + } |
| 54 | + |
| 55 | + const defaultExtension = extension ?? 'txt'; |
| 56 | + const { mimeType } = getMimeTypeFromBase64({ base64String: sourceValue }); |
| 57 | + let base64String = sourceValue; |
| 58 | + if (!mimeType) { |
| 59 | + const targetMimeType = fileMimeType ?? getMimeTypeFromExtension(defaultExtension); |
| 60 | + base64String = `data:${targetMimeType};base64,${sourceValue}`; |
| 61 | + } |
| 62 | + |
| 63 | + const cleanExtension = extension ?? getFileExtensionFromMimeType( |
| 64 | + { mimeType, defaultExtension }); |
| 65 | + let cleanFileName = filename ?? `file.${cleanExtension}`; |
| 66 | + if (extension && !cleanFileName.endsWith(`.${extension}`)) { |
| 67 | + cleanFileName = `${cleanFileName}.${cleanExtension}`; |
| 68 | + } |
51 | 69 |
|
52 | | - const { mimeType } = getMimeTypeFromBase64({ base64String: source.value }); |
53 | | - const base64String = mimeType |
54 | | - ? source.value |
55 | | - : `data:text/plain;base64,${source.value}`; |
| 70 | + const a = document.createElement('a'); |
| 71 | + a.href = base64String; |
| 72 | + a.download = cleanFileName; |
| 73 | + a.click(); |
| 74 | +} |
56 | 75 |
|
57 | | - const cleanFileName = filename ?? `file.${getFileExtensionFromMimeType({ mimeType })}`; |
| 76 | +function useDownloadFileFromBase64( |
| 77 | + { source, filename, extension, fileMimeType }: |
| 78 | + { source: Ref<string>; filename?: string; extension?: string; fileMimeType?: string }) { |
| 79 | + return { |
| 80 | + download() { |
| 81 | + downloadFromBase64({ sourceValue: source.value, filename, extension, fileMimeType }); |
| 82 | + }, |
| 83 | + }; |
| 84 | +} |
58 | 85 |
|
59 | | - const a = document.createElement('a'); |
60 | | - a.href = base64String; |
61 | | - a.download = cleanFileName; |
62 | | - a.click(); |
| 86 | +function useDownloadFileFromBase64Refs( |
| 87 | + { source, filename, extension }: |
| 88 | + { source: Ref<string>; filename?: Ref<string>; extension?: Ref<string> }) { |
| 89 | + return { |
| 90 | + download() { |
| 91 | + downloadFromBase64({ sourceValue: source.value, filename: filename?.value, extension: extension?.value }); |
63 | 92 | }, |
64 | 93 | }; |
65 | 94 | } |
0 commit comments