77 lines
2.0 KiB
TypeScript
77 lines
2.0 KiB
TypeScript
|
import { Multihash } from "./multihash.js";
|
||
|
|
import NodeId from "./nodeId.js";
|
||
|
|
import { S5Config } from "./types.js";
|
||
|
|
import Unpacker from "./serialization/unpack.js";
|
||
|
|
import Packer from "./serialization/pack.js";
|
||
|
|
import StorageLocation from "./storage.js";
|
||
|
|
|
||
|
|
export async function readStorageLocationsFromDB({
|
||
|
|
hash,
|
||
|
|
config,
|
||
|
|
}: {
|
||
|
|
hash: Multihash;
|
||
|
|
config: S5Config;
|
||
|
|
}): Promise<Map<number, Map<NodeId, Map<number, any>>>> {
|
||
|
|
const map = new Map<number, Map<NodeId, Map<number, any>>>();
|
||
|
|
const bytes = await config.db.get(stringifyHash(hash));
|
||
|
|
if (bytes === null) {
|
||
|
|
return map;
|
||
|
|
}
|
||
|
|
const unpacker = Unpacker.fromPacked(bytes);
|
||
|
|
const mapLength = unpacker.unpackMapLength();
|
||
|
|
for (let i = 0; i < mapLength; i++) {
|
||
|
|
const type = unpacker.unpackInt() as number;
|
||
|
|
const innerMap = new Map<NodeId, Map<number, any>>();
|
||
|
|
map.set(type, innerMap);
|
||
|
|
const innerMapLength = unpacker.unpackMapLength();
|
||
|
|
for (let j = 0; j < innerMapLength; j++) {
|
||
|
|
const nodeId = new NodeId(unpacker.unpackBinary());
|
||
|
|
innerMap.set(nodeId, new Map(unpacker.unpackMap() as [number, any][]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return map;
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function addStorageLocation({
|
||
|
|
hash,
|
||
|
|
nodeId,
|
||
|
|
location,
|
||
|
|
message,
|
||
|
|
config,
|
||
|
|
}: {
|
||
|
|
hash: Multihash;
|
||
|
|
nodeId: NodeId;
|
||
|
|
location: StorageLocation;
|
||
|
|
message?: Uint8Array;
|
||
|
|
config: S5Config;
|
||
|
|
}) {
|
||
|
|
const map = this.readStorageLocationsFromDB(hash);
|
||
|
|
const innerMap =
|
||
|
|
map.get(location.type) || new Map<NodeId, Map<number, any>>();
|
||
|
|
map.set(location.type, innerMap);
|
||
|
|
|
||
|
|
const locationMap = new Map<number, any>([
|
||
|
|
[1, location.parts],
|
||
|
|
// [2, location.binaryParts],
|
||
|
|
[3, location.expiry],
|
||
|
|
[4, message],
|
||
|
|
]);
|
||
|
|
|
||
|
|
innerMap.set(nodeId, locationMap);
|
||
|
|
await config.cacheDb.put(
|
||
|
|
stringifyHash(hash),
|
||
|
|
new Packer().pack(map).takeBytes(),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
export function stringifyBytes(data: Uint8Array) {
|
||
|
|
return String.fromCharCode(...data);
|
||
|
|
}
|
||
|
|
|
||
|
|
function stringifyHash(hash: Multihash) {
|
||
|
|
return stringifyBytes(hash.fullBytes);
|
||
|
|
}
|
||
|
|
export function stringifyNode(node: NodeId) {
|
||
|
|
return stringifyBytes(node.bytes);
|
||
|
|
}
|