init

马宇豪

2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const Hash = require("../Hash");
const MAX_SHORT_STRING = require("./wasm-hash").MAX_SHORT_STRING;
 
class BatchedHash extends Hash {
    constructor(hash) {
        super();
        this.string = undefined;
        this.encoding = undefined;
        this.hash = hash;
    }
 
    /**
     * Update hash {@link https://nodejs.org/api/crypto.html#crypto_hash_update_data_inputencoding}
     * @param {string|Buffer} data data
     * @param {string=} inputEncoding data encoding
     * @returns {this} updated hash
     */
    update(data, inputEncoding) {
        if (this.string !== undefined) {
            if (
                typeof data === "string" &&
                inputEncoding === this.encoding &&
                this.string.length + data.length < MAX_SHORT_STRING
            ) {
                this.string += data;
                return this;
            }
            this.hash.update(this.string, this.encoding);
            this.string = undefined;
        }
        if (typeof data === "string") {
            if (
                data.length < MAX_SHORT_STRING &&
                // base64 encoding is not valid since it may contain padding chars
                (!inputEncoding || !inputEncoding.startsWith("ba"))
            ) {
                this.string = data;
                this.encoding = inputEncoding;
            } else {
                this.hash.update(data, inputEncoding);
            }
        } else {
            this.hash.update(data);
        }
        return this;
    }
 
    /**
     * Calculates the digest {@link https://nodejs.org/api/crypto.html#crypto_hash_digest_encoding}
     * @param {string=} encoding encoding of the return value
     * @returns {string|Buffer} digest
     */
    digest(encoding) {
        if (this.string !== undefined) {
            this.hash.update(this.string, this.encoding);
        }
        return this.hash.digest(encoding);
    }
}
 
module.exports = BatchedHash;