cvitool 1.0.71 → 1.0.72

package/build/src/cutil.d.ts

@@ -1,6 +1,8 @@
 /// <reference types="node" />
 /// <reference types="node" />
+/// <reference types="node" />
 import { Encoding } from 'crypto';
+import { ReadStream } from 'fs';
 interface randomStringOptions {
     special?: boolean;
     lowercase?: boolean;
@@ -8,9 +10,15 @@ interface randomStringOptions {
     number?: boolean;
     specials?: string;
 }
+interface getMediaFileTypeRes {
+    type: string;
+    extname: string;
+}
 declare function randomString(length: number, options?: randomStringOptions): string;
 declare function encryptCBC(data: string, length: 128 | 192 | 256, key: Buffer, iv: Buffer, inputEncoding?: Encoding): string;
 declare function decryptCBC(encryptData: string, length: 128 | 192 | 256, key: Buffer, iv: Buffer, outputEncoding?: Encoding): string;
 declare function md5(data: string, inputEncoding?: Encoding): string;
 declare function execCmdCommand(command: string): Promise<unknown>;
-export { randomStringOptions, randomString, encryptCBC, decryptCBC, md5, execCmdCommand };
+declare function getValueType(value: any): "string" | "number" | "undefined" | "object" | "array" | "null" | "unkonw";
+declare function getMediaFileType(target: string | Buffer | ReadStream): Promise<getMediaFileTypeRes>;
+export { randomStringOptions, getMediaFileTypeRes, randomString, encryptCBC, decryptCBC, md5, execCmdCommand, getValueType, getMediaFileType };

package/build/src/cutil.js

@@ -9,9 +9,10 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.execCmdCommand = exports.md5 = exports.decryptCBC = exports.encryptCBC = exports.randomString = void 0;
+exports.getMediaFileType = exports.getValueType = exports.execCmdCommand = exports.md5 = exports.decryptCBC = exports.encryptCBC = exports.randomString = void 0;
 const crypto_1 = require("crypto");
 const child_process_1 = require("child_process");
+const hgo = require("./hgo");
 function randomString(length, options) {
     let { special = false, lowercase = true, upperCase = true, number = true, specials } = options || {};
     if (specials) {
@@ -92,3 +93,146 @@ function execCmdCommand(command) {
     });
 }
 exports.execCmdCommand = execCmdCommand;
+function getValueType(value) {
+    const typeStr = Object.prototype.toString.call(value);
+    let type;
+    switch (typeStr) {
+        case '[object Number]':
+            type = 'number';
+            break;
+        case '[object String]':
+            type = 'string';
+            break;
+        case '[object Object]':
+            type = 'object';
+            break;
+        case '[object Array]':
+            type = 'array';
+            break;
+        case '[object Undefined]':
+            type = 'undefined';
+            break;
+        case '[object Null]':
+            type = 'null';
+            break;
+        default:
+            type = 'unkonw';
+            break;
+    }
+    ;
+    return type;
+}
+exports.getValueType = getValueType;
+function getMediaFileType(target) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const readStreamOnceBuffer = (stream) => __awaiter(this, void 0, void 0, function* () {
+            return new Promise((resolve, reject) => {
+                stream.once('data', chunk => {
+                    stream.close();
+                    resolve(chunk);
+                });
+                stream.on('error', e => {
+                    reject(e);
+                });
+            });
+        });
+        const checkString = (waitMatchStr, byteStart, byteEnd, buffer) => {
+            return waitMatchStr === Buffer.from(buffer.slice(byteStart, byteEnd + 1)).toString();
+        };
+        const checkBuffer = (waitMatchBuffer, byteStart, byteEnd, buffer) => {
+            let equal = true;
+            const sliceBufferList = buffer.slice(byteStart, byteEnd + 1);
+            for (const [index, item] of waitMatchBuffer.entries()) {
+                if (item !== sliceBufferList[index]) {
+                    equal = false;
+                    break;
+                }
+            }
+            return equal;
+        };
+        const getResult = (typeStr) => {
+            const arr = typeStr.split('|');
+            return {
+                type: arr[0],
+                extname: arr[1]
+            };
+        };
+        let buffer;
+        const targetType = this.getValueType(target);
+        if (targetType === 'string') {
+            const res = yield hgo.get(target, {
+                resType: 'buffer',
+                headers: {
+                    Range: 'bytes=0-20'
+                }
+            });
+            buffer = res.resBody;
+        }
+        else if (targetType === 'array') {
+            buffer = target;
+        }
+        else {
+            buffer = yield readStreamOnceBuffer(target);
+        }
+        buffer = buffer.slice(0, 20);
+        if (checkBuffer([0xff, 0xd8], 0, 1, buffer)) {
+            return getResult('image|jpg');
+        }
+        if (checkString('PNG', 1, 3, buffer)) {
+            return getResult('image|png');
+        }
+        if (checkBuffer([0x49, 0x44, 0x33], 0, 2, buffer) || checkBuffer([0xff, 0xf3], 0, 1, buffer)) {
+            return getResult('audio|mp3');
+        }
+        if (checkString('WAV', 8, 10, buffer)) {
+            return getResult('audio|wav');
+        }
+        if (checkString('M4A', 8, 10, buffer)) {
+            return getResult('audio|m4a');
+        }
+        if (checkString('heic', 8, 11, buffer) || checkString('mif1', 8, 11, buffer)) {
+            return getResult('image|heic');
+        }
+        if (checkString('ftyp', 4, 7, buffer)) {
+            if (checkString('ftypqt', 4, 9, buffer)) {
+                return getResult('video|mov');
+            }
+            return getResult('video|mp4');
+        }
+        if (checkString('WEBP', 8, 11, buffer)) {
+            return getResult('image|webp');
+        }
+        if (checkBuffer([0x47, 0x49, 0x46], 0, 2, buffer)) {
+            return getResult('image|gif');
+        }
+        if (checkBuffer([0x42, 0x4d], 0, 1, buffer)) {
+            return getResult('image|bmp');
+        }
+        if (checkBuffer([0x46, 0x4c, 0x56], 0, 2, buffer)) {
+            return getResult('video|flv');
+        }
+        if (checkBuffer([0x4f, 0x67, 0x67, 0x53], 0, 3, buffer)) {
+            return getResult('video|ogg');
+        }
+        if (checkBuffer([0x1a, 0x45, 0xdf, 0xa3], 0, 3, buffer)) {
+            return getResult('video|mkv');
+        }
+        if (checkBuffer([0x30, 0x26, 0xb2], 0, 2, buffer)) {
+            return getResult('video|wmv');
+        }
+        if (checkString('AVI', 8, 10, buffer)) {
+            return getResult('video|avi');
+        }
+        if (checkBuffer([0xff, 0xf1], 0, 1, buffer)) {
+            return getResult('audio|aac');
+        }
+        if (checkString('PCM', 10, 12, buffer)) {
+            return getResult('audio|pcm');
+        }
+        if (checkBuffer([0x66, 0x4c, 0x61, 0x43], 0, 3, buffer)) {
+            return getResult('audio|flac');
+        }
+        return getResult('unknow|unknow');
+    });
+}
+exports.getMediaFileType = getMediaFileType;

package/index.d.ts

@@ -14,7 +14,8 @@ import {
 } from './src/streamhelper';
 
 import {
-    randomStringOptions
+    randomStringOptions,
+    getMediaFileTypeRes
 } from './src/cutil';
 
 interface Hgo {
@@ -119,7 +120,17 @@ interface Cutil {
      * 调用终端执行命令
      * @param command 
      */
-    execCmdCommand(command: string): string
+    execCmdCommand(command: string): Promise<string>
+    /**
+     * 获取给定值node类型
+     * @param value 
+     */
+    getValueType(value: any): string
+    /**
+     * 获取媒体文件类型
+     * @param target 文件url/buffer数据/可读流
+     */
+    getMediaFileType(target: string | Buffer | ReadStream): Promise<getMediaFileTypeRes>
 }
 
 declare const hgo: Hgo;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cvitool",
-  "version": "1.0.71",
+  "version": "1.0.72",
   "description": "cvitool",
   "main": "index.js",
   "scripts": {
@@ -15,6 +15,9 @@
     "eslint-config-standard": "16.0.3",
     "@types/node": "20.10.4"
   },
+  "engines": {
+    "node": ">=16.0.0"
+  },
   "files": [
     "build/src",
     "src",

package/src/cutil.ts

@@ -8,6 +8,8 @@ import {
 import {
     spawn
 } from 'child_process';
+import { ReadStream } from 'fs';
+import * as hgo from './hgo';
 
 interface randomStringOptions {
     special?: boolean,
@@ -17,6 +19,11 @@ interface randomStringOptions {
     specials?: string
 }
 
+interface getMediaFileTypeRes {
+    type: string,
+    extname: string
+}
+
 function randomString(length: number, options?: randomStringOptions) {
     let { special = false, lowercase = true, upperCase = true, number = true, specials } = options || {};
     if (specials) {
@@ -93,11 +100,155 @@ async function execCmdCommand(command: string) {
     });
 }
 
+function getValueType(value: any) {
+    const typeStr = Object.prototype.toString.call(value);
+    let type: 'number' | 'string' | 'array' | 'object' | 'undefined' | 'null' | 'unkonw';
+    switch (typeStr) {
+        case '[object Number]':
+            type = 'number';
+            break;
+        case '[object String]':
+            type = 'string';
+            break;
+        case '[object Object]':
+            type = 'object';
+            break;
+        case '[object Array]':
+            type = 'array';
+            break;
+        case '[object Undefined]':
+            type = 'undefined';
+            break;
+        case '[object Null]':
+            type = 'null';
+            break;
+        default:
+            type = 'unkonw';
+            break;
+    };
+    return type;
+}
+
+async function getMediaFileType(target: string | Buffer | ReadStream): Promise<getMediaFileTypeRes> {
+    const readStreamOnceBuffer = async (stream: ReadStream): Promise<Buffer> => {
+        return new Promise((resolve, reject) => {
+            stream.once('data', chunk => {
+                stream.close();
+                resolve(chunk as Buffer);
+            });
+            stream.on('error', e => {
+                reject(e);
+            });
+        });
+    };
+    const checkString = (waitMatchStr: string, byteStart: number, byteEnd: number, buffer: Buffer) => {
+        return waitMatchStr === Buffer.from(buffer.slice(byteStart, byteEnd + 1)).toString();
+    };
+    const checkBuffer = (waitMatchBuffer: number[], byteStart: number, byteEnd: number, buffer: Buffer) => {
+        let equal = true;
+        const sliceBufferList = buffer.slice(byteStart, byteEnd + 1);
+        for (const [index, item] of waitMatchBuffer.entries()) {
+            if (item !== sliceBufferList[index]) {
+                equal = false;
+                break;
+            }
+        }
+        return equal;
+    };
+    const getResult = (typeStr: string) => {
+        const arr = typeStr.split('|');
+        return {
+            type: arr[0],
+            extname: arr[1]
+        };
+    };
+    let buffer: Buffer;
+    const targetType = this.getValueType(target);
+    if (targetType === 'string') {
+        const res = await hgo.get(
+            target as string,
+            {
+                resType: 'buffer',
+                headers: {
+                    Range: 'bytes=0-20'
+                }
+            }
+        );
+        buffer = res.resBody as Buffer;
+    } else if (targetType === 'array') {
+        buffer = target as Buffer;
+    } else {
+        buffer = await readStreamOnceBuffer(target as ReadStream);
+    }
+    buffer = buffer.slice(0, 20);
+    if (checkBuffer([0xff, 0xd8], 0, 1, buffer)) {
+        return getResult('image|jpg');
+    }
+    if (checkString('PNG', 1, 3, buffer)) {
+        return getResult('image|png');
+    }
+    if (checkBuffer([0x49, 0x44, 0x33], 0, 2, buffer) || checkBuffer([0xff, 0xf3], 0, 1, buffer)) {
+        return getResult('audio|mp3');
+    }
+    if (checkString('WAV', 8, 10, buffer)) {
+        return getResult('audio|wav');
+    }
+    if (checkString('M4A', 8, 10, buffer)) {
+        return getResult('audio|m4a');
+    }
+    if (checkString('heic', 8, 11, buffer) || checkString('mif1', 8, 11, buffer)) {
+        return getResult('image|heic');
+    }
+    if (checkString('ftyp', 4, 7, buffer)) {
+        if (checkString('ftypqt', 4, 9, buffer)) {
+            return getResult('video|mov');
+        }
+        return getResult('video|mp4');
+    }
+    if (checkString('WEBP', 8, 11, buffer)) {
+        return getResult('image|webp');
+    }
+    if (checkBuffer([0x47, 0x49, 0x46], 0, 2, buffer)) {
+        return getResult('image|gif');
+    }
+    if (checkBuffer([0x42, 0x4d], 0, 1, buffer)) {
+        return getResult('image|bmp');
+    }
+    if (checkBuffer([0x46, 0x4c, 0x56], 0, 2, buffer)) {
+        return getResult('video|flv');
+    }
+    if (checkBuffer([0x4f, 0x67, 0x67, 0x53], 0, 3, buffer)) {
+        return getResult('video|ogg');
+    }
+    if (checkBuffer([0x1a, 0x45, 0xdf, 0xa3], 0, 3, buffer)) {
+        return getResult('video|mkv');
+    }
+    if (checkBuffer([0x30, 0x26, 0xb2], 0, 2, buffer)) {
+        return getResult('video|wmv');
+    }
+    if (checkString('AVI', 8, 10, buffer)) {
+        return getResult('video|avi');
+    }
+    if (checkBuffer([0xff, 0xf1], 0, 1, buffer)) {
+        return getResult('audio|aac');
+    }
+    if (checkString('PCM', 10, 12, buffer)) {
+        return getResult('audio|pcm');
+    }
+    if (checkBuffer([0x66, 0x4c, 0x61, 0x43], 0, 3, buffer)) {
+        return getResult('audio|flac');
+    }
+    return getResult('unknow|unknow');
+}
+
 export {
     randomStringOptions,
+    getMediaFileTypeRes,
     randomString,
     encryptCBC,
     decryptCBC,
     md5,
-    execCmdCommand
+    execCmdCommand,
+    getValueType,
+    getMediaFileType
 };