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