@fizzyflow/suisql
Version:
SuiSQL is a library and set of tools for working with decentralized SQL databases on the Sui blockchain and Walrus protocol.
123 lines (122 loc) • 3.4 kB
JavaScript
import pako from "pako";
import { bcs } from "@mysten/sui/bcs";
const compress = async (input) => {
return pako.deflate(input);
};
const decompress = async (compressed) => {
return pako.inflate(compressed);
};
const anyShallowCopy = (input) => {
if (Array.isArray(input)) {
return [...input];
} else if (typeof input === "object" && input !== null) {
return { ...input };
} else {
return input;
}
};
const isSureWriteSql = (sql) => {
const checks = ["CREATE", "ALTER", "INSERT", "UPDATE", "DELETE", "DROP"];
for (const check of checks) {
if (sql.trim().toUpperCase().startsWith(check)) {
return true;
}
}
return false;
};
const getFieldsFromCreateTableSql = (sql) => {
const inParentheses = extractTopLevelParenthesesText(sql.split("\n").join(" "));
if (!inParentheses || !inParentheses[0]) {
return null;
}
const fields = inParentheses[0].split(",");
const ret = [];
for (const field of fields) {
const definition = field.trim().toLowerCase();
ret.push(definition);
}
console.log(ret);
console.log(ret);
return ret;
};
const extractTopLevelParenthesesText = (str) => {
let result = [];
let stack = [];
let startIndex = -1;
for (let i = 0; i < str.length; i++) {
if (str[i] === "(") {
if (stack.length === 0) {
startIndex = i + 1;
}
stack.push("(");
} else if (str[i] === ")") {
stack.pop();
if (stack.length === 0 && startIndex !== -1) {
result.push(str.substring(startIndex, i));
startIndex = -1;
}
}
}
return result;
};
const int32ToUint8ArrayBE = (num) => Uint8Array.from([num >>> 24, num >>> 16 & 255, num >>> 8 & 255, num & 255]);
const bigintToUint8Array = (bigint) => {
return bcs.u256().serialize(bigint).toBytes();
};
const idTo64 = (id) => {
const asA = Array.from(bigintToUint8Array(BigInt(id)));
let base64String = btoa(String.fromCharCode.apply(null, asA));
return base64String.replaceAll("/", "_").replaceAll("+", "-").replaceAll("=", "");
};
const walrus64ToBigInt = (v) => {
const base64 = v.replaceAll("_", "/").replaceAll("-", "+");
const raw = atob(base64);
const hex = [];
raw.split("").forEach(function(ch) {
var h = ch.charCodeAt(0).toString(16);
if (h.length % 2) {
h = "0" + h;
}
hex.unshift(h);
});
return BigInt("0x" + hex.join(""));
};
const concatUint8Arrays = (arrays) => {
const totalLength = arrays.reduce((sum, arr) => sum + arr.length, 0);
const result = new Uint8Array(totalLength);
let offset = 0;
for (const arr of arrays) {
result.set(arr, offset);
offset += arr.length;
}
return result;
};
function blobIdFromInt(blobId) {
return bcs.u256().serialize(blobId).toBase64().replace(/=*$/, "").replaceAll("+", "-").replaceAll("/", "_");
}
function blobIdFromBytes(blobId) {
return blobIdFromInt(bcs.u256().parse(blobId));
}
function blobIdIntFromBytes(blobId) {
return BigInt(bcs.u256().parse(blobId));
}
function blobIdToInt(blobId) {
return BigInt(bcs.u256().fromBase64(blobId.replaceAll("-", "+").replaceAll("_", "/")));
}
export {
anyShallowCopy,
bigintToUint8Array,
blobIdFromBytes,
blobIdFromInt,
blobIdIntFromBytes,
blobIdToInt,
compress,
concatUint8Arrays,
decompress,
getFieldsFromCreateTableSql,
idTo64,
int32ToUint8ArrayBE,
isSureWriteSql,
walrus64ToBigInt
};
//# sourceMappingURL=SuiSqlUtils.js.map