/**
 * Copyright (c) Whales Corp.
 * All Rights Reserved.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

import { Address } from "@ton/core";
import { BitString } from "@ton/core";
import { paddedBufferToBits } from "@/asm/dict/paddedBits";

export function serializeInternalKey(value: any): string {
    if (typeof value === "number") {
        if (!Number.isSafeInteger(value)) {
            throw Error("Invalid key type: not a safe integer: " + value);
        }
        return "n:" + value.toString(10);
    } else if (typeof value === "bigint") {
        return "b:" + value.toString(10);
    } else if (Address.isAddress(value)) {
        return "a:" + value.toString();
    } else if (Buffer.isBuffer(value)) {
        return "f:" + value.toString("hex");
    } else if (BitString.isBitString(value)) {
        return "B:" + value.toString();
    } else {
        throw Error("Invalid key type");
    }
}

export function deserializeInternalKey(value: string): any {
    let k = value.slice(0, 2);
    let v = value.slice(2);
    if (k === "n:") {
        return Number.parseInt(v, 10);
    } else if (k === "b:") {
        return BigInt(v);
    } else if (k === "a:") {
        return Address.parse(v);
    } else if (k === "f:") {
        return Buffer.from(v, "hex");
    } else if (k === "B:") {
        const lastDash = v.slice(-1) == "_";
        const isPadded = lastDash || v.length % 2 != 0;
        if (isPadded) {
            let charLen = lastDash ? v.length - 1 : v.length;
            const padded = v.substr(0, charLen) + "0"; //Padding
            if (!lastDash && (charLen & 1) !== 0) {
                // Four bit nimble without padding
                return new BitString(
                    Buffer.from(padded, "hex"),
                    0,
                    charLen << 2,
                );
            } else {
                return paddedBufferToBits(Buffer.from(padded, "hex"));
            }
        } else {
            return new BitString(Buffer.from(v, "hex"), 0, v.length << 2);
        }
    }
    throw Error("Invalid key type: " + k);
}
