@peerbit/program 1.0.6 → 2.0.0

package/src/program.ts

@@ -0,0 +1,470 @@
+import { PublicSignKey, getPublicKeyFromPeerId } from "@peerbit/crypto";
+import { Constructor, getSchema, variant } from "@dao-xyz/borsh";
+import { getValuesWithType } from "./utils.js";
+import { serialize, deserialize } from "@dao-xyz/borsh";
+import { CustomEvent, EventEmitter } from "@libp2p/interfaces/events";
+import { Client } from "./client.js";
+import { waitForAsync } from "@peerbit/time";
+import { Blocks } from "@peerbit/blocks-interface";
+import { PeerId as Libp2pPeerId } from "@libp2p/interface-peer-id";
+import {
+	SubscriptionEvent,
+	UnsubcriptionEvent,
+} from "@peerbit/pubsub-interface";
+import { Address } from "./address.js";
+import {
+	EventOptions,
+	Handler,
+	Manageable,
+	ProgramInitializationOptions,
+} from "./handler.js";
+
+const intersection = (
+	a: Set<string> | undefined,
+	b: Set<string> | IterableIterator<string>
+) => {
+	const newSet = new Set<string>();
+	for (const el of b) {
+		if (!a || a.has(el)) {
+			newSet.add(el);
+		}
+	}
+	return newSet;
+};
+
+export type OpenProgram = (program: Program) => Promise<Program>;
+
+export interface NetworkEvents {
+	join: CustomEvent<PublicSignKey>;
+	leave: CustomEvent<PublicSignKey>;
+}
+
+export interface LifeCycleEvents {
+	drop: CustomEvent<Program>;
+	open: CustomEvent<Program>;
+	close: CustomEvent<Program>;
+}
+
+export interface ProgramEvents extends NetworkEvents, LifeCycleEvents {}
+
+const getAllParentAddresses = (p: Program): string[] => {
+	return getAllParent(p, [])
+		.filter((x) => x instanceof Program)
+		.map((x) => (x as Program).address);
+};
+
+const getAllParent = (a: Program, arr: Program[] = [], includeThis = false) => {
+	includeThis && arr.push(a);
+	if (a.parents) {
+		for (const p of a.parents) {
+			if (p) {
+				getAllParent(p, arr, true);
+			}
+		}
+	}
+	return arr;
+};
+
+export type ProgramClient = Client<Program>;
+class ProgramHandler extends Handler<Program> {
+	constructor(properties: { client: ProgramClient }) {
+		super({
+			client: properties.client,
+			shouldMonitor: (p) => p instanceof Program,
+			load: Program.load,
+		});
+	}
+}
+export { ProgramHandler };
+
+@variant(0)
+export abstract class Program<
+	Args = any,
+	Events extends ProgramEvents = ProgramEvents
+> implements Manageable<Args>
+{
+	private _node: ProgramClient;
+	private _allPrograms: Program[] | undefined;
+
+	private _events: EventEmitter<ProgramEvents>;
+	private _closed: boolean;
+
+	parents: (Program<any> | undefined)[];
+	children: Program<Args>[];
+
+	private _address?: Address;
+
+	get address(): Address {
+		if (!this._address) {
+			throw new Error(
+				"Address does not exist, please open or save this program once to obtain it"
+			);
+		}
+		return this._address;
+	}
+
+	set address(address: Address) {
+		this._address = address;
+	}
+
+	addParent(program: Program<any> | undefined) {
+		(this.parents || (this.parents = [])).push(program);
+		if (program) {
+			(program.children || (program.children = [])).push(this);
+		}
+	}
+
+	get events(): EventEmitter<Events> {
+		return this._events || (this._events = new EventEmitter());
+	}
+
+	get closed(): boolean {
+		if (this._closed == null) {
+			return true;
+		}
+		return this._closed;
+	}
+	set closed(closed: boolean) {
+		this._closed = closed;
+	}
+
+	get node(): ProgramClient {
+		return this._node;
+	}
+
+	set node(node: ProgramClient) {
+		this._node = node;
+	}
+
+	private _eventOptions: EventOptions | undefined;
+
+	async beforeOpen(
+		node: ProgramClient,
+		options?: ProgramInitializationOptions<Args, this>
+	) {
+		// check that a  discriminator exist
+		const schema = getSchema(this.constructor);
+		if (!schema || typeof schema.variant !== "string") {
+			throw new Error(
+				`Expecting class to be decorated with a string variant. Example:\n\'import { variant } "@dao-xyz/borsh"\n@variant("example-db")\nclass ${this.constructor.name} { ...`
+			);
+		}
+
+		await this.save(node.services.blocks);
+		if (getAllParentAddresses(this as Program).includes(this.address)) {
+			throw new Error(
+				"Subprogram has same address as some parent program. This is not currently supported"
+			);
+		}
+
+		if (!this.closed) {
+			this.addParent(options?.parent);
+			return;
+		} else {
+			this.addParent(options?.parent);
+		}
+		this._eventOptions = options;
+		this.node = node;
+		const nexts = this.programs;
+		for (const next of nexts) {
+			await next.beforeOpen(node, { ...options, parent: this });
+		}
+
+		await this.node.services.pubsub.addEventListener(
+			"subscribe",
+			this._subscriptionEventListener ||
+				(this._subscriptionEventListener = (s) =>
+					!this.closed && this._emitJoinNetworkEvents(s.detail))
+		);
+		await this.node.services.pubsub.addEventListener(
+			"unsubscribe",
+			this._unsubscriptionEventListener ||
+				(this._unsubscriptionEventListener = (s) =>
+					!this.closed && this._emitLeaveNetworkEvents(s.detail))
+		);
+
+		await this._eventOptions?.onBeforeOpen?.(this);
+	}
+
+	async afterOpen() {
+		this.emitEvent(new CustomEvent("open", { detail: this }), true);
+		await this._eventOptions?.onOpen?.(this);
+		this.closed = false;
+		const nexts = this.programs;
+		for (const next of nexts) {
+			await next.afterOpen();
+		}
+	}
+
+	abstract open(args?: Args): Promise<void>;
+
+	private _clear() {
+		this._allPrograms = undefined;
+	}
+
+	private async _emitJoinNetworkEvents(s: SubscriptionEvent) {
+		const allTopics = this.programs
+			.map((x) => x.getTopics?.())
+			.filter((x) => x)
+			.flat() as string[];
+
+		// if subscribing to all topics, emit "join" event
+		for (const topic of allTopics) {
+			if (
+				!(await this.node.services.pubsub.getSubscribers(topic))?.has(
+					s.from.hashcode()
+				)
+			) {
+				return;
+			}
+		}
+		this.events.dispatchEvent(new CustomEvent("join", { detail: s.from }));
+	}
+
+	private async _emitLeaveNetworkEvents(s: UnsubcriptionEvent) {
+		const allTopics = this.programs
+			.map((x) => x.getTopics?.())
+			.filter((x) => x)
+			.flat() as string[];
+
+		// if subscribing not subscribing to any topics, emit "leave" event
+		for (const topic of allTopics) {
+			if (
+				(await this.node.services.pubsub.getSubscribers(topic))?.has(
+					s.from.hashcode()
+				)
+			) {
+				return;
+			}
+		}
+		this.events.dispatchEvent(new CustomEvent("leave", { detail: s.from }));
+	}
+
+	private _subscriptionEventListener: (
+		e: CustomEvent<SubscriptionEvent>
+	) => void;
+	private _unsubscriptionEventListener: (
+		e: CustomEvent<UnsubcriptionEvent>
+	) => void;
+
+	private async processEnd(type: "drop" | "close") {
+		if (!this.closed) {
+			this.emitEvent(new CustomEvent(type, { detail: this }), true);
+			if (type === "close") {
+				this._eventOptions?.onClose?.(this);
+			} else if (type === "drop") {
+				this._eventOptions?.onDrop?.(this);
+			} else {
+				throw new Error("Unsupported event type: " + type);
+			}
+
+			const promises: Promise<void | boolean>[] = [];
+
+			if (this.children) {
+				for (const program of this.children) {
+					promises.push(program[type](this as Program)); // TODO types
+				}
+				this.children = [];
+			}
+			await Promise.all(promises);
+
+			this._clear();
+			this.closed = true;
+			return true;
+		} else {
+			this._clear();
+			return true;
+		}
+	}
+
+	private async end(type: "drop" | "close", from?: Program): Promise<boolean> {
+		if (this.closed) {
+			return true;
+		}
+
+		let parentIdx = -1;
+		let close = true;
+		if (this.parents) {
+			parentIdx = this.parents.findIndex((x) => x == from);
+			if (parentIdx !== -1) {
+				if (this.parents.length === 1) {
+					close = true;
+				} else {
+					this.parents.splice(parentIdx, 1);
+					close = false;
+				}
+			} else if (from) {
+				throw new Error("Could not find from in parents");
+			}
+		}
+
+		const end = close && (await this.processEnd(type));
+		if (end) {
+			this.node?.services.pubsub.removeEventListener(
+				"subscribe",
+				this._subscriptionEventListener
+			);
+			this.node?.services.pubsub.removeEventListener(
+				"unsubscribe",
+				this._unsubscriptionEventListener
+			);
+
+			this._eventOptions = undefined;
+
+			if (parentIdx !== -1) {
+				this.parents.splice(parentIdx, 1); // We splice this here because this._end depends on this parent to exist
+			}
+		}
+
+		return end;
+	}
+	async close(from?: Program): Promise<boolean> {
+		return this.end("close", from);
+	}
+
+	async drop(from?: Program): Promise<boolean> {
+		const dropped = await this.end("drop", from);
+		if (dropped) {
+			await this.delete();
+		}
+		return dropped;
+	}
+
+	emitEvent(event: CustomEvent, parents = false) {
+		this.events.dispatchEvent(event);
+		if (parents) {
+			if (this.parents) {
+				for (const parent of this.parents) {
+					parent?.emitEvent(event);
+				}
+			}
+		}
+	}
+
+	/**
+	 * Wait for another peer to be 'ready' to talk with you for this particular program
+	 * @param other
+	 */
+	async waitFor(...other: (PublicSignKey | Libp2pPeerId)[]): Promise<void> {
+		const expectedHashes = new Set(
+			other.map((x) =>
+				x instanceof PublicSignKey
+					? x.hashcode()
+					: getPublicKeyFromPeerId(x).hashcode()
+			)
+		);
+		await waitForAsync(
+			async () => {
+				return (
+					intersection(expectedHashes, await this.getReady()).size ===
+					expectedHashes.size
+				);
+			},
+			{ delayInterval: 200, timeout: 10 * 1000 }
+		); // 200 ms delay since this is an expensive op. TODO, make event based instead
+	}
+
+	async getReady(): Promise<Set<string>> {
+		// all peers that subscribe to all topics
+		let ready: Set<string> | undefined = undefined; // the interesection of all ready
+		for (const program of this.allPrograms) {
+			if (program.getTopics) {
+				const topics = program.getTopics();
+				for (const topic of topics) {
+					const subscribers = await this.node.services.pubsub.getSubscribers(
+						topic
+					);
+					if (!subscribers) {
+						throw new Error(
+							"client is not subscriber to topic data, do not have any info about peer readiness"
+						);
+					}
+					ready = intersection(ready, subscribers.keys());
+				}
+			}
+		}
+		if (ready == null) {
+			throw new Error("Do not have any info about peer readiness");
+		}
+		return ready;
+	}
+
+	get allPrograms(): Program[] {
+		if (this._allPrograms) {
+			return this._allPrograms;
+		}
+		const arr: Program[] = this.programs;
+		const nexts = this.programs;
+		for (const next of nexts) {
+			arr.push(...next.allPrograms);
+		}
+		this._allPrograms = arr;
+		return this._allPrograms;
+	}
+
+	get programs(): Program[] {
+		return getValuesWithType(this, Program);
+	}
+
+	clone(): this {
+		return deserialize(serialize(this), this.constructor);
+	}
+
+	getTopics?(): string[];
+
+	async save(store: Blocks = this.node.services.blocks): Promise<Address> {
+		const existingAddress = this._address;
+		const hash = await store.put(serialize(this));
+
+		this._address = hash;
+		if (!this.address) {
+			throw new Error("Unexpected");
+		}
+
+		if (existingAddress && existingAddress !== this.address) {
+			throw new Error(
+				"Program properties has been changed after constructor so that the hash has changed. Make sure that the 'setup(...)' function does not modify any properties that are to be serialized"
+			);
+		}
+
+		return this._address!;
+	}
+
+	async delete(): Promise<void> {
+		if (this.address) {
+			return this.node.services.blocks.rm(this.address);
+		}
+		// Not saved
+	}
+
+	static async load<P extends Program<any>>(
+		address: Address,
+		store: Blocks,
+		options?: {
+			timeout?: number;
+		}
+	): Promise<P | undefined> {
+		const bytes = await store.get(address, options);
+		if (!bytes) {
+			return undefined;
+		}
+		const der = deserialize(bytes, Program);
+		der.address = address;
+		return der as P;
+	}
+
+	static async open<T extends Program<Args>, Args = any>(
+		this: Constructor<T>,
+		address: Address,
+		node: ProgramClient,
+		options?: ProgramInitializationOptions<Args, T>
+	): Promise<T> {
+		const p = await Program.load<T>(address, node.services.blocks);
+
+		if (!p) {
+			throw new Error("Failed to load program");
+		}
+		await node.open(p, options);
+		return p as T;
+	}
+}

package/lib/esm/node.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=node.js.map

package/lib/esm/node.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"node.js","sourceRoot":"","sources":["../../src/node.ts"],"names":[],"mappings":""}