Files
query-awesome/query/query-resources/utils.ts

69 lines
2.0 KiB
TypeScript

import MD5 from 'crypto-js/md5';
export const hashContent = (str: string | Blob | Buffer): Promise<string> | string => {
if (typeof str === 'string') {
return MD5(str).toString();
} else if (str instanceof Blob) {
return hashBlob(str);
} else if (Buffer.isBuffer(str)) {
return MD5(str.toString()).toString();
}
console.error('hashContent error: input must be a string, Blob, or Buffer');
return '';
};
export const hashBlob = (blob: Blob): Promise<string> => {
return new Promise((resolve, reject) => {
const reader = new FileReader();
reader.onload = async () => {
try {
const content = reader.result;
if (typeof content === 'string') {
resolve(MD5(content).toString());
} else if (content) {
const contentString = new TextDecoder().decode(content);
resolve(MD5(contentString).toString());
} else {
reject(new Error('Empty content'));
}
} catch (error) {
console.error('hashBlob error', error);
reject(error);
}
};
reader.onerror = (error) => reject(error);
reader.readAsArrayBuffer(blob);
});
};
export const hashFile = (file: File): Promise<string> => {
return new Promise((resolve, reject) => {
const reader = new FileReader();
reader.onload = async (event) => {
try {
const content = event.target?.result;
if (content instanceof ArrayBuffer) {
const contentString = new TextDecoder().decode(content);
const hashHex = MD5(contentString).toString();
resolve(hashHex);
} else if (typeof content === 'string') {
const hashHex = MD5(content).toString();
resolve(hashHex);
} else {
throw new Error('Invalid content type');
}
} catch (error) {
console.error('hashFile error', error);
reject(error);
}
};
reader.onerror = (error) => {
reject(error);
};
// 读取文件为 ArrayBuffer
reader.readAsArrayBuffer(file);
});
};