89 lines
2.9 KiB
TypeScript
89 lines
2.9 KiB
TypeScript
|
"use strict";
|
||
|
|
|
||
|
|
import { BigNumber } from "@ethersproject/bignumber";
|
||
|
|
import { arrayify, concat, hexlify, zeroPad } from "@ethersproject/bytes";
|
||
|
|
import { keccak256 as hashKeccak256 } from "@ethersproject/keccak256";
|
||
|
|
import { sha256 as hashSha256 } from "@ethersproject/sha2";
|
||
|
|
import { toUtf8Bytes } from "@ethersproject/strings";
|
||
|
|
|
||
|
|
const regexBytes = new RegExp("^bytes([0-9]+)$");
|
||
|
|
const regexNumber = new RegExp("^(u?int)([0-9]*)$");
|
||
|
|
const regexArray = new RegExp("^(.*)\\[([0-9]*)\\]$");
|
||
|
|
|
||
|
|
const Zeros = "0000000000000000000000000000000000000000000000000000000000000000";
|
||
|
|
|
||
|
|
function _pack(type: string, value: any, isArray?: boolean): Uint8Array {
|
||
|
|
switch(type) {
|
||
|
|
case "address":
|
||
|
|
if (isArray) { return zeroPad(value, 32); }
|
||
|
|
return arrayify(value);
|
||
|
|
case "string":
|
||
|
|
return toUtf8Bytes(value);
|
||
|
|
case "bytes":
|
||
|
|
return arrayify(value);
|
||
|
|
case "bool":
|
||
|
|
value = (value ? "0x01": "0x00");
|
||
|
|
if (isArray) { return zeroPad(value, 32); }
|
||
|
|
return arrayify(value);
|
||
|
|
}
|
||
|
|
|
||
|
|
let match = type.match(regexNumber);
|
||
|
|
if (match) {
|
||
|
|
//let signed = (match[1] === "int")
|
||
|
|
let size = parseInt(match[2] || "256")
|
||
|
|
if ((size % 8 != 0) || size === 0 || size > 256) {
|
||
|
|
throw new Error("invalid number type - " + type);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (isArray) { size = 256; }
|
||
|
|
|
||
|
|
value = BigNumber.from(value).toTwos(size);
|
||
|
|
|
||
|
|
return zeroPad(value, size / 8);
|
||
|
|
}
|
||
|
|
|
||
|
|
match = type.match(regexBytes);
|
||
|
|
if (match) {
|
||
|
|
let size = parseInt(match[1]);
|
||
|
|
if (String(size) != match[1] || size === 0 || size > 32) {
|
||
|
|
throw new Error("invalid number type - " + type);
|
||
|
|
}
|
||
|
|
if (arrayify(value).byteLength !== size) { throw new Error("invalid value for " + type); }
|
||
|
|
if (isArray) { return arrayify((value + Zeros).substring(0, 66)); }
|
||
|
|
return value;
|
||
|
|
}
|
||
|
|
|
||
|
|
match = type.match(regexArray);
|
||
|
|
if (match && Array.isArray(value)) {
|
||
|
|
let baseType = match[1];
|
||
|
|
let count = parseInt(match[2] || String(value.length));
|
||
|
|
if (count != value.length) { throw new Error("invalid value for " + type); }
|
||
|
|
let result: Array<Uint8Array> = [];
|
||
|
|
value.forEach(function(value) {
|
||
|
|
result.push(_pack(baseType, value, true));
|
||
|
|
});
|
||
|
|
return concat(result);
|
||
|
|
}
|
||
|
|
|
||
|
|
throw new Error("unknown type - " + type);
|
||
|
|
}
|
||
|
|
|
||
|
|
// @TODO: Array Enum
|
||
|
|
|
||
|
|
export function pack(types: Array<string>, values: Array<any>) {
|
||
|
|
if (types.length != values.length) { throw new Error("type/value count mismatch"); }
|
||
|
|
let tight: Array<Uint8Array> = [];
|
||
|
|
types.forEach(function(type, index) {
|
||
|
|
tight.push(_pack(type, values[index]));
|
||
|
|
});
|
||
|
|
return hexlify(concat(tight));
|
||
|
|
}
|
||
|
|
|
||
|
|
export function keccak256(types: Array<string>, values: Array<any>) {
|
||
|
|
return hashKeccak256(pack(types, values));
|
||
|
|
}
|
||
|
|
|
||
|
|
export function sha256(types: Array<string>, values: Array<any>) {
|
||
|
|
return hashSha256(pack(types, values));
|
||
|
|
}
|