Move ProtonStart lifecycle

Author: Sleitnick

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rbxts/proton",
-  "version": "0.5.8",
+  "version": "0.5.9",
   "description": "Framework for Roblox game development",
   "main": "out/init.lua",
   "scripts": {

src/core.ts

@@ -1,11 +1,8 @@
-import { LifecycleBehavior, ProtonLifecycle } from "./lifecycle";
-
 const providerClasses = new Map<new () => unknown, unknown>();
 
 let started = false;
 const awaitStartThreads: thread[] = [];
-
-export const ProtonStart = new ProtonLifecycle<() => void>(LifecycleBehavior.Concurrent);
+const awaitCallbacks: (() => void)[] = [];
 
 /**
  * Provider decorator.
@@ -36,8 +33,10 @@ export namespace Proton {
 	 */
 	export function start() {
 		if (started) return;
-		ProtonStart.fire();
 		started = true;
+		for (const callback of awaitCallbacks) {
+			task.spawn(callback);
+		}
 		for (const awaitThread of awaitStartThreads) {
 			task.spawn(awaitThread);
 		}
@@ -62,6 +61,20 @@ export namespace Proton {
 		coroutine.yield();
 	}
 
+	/**
+	 * Calls the callback once Proton has fully started.
+	 * If Proton is already started, the callback will
+	 * be spawned immediately.
+	 * @param callback Callback
+	 */
+	export function onStart(callback: () => void) {
+		if (started) {
+			task.spawn(callback);
+			return;
+		}
+		awaitCallbacks.push(callback);
+	}
+
 	/**
 	 * Gets a provider within Proton.
 	 *

src/lifecycle.ts

@@ -1,4 +1,4 @@
-import { Proton, ProtonStart, Provider } from "./core";
+import { Proton } from "./core";
 
 type LifecycleCallback<T> = (...args: Parameters<T>) => void;
 
@@ -160,3 +160,6 @@ export function Lifecycle<T extends LifecycleCallback<T>>(lifecycle: ProtonLifec
 		);
 	};
 }
+
+export const ProtonStart = new ProtonLifecycle<() => void>(LifecycleBehavior.Concurrent);
+Proton.onStart(() => ProtonStart.fire());