|
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 | + previewImageFromBase64, |
| 10 | +}; |
6 | 11 |
|
7 | 12 | const commonMimeTypesSignatures = {
|
8 | 13 | 'JVBERi0': 'application/pdf',
|
@@ -36,30 +41,78 @@ function getFileExtensionFromMimeType({
|
36 | 41 | defaultExtension?: string
|
37 | 42 | }) {
|
38 | 43 | if (mimeType) {
|
39 |
| - return getExtensionFromMime(mimeType) ?? defaultExtension; |
| 44 | + return getExtensionFromMimeType(mimeType) ?? defaultExtension; |
40 | 45 | }
|
41 | 46 |
|
42 | 47 | return defaultExtension;
|
43 | 48 | }
|
44 | 49 |
|
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 |
| - } |
| 50 | +function downloadFromBase64({ sourceValue, filename, extension, fileMimeType }: |
| 51 | +{ sourceValue: string; filename?: string; extension?: string; fileMimeType?: string }) { |
| 52 | + if (sourceValue === '') { |
| 53 | + throw new Error('Base64 string is empty'); |
| 54 | + } |
51 | 55 |
|
52 |
| - const { mimeType } = getMimeTypeFromBase64({ base64String: source.value }); |
53 |
| - const base64String = mimeType |
54 |
| - ? source.value |
55 |
| - : `data:text/plain;base64,${source.value}`; |
| 56 | + const defaultExtension = extension ?? 'txt'; |
| 57 | + const { mimeType } = getMimeTypeFromBase64({ base64String: sourceValue }); |
| 58 | + let base64String = sourceValue; |
| 59 | + if (!mimeType) { |
| 60 | + const targetMimeType = fileMimeType ?? getMimeTypeFromExtension(defaultExtension); |
| 61 | + base64String = `data:${targetMimeType};base64,${sourceValue}`; |
| 62 | + } |
56 | 63 |
|
57 |
| - const cleanFileName = filename ?? `file.${getFileExtensionFromMimeType({ mimeType })}`; |
| 64 | + const cleanExtension = extension ?? getFileExtensionFromMimeType( |
| 65 | + { mimeType, defaultExtension }); |
| 66 | + let cleanFileName = filename ?? `file.${cleanExtension}`; |
| 67 | + if (extension && !cleanFileName.endsWith(`.${extension}`)) { |
| 68 | + cleanFileName = `${cleanFileName}.${cleanExtension}`; |
| 69 | + } |
58 | 70 |
|
59 |
| - const a = document.createElement('a'); |
60 |
| - a.href = base64String; |
61 |
| - a.download = cleanFileName; |
62 |
| - a.click(); |
| 71 | + const a = document.createElement('a'); |
| 72 | + a.href = base64String; |
| 73 | + a.download = cleanFileName; |
| 74 | + a.click(); |
| 75 | +} |
| 76 | + |
| 77 | +function useDownloadFileFromBase64( |
| 78 | + { source, filename, extension, fileMimeType }: |
| 79 | + { source: Ref<string>; filename?: string; extension?: string; fileMimeType?: string }) { |
| 80 | + return { |
| 81 | + download() { |
| 82 | + downloadFromBase64({ sourceValue: source.value, filename, extension, fileMimeType }); |
63 | 83 | },
|
64 | 84 | };
|
65 | 85 | }
|
| 86 | + |
| 87 | +function useDownloadFileFromBase64Refs( |
| 88 | + { source, filename, extension }: |
| 89 | + { source: Ref<string>; filename?: Ref<string>; extension?: Ref<string> }) { |
| 90 | + return { |
| 91 | + download() { |
| 92 | + downloadFromBase64({ sourceValue: source.value, filename: filename?.value, extension: extension?.value }); |
| 93 | + }, |
| 94 | + }; |
| 95 | +} |
| 96 | + |
| 97 | +function previewImageFromBase64(base64String: string): HTMLImageElement { |
| 98 | + if (base64String === '') { |
| 99 | + throw new Error('Base64 string is empty'); |
| 100 | + } |
| 101 | + |
| 102 | + const img = document.createElement('img'); |
| 103 | + img.src = base64String; |
| 104 | + |
| 105 | + const container = document.createElement('div'); |
| 106 | + container.appendChild(img); |
| 107 | + |
| 108 | + const previewContainer = document.getElementById('previewContainer'); |
| 109 | + if (previewContainer) { |
| 110 | + previewContainer.innerHTML = ''; |
| 111 | + previewContainer.appendChild(container); |
| 112 | + } |
| 113 | + else { |
| 114 | + throw new Error('Preview container element not found'); |
| 115 | + } |
| 116 | + |
| 117 | + return img; |
| 118 | +} |
0 commit comments